This commit is contained in:
54
ChatBot.Tests/Services/Telegram/BotInfoServiceSimpleTests.cs
Normal file
54
ChatBot.Tests/Services/Telegram/BotInfoServiceSimpleTests.cs
Normal file
@@ -0,0 +1,54 @@
|
||||
using ChatBot.Services.Telegram.Services;
|
||||
using FluentAssertions;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Moq;
|
||||
using Telegram.Bot;
|
||||
|
||||
namespace ChatBot.Tests.Services.Telegram;
|
||||
|
||||
/// <summary>
|
||||
/// Simple tests for BotInfoService that don't rely on mocking extension methods
|
||||
/// </summary>
|
||||
public class BotInfoServiceSimpleTests
|
||||
{
|
||||
[Fact]
|
||||
public void Constructor_ShouldCreateInstance()
|
||||
{
|
||||
// Arrange
|
||||
var botClientMock = new Mock<ITelegramBotClient>();
|
||||
var loggerMock = new Mock<ILogger<BotInfoService>>();
|
||||
|
||||
// Act
|
||||
var service = new BotInfoService(botClientMock.Object, loggerMock.Object);
|
||||
|
||||
// Assert
|
||||
service.Should().NotBeNull();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IsCacheValid_InitiallyFalse()
|
||||
{
|
||||
// Arrange
|
||||
var botClientMock = new Mock<ITelegramBotClient>();
|
||||
var loggerMock = new Mock<ILogger<BotInfoService>>();
|
||||
var service = new BotInfoService(botClientMock.Object, loggerMock.Object);
|
||||
|
||||
// Act & Assert
|
||||
service.IsCacheValid().Should().BeFalse();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void InvalidateCache_ShouldWork()
|
||||
{
|
||||
// Arrange
|
||||
var botClientMock = new Mock<ITelegramBotClient>();
|
||||
var loggerMock = new Mock<ILogger<BotInfoService>>();
|
||||
var service = new BotInfoService(botClientMock.Object, loggerMock.Object);
|
||||
|
||||
// Act
|
||||
service.InvalidateCache();
|
||||
|
||||
// Assert
|
||||
service.IsCacheValid().Should().BeFalse();
|
||||
}
|
||||
}
|
||||
100
ChatBot.Tests/Services/Telegram/StatusCommandEdgeCaseTests.cs
Normal file
100
ChatBot.Tests/Services/Telegram/StatusCommandEdgeCaseTests.cs
Normal file
@@ -0,0 +1,100 @@
|
||||
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.Options;
|
||||
using Moq;
|
||||
|
||||
namespace ChatBot.Tests.Services.Telegram;
|
||||
|
||||
/// <summary>
|
||||
/// Additional edge case tests for StatusCommand to improve code coverage
|
||||
/// </summary>
|
||||
public class StatusCommandEdgeCaseTests : UnitTestBase
|
||||
{
|
||||
private readonly Mock<IOllamaClient> _ollamaClientMock;
|
||||
private readonly StatusCommand _statusCommand;
|
||||
|
||||
public StatusCommandEdgeCaseTests()
|
||||
{
|
||||
_ollamaClientMock = TestDataBuilder.Mocks.CreateOllamaClientMock();
|
||||
var ollamaSettings = TestDataBuilder.Configurations.CreateOllamaSettings();
|
||||
var ollamaSettingsMock = TestDataBuilder.Mocks.CreateOptionsMock(ollamaSettings);
|
||||
|
||||
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
|
||||
);
|
||||
|
||||
var modelServiceMock = new Mock<ModelService>(
|
||||
TestDataBuilder.Mocks.CreateLoggerMock<ModelService>().Object,
|
||||
ollamaSettingsMock.Object
|
||||
);
|
||||
|
||||
var aiSettingsMock = TestDataBuilder.Mocks.CreateOptionsMock(new AISettings());
|
||||
|
||||
_statusCommand = new StatusCommand(
|
||||
chatServiceMock.Object,
|
||||
modelServiceMock.Object,
|
||||
aiSettingsMock.Object,
|
||||
_ollamaClientMock.Object
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ExecuteAsync_WhenOllamaThrowsHttpRequestException_ShouldHandleGracefully()
|
||||
{
|
||||
// Arrange
|
||||
var context = new TelegramCommandContext
|
||||
{
|
||||
ChatId = 12345,
|
||||
Username = "testuser",
|
||||
MessageText = "/status",
|
||||
ChatType = "private",
|
||||
ChatTitle = "Test Chat"
|
||||
};
|
||||
|
||||
_ollamaClientMock
|
||||
.Setup(x => x.ListLocalModelsAsync())
|
||||
.ThrowsAsync(new HttpRequestException("502 Bad Gateway"));
|
||||
|
||||
// Act
|
||||
var result = await _statusCommand.ExecuteAsync(context);
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNullOrEmpty();
|
||||
result.Should().Contain("Статус системы");
|
||||
// StatusCommand handles exceptions internally and returns formatted status
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ExecuteAsync_WhenOllamaThrowsTaskCanceledException_ShouldHandleGracefully()
|
||||
{
|
||||
// Arrange
|
||||
var context = new TelegramCommandContext
|
||||
{
|
||||
ChatId = 12345,
|
||||
Username = "testuser",
|
||||
MessageText = "/status",
|
||||
ChatType = "private",
|
||||
ChatTitle = "Test Chat"
|
||||
};
|
||||
|
||||
_ollamaClientMock
|
||||
.Setup(x => x.ListLocalModelsAsync())
|
||||
.ThrowsAsync(new TaskCanceledException("Operation timed out"));
|
||||
|
||||
// Act
|
||||
var result = await _statusCommand.ExecuteAsync(context);
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNullOrEmpty();
|
||||
result.Should().Contain("Статус системы");
|
||||
// StatusCommand handles timeouts internally and returns formatted status
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user