140 lines
4.7 KiB
Erlang
140 lines
4.7 KiB
Erlang
-module(logic_event_recurring_tests).
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
-include("records.hrl").
|
|
|
|
-define(TABLES, [user, calendar, event, recurrence_exception]).
|
|
|
|
setup() ->
|
|
eh_test_support:start_mnesia(),
|
|
eh_test_support:ensure_tables(?TABLES),
|
|
ok.
|
|
|
|
cleanup(_) ->
|
|
eh_test_support:delete_tables(?TABLES),
|
|
eh_test_support:stop_mnesia(),
|
|
ok.
|
|
|
|
logic_event_recurring_test_() ->
|
|
{foreach,
|
|
fun setup/0,
|
|
fun cleanup/1,
|
|
[
|
|
{"Create recurring event test", fun test_create_recurring_event/0},
|
|
{"Create recurring event invalid RRULE test", fun test_create_recurring_invalid/0},
|
|
{"Get occurrences test", fun test_get_occurrences/0},
|
|
{"Cancel occurrence test", fun test_cancel_occurrence/0},
|
|
{"Get occurrences with cancelled test", fun test_occurrences_with_cancelled/0},
|
|
{"Materialize for booking test", fun test_materialize_for_booking/0}
|
|
]}.
|
|
|
|
create_test_user_and_calendar() ->
|
|
UserId = base64:encode(crypto:strong_rand_bytes(16), #{mode => urlsafe, padding => false}),
|
|
User = #user{
|
|
id = UserId,
|
|
email = <<"test@example.com">>,
|
|
password_hash = <<"hash">>,
|
|
role = user,
|
|
status = active,
|
|
created_at = calendar:universal_time(),
|
|
updated_at = calendar:universal_time()
|
|
},
|
|
mnesia:dirty_write(User),
|
|
|
|
{ok, Calendar} = logic_calendar:create_calendar(UserId, <<"Test Calendar">>, <<"">>, manual),
|
|
{UserId, Calendar#calendar.id}.
|
|
|
|
add_days(DateTime, Days) ->
|
|
Sec = calendar:datetime_to_gregorian_seconds(DateTime) + Days * 86400,
|
|
calendar:gregorian_seconds_to_datetime(Sec).
|
|
|
|
test_create_recurring_event() ->
|
|
{UserId, CalendarId} = create_test_user_and_calendar(),
|
|
Title = <<"Weekly Meeting">>,
|
|
StartTime = eh_test_support:future_start(),
|
|
Duration = 60,
|
|
RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},
|
|
|
|
{ok, Event} = logic_event:create_recurring_event(UserId, CalendarId, Title, StartTime, Duration, RRule),
|
|
|
|
?assertEqual(CalendarId, Event#event.calendar_id),
|
|
?assertEqual(Title, Event#event.title),
|
|
?assertEqual(recurring, Event#event.event_type).
|
|
|
|
test_create_recurring_invalid() ->
|
|
{UserId, CalendarId} = create_test_user_and_calendar(),
|
|
Title = <<"Invalid">>,
|
|
StartTime = eh_test_support:future_start(),
|
|
Duration = 60,
|
|
InvalidRRule = #{<<"freq">> => <<"YEARLY">>, <<"interval">> => 1},
|
|
|
|
{error, invalid_rrule} = logic_event:create_recurring_event(
|
|
UserId, CalendarId, Title, StartTime, Duration, InvalidRRule
|
|
).
|
|
|
|
test_get_occurrences() ->
|
|
{UserId, CalendarId} = create_test_user_and_calendar(),
|
|
StartTime = eh_test_support:future_start(),
|
|
RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},
|
|
|
|
{ok, Event} = logic_event:create_recurring_event(
|
|
UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
|
|
),
|
|
|
|
RangeEnd = add_days(StartTime, 28),
|
|
{ok, Occurrences} = logic_event:get_occurrences(UserId, Event#event.id, RangeEnd),
|
|
|
|
?assertEqual(5, length(Occurrences)).
|
|
|
|
test_cancel_occurrence() ->
|
|
{UserId, CalendarId} = create_test_user_and_calendar(),
|
|
StartTime = eh_test_support:future_start(),
|
|
RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},
|
|
|
|
{ok, Event} = logic_event:create_recurring_event(
|
|
UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
|
|
),
|
|
|
|
CancelTime = add_days(StartTime, 7),
|
|
{ok, cancelled} = logic_event:cancel_occurrence(UserId, Event#event.id, CancelTime).
|
|
|
|
test_occurrences_with_cancelled() ->
|
|
{UserId, CalendarId} = create_test_user_and_calendar(),
|
|
StartTime = eh_test_support:future_start(),
|
|
RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},
|
|
|
|
{ok, Event} = logic_event:create_recurring_event(
|
|
UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
|
|
),
|
|
|
|
CancelTime = add_days(StartTime, 7),
|
|
{ok, cancelled} = logic_event:cancel_occurrence(UserId, Event#event.id, CancelTime),
|
|
|
|
RangeEnd = add_days(StartTime, 28),
|
|
{ok, Occurrences} = logic_event:get_occurrences(UserId, Event#event.id, RangeEnd),
|
|
|
|
?assertEqual(4, length(Occurrences)),
|
|
|
|
Starts = [O || {virtual, O} <- Occurrences],
|
|
?assertNot(lists:member(CancelTime, Starts)).
|
|
|
|
test_materialize_for_booking() ->
|
|
{UserId, CalendarId} = create_test_user_and_calendar(),
|
|
StartTime = eh_test_support:future_start(),
|
|
RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},
|
|
|
|
{ok, Event} = logic_event:create_recurring_event(
|
|
UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
|
|
),
|
|
|
|
OccurrenceStart = add_days(StartTime, 7),
|
|
SpecialistId = <<"specialist123">>,
|
|
|
|
{ok, Instance} = logic_event:materialize_for_booking(
|
|
Event#event.id, OccurrenceStart, SpecialistId
|
|
),
|
|
|
|
?assertEqual(Event#event.id, Instance#event.master_id),
|
|
?assertEqual(OccurrenceStart, Instance#event.start_time),
|
|
?assertEqual(SpecialistId, Instance#event.specialist_id),
|
|
?assertEqual(true, Instance#event.is_instance).
|