Refactor LibraryDebugCollector and TelegramApi for improved data handling and clarity

Updated the LibraryDebugCollector to introduce a new CompositionEntry record for better organization of group composition data. Refactored the composition logic to utilize this new structure, enhancing readability and maintainability. In the TelegramApi, replaced inline allowed_updates array with a static readonly field to reduce redundancy and improve clarity in API call configurations. Adjusted related tests to ensure proper handling of message assertions, enhancing overall test reliability.
2026-07-31 10:07:28 +03:00
parent 08f56e14c2
commit 4a2e6ac7f1
8 changed files with 991 additions and 18 deletions
@@ -0,0 +1,175 @@
+using NSubstitute;
+using TeleWave.Application.Common.Interfaces;
+using TeleWave.Application.Notifications.Link;
+using TeleWave.Application.Notifications.Settings;
+using TeleWave.Application.Notifications.Subscribers;
+using TeleWave.Application.Tests.Support;
+using TeleWave.Domain.Broadcast;
+using TeleWave.Domain.Notifications;
+using Xunit;
+
+namespace TeleWave.Application.Tests.Notifications;
+
+/// <summary>
+/// Что видит и делает админ вокруг бота: выдача ссылки зрителю и список подписчиков. Ссылка —
+/// единственная дверь в подписку, поэтому проверяем, что старые коды при выдаче новой гаснут.
+/// </summary>
+public class TelegramAdminTests
+{
+    private static readonly DateTimeOffset T0 = new(2026, 8, 1, 12, 0, 0, TimeSpan.Zero);
+
+    [Fact]
+    public async Task Link_BurnsPreviousCodesOfTheSameUser()
+    {
+        var fixture = new TestDb();
+        var userId = Guid.NewGuid();
+        var stale = TelegramLinkCode.Issue("old", userId, T0);
+        var foreign = TelegramLinkCode.Issue("alien", Guid.NewGuid(), T0);
+
+        await using (var seed = fixture.New())
+        {
+            seed.TelegramSettings.Add(Configured());
+            seed.TelegramLinkCodes.AddRange(stale, foreign);
+            await seed.SaveChangesAsync(CancellationToken.None);
+        }
+
+        await using var db = fixture.New();
+        var result = await new IssueTelegramLinkCommandHandler(
+            db,
+            new TelegramSettingsService(db),
+            CurrentUser(userId)
+        ).Handle(new IssueTelegramLinkCommand(), CancellationToken.None);
+        await db.SaveChangesAsync(CancellationToken.None);
+
+        Assert.True(result.IsSuccess);
+        Assert.StartsWith("https://t.me/telewave_bot?start=", result.Value.Url);
+
+        await using var check = fixture.New();
+        var codes = check.TelegramLinkCodes.ToList();
+        // Своя старая ссылка гасится — две живые двери в один аккаунт не нужны; чужая не трогается.
+        Assert.NotNull(codes.Single(c => c.Code == "old").UsedAt);
+        Assert.Null(codes.Single(c => c.Code == "alien").UsedAt);
+        var issued = codes.Single(c => c.UserId == userId && c.UsedAt is null);
+        Assert.EndsWith(issued.Code, result.Value.Url);
+        Assert.Equal(TelegramLinkCode.Lifetime, issued.ExpiresAt - issued.CreatedAt);
+    }
+
+    [Fact]
+    public async Task Link_WithoutConfiguredBot_IsRefused()
+    {
+        var fixture = new TestDb();
+
+        await using var db = fixture.New();
+        var result = await new IssueTelegramLinkCommandHandler(
+            db,
+            new TelegramSettingsService(db),
+            CurrentUser(Guid.NewGuid())
+        ).Handle(new IssueTelegramLinkCommand(), CancellationToken.None);
+
+        // Ссылку не из чего собрать: имя бота приходит только после успешной проверки токена.
+        Assert.False(result.IsSuccess);
+        Assert.Equal("Telegram.NotConfigured", result.Error.Code);
+        Assert.Empty(db.TelegramLinkCodes);
+    }
+
+    [Fact]
+    public async Task Link_WithoutSignedInViewer_IsRefused()
+    {
+        var fixture = new TestDb();
+
+        await using (var seed = fixture.New())
+        {
+            seed.TelegramSettings.Add(Configured());
+            await seed.SaveChangesAsync(CancellationToken.None);
+        }
+
+        await using var db = fixture.New();
+        var anonymous = Substitute.For<ICurrentUser>();
+        anonymous.UserId.Returns((Guid?)null);
+
+        var result = await new IssueTelegramLinkCommandHandler(
+            db,
+            new TelegramSettingsService(db),
+            anonymous
+        ).Handle(new IssueTelegramLinkCommand(), CancellationToken.None);
+
+        Assert.False(result.IsSuccess);
+        Assert.Empty(db.TelegramLinkCodes);
+    }
+
+    [Fact]
+    public async Task Subscribers_ShowChannelsAndUserNamesNewestFirst()
+    {
+        var fixture = new TestDb();
+        var channel = Channel.Create("Мультреалити", "mult", T0);
+        var userId = Guid.NewGuid();
+
+        var active = TelegramSubscriber.Create(100, userId, "viewer", T0);
+        active.Touch(T0.AddHours(2));
+        active.Toggle(channel.Id, TelegramNotificationKind.ShowStart);
+        active.Toggle(channel.Id, TelegramNotificationKind.UpNext);
+        // Канал удалили, а подписка осталась висеть — это должно быть видно, а не пропасть.
+        active.Toggle(Guid.NewGuid(), TelegramNotificationKind.ShowStart);
+
+        var older = TelegramSubscriber.Create(101, Guid.NewGuid(), "gone", T0.AddHours(-1));
+
+        await using (var seed = fixture.New())
+        {
+            seed.Channels.Add(channel);
+            seed.TelegramSubscribers.AddRange(active, older);
+            await seed.SaveChangesAsync(CancellationToken.None);
+        }
+
+        var identity = Substitute.For<IIdentityService>();
+        identity
+            .GetProfileAsync(userId, Arg.Any<CancellationToken>())
+            .Returns(new CurrentUserProfile(userId, "leonid", "user", false));
+
+        await using var db = fixture.New();
+        var result = await new ListTelegramSubscribersQueryHandler(db, identity).Handle(
+            new ListTelegramSubscribersQuery(),
+            CancellationToken.None
+        );
+
+        var subscribers = result.Value;
+        Assert.Equal([100L, 101L], subscribers.Select(s => s.ChatId).ToArray());
+        Assert.Equal("leonid", subscribers[0].UserName);
+        // Учётку без профиля показываем как есть: чат жив, а пользователя уже нет.
+        Assert.Null(subscribers[1].UserName);
+
+        var live = subscribers[0].Channels.Single(c => c.ChannelId == channel.Id);
+        Assert.Equal("Мультреалити", live.ChannelName);
+        Assert.Equal(["ShowStart", "UpNext"], live.Kinds);
+        Assert.Equal(
+            "—",
+            subscribers[0].Channels.Single(c => c.ChannelId != channel.Id).ChannelName
+        );
+    }
+
+    private static TelegramSettings Configured()
+    {
+        var settings = TelegramSettings.CreateDefault();
+        settings.Update(
+            new TelegramSettingsUpdate(
+                true,
+                TelegramTransport.Polling,
+                "123:ABC",
+                null,
+                TelegramProxyKind.None,
+                null,
+                null,
+                null,
+                null
+            )
+        );
+        settings.SetBotUsername("telewave_bot");
+        return settings;
+    }
+
+    private static ICurrentUser CurrentUser(Guid userId)
+    {
+        var currentUser = Substitute.For<ICurrentUser>();
+        currentUser.UserId.Returns(userId);
+        return currentUser;
+    }
+}
@@ -77,7 +77,9 @@ public class TelegramBotTests
            .SendMessageAsync(
                Arg.Any<TelegramSettings>(),
                100,
-                Arg.Is<string>(s => s.Contains("код")),
+                // Предикат NSubstitute принимает аргумент как допускающий null — само сообщение
+                // объявлено обязательным, потому и разыменовываем без проверки.
+                Arg.Is<string>(s => s!.Contains("код")),
                null,
                Arg.Any<CancellationToken>()
            );
@@ -136,7 +136,9 @@ public class TelegramNotifierTests
            .SendMessageAsync(
                Arg.Any<TelegramSettings>(),
                100,
-                Arg.Is<string>(s => s.Contains("Футурама")),
+                // Предикат NSubstitute принимает аргумент как допускающий null — само сообщение
+                // объявлено обязательным, потому и разыменовываем без проверки.
+                Arg.Is<string>(s => s!.Contains("Футурама")),
                null,
                Arg.Any<CancellationToken>()
            );
@@ -0,0 +1,294 @@
+using NSubstitute;
+using TeleWave.Application.Common.Interfaces;
+using TeleWave.Application.Notifications;
+using TeleWave.Application.Notifications.Settings;
+using TeleWave.Application.Tests.Support;
+using TeleWave.Domain.Notifications;
+using TeleWave.Infrastructure.Persistence;
+using Xunit;
+
+namespace TeleWave.Application.Tests.Notifications;
+
+/// <summary>
+/// Настройки бота: сохранение проверяет связь делом и приводит режим доставки в соответствие.
+/// Главное свойство — админка никогда не видит секретов, только признак «задан».
+/// </summary>
+public class TelegramSettingsTests
+{
+    private const string Token = "123:ABC";
+
+    [Fact]
+    public async Task Update_WithPolling_RemovesWebhookAndRemembersBotName()
+    {
+        var fixture = new TestDb();
+        var api = Api("telewave_bot");
+
+        await using var db = fixture.New();
+        var result = await Handler(db, api).Handle(Command(token: Token), CancellationToken.None);
+
+        Assert.True(result.IsSuccess);
+        Assert.Equal("telewave_bot", result.Value.BotUsername);
+        Assert.Equal("https://t.me/telewave_bot", result.Value.BotLink);
+        Assert.True(result.Value.HasToken);
+        Assert.NotNull(result.Value.LastContactAt);
+        Assert.Null(result.Value.LastError);
+        // Пока стоит вебхук, Telegram не отдаёт обновления опросом — снимаем его явно.
+        await api.Received(1)
+            .DeleteWebhookAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>());
+        await api.DidNotReceive()
+            .SetWebhookAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>());
+    }
+
+    [Fact]
+    public async Task Update_WithWebhook_GeneratesSecretOnce()
+    {
+        var fixture = new TestDb();
+        var api = Api("telewave_bot");
+
+        await using var db = fixture.New();
+        var handler = Handler(db, api);
+        var command = Command(
+            token: Token,
+            transport: TelegramTransport.Webhook,
+            webhookUrl: "https://tv.example.com/api/telegram/webhook"
+        );
+
+        Assert.True((await handler.Handle(command, CancellationToken.None)).IsSuccess);
+        await db.SaveChangesAsync(CancellationToken.None);
+        var secret = db.TelegramSettings.Single().WebhookSecret;
+        Assert.False(string.IsNullOrWhiteSpace(secret));
+
+        // Повторное сохранение секрет не меняет: он уже стоит на стороне Telegram, и смена
+        // на каждое нажатие «Сохранить» рвала бы приём обновлений до следующего setWebhook.
+        Assert.True((await handler.Handle(command, CancellationToken.None)).IsSuccess);
+        Assert.Equal(secret, db.TelegramSettings.Single().WebhookSecret);
+        await api.Received(2)
+            .SetWebhookAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>());
+    }
+
+    [Fact]
+    public async Task Update_WhenTelegramFails_KeepsWhatAdminTypedAndReportsWhy()
+    {
+        var fixture = new TestDb();
+        var api = Substitute.For<ITelegramApi>();
+        api.GetMeAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>())
+            .Returns<TelegramBotInfo?>(_ => throw new InvalidOperationException("Unauthorized"));
+
+        await using (var db = fixture.New())
+        {
+            var result = await Handler(db, api)
+                .Handle(Command(token: Token, proxyHost: "127.0.0.1"), CancellationToken.None);
+
+            Assert.False(result.IsSuccess);
+            Assert.Equal("Telegram.Unreachable", result.Error.Code);
+        }
+
+        // Настройки при этом сохранены: иначе админ набирал бы форму заново из-за одной опечатки.
+        await using var check = fixture.New();
+        var stored = check.TelegramSettings.Single();
+        Assert.Equal("127.0.0.1", stored.ProxyHost);
+        Assert.Equal("Unauthorized", stored.LastError);
+    }
+
+    [Fact]
+    public async Task Update_WithoutToken_DoesNotEnableTheBot()
+    {
+        var fixture = new TestDb();
+        var api = Api("telewave_bot");
+
+        await using var db = fixture.New();
+        var result = await Handler(db, api).Handle(Command(token: null), CancellationToken.None);
+
+        Assert.False(result.IsSuccess);
+        Assert.Equal("Telegram.NotConfigured", result.Error.Code);
+        await api.DidNotReceive()
+            .GetMeAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>());
+    }
+
+    [Theory]
+    [InlineData(
+        TelegramTransport.Webhook,
+        null,
+        TelegramProxyKind.None,
+        null,
+        "Telegram.WebhookUrlRequired"
+    )]
+    [InlineData(
+        TelegramTransport.Polling,
+        null,
+        TelegramProxyKind.Socks5,
+        null,
+        "Telegram.ProxyHostRequired"
+    )]
+    public async Task Update_WithHalfFilledMode_IsRejectedBeforeAnyCall(
+        TelegramTransport transport,
+        string? webhookUrl,
+        TelegramProxyKind proxyKind,
+        string? proxyHost,
+        string expected
+    )
+    {
+        var fixture = new TestDb();
+        var api = Api("telewave_bot");
+
+        await using var db = fixture.New();
+        var result = await Handler(db, api)
+            .Handle(
+                Command(
+                    token: Token,
+                    transport: transport,
+                    webhookUrl: webhookUrl,
+                    proxyKind: proxyKind,
+                    proxyHost: proxyHost
+                ),
+                CancellationToken.None
+            );
+
+        Assert.False(result.IsSuccess);
+        Assert.Equal(expected, result.Error.Code);
+        await api.DidNotReceive()
+            .GetMeAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>());
+    }
+
+    [Fact]
+    public async Task Update_WhileDisabled_SavesWithoutTouchingTelegram()
+    {
+        var fixture = new TestDb();
+        var api = Api("telewave_bot");
+
+        await using var db = fixture.New();
+        var result = await Handler(db, api)
+            .Handle(Command(token: Token, isEnabled: false), CancellationToken.None);
+
+        // Выключенный бот в сеть не ходит: проверять нечего, а лишний запрос через прокси
+        // ждал бы таймаута на каждом сохранении.
+        Assert.True(result.IsSuccess);
+        Assert.False(result.Value.IsEnabled);
+        Assert.Null(result.Value.BotLink);
+        await api.DidNotReceive()
+            .GetMeAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>());
+    }
+
+    [Fact]
+    public async Task Get_WithoutRow_ReturnsDefaultsAndCreatesNothing()
+    {
+        var fixture = new TestDb();
+
+        await using var db = fixture.New();
+        var result = await new GetTelegramSettingsQueryHandler(
+            new TelegramSettingsService(db)
+        ).Handle(new GetTelegramSettingsQuery(), CancellationToken.None);
+
+        Assert.True(result.IsSuccess);
+        Assert.False(result.Value.IsEnabled);
+        Assert.False(result.Value.HasToken);
+        Assert.Equal(TelegramTransport.Polling, result.Value.Transport);
+        Assert.Equal(0, result.Value.Subscribers);
+        // Чтение не должно заводить строку: GET в базу не пишет.
+        Assert.Empty(db.TelegramSettings);
+    }
+
+    [Fact]
+    public async Task Get_CountsLiveSubscribersAndHidesSecrets()
+    {
+        var fixture = new TestDb();
+        var settings = TelegramSettings.CreateDefault();
+        settings.Update(
+            new TelegramSettingsUpdate(
+                true,
+                TelegramTransport.Polling,
+                Token,
+                null,
+                TelegramProxyKind.Socks5,
+                "127.0.0.1",
+                1080,
+                "user",
+                "secret"
+            )
+        );
+        settings.SetBotUsername("telewave_bot");
+
+        var live = TelegramSubscriber.Create(100, Guid.NewGuid(), "viewer", DateTimeOffset.UtcNow);
+        var stopped = TelegramSubscriber.Create(101, Guid.NewGuid(), "gone", DateTimeOffset.UtcNow);
+        stopped.Stop();
+
+        await using (var seed = fixture.New())
+        {
+            seed.TelegramSettings.Add(settings);
+            seed.TelegramSubscribers.AddRange(live, stopped);
+            await seed.SaveChangesAsync(CancellationToken.None);
+        }
+
+        await using var db = fixture.New();
+        var result = await new GetTelegramSettingsQueryHandler(
+            new TelegramSettingsService(db)
+        ).Handle(new GetTelegramSettingsQuery(), CancellationToken.None);
+
+        var dto = result.Value;
+        // Ушедший в «стоп» в счётчик не идёт: админ смотрит, скольким реально уходит рассылка.
+        Assert.Equal(1, dto.Subscribers);
+        Assert.True(dto.HasToken);
+        Assert.True(dto.HasProxyPassword);
+        Assert.Equal("user", dto.ProxyUsername);
+        Assert.Equal(1080, dto.ProxyPort);
+    }
+
+    [Fact]
+    public void BotLink_NeedsBothNameAndEnabledFlag()
+    {
+        var settings = TelegramSettings.CreateDefault();
+        Assert.Null(TelegramSettingsService.BotLink(settings));
+
+        settings.SetBotUsername("telewave_bot");
+        // Имя есть, но бот выключен — кнопки у зрителя быть не должно.
+        Assert.Null(TelegramSettingsService.BotLink(settings));
+
+        settings.Update(
+            new TelegramSettingsUpdate(
+                true,
+                TelegramTransport.Polling,
+                Token,
+                null,
+                TelegramProxyKind.None,
+                null,
+                null,
+                null,
+                null
+            )
+        );
+        Assert.Equal("https://t.me/telewave_bot", TelegramSettingsService.BotLink(settings));
+    }
+
+    private static ITelegramApi Api(string username)
+    {
+        var api = Substitute.For<ITelegramApi>();
+        api.GetMeAsync(Arg.Any<TelegramSettings>(), Arg.Any<CancellationToken>())
+            .Returns(new TelegramBotInfo(1, username));
+        return api;
+    }
+
+    private static UpdateTelegramSettingsCommandHandler Handler(
+        AppDbContext db,
+        ITelegramApi api
+    ) => new(db, new TelegramSettingsService(db), api);
+
+    private static UpdateTelegramSettingsCommand Command(
+        string? token,
+        bool isEnabled = true,
+        TelegramTransport transport = TelegramTransport.Polling,
+        string? webhookUrl = null,
+        TelegramProxyKind proxyKind = TelegramProxyKind.None,
+        string? proxyHost = null
+    ) =>
+        new(
+            isEnabled,
+            transport,
+            token,
+            webhookUrl,
+            proxyKind,
+            proxyHost,
+            proxyHost is null ? null : 1080,
+            null,
+            null
+        );
+}