parent
3ce658b930
commit
7dd986a563
@ -1,36 +0,0 @@
|
||||
# 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.Token.CleanWorker do
|
||||
@moduledoc """
|
||||
The module represents functions to clean an expired OAuth and MFA tokens.
|
||||
"""
|
||||
use GenServer
|
||||
|
||||
@ten_seconds 10_000
|
||||
@one_day 86_400_000
|
||||
|
||||
alias Pleroma.MFA
|
||||
alias Pleroma.Workers.BackgroundWorker
|
||||
|
||||
def start_link(_), do: GenServer.start_link(__MODULE__, %{})
|
||||
|
||||
def init(_) do
|
||||
Process.send_after(self(), :perform, @ten_seconds)
|
||||
{:ok, nil}
|
||||
end
|
||||
|
||||
@doc false
|
||||
def handle_info(:perform, state) do
|
||||
BackgroundWorker.enqueue("clean_expired_tokens", %{})
|
||||
interval = Pleroma.Config.get([:oauth2, :clean_expired_tokens_interval], @one_day)
|
||||
|
||||
Process.send_after(self(), :perform, interval)
|
||||
{:noreply, state}
|
||||
end
|
||||
|
||||
def perform(:clean) do
|
||||
MFA.Token.delete_expired_tokens()
|
||||
end
|
||||
end
|
@ -1,27 +0,0 @@
|
||||
# Pleroma: A lightweight social networking server
|
||||
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
|
||||
# SPDX-License-Identifier: AGPL-3.0-only
|
||||
|
||||
defmodule Pleroma.Workers.PurgeExpiredOAuthTokenTest do
|
||||
use Pleroma.DataCase, async: true
|
||||
use Oban.Testing, repo: Pleroma.Repo
|
||||
|
||||
import Pleroma.Factory
|
||||
|
||||
setup do: clear_config([:oauth2, :clean_expired_tokens], true)
|
||||
|
||||
test "purges expired token" do
|
||||
user = insert(:user)
|
||||
app = insert(:oauth_app)
|
||||
|
||||
{:ok, %{id: id}} = Pleroma.Web.OAuth.Token.create(app, user)
|
||||
|
||||
assert_enqueued(
|
||||
worker: Pleroma.Workers.PurgeExpiredOAuthToken,
|
||||
args: %{token_id: id}
|
||||
)
|
||||
|
||||
assert {:ok, %{id: ^id}} =
|
||||
perform_job(Pleroma.Workers.PurgeExpiredOAuthToken, %{token_id: id})
|
||||
end
|
||||
end
|
@ -0,0 +1,51 @@
|
||||
# Pleroma: A lightweight social networking server
|
||||
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
|
||||
# SPDX-License-Identifier: AGPL-3.0-only
|
||||
|
||||
defmodule Pleroma.Workers.PurgeExpiredTokenTest do
|
||||
use Pleroma.DataCase, async: true
|
||||
use Oban.Testing, repo: Pleroma.Repo
|
||||
|
||||
import Pleroma.Factory
|
||||
|
||||
setup do: clear_config([:oauth2, :clean_expired_tokens], true)
|
||||
|
||||
test "purges expired oauth token" do
|
||||
user = insert(:user)
|
||||
app = insert(:oauth_app)
|
||||
|
||||
{:ok, %{id: id}} = Pleroma.Web.OAuth.Token.create(app, user)
|
||||
|
||||
assert_enqueued(
|
||||
worker: Pleroma.Workers.PurgeExpiredToken,
|
||||
args: %{token_id: id, mod: Pleroma.Web.OAuth.Token}
|
||||
)
|
||||
|
||||
assert {:ok, %{id: ^id}} =
|
||||
perform_job(Pleroma.Workers.PurgeExpiredToken, %{
|
||||
token_id: id,
|
||||
mod: Pleroma.Web.OAuth.Token
|
||||
})
|
||||
|
||||
assert Repo.aggregate(Pleroma.Web.OAuth.Token, :count, :id) == 0
|
||||
end
|
||||
|
||||
test "purges expired mfa token" do
|
||||
authorization = insert(:oauth_authorization)
|
||||
|
||||
{:ok, %{id: id}} = Pleroma.MFA.Token.create(authorization.user, authorization)
|
||||
|
||||
assert_enqueued(
|
||||
worker: Pleroma.Workers.PurgeExpiredToken,
|
||||
args: %{token_id: id, mod: Pleroma.MFA.Token}
|
||||
)
|
||||
|
||||
assert {:ok, %{id: ^id}} =
|
||||
perform_job(Pleroma.Workers.PurgeExpiredToken, %{
|
||||
token_id: id,
|
||||
mod: Pleroma.MFA.Token
|
||||
})
|
||||
|
||||
assert Repo.aggregate(Pleroma.MFA.Token, :count, :id) == 0
|
||||
end
|
||||
end
|
Loading…
Reference in new issue