add tests
This commit is contained in:
90
ChatBot.Tests/Telegram/Commands/ClearCommandTests.cs
Normal file
90
ChatBot.Tests/Telegram/Commands/ClearCommandTests.cs
Normal file
@@ -0,0 +1,90 @@
|
||||
using ChatBot.Models;
|
||||
using ChatBot.Models.Configuration;
|
||||
using ChatBot.Services;
|
||||
using ChatBot.Services.Interfaces;
|
||||
using ChatBot.Services.Telegram.Commands;
|
||||
using ChatBot.Tests.TestUtilities;
|
||||
using FluentAssertions;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Moq;
|
||||
|
||||
namespace ChatBot.Tests.Telegram.Commands;
|
||||
|
||||
public class ClearCommandTests : UnitTestBase
|
||||
{
|
||||
private readonly Mock<ChatService> _chatServiceMock;
|
||||
private readonly Mock<ModelService> _modelServiceMock;
|
||||
private readonly ClearCommand _clearCommand;
|
||||
|
||||
public ClearCommandTests()
|
||||
{
|
||||
_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
|
||||
);
|
||||
_modelServiceMock = new Mock<ModelService>(
|
||||
TestDataBuilder.Mocks.CreateLoggerMock<ModelService>().Object,
|
||||
TestDataBuilder
|
||||
.Mocks.CreateOptionsMock(TestDataBuilder.Configurations.CreateOllamaSettings())
|
||||
.Object
|
||||
);
|
||||
_clearCommand = new ClearCommand(_chatServiceMock.Object, _modelServiceMock.Object);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ExecuteAsync_ShouldClearSession_WhenSessionExists()
|
||||
{
|
||||
// Arrange
|
||||
var chatId = 12345L;
|
||||
_chatServiceMock.Setup(x => x.ClearHistoryAsync(chatId)).Returns(Task.CompletedTask);
|
||||
|
||||
var context = new TelegramCommandContext
|
||||
{
|
||||
ChatId = chatId,
|
||||
Username = "testuser",
|
||||
MessageText = "/clear",
|
||||
ChatType = "private",
|
||||
ChatTitle = "Test Chat",
|
||||
};
|
||||
|
||||
// Act
|
||||
var result = await _clearCommand.ExecuteAsync(context);
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNull();
|
||||
result.Should().Contain("очищена");
|
||||
_chatServiceMock.Verify(x => x.ClearHistoryAsync(chatId), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ExecuteAsync_ShouldReturnMessage_WhenSessionDoesNotExist()
|
||||
{
|
||||
// Arrange
|
||||
var chatId = 12345L;
|
||||
_chatServiceMock
|
||||
.Setup(x => x.ClearHistoryAsync(chatId))
|
||||
.ThrowsAsync(new InvalidOperationException("Session not found"));
|
||||
|
||||
var context = new TelegramCommandContext
|
||||
{
|
||||
ChatId = chatId,
|
||||
Username = "testuser",
|
||||
MessageText = "/clear",
|
||||
ChatType = "private",
|
||||
ChatTitle = "Test Chat",
|
||||
};
|
||||
|
||||
// Act
|
||||
var result = await _clearCommand.ExecuteAsync(context);
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNull();
|
||||
result.Should().Contain("очищена");
|
||||
_chatServiceMock.Verify(x => x.ClearHistoryAsync(chatId), Times.Once);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user