Files
ChatBot/ChatBot.Tests/Services/Telegram/TelegramMessageHandlerTests.cs
Leonid Pershin c9eac74e35
Some checks failed
SonarQube / Build and analyze (push) Failing after 3m2s
Unit Tests / Run Tests (push) Failing after 2m23s
Add more tests
2025-10-20 08:20:55 +03:00

833 lines
26 KiB
C#

using ChatBot.Services.Telegram.Commands;
using ChatBot.Services.Telegram.Interfaces;
using ChatBot.Services.Telegram.Services;
using ChatBot.Tests.TestUtilities;
using FluentAssertions;
using Microsoft.Extensions.Logging;
using Moq;
using Telegram.Bot;
using Telegram.Bot.Types;
using Telegram.Bot.Types.Enums;
using Xunit;
namespace ChatBot.Tests.Services.Telegram;
public class TelegramMessageHandlerTests : UnitTestBase
{
private readonly Mock<ILogger<TelegramMessageHandler>> _loggerMock;
private readonly Mock<ITelegramCommandProcessor> _commandProcessorMock;
private readonly Mock<ITelegramMessageSender> _messageSenderMock;
private readonly Mock<ITelegramBotClient> _botClientMock;
private readonly TelegramMessageHandler _handler;
public TelegramMessageHandlerTests()
{
_loggerMock = TestDataBuilder.Mocks.CreateLoggerMock<TelegramMessageHandler>();
_commandProcessorMock = new Mock<ITelegramCommandProcessor>();
_messageSenderMock = new Mock<ITelegramMessageSender>();
_botClientMock = TestDataBuilder.Mocks.CreateTelegramBotClient();
_handler = new TelegramMessageHandler(
_loggerMock.Object,
_commandProcessorMock.Object,
_messageSenderMock.Object
);
}
private static Message CreateMessage(
string text,
long chatId,
string username,
string chatTitle,
int messageId,
User? from = null,
Message? replyToMessage = null
)
{
var message = new Message
{
Text = text,
Chat = new Chat
{
Id = chatId,
Type = ChatType.Private,
Title = chatTitle,
},
From = from ?? new User { Id = 67890, Username = username },
ReplyToMessage = replyToMessage,
};
// Note: MessageId is read-only, so we can't set it directly
// The actual MessageId will be 0, but this is sufficient for testing
return message;
}
[Fact]
public void Constructor_ShouldInitializeCorrectly()
{
// Arrange
var logger = TestDataBuilder.Mocks.CreateLoggerMock<TelegramMessageHandler>().Object;
var commandProcessor = new Mock<ITelegramCommandProcessor>().Object;
var messageSender = new Mock<ITelegramMessageSender>().Object;
// Act
var handler = new TelegramMessageHandler(logger, commandProcessor, messageSender);
// Assert
handler.Should().NotBeNull();
}
[Fact]
public void Constructor_ShouldNotThrow_WhenLoggerIsNull()
{
// Arrange
ILogger<TelegramMessageHandler>? logger = null;
var commandProcessor = new Mock<ITelegramCommandProcessor>().Object;
var messageSender = new Mock<ITelegramMessageSender>().Object;
// Act & Assert
var act = () => new TelegramMessageHandler(logger!, commandProcessor, messageSender);
act.Should().NotThrow();
}
[Fact]
public void Constructor_ShouldNotThrow_WhenCommandProcessorIsNull()
{
// Arrange
var logger = TestDataBuilder.Mocks.CreateLoggerMock<TelegramMessageHandler>().Object;
ITelegramCommandProcessor? commandProcessor = null;
var messageSender = new Mock<ITelegramMessageSender>().Object;
// Act & Assert
var act = () => new TelegramMessageHandler(logger, commandProcessor!, messageSender);
act.Should().NotThrow();
}
[Fact]
public void Constructor_ShouldNotThrow_WhenMessageSenderIsNull()
{
// Arrange
var logger = TestDataBuilder.Mocks.CreateLoggerMock<TelegramMessageHandler>().Object;
var commandProcessor = new Mock<ITelegramCommandProcessor>().Object;
ITelegramMessageSender? messageSender = null;
// Act & Assert
var act = () => new TelegramMessageHandler(logger, commandProcessor, messageSender!);
act.Should().NotThrow();
}
[Fact]
public async Task HandleUpdateAsync_ShouldReturnEarly_WhenUpdateMessageIsNull()
{
// Arrange
var update = new Update { Message = null };
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
),
Times.Never
);
_messageSenderMock.Verify(
x =>
x.SendMessageWithRetry(
It.IsAny<ITelegramBotClient>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<int>(),
It.IsAny<CancellationToken>(),
It.IsAny<int>()
),
Times.Never
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldReturnEarly_WhenMessageTextIsNull()
{
// Arrange
var message = CreateMessage(null!, 12345, "testuser", "Test Chat", 1);
var update = new Update { Message = message };
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
),
Times.Never
);
_messageSenderMock.Verify(
x =>
x.SendMessageWithRetry(
It.IsAny<ITelegramBotClient>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<int>(),
It.IsAny<CancellationToken>(),
It.IsAny<int>()
),
Times.Never
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldProcessMessage_WhenValidMessage()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var chatType = "Private";
var chatTitle = "Test Chat";
var messageId = 1;
var message = CreateMessage(messageText, chatId, username, chatTitle, messageId);
var update = new Update { Message = message };
var expectedResponse = "Hello! How can I help you?";
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
messageText,
chatId,
username,
chatType,
chatTitle,
null,
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync(expectedResponse);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
messageText,
chatId,
username,
chatType,
chatTitle,
null,
It.IsAny<CancellationToken>()
),
Times.Once
);
_messageSenderMock.Verify(
x =>
x.SendMessageWithRetry(
_botClientMock.Object,
chatId,
expectedResponse,
It.IsAny<int>(),
It.IsAny<CancellationToken>(),
3
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldNotSendMessage_WhenResponseIsEmpty()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var chatType = "Private";
var chatTitle = "Test Chat";
var message = CreateMessage(messageText, chatId, username, chatTitle, 1);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync(string.Empty);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
),
Times.Once
);
_messageSenderMock.Verify(
x =>
x.SendMessageWithRetry(
It.IsAny<ITelegramBotClient>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<int>(),
It.IsAny<CancellationToken>(),
It.IsAny<int>()
),
Times.Never
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldNotSendMessage_WhenResponseIsNull()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var chatType = "Private";
var chatTitle = "Test Chat";
var message = CreateMessage(messageText, chatId, username, chatTitle, 1);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync((string)null!);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
),
Times.Once
);
_messageSenderMock.Verify(
x =>
x.SendMessageWithRetry(
It.IsAny<ITelegramBotClient>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<int>(),
It.IsAny<CancellationToken>(),
It.IsAny<int>()
),
Times.Never
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldUseUsername_WhenFromHasUsername()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var chatType = "Private";
var chatTitle = "Test Chat";
var message = CreateMessage(messageText, chatId, username, chatTitle, 1);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync("Response");
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
messageText,
chatId,
username,
chatType,
chatTitle,
null,
It.IsAny<CancellationToken>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldUseFirstName_WhenFromHasNoUsername()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var firstName = "TestUser";
var chatType = "Private";
var chatTitle = "Test Chat";
var from = new User
{
Id = 67890,
Username = null,
FirstName = firstName,
};
var message = CreateMessage(messageText, chatId, firstName, chatTitle, 1, from);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync("Response");
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
messageText,
chatId,
firstName,
chatType,
chatTitle,
null,
It.IsAny<CancellationToken>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldUseUnknown_WhenFromIsNull()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var chatType = "Private";
var chatTitle = "Test Chat";
var message = CreateMessage(messageText, chatId, "Unknown", chatTitle, 1, null);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync("Response");
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
messageText,
chatId,
"Unknown",
chatType,
chatTitle,
null,
It.IsAny<CancellationToken>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldHandleReplyMessage()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var chatType = "Private";
var chatTitle = "Test Chat";
var messageId = 1;
var replyToMessageId = 2;
var replyToUserId = 67890L;
var replyToUsername = "originaluser";
var replyToMessage = CreateMessage(
"Original message",
chatId,
replyToUsername,
chatTitle,
replyToMessageId
);
var message = CreateMessage(
messageText,
chatId,
username,
chatTitle,
messageId,
null,
replyToMessage
);
var update = new Update { Message = message };
var expectedResponse = "Response to reply";
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync(expectedResponse);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
messageText,
chatId,
username,
chatType,
chatTitle,
It.Is<ReplyInfo>(r =>
r.UserId == replyToUserId && r.Username == replyToUsername
),
It.IsAny<CancellationToken>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldPassCancellationToken()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var chatType = "Private";
var chatTitle = "Test Chat";
var cancellationToken = new CancellationToken();
var message = CreateMessage(messageText, chatId, username, chatTitle, 1);
var update = new Update { Message = message };
var expectedResponse = "Response";
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync(expectedResponse);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, cancellationToken);
// Assert
_commandProcessorMock.Verify(
x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
cancellationToken
),
Times.Once
);
_messageSenderMock.Verify(
x =>
x.SendMessageWithRetry(
It.IsAny<ITelegramBotClient>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<int>(),
cancellationToken,
It.IsAny<int>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldLogError_WhenExceptionOccurs()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var chatType = "Private";
var chatTitle = "Test Chat";
var message = CreateMessage(messageText, chatId, username, chatTitle, 1);
var update = new Update { Message = message };
var exception = new Exception("Test exception");
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ThrowsAsync(exception);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_loggerMock.Verify(
x =>
x.Log(
LogLevel.Error,
It.IsAny<EventId>(),
It.Is<It.IsAnyType>(
(v, t) => v.ToString()!.Contains("Error handling update from chat")
),
It.IsAny<Exception>(),
It.IsAny<Func<It.IsAnyType, Exception?, string>>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldLogInformation_WhenMessageReceived()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var message = CreateMessage(messageText, chatId, username, "Test Chat", 1);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync("Response");
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_loggerMock.Verify(
x =>
x.Log(
LogLevel.Information,
It.IsAny<EventId>(),
It.Is<It.IsAnyType>(
(v, t) =>
v.ToString()!
.Contains(
$"Message from @{username} in chat {chatId}: \"{messageText}\""
)
),
It.IsAny<Exception>(),
It.IsAny<Func<It.IsAnyType, Exception?, string>>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldLogInformation_WhenResponseSent()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var response = "Hello! How can I help you?";
var message = CreateMessage(messageText, chatId, username, "Test Chat", 1);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync(response);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_loggerMock.Verify(
x =>
x.Log(
LogLevel.Information,
It.IsAny<EventId>(),
It.Is<It.IsAnyType>(
(v, t) =>
v.ToString()!
.Contains(
$"Response sent to @{username} in chat {chatId}: \"{response}\""
)
),
It.IsAny<Exception>(),
It.IsAny<Func<It.IsAnyType, Exception?, string>>()
),
Times.Once
);
}
[Fact]
public async Task HandleUpdateAsync_ShouldLogInformation_WhenNoResponseSent()
{
// Arrange
var messageText = "Hello bot";
var chatId = 12345L;
var username = "testuser";
var message = CreateMessage(messageText, chatId, username, "Test Chat", 1);
var update = new Update { Message = message };
_commandProcessorMock
.Setup(x =>
x.ProcessMessageAsync(
It.IsAny<string>(),
It.IsAny<long>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<ReplyInfo?>(),
It.IsAny<CancellationToken>()
)
)
.ReturnsAsync(string.Empty);
// Act
await _handler.HandleUpdateAsync(_botClientMock.Object, update, CancellationToken.None);
// Assert
_loggerMock.Verify(
x =>
x.Log(
LogLevel.Information,
It.IsAny<EventId>(),
It.Is<It.IsAnyType>(
(v, t) =>
v.ToString()!
.Contains(
$"No response sent to @{username} in chat {chatId} (AI chose to ignore message)"
)
),
It.IsAny<Exception>(),
It.IsAny<Func<It.IsAnyType, Exception?, string>>()
),
Times.Once
);
}
}