Stage 8

2026-04-21 16:35:05 +03:00
parent a4a7daa5e0
commit 3b73439b49
11 changed files with 981 additions and 2 deletions
--- a/test/core_subscription_tests.erl
+++ b/test/core_subscription_tests.erl
@@ -0,0 +1,90 @@
+-module(core_subscription_tests).
+-include_lib("eunit/include/eunit.hrl").
+-include("records.hrl").
+
+setup() ->
+  mnesia:start(),
+  mnesia:create_table(subscription, [
+    {attributes, record_info(fields, subscription)},
+    {ram_copies, [node()]}
+  ]),
+  mnesia:create_table(calendar, [
+    {attributes, record_info(fields, calendar)},
+    {ram_copies, [node()]}
+  ]),
+  ok.
+
+cleanup(_) ->
+  mnesia:delete_table(calendar),
+  mnesia:delete_table(subscription),
+  mnesia:stop(),
+  ok.
+
+core_subscription_test_() ->
+  {foreach,
+    fun setup/0,
+    fun cleanup/1,
+    [
+      {"Create trial subscription", fun test_create_trial/0},
+      {"Create paid subscription", fun test_create_paid/0},
+      {"Get active by user", fun test_get_active_by_user/0},
+      {"List by user", fun test_list_by_user/0},
+      {"Update status", fun test_update_status/0},
+      {"Check expired", fun test_check_expired/0}
+    ]}.
+
+test_create_trial() ->
+  UserId = <<"user123">>,
+  {ok, Sub} = core_subscription:create(UserId, trial, false),
+
+  ?assertEqual(UserId, Sub#subscription.user_id),
+  ?assertEqual(trial, Sub#subscription.plan),
+  ?assertEqual(false, Sub#subscription.trial_used),
+  ?assertEqual(active, Sub#subscription.status).
+
+test_create_paid() ->
+  UserId = <<"user123">>,
+  {ok, Sub} = core_subscription:create(UserId, monthly, true),
+
+  ?assertEqual(UserId, Sub#subscription.user_id),
+  ?assertEqual(monthly, Sub#subscription.plan),
+  ?assertEqual(true, Sub#subscription.trial_used),
+  ?assertEqual(active, Sub#subscription.status).
+
+test_get_active_by_user() ->
+  UserId = <<"user123">>,
+  {ok, Sub1} = core_subscription:create(UserId, trial, false),
+  {ok, Sub2} = core_subscription:create(UserId, monthly, true),
+
+  core_subscription:update_status(Sub1#subscription.id, expired),
+
+  {ok, Active} = core_subscription:get_active_by_user(UserId),
+  ?assertEqual(Sub2#subscription.id, Active#subscription.id).
+
+test_list_by_user() ->
+  UserId = <<"user123">>,
+  {ok, _} = core_subscription:create(UserId, trial, false),
+  {ok, _} = core_subscription:create(UserId, monthly, true),
+
+  {ok, Subs} = core_subscription:list_by_user(UserId),
+  ?assertEqual(2, length(Subs)).
+
+test_update_status() ->
+  UserId = <<"user123">>,
+  {ok, Sub} = core_subscription:create(UserId, trial, false),
+
+  {ok, Cancelled} = core_subscription:update_status(Sub#subscription.id, cancelled),
+  ?assertEqual(cancelled, Cancelled#subscription.status).
+
+test_check_expired() ->
+  UserId = <<"user123">>,
+  % Создаём подписку с истёкшим сроком
+  {ok, Sub} = core_subscription:create(UserId, trial, false),
+  % Принудительно устанавливаем expires_at в прошлое
+  Past = {{2020, 1, 1}, {0, 0, 0}},
+  mnesia:dirty_write(Sub#subscription{expires_at = Past}),
+
+  core_subscription:check_expired(),
+
+  {ok, Updated} = core_subscription:get_by_id(Sub#subscription.id),
+  ?assertEqual(expired, Updated#subscription.status).
--- a/test/logic_subscription_tests.erl
+++ b/test/logic_subscription_tests.erl
@@ -0,0 +1,120 @@
+-module(logic_subscription_tests).
+-include_lib("eunit/include/eunit.hrl").
+-include("records.hrl").
+
+logic_subscription_test_() ->
+  {foreach,
+    fun setup/0,
+    fun cleanup/1,
+    [
+      {"Start trial", fun test_start_trial/0},
+      {"Start trial duplicate", fun test_start_trial_duplicate/0},
+      {"Activate subscription (no trial)", fun test_activate_subscription_no_trial/0},
+      {"Activate subscription (after trial)", fun test_activate_subscription_after_trial/0},
+      {"Check subscription - free", fun test_check_free/0},
+      {"Check subscription - trial", fun test_check_trial/0},
+      {"Check subscription - paid", fun test_check_paid/0},
+      {"Can create commercial - free (auto-trial)", fun test_can_create_commercial_free/0},
+      {"Can create commercial - trial", fun test_can_create_commercial_trial/0},
+      {"Can create commercial - paid", fun test_can_create_commercial_paid/0},
+      {"Get user subscription - free", fun test_get_user_subscription_free/0},
+      {"Get user subscription - trial", fun test_get_user_subscription_trial/0},
+      {"Get user subscription - paid", fun test_get_user_subscription_paid/0}
+    ]}.
+
+setup() ->
+  mnesia:start(),
+  mnesia:create_table(user, [{attributes, record_info(fields, user)}, {ram_copies, [node()]}]),
+  mnesia:create_table(calendar, [{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(),
+  ok.
+
+create_test_user() ->
+  UserId = base64:encode(crypto:strong_rand_bytes(16), #{mode => urlsafe, padding => false}),
+  User = #user{
+    id = UserId,
+    email = <<UserId/binary, "@test.com">>,
+    password_hash = <<"hash">>,
+    role = user,
+    status = active,
+    created_at = calendar:universal_time(),
+    updated_at = calendar:universal_time()
+  },
+  mnesia:dirty_write(User),
+  UserId.
+
+%% ============ Тесты ============
+
+test_start_trial() ->
+  UserId = create_test_user(),
+  {ok, Sub} = logic_subscription:start_trial(UserId),
+  ?assertEqual(trial, Sub#subscription.plan),
+  ?assertEqual(true, Sub#subscription.trial_used).
+
+test_start_trial_duplicate() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:start_trial(UserId),
+  {error, already_has_subscription} = logic_subscription:start_trial(UserId).
+
+test_activate_subscription_no_trial() ->
+  UserId = create_test_user(),
+  {ok, Sub} = logic_subscription:activate_subscription(UserId, monthly, #{card => "4242"}),
+  ?assertEqual(monthly, Sub#subscription.plan),
+  ?assertEqual(false, Sub#subscription.trial_used).
+
+test_activate_subscription_after_trial() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:start_trial(UserId),
+  {ok, Sub} = logic_subscription:activate_subscription(UserId, monthly, #{card => "4242"}),
+  ?assertEqual(monthly, Sub#subscription.plan),
+  ?assertEqual(true, Sub#subscription.trial_used).
+
+test_check_free() ->
+  UserId = create_test_user(),
+  {ok, free, free} = logic_subscription:check_user_subscription(UserId).
+
+test_check_trial() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:start_trial(UserId),
+  {ok, active, trial} = logic_subscription:check_user_subscription(UserId).
+
+test_check_paid() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:activate_subscription(UserId, monthly, #{card => "4242"}),
+  {ok, active, monthly} = logic_subscription:check_user_subscription(UserId).
+
+test_can_create_commercial_free() ->
+  % Новый пользователь автоматически получает пробный период при проверке
+  UserId = create_test_user(),
+  ?assert(logic_subscription:can_create_commercial_calendar(UserId)).
+
+test_can_create_commercial_trial() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:start_trial(UserId),
+  ?assert(logic_subscription:can_create_commercial_calendar(UserId)).
+
+test_can_create_commercial_paid() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:activate_subscription(UserId, monthly, #{card => "4242"}),
+  ?assert(logic_subscription:can_create_commercial_calendar(UserId)).
+
+test_get_user_subscription_free() ->
+  UserId = create_test_user(),
+  {ok, #{status := free}} = logic_subscription:get_user_subscription(UserId).
+
+test_get_user_subscription_trial() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:start_trial(UserId),
+  {ok, #{status := active, plan := trial}} = logic_subscription:get_user_subscription(UserId).
+
+test_get_user_subscription_paid() ->
+  UserId = create_test_user(),
+  {ok, _} = logic_subscription:activate_subscription(UserId, annual, #{card => "4242"}),
+  {ok, #{status := active, plan := annual}} = logic_subscription:get_user_subscription(UserId).
--- a/test/scripts/test_subscription_api.sh
+++ b/test/scripts/test_subscription_api.sh
@@ -0,0 +1,217 @@
+#!/bin/bash
+
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+BLUE='\033[0;34m'
+NC='\033[0m'
+
+BASE_URL="http://localhost:8080"
+
+log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
+log_success() { echo -e "${GREEN}[SUCCESS]${NC} $1"; }
+log_error() { echo -e "${RED}[ERROR]${NC} $1"; }
+
+extract_json() {
+    echo "$1" | grep -o "\"$2\":\"[^\"]*\"" | head -1 | sed "s/\"$2\":\"//;s/\"$//"
+}
+
+http_post() {
+    local url=$1; local data=$2; local token=$3
+    if [ -n "$token" ]; then
+        curl -s -X POST "$url" -H "Content-Type: application/json" -H "Authorization: Bearer $token" -d "$data"
+    else
+        curl -s -X POST "$url" -H "Content-Type: application/json" -d "$data"
+    fi
+}
+
+http_get() {
+    local url=$1; local token=$2
+    if [ -n "$token" ]; then
+        curl -s -X GET "$url" -H "Authorization: Bearer $token"
+    else
+        curl -s -X GET "$url"
+    fi
+}
+
+echo "============================================================"
+echo "          EVENTHUB SUBSCRIPTION API TEST SCRIPT"
+echo "============================================================"
+echo ""
+
+log_info "Checking if server is running..."
+if ! curl -s "$BASE_URL/health" | grep -q "ok"; then
+    log_error "Server is not running"
+    exit 1
+fi
+log_success "Server is running"
+
+echo ""
+log_info "============================================================"
+log_info "STEP 1: Create test users"
+log_info "============================================================"
+
+# Пользователь 1 (будет использовать пробный период через commercial календарь)
+USER1_EMAIL="sub_user1_$(date +%s)@example.com"
+USER1_PASSWORD="user123"
+
+log_info "Creating user 1..."
+response=$(http_post "$BASE_URL/v1/register" "{\"email\":\"$USER1_EMAIL\",\"password\":\"$USER1_PASSWORD\"}" "")
+USER1_TOKEN=$(extract_json "$response" "token")
+USER1_ID=$(extract_json "$response" "id")
+log_success "User 1 created"
+
+# Пользователь 2 (для проверки, что пробный период используется один раз)
+USER2_EMAIL="sub_user2_$(date +%s)@example.com"
+USER2_PASSWORD="user123"
+
+log_info "Creating user 2..."
+response=$(http_post "$BASE_URL/v1/register" "{\"email\":\"$USER2_EMAIL\",\"password\":\"$USER2_PASSWORD\"}" "")
+USER2_TOKEN=$(extract_json "$response" "token")
+USER2_ID=$(extract_json "$response" "id")
+log_success "User 2 created"
+
+echo ""
+log_info "============================================================"
+log_info "TEST 1: Get subscription (free)"
+log_info "============================================================"
+
+response=$(http_get "$BASE_URL/v1/subscription" "$USER1_TOKEN")
+if echo "$response" | grep -q "free"; then
+    log_success "User has free subscription"
+else
+    log_error "Expected free subscription: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 2: Create personal calendar (should work with free)"
+log_info "============================================================"
+
+response=$(http_post "$BASE_URL/v1/calendars" "{\"title\":\"Personal Calendar\",\"type\":\"personal\"}" "$USER1_TOKEN")
+PERSONAL_CALENDAR_ID=$(extract_json "$response" "id")
+
+if [ -n "$PERSONAL_CALENDAR_ID" ]; then
+    log_success "Personal calendar created (free subscription allows this)"
+else
+    log_error "Failed to create personal calendar: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 3: Create commercial calendar (auto-activates trial)"
+log_info "============================================================"
+
+log_info "User 1 creating commercial calendar (should auto-start trial)..."
+response=$(http_post "$BASE_URL/v1/calendars" "{\"title\":\"Commercial Calendar\",\"type\":\"commercial\"}" "$USER1_TOKEN")
+COMMERCIAL_CALENDAR_ID=$(extract_json "$response" "id")
+
+if [ -n "$COMMERCIAL_CALENDAR_ID" ]; then
+    log_success "Commercial calendar created - trial auto-activated"
+else
+    log_error "Failed to create commercial calendar: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 4: Get subscription (should be trial now)"
+log_info "============================================================"
+
+response=$(http_get "$BASE_URL/v1/subscription" "$USER1_TOKEN")
+if echo "$response" | grep -q "trial"; then
+    log_success "User now has trial subscription"
+else
+    log_error "Expected trial subscription: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 5: Create second commercial calendar (should still work)"
+log_info "============================================================"
+
+response=$(http_post "$BASE_URL/v1/calendars" "{\"title\":\"Second Commercial\",\"type\":\"commercial\"}" "$USER1_TOKEN")
+SECOND_COMMERCIAL_ID=$(extract_json "$response" "id")
+
+if [ -n "$SECOND_COMMERCIAL_ID" ]; then
+    log_success "Second commercial calendar created"
+else
+    log_error "Failed to create second commercial calendar: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 6: User 2 creates commercial calendar (gets trial)"
+log_info "============================================================"
+
+log_info "User 2 creating commercial calendar..."
+response=$(http_post "$BASE_URL/v1/calendars" "{\"title\":\"User2 Commercial\",\"type\":\"commercial\"}" "$USER2_TOKEN")
+USER2_CALENDAR_ID=$(extract_json "$response" "id")
+
+if [ -n "$USER2_CALENDAR_ID" ]; then
+    log_success "User 2 commercial calendar created - trial activated"
+else
+    log_error "User 2 failed: $response"
+fi
+
+response=$(http_get "$BASE_URL/v1/subscription" "$USER2_TOKEN")
+if echo "$response" | grep -q "trial"; then
+    log_success "User 2 has trial subscription"
+else
+    log_error "User 2 subscription: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 7: Activate paid subscription"
+log_info "============================================================"
+
+log_info "User 1 activating paid subscription..."
+response=$(http_post "$BASE_URL/v1/subscription" "{\"action\":\"activate\",\"plan\":\"monthly\",\"payment_info\":{\"card\":\"4242424242424242\"}}" "$USER1_TOKEN")
+if echo "$response" | grep -q "monthly"; then
+    log_success "Paid subscription activated"
+else
+    log_error "Failed to activate: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 8: Get subscription (should be active paid)"
+log_info "============================================================"
+
+response=$(http_get "$BASE_URL/v1/subscription" "$USER1_TOKEN")
+if echo "$response" | grep -q "active" && echo "$response" | grep -q "monthly"; then
+    log_success "User has active paid subscription"
+else
+    log_error "Expected active paid subscription: $response"
+fi
+
+echo ""
+log_info "============================================================"
+log_info "TEST 9: User 3 (free) tries to use already consumed trial"
+log_info "============================================================"
+
+# Создаём пользователя 3, который сначала использует trial, потом отменяет подписку
+USER3_EMAIL="sub_user3_$(date +%s)@example.com"
+USER3_PASSWORD="user123"
+
+log_info "Creating user 3..."
+response=$(http_post "$BASE_URL/v1/register" "{\"email\":\"$USER3_EMAIL\",\"password\":\"$USER3_PASSWORD\"}" "")
+USER3_TOKEN=$(extract_json "$response" "token")
+log_success "User 3 created"
+
+log_info "User 3 creating commercial calendar (uses trial)..."
+response=$(http_post "$BASE_URL/v1/calendars" "{\"title\":\"User3 Commercial\",\"type\":\"commercial\"}" "$USER3_TOKEN")
+USER3_CALENDAR_ID=$(extract_json "$response" "id")
+log_success "Commercial calendar created"
+
+log_info "Simulating trial expiration (requires admin or time travel - skipped)"
+
+echo ""
+echo "============================================================"
+log_success "SUBSCRIPTION API TESTS COMPLETED!"
+echo "============================================================"
+echo ""
+echo "Summary:"
+echo "  User 1: $USER1_EMAIL (trial -> paid)"
+echo "  User 2: $USER2_EMAIL (trial)"
+echo "  User 3: $USER3_EMAIL (trial)"
+echo ""