Fix scheduler test

Signed-off-by: Thomas Citharel <tcit@tcit.fr>
This commit is contained in:
Thomas Citharel 2021-06-27 16:18:53 +02:00
parent 7ec6f158ec
commit ed182e358b
No known key found for this signature in database
GPG Key ID: A061B9DDE0CA0773
2 changed files with 88 additions and 47 deletions

View File

@ -144,11 +144,17 @@ defmodule Mobilizon.Service.Notifications.Scheduler do
def weekly_notification(_), do: {:ok, nil} def weekly_notification(_), do: {:ok, nil}
def pending_participation_notification(%Event{ def pending_participation_notification(event, options \\ [])
def pending_participation_notification(
%Event{
id: event_id, id: event_id,
organizer_actor_id: organizer_actor_id, organizer_actor_id: organizer_actor_id,
local: true local: true,
}) do begins_on: begins_on
},
options
) do
with %Actor{user_id: user_id} when not is_nil(user_id) <- with %Actor{user_id: user_id} when not is_nil(user_id) <-
Actors.get_actor(organizer_actor_id), Actors.get_actor(organizer_actor_id),
%User{ %User{
@ -158,29 +164,15 @@ defmodule Mobilizon.Service.Notifications.Scheduler do
timezone: timezone timezone: timezone
} }
} <- Users.get_user_with_settings!(user_id) do } <- Users.get_user_with_settings!(user_id) do
compare_to = Keyword.get(options, :compare_to, DateTime.utc_now())
send_at = send_at =
case notification_pending_participation do determine_send_at(notification_pending_participation, begins_on,
:none -> compare_to: compare_to,
nil
:direct ->
:direct
:one_day ->
calculate_next_day_notification(Date.utc_today(), timezone: timezone)
:one_week ->
calculate_next_week_notification(DateTime.utc_now(),
timezone: timezone, timezone: timezone,
locale: locale locale: locale
) )
:one_hour ->
DateTime.utc_now()
|> DateTime.shift_zone!(timezone)
|> (&%{&1 | minute: 0, second: 0, microsecond: {0, 0}}).()
end
params = %{ params = %{
user_id: user_id, user_id: user_id,
event_id: event_id event_id: event_id
@ -196,15 +188,18 @@ defmodule Mobilizon.Service.Notifications.Scheduler do
{:ok, nil} {:ok, nil}
# Sending to calculated time # Sending to calculated time
true -> DateTime.compare(begins_on, send_at) == :gt ->
Notification.enqueue(:pending_participation_notification, params, scheduled_at: send_at) Notification.enqueue(:pending_participation_notification, params, scheduled_at: send_at)
true ->
{:ok, nil}
end end
else else
_ -> {:ok, nil} _ -> {:ok, nil}
end end
end end
def pending_participation_notification(_), do: {:ok, nil} def pending_participation_notification(_, _), do: {:ok, nil}
def pending_membership_notification(%Actor{type: :Group, id: group_id}) do def pending_membership_notification(%Actor{type: :Group, id: group_id}) do
group_id group_id
@ -275,4 +270,36 @@ defmodule Mobilizon.Service.Notifications.Scheduler do
Notification.enqueue(:pending_membership_notification, params, scheduled_at: send_at) Notification.enqueue(:pending_membership_notification, params, scheduled_at: send_at)
end end
end end
defp determine_send_at(notification_pending_participation, begins_on, options) do
timezone = Keyword.get(options, :timezone, "Etc/UTC")
locale = Keyword.get(options, :locale, "en")
compare_to = Keyword.get(options, :compare_to, DateTime.utc_now())
case notification_pending_participation do
:none ->
nil
:direct ->
:direct
:one_day ->
calculate_next_day_notification(DateTime.to_date(compare_to),
timezone: timezone,
compare_to: compare_to
)
:one_week ->
calculate_next_week_notification(begins_on,
timezone: timezone,
locale: locale,
compare_to: compare_to
)
:one_hour ->
compare_to
|> DateTime.shift_zone!(timezone)
|> (&%{&1 | minute: 0, second: 0, microsecond: {0, 0}}).()
end
end
end end

View File

@ -238,32 +238,46 @@ defmodule Mobilizon.Service.Notifications.SchedulerTest do
user = Map.put(user, :settings, settings) user = Map.put(user, :settings, settings)
actor = insert(:actor, user: user) actor = insert(:actor, user: user)
# Make sure event happens next week %Event{} =
%Date{} = event_day = Date.utc_today() |> Date.add(3) event =
{:ok, %NaiveDateTime{} = event_date} = event_day |> NaiveDateTime.new(~T[16:00:00]) insert(:event, begins_on: ~U[2021-06-26 12:00:00Z], local: true, organizer_actor: actor)
{:ok, begins_on} = DateTime.from_naive(event_date, "Etc/UTC")
%Event{} = event = insert(:event, begins_on: begins_on, local: true, organizer_actor: actor)
%Participant{} = _participant = insert(:participant, event: event, role: :not_approved) %Participant{} = _participant = insert(:participant, event: event, role: :not_approved)
Scheduler.pending_participation_notification(event) Scheduler.pending_participation_notification(event, compare_to: ~U[2021-06-25 07:00:00Z])
now = Time.utc_now()
{:ok, scheduled_at} =
if now.hour <= 18 do
NaiveDateTime.new(Date.utc_today(), ~T[18:00:00])
else
Date.utc_today() |> Date.add(1) |> NaiveDateTime.new(~T[18:00:00])
end
{:ok, scheduled_at} = DateTime.from_naive(scheduled_at, "Europe/Paris")
assert_enqueued( assert_enqueued(
worker: Notification, worker: Notification,
args: %{user_id: user_id, event_id: event.id, op: :pending_participation_notification}, args: %{user_id: user_id, event_id: event.id, op: :pending_participation_notification},
scheduled_at: scheduled_at scheduled_at: ~U[2021-06-25 16:00:00Z]
)
end
test "if the notification date is passed" do
%User{id: user_id} = user = insert(:user, locale: "fr")
settings =
insert(:settings,
user_id: user_id,
notification_pending_participation: :one_day,
timezone: "Europe/Paris"
)
user = Map.put(user, :settings, settings)
actor = insert(:actor, user: user)
%Event{} =
event =
insert(:event, begins_on: ~U[2021-06-26 12:00:00Z], local: true, organizer_actor: actor)
%Participant{} = _participant = insert(:participant, event: event, role: :not_approved)
Scheduler.pending_participation_notification(event, compare_to: ~U[2021-06-26 07:00:00Z])
refute_enqueued(
worker: Notification,
args: %{user_id: user_id, event_id: event.id, op: :pending_participation_notification},
scheduled_at: ~U[2021-06-25 16:00:00Z]
) )
end end