add tests

ChatBot.Tests/Telegram/Commands/TelegramCommandProcessorTests.cs

@@ -6,6 +6,7 @@ using ChatBot.Tests.TestUtilities;
 using FluentAssertions;
 using Microsoft.Extensions.Logging;
 using Moq;
+using Telegram.Bot;
 using Telegram.Bot.Types;
 
 namespace ChatBot.Tests.Telegram.Commands;
@@ -417,8 +418,8 @@ public class TelegramCommandProcessorTests : UnitTestBase
         var username = "testuser";
         var chatType = "private";
         var chatTitle = "Test Chat";
-        var cts = new CancellationTokenSource();
-        cts.Cancel();
+        using var cts = new CancellationTokenSource();
+        await cts.CancelAsync();
 
         // Act
         var result = await _processor.ProcessMessageAsync(
@@ -488,4 +489,347 @@ public class TelegramCommandProcessorTests : UnitTestBase
         // Assert
         result.Should().NotBeNull();
     }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithReplyToBot_ShouldProcessMessage()
+    {
+        // Arrange
+        var botUser = new User
+        {
+            Id = 999,
+            Username = "testbot",
+            IsBot = true,
+        };
+
+        var botInfoServiceMock = new Mock<BotInfoService>(
+            TestDataBuilder.Mocks.CreateTelegramBotClient().Object,
+            TestDataBuilder.Mocks.CreateLoggerMock<BotInfoService>().Object
+        );
+        botInfoServiceMock.Setup(x => x.GetBotInfoAsync(It.IsAny<CancellationToken>()))
+            .ReturnsAsync(botUser);
+
+        var processor = new TelegramCommandProcessor(
+            _commandRegistry,
+            _chatService,
+            _loggerMock.Object,
+            botInfoServiceMock.Object
+        );
+
+        var replyInfo = new ReplyInfo
+        {
+            MessageId = 1,
+            UserId = 999,
+            Username = "testbot",
+        };
+
+        // Act
+        var result = await processor.ProcessMessageAsync(
+            "Test reply",
+            12345L,
+            "user",
+            "private",
+            "Test Chat",
+            replyInfo
+        );
+
+        // Assert
+        result.Should().NotBeNullOrEmpty();
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithReplyToOtherUser_ShouldReturnEmpty()
+    {
+        // Arrange
+        var botUser = new User
+        {
+            Id = 999,
+            Username = "testbot",
+            IsBot = true,
+        };
+
+        var botInfoServiceMock = new Mock<BotInfoService>(
+            TestDataBuilder.Mocks.CreateTelegramBotClient().Object,
+            TestDataBuilder.Mocks.CreateLoggerMock<BotInfoService>().Object
+        );
+        botInfoServiceMock.Setup(x => x.GetBotInfoAsync(It.IsAny<CancellationToken>()))
+            .ReturnsAsync(botUser);
+
+        var processor = new TelegramCommandProcessor(
+            _commandRegistry,
+            _chatService,
+            _loggerMock.Object,
+            botInfoServiceMock.Object
+        );
+
+        var replyInfo = new ReplyInfo
+        {
+            MessageId = 1,
+            UserId = 123,
+            Username = "otheruser",
+        };
+
+        // Act
+        var result = await processor.ProcessMessageAsync(
+            "Test reply",
+            12345L,
+            "user",
+            "private",
+            "Test Chat",
+            replyInfo
+        );
+
+        // Assert
+        result.Should().BeEmpty();
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithBotMention_ShouldProcessMessage()
+    {
+        // Arrange
+        var botUser = new User
+        {
+            Id = 999,
+            Username = "testbot",
+            IsBot = true,
+        };
+
+        var botInfoServiceMock = new Mock<BotInfoService>(
+            TestDataBuilder.Mocks.CreateTelegramBotClient().Object,
+            TestDataBuilder.Mocks.CreateLoggerMock<BotInfoService>().Object
+        );
+        botInfoServiceMock.Setup(x => x.GetBotInfoAsync(It.IsAny<CancellationToken>()))
+            .ReturnsAsync(botUser);
+
+        var processor = new TelegramCommandProcessor(
+            _commandRegistry,
+            _chatService,
+            _loggerMock.Object,
+            botInfoServiceMock.Object
+        );
+
+        // Act
+        var result = await processor.ProcessMessageAsync(
+            "Hello @testbot",
+            12345L,
+            "user",
+            "private",
+            "Test Chat"
+        );
+
+        // Assert
+        result.Should().NotBeNullOrEmpty();
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithOtherUserMention_ShouldReturnEmpty()
+    {
+        // Arrange
+        var botUser = new User
+        {
+            Id = 999,
+            Username = "testbot",
+            IsBot = true,
+        };
+
+        var botInfoServiceMock = new Mock<BotInfoService>(
+            TestDataBuilder.Mocks.CreateTelegramBotClient().Object,
+            TestDataBuilder.Mocks.CreateLoggerMock<BotInfoService>().Object
+        );
+        botInfoServiceMock.Setup(x => x.GetBotInfoAsync(It.IsAny<CancellationToken>()))
+            .ReturnsAsync(botUser);
+
+        var processor = new TelegramCommandProcessor(
+            _commandRegistry,
+            _chatService,
+            _loggerMock.Object,
+            botInfoServiceMock.Object
+        );
+
+        // Act
+        var result = await processor.ProcessMessageAsync(
+            "Hello @otheruser",
+            12345L,
+            "user",
+            "private",
+            "Test Chat"
+        );
+
+        // Assert
+        result.Should().BeEmpty();
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithCommand_ShouldExecuteCommand()
+    {
+        // Arrange
+        var commandMock = new Mock<ITelegramCommand>();
+        commandMock.Setup(x => x.CommandName).Returns("/test");
+        commandMock
+            .Setup(x => x.CanHandle(It.IsAny<string>()))
+            .Returns((string msg) => msg.StartsWith("/test"));
+        commandMock
+            .Setup(x =>
+                x.ExecuteAsync(It.IsAny<TelegramCommandContext>(), It.IsAny<CancellationToken>())
+            )
+            .ReturnsAsync("Command executed");
+
+        var commandRegistry = new CommandRegistry(
+            TestDataBuilder.Mocks.CreateLoggerMock<CommandRegistry>().Object,
+            new[] { commandMock.Object }
+        );
+
+        var processor = new TelegramCommandProcessor(
+            commandRegistry,
+            _chatService,
+            _loggerMock.Object,
+            _botInfoService
+        );
+
+        // Act
+        var result = await processor.ProcessMessageAsync(
+            "/test argument",
+            12345L,
+            "user",
+            "private",
+            "Test Chat"
+        );
+
+        // Assert
+        result.Should().Be("Command executed");
+        commandMock.Verify(
+            x => x.ExecuteAsync(It.IsAny<TelegramCommandContext>(), It.IsAny<CancellationToken>()),
+            Times.Once
+        );
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithException_ShouldReturnErrorMessage()
+    {
+        // Arrange
+        var chatServiceMock = new Mock<ChatService>(
+            TestDataBuilder.Mocks.CreateLoggerMock<ChatService>().Object,
+            TestDataBuilder.Mocks.CreateAIServiceMock().Object,
+            TestDataBuilder.Mocks.CreateSessionStorageMock().Object,
+            TestDataBuilder
+                .Mocks.CreateOptionsMock(TestDataBuilder.Configurations.CreateAISettings())
+                .Object,
+            TestDataBuilder.Mocks.CreateCompressionServiceMock().Object
+        );
+
+        chatServiceMock
+            .Setup(x =>
+                x.ProcessMessageAsync(
+                    It.IsAny<long>(),
+                    It.IsAny<string>(),
+                    It.IsAny<string>(),
+                    It.IsAny<string>(),
+                    It.IsAny<string>(),
+                    It.IsAny<CancellationToken>()
+                )
+            )
+            .ThrowsAsync(new Exception("Test error"));
+
+        var processor = new TelegramCommandProcessor(
+            _commandRegistry,
+            chatServiceMock.Object,
+            _loggerMock.Object,
+            _botInfoService
+        );
+
+        // Act
+        var result = await processor.ProcessMessageAsync(
+            "Test message",
+            12345L,
+            "user",
+            "private",
+            "Test Chat"
+        );
+
+        // Assert
+        result.Should().Contain("Произошла ошибка");
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithGroupChat_ShouldProcessCorrectly()
+    {
+        // Arrange
+        var messageText = "Hello";
+        var chatId = -100123456789L; // Group chat ID
+        var username = "testuser";
+        var chatType = "group";
+        var chatTitle = "Test Group";
+
+        // Act
+        var result = await _processor.ProcessMessageAsync(
+            messageText,
+            chatId,
+            username,
+            chatType,
+            chatTitle
+        );
+
+        // Assert
+        result.Should().NotBeNull();
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithSupergroupChat_ShouldProcessCorrectly()
+    {
+        // Arrange
+        var messageText = "Hello";
+        var chatId = -100123456789L;
+        var username = "testuser";
+        var chatType = "supergroup";
+        var chatTitle = "Test Supergroup";
+
+        // Act
+        var result = await _processor.ProcessMessageAsync(
+            messageText,
+            chatId,
+            username,
+            chatType,
+            chatTitle
+        );
+
+        // Assert
+        result.Should().NotBeNull();
+    }
+
+    [Fact]
+    public async Task ProcessMessageAsync_WithMultipleMentions_IncludingBot_ShouldProcessMessage()
+    {
+        // Arrange
+        var botUser = new User
+        {
+            Id = 999,
+            Username = "testbot",
+            IsBot = true,
+        };
+
+        var botInfoServiceMock = new Mock<BotInfoService>(
+            TestDataBuilder.Mocks.CreateTelegramBotClient().Object,
+            TestDataBuilder.Mocks.CreateLoggerMock<BotInfoService>().Object
+        );
+        botInfoServiceMock.Setup(x => x.GetBotInfoAsync(It.IsAny<CancellationToken>()))
+            .ReturnsAsync(botUser);
+
+        var processor = new TelegramCommandProcessor(
+            _commandRegistry,
+            _chatService,
+            _loggerMock.Object,
+            botInfoServiceMock.Object
+        );
+
+        // Act
+        var result = await processor.ProcessMessageAsync(
+            "Hello @testbot and @otheruser",
+            12345L,
+            "user",
+            "group",
+            "Test Group"
+        );
+
+        // Assert
+        result.Should().NotBeNullOrEmpty();
+    }
 }