Page MenuHomePhorge
Files F85826043

No OneTemporary
Actions

Authored By
Unknown
Size
3 KB
Referenced Files
None
Subscribers
None

View Options

diff --git a/test/pleroma/workers/poll_worker_test.exs b/test/pleroma/workers/poll_worker_test.exs
index 56a338bac..0fafcae11 100644
--- a/test/pleroma/workers/poll_worker_test.exs
+++ b/test/pleroma/workers/poll_worker_test.exs
@@ -1,84 +1,103 @@
# Pleroma: A lightweight social networking server
# Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Workers.PollWorkerTest do
use Pleroma.DataCase
use Oban.Testing, repo: Pleroma.Repo
import Mock
import Pleroma.Factory
alias Pleroma.Workers.PollWorker
test "poll notification job" do
user = insert(:user)
question = insert(:question, user: user)
activity = insert(:question_activity, question: question)
PollWorker.schedule_poll_end(activity)
expected_job_args = %{"activity_id" => activity.id, "op" => "poll_end"}
assert_enqueued(args: expected_job_args)
with_mocks([
{
Pleroma.Web.Streamer,
[],
[
stream: fn _, _ -> nil end
]
},
{
Pleroma.Web.Push,
[],
[
send: fn _ -> nil end
]
}
]) do
[job] = all_enqueued(worker: PollWorker)
PollWorker.perform(job)
# Ensure notifications were streamed out when job executes
assert called(Pleroma.Web.Streamer.stream(["user", "user:notification"], :_))
assert called(Pleroma.Web.Push.send(:_))
# Ensure we scheduled a final refresh of the poll
assert_enqueued(
worker: PollWorker,
args: %{"op" => "refresh", "activity_id" => activity.id}
)
end
end
- test "poll refresh job" do
- user = insert(:user, local: false)
- question = insert(:question, user: user)
- activity = insert(:question_activity, question: question)
+ describe "poll refresh" do
+ test "normal job" do
+ user = insert(:user, local: false)
+ question = insert(:question, user: user)
+ activity = insert(:question_activity, question: question)
+
+ PollWorker.new(%{"op" => "refresh", "activity_id" => activity.id})
+ |> Oban.insert()
+
+ expected_job_args = %{"activity_id" => activity.id, "op" => "refresh"}
+
+ assert_enqueued(args: expected_job_args)
+
+ with_mocks([
+ {
+ Pleroma.Web.Streamer,
+ [],
+ [
+ stream: fn _, _ -> nil end
+ ]
+ }
+ ]) do
+ [job] = all_enqueued(worker: PollWorker)
+ PollWorker.perform(job)
+
+ # Ensure updates are streamed out
+ assert called(Pleroma.Web.Streamer.stream(["user", "list", "public", "public:local"], :_))
+ end
+ end
- PollWorker.new(%{"op" => "refresh", "activity_id" => activity.id})
- |> Oban.insert()
+ test "when updated_at is after poll closing" do
+ poll_closed = DateTime.utc_now() |> DateTime.add(-86_400) |> DateTime.to_iso8601()
+ user = insert(:user, local: false)
+ question = insert(:question, user: user, closed: poll_closed)
+ activity = insert(:question_activity, question: question)
- expected_job_args = %{"activity_id" => activity.id, "op" => "refresh"}
+ PollWorker.new(%{"op" => "refresh", "activity_id" => activity.id})
+ |> Oban.insert()
- assert_enqueued(args: expected_job_args)
+ expected_job_args = %{"activity_id" => activity.id, "op" => "refresh"}
- with_mocks([
- {
- Pleroma.Web.Streamer,
- [],
- [
- stream: fn _, _ -> nil end
- ]
- }
- ]) do
- [job] = all_enqueued(worker: PollWorker)
- PollWorker.perform(job)
+ assert_enqueued(args: expected_job_args)
- # Ensure updates are streamed out
- assert called(Pleroma.Web.Streamer.stream(["user", "list", "public", "public:local"], :_))
+ [job] = all_enqueued(worker: PollWorker)
+ assert {:cancel, :poll_finalized} = PollWorker.perform(job)
end
end
end

File Metadata

Mime Type
text/x-diff
Expires
Fri, Oct 30, 4:47 AM (1 d, 2 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
1799483
Default Alt Text
(3 KB)

Event Timeline