85 lines
		
	
	
	
		
			2.4 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			85 lines
		
	
	
	
		
			2.4 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
| # Pleroma: A lightweight social networking server
 | |
| # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
 | |
| # SPDX-License-Identifier: AGPL-3.0-only
 | |
| 
 | |
| defmodule Pleroma.Web.OAuth.TokenTest do
 | |
|   use Pleroma.DataCase
 | |
|   alias Pleroma.Repo
 | |
|   alias Pleroma.Web.OAuth.App
 | |
|   alias Pleroma.Web.OAuth.Authorization
 | |
|   alias Pleroma.Web.OAuth.Token
 | |
| 
 | |
|   import Pleroma.Factory
 | |
| 
 | |
|   test "exchanges a auth token for an access token, preserving `scopes`" do
 | |
|     {:ok, app} =
 | |
|       Repo.insert(
 | |
|         App.register_changeset(%App{}, %{
 | |
|           client_name: "client",
 | |
|           scopes: ["read", "write"],
 | |
|           redirect_uris: "url"
 | |
|         })
 | |
|       )
 | |
| 
 | |
|     user = insert(:user)
 | |
| 
 | |
|     {:ok, auth} = Authorization.create_authorization(app, user, ["read"])
 | |
|     assert auth.scopes == ["read"]
 | |
| 
 | |
|     {:ok, token} = Token.exchange_token(app, auth)
 | |
| 
 | |
|     assert token.app_id == app.id
 | |
|     assert token.user_id == user.id
 | |
|     assert token.scopes == auth.scopes
 | |
|     assert String.length(token.token) > 10
 | |
|     assert String.length(token.refresh_token) > 10
 | |
| 
 | |
|     auth = Repo.get(Authorization, auth.id)
 | |
|     {:error, "already used"} = Token.exchange_token(app, auth)
 | |
|   end
 | |
| 
 | |
|   test "deletes all tokens of a user" do
 | |
|     {:ok, app1} =
 | |
|       Repo.insert(
 | |
|         App.register_changeset(%App{}, %{
 | |
|           client_name: "client1",
 | |
|           scopes: ["scope"],
 | |
|           redirect_uris: "url"
 | |
|         })
 | |
|       )
 | |
| 
 | |
|     {:ok, app2} =
 | |
|       Repo.insert(
 | |
|         App.register_changeset(%App{}, %{
 | |
|           client_name: "client2",
 | |
|           scopes: ["scope"],
 | |
|           redirect_uris: "url"
 | |
|         })
 | |
|       )
 | |
| 
 | |
|     user = insert(:user)
 | |
| 
 | |
|     {:ok, auth1} = Authorization.create_authorization(app1, user)
 | |
|     {:ok, auth2} = Authorization.create_authorization(app2, user)
 | |
| 
 | |
|     {:ok, _token1} = Token.exchange_token(app1, auth1)
 | |
|     {:ok, _token2} = Token.exchange_token(app2, auth2)
 | |
| 
 | |
|     {tokens, _} = Token.delete_user_tokens(user)
 | |
| 
 | |
|     assert tokens == 2
 | |
|   end
 | |
| 
 | |
|   test "deletes expired tokens" do
 | |
|     insert(:oauth_token, valid_until: Timex.shift(Timex.now(), days: -3))
 | |
|     insert(:oauth_token, valid_until: Timex.shift(Timex.now(), days: -3))
 | |
|     t3 = insert(:oauth_token)
 | |
|     t4 = insert(:oauth_token, valid_until: Timex.shift(Timex.now(), minutes: 10))
 | |
|     {tokens, _} = Token.delete_expired_tokens()
 | |
|     assert tokens == 2
 | |
|     available_tokens = Pleroma.Repo.all(Token)
 | |
| 
 | |
|     token_ids = available_tokens |> Enum.map(& &1.id)
 | |
|     assert token_ids == [t3.id, t4.id]
 | |
|   end
 | |
| end
 | 
