Implement commercial calendars: booking_open, specialists, expire without type downgrade.

Refs EventHub/EventHubBack#54

test/unit/booking_integration_tests.erl

@@ -2,7 +2,7 @@
 -include_lib("eunit/include/eunit.hrl").
 -include("records.hrl").
 
--define(TABLES, [user, calendar, event, booking, admin]).
+-define(TABLES, [user, calendar, event, booking, admin, subscription]).
 
 setup() ->
   eh_test_support:start_mnesia(),
@@ -19,7 +19,7 @@ booking_integration_test_() ->
     fun setup/0,
     fun cleanup/1,
     [
-      {"Booking create stays pending", fun test_pending_booking_flow/0},
+      {"Booking auto confirms", fun test_auto_booking_flow/0},
       {"Full booking flow with manual confirmation", fun test_manual_booking_flow/0},
       {"Capacity management test", fun test_capacity_management/0},
       {"Multiple bookings test", fun test_multiple_bookings/0}
@@ -39,24 +39,25 @@ create_user() ->
   mnesia:dirty_write(User),
   UserId.
 
+create_commercial(OwnerId, Confirmation) ->
+  {ok, _} = core_subscription:create(OwnerId, monthly, true),
+  {ok, Calendar} = core_calendar:create(OwnerId, <<"Cal">>, <<"">>, Confirmation, commercial),
+  Calendar.
+
 add_days(DateTime, Days) ->
   Sec = calendar:datetime_to_gregorian_seconds(DateTime) + Days * 86400,
   calendar:gregorian_seconds_to_datetime(Sec).
 
-test_pending_booking_flow() ->
+test_auto_booking_flow() ->
   OwnerId = create_user(),
   ParticipantId = create_user(),
-
-  {ok, Calendar} = core_calendar:create(OwnerId, <<"Auto">>, <<"">>, auto),
+  Calendar = create_commercial(OwnerId, auto),
 
   StartTime = eh_test_support:future_start(),
   {ok, Event} = core_event:create(Calendar#calendar.id, <<"Event">>, StartTime, 60),
 
   {ok, Booking} = logic_booking:create_booking(ParticipantId, Event#event.id),
-  ?assertEqual(pending, Booking#booking.status),
-
-  {ok, Stored} = core_booking:get_by_id(Booking#booking.id),
-  ?assertEqual(pending, Stored#booking.status),
+  ?assertEqual(confirmed, Booking#booking.status),
 
   {ok, EventBookings} = logic_booking:list_event_bookings(Event#event.id),
   ?assertEqual(1, length(EventBookings)),
@@ -67,8 +68,7 @@ test_pending_booking_flow() ->
 test_manual_booking_flow() ->
   OwnerId = create_user(),
   ParticipantId = create_user(),
-
-  {ok, Calendar} = core_calendar:create(OwnerId, <<"Manual">>, <<"">>, manual),
+  Calendar = create_commercial(OwnerId, manual),
 
   StartTime = eh_test_support:future_start(),
   {ok, Event} = core_event:create(Calendar#calendar.id, <<"Event">>, StartTime, 60),
@@ -88,7 +88,7 @@ test_capacity_management() ->
   Participant2Id = create_user(),
   Participant3Id = create_user(),
 
-  {ok, Calendar} = core_calendar:create(OwnerId, <<"Test">>, <<"">>, manual),
+  Calendar = create_commercial(OwnerId, manual),
 
   StartTime = eh_test_support:future_start(),
   {ok, Event} = core_event:create(Calendar#calendar.id, <<"Event">>, StartTime, 60),
@@ -108,8 +108,7 @@ test_capacity_management() ->
 test_multiple_bookings() ->
   OwnerId = create_user(),
   ParticipantId = create_user(),
-
-  {ok, Calendar} = core_calendar:create(OwnerId, <<"Test">>, <<"">>, manual),
+  Calendar = create_commercial(OwnerId, manual),
 
   Base = eh_test_support:future_start(),
   StartTime1 = Base,

test/unit/eh_test_support.erl

@@ -129,7 +129,8 @@ table_opts(calendar_share) ->
 table_opts(calendar_follow) ->
   [{ram_copies, [node()]}, {attributes, record_info(fields, calendar_follow)}];
 table_opts(calendar_specialist) ->
-  [{ram_copies, [node()]}, {attributes, record_info(fields, calendar_specialist)}];
+  [{ram_copies, [node()]}, {attributes, record_info(fields, calendar_specialist)},
+   {index, [calendar_id, user_id]}];
 table_opts(event) ->
   [{ram_copies, [node()]}, {attributes, record_info(fields, event)}];
 table_opts(recurrence_exception) ->

test/unit/logic_booking_tests.erl

@@ -2,7 +2,7 @@
 -include_lib("eunit/include/eunit.hrl").
 -include("records.hrl").
 
--define(TABLES, [user, calendar, event, booking, admin]).
+-define(TABLES, [user, calendar, event, booking, admin, subscription, calendar_specialist]).
 
 setup() ->
   eh_test_support:start_mnesia(),
@@ -19,14 +19,17 @@ logic_booking_test_() ->
     fun setup/0,
     fun cleanup/1,
     [
-      {"Create booking returns pending", fun test_create_booking_pending/0},
+      {"Create booking auto confirms", fun test_create_booking_auto/0},
+      {"Create booking manual pending", fun test_create_booking_pending/0},
+      {"Create booking personal denied", fun test_booking_personal_denied/0},
       {"Create duplicate booking", fun test_create_duplicate_booking/0},
       {"Create booking for missing event", fun test_booking_missing_event/0},
       {"Create booking when event is full", fun test_booking_event_full/0},
-      {"Pending bookings do not fill capacity", fun test_pending_does_not_fill/0},
+      {"Pending bookings fill capacity", fun test_pending_fills_capacity/0},
       {"Confirm booking", fun test_confirm_booking/0},
       {"Confirm booking as booker denied", fun test_confirm_booker_denied/0},
       {"Confirm booking as stranger denied", fun test_confirm_stranger_denied/0},
+      {"Confirm booking as specialist", fun test_confirm_booking_specialist/0},
       {"Confirm booking as admin", fun test_confirm_booking_admin/0},
       {"Decline booking by owner", fun test_decline_booking/0},
       {"Decline booking as booker denied", fun test_decline_booker_denied/0},
@@ -59,8 +62,17 @@ create_test_admin() ->
   Admin = eh_test_support:seed_admin(#{id => AdminId, email => <<AdminId/binary, "@admin.test">>}),
   Admin#admin.id.
 
+ensure_subscription(OwnerId) ->
+  {ok, _} = core_subscription:create(OwnerId, monthly, true),
+  ok.
+
 create_test_calendar(OwnerId, Confirmation) ->
-  {ok, Calendar} = core_calendar:create(OwnerId, <<"Test Calendar">>, <<"">>, Confirmation),
+  ensure_subscription(OwnerId),
+  {ok, Calendar} = core_calendar:create(OwnerId, <<"Test Calendar">>, <<"">>, Confirmation, commercial),
+  Calendar#calendar.id.
+
+create_personal_calendar(OwnerId) ->
+  {ok, Calendar} = core_calendar:create(OwnerId, <<"Personal">>, <<"">>, manual, personal),
   Calendar#calendar.id.
 
 create_test_event(CalendarId) ->
@@ -75,18 +87,34 @@ create_test_event_with_capacity(CalendarId, Capacity) ->
   Updated#event.id.
 
 %% Тесты
-test_create_booking_pending() ->
+test_create_booking_auto() ->
   OwnerId = create_test_user(user),
   ParticipantId = create_test_user(user),
   CalendarId = create_test_calendar(OwnerId, auto),
   EventId = create_test_event(CalendarId),
 
+  {ok, Booking} = logic_booking:create_booking(ParticipantId, EventId),
+  ?assertEqual(confirmed, Booking#booking.status).
+
+test_create_booking_pending() ->
+  OwnerId = create_test_user(user),
+  ParticipantId = create_test_user(user),
+  CalendarId = create_test_calendar(OwnerId, manual),
+  EventId = create_test_event(CalendarId),
+
   {ok, Booking} = logic_booking:create_booking(ParticipantId, EventId),
   ?assertEqual(pending, Booking#booking.status),
 
   {ok, Stored} = core_booking:get_by_id(Booking#booking.id),
   ?assertEqual(pending, Stored#booking.status).
 
+test_booking_personal_denied() ->
+  OwnerId = create_test_user(user),
+  ParticipantId = create_test_user(user),
+  CalendarId = create_personal_calendar(OwnerId),
+  EventId = create_test_event(CalendarId),
+  {error, personal_calendar} = logic_booking:create_booking(ParticipantId, EventId).
+
 test_create_duplicate_booking() ->
   OwnerId = create_test_user(user),
   ParticipantId = create_test_user(user),
@@ -111,7 +139,7 @@ test_booking_event_full() ->
   {ok, _} = logic_booking:confirm_booking(OwnerId, B1#booking.id, confirm),
   {error, full} = logic_booking:create_booking(Participant2Id, EventId).
 
-test_pending_does_not_fill() ->
+test_pending_fills_capacity() ->
   OwnerId = create_test_user(user),
   Participant1Id = create_test_user(user),
   Participant2Id = create_test_user(user),
@@ -119,8 +147,7 @@ test_pending_does_not_fill() ->
   EventId = create_test_event_with_capacity(CalendarId, 1),
 
   {ok, _} = logic_booking:create_booking(Participant1Id, EventId),
-  {ok, B2} = logic_booking:create_booking(Participant2Id, EventId),
-  ?assertEqual(pending, B2#booking.status).
+  {error, full} = logic_booking:create_booking(Participant2Id, EventId).
 
 test_confirm_booking() ->
   OwnerId = create_test_user(user),
@@ -151,6 +178,19 @@ test_confirm_stranger_denied() ->
   {ok, Booking} = logic_booking:create_booking(ParticipantId, EventId),
   {error, access_denied} = logic_booking:confirm_booking(StrangerId, Booking#booking.id, confirm).
 
+test_confirm_booking_specialist() ->
+  OwnerId = create_test_user(user),
+  SpecId = create_test_user(user),
+  ParticipantId = create_test_user(user),
+  CalendarId = create_test_calendar(OwnerId, manual),
+  {ok, _} = core_calendar_specialist:create(CalendarId, SpecId, <<"Spec">>, []),
+  EventId = create_test_event(CalendarId),
+  {ok, _} = core_event:update(EventId, [{specialist_id, SpecId}]),
+
+  {ok, Booking} = logic_booking:create_booking(ParticipantId, EventId),
+  {ok, Confirmed} = logic_booking:confirm_booking(SpecId, Booking#booking.id, confirm),
+  ?assertEqual(confirmed, Confirmed#booking.status).
+
 test_confirm_booking_admin() ->
   OwnerId = create_test_user(user),
   ParticipantId = create_test_user(user),

test/unit/logic_calendar_tests.erl

@@ -12,9 +12,14 @@ setup() ->
     {attributes, record_info(fields, calendar)},
     {ram_copies, [node()]}
   ]),
+  mnesia:create_table(subscription, [
+    {attributes, record_info(fields, subscription)},
+    {ram_copies, [node()]}
+  ]),
   ok.
 
 cleanup(_) ->
+  mnesia:delete_table(subscription),
   mnesia:delete_table(calendar),
   mnesia:delete_table(user),
   mnesia:stop(),

test/unit/logic_search_tests.erl

@@ -2,7 +2,7 @@
 -include_lib("eunit/include/eunit.hrl").
 -include("records.hrl").
 
--define(TABLES, [user, calendar, event]).
+-define(TABLES, [user, calendar, event, subscription]).
 
 setup() ->
   eh_test_support:start_mnesia(),
@@ -50,6 +50,12 @@ create_test_user(Role) ->
   UserId.
 
 create_test_calendar(OwnerId, Type, Tags) ->
+  case Type of
+    commercial ->
+      _ = core_subscription:create(OwnerId, monthly, true);
+    _ ->
+      ok
+  end,
   {ok, Calendar} = core_calendar:create(OwnerId, <<"Test Calendar">>, <<"Description">>, manual),
   core_calendar:update(Calendar#calendar.id, [{type, Type}, {tags, Tags}]),
   {ok, Updated} = core_calendar:get_by_id(Calendar#calendar.id),

test/unit/migration_engine_tests.erl

@@ -11,7 +11,8 @@
   "20260717180000_stats_counters",
   "20260717190000_admin_stats_indexes",
   "20260719210000_review_vote",
-  "20260720210000_calendar_follow"
+  "20260720210000_calendar_follow",
+  "20260722150000_calendar_specialist_id"
 ]).
 
 setup() ->

test/unit/stats_tops_tests.erl

@@ -45,7 +45,10 @@ test_event_rating_top() ->
   ok = mnesia:dirty_write(make_event(<<"e_low">>, 2.0, Now)),
   ok = mnesia:dirty_write(make_event(<<"e_high">>, 5.0, Now)),
   ok = mnesia:dirty_write(make_event(<<"e_mid">>, 3.5, Now)),
-  wait_top_event(<<"e_high">>, 30),
+  wait_until(fun() ->
+    [E#event.id || E <- core_event:get_top_events_by_rating(2)]
+      =:= [<<"e_high">>, <<"e_mid">>]
+  end, 40),
   Top = core_event:get_top_events_by_rating(2),
   Ids = [E#event.id || E <- Top],
   ?assertEqual([<<"e_high">>, <<"e_mid">>], Ids).