add latest tests

ChatBot.Tests/Services/AIServiceTests.cs

@@ -392,14 +392,13 @@ public class AIServiceTests : UnitTestBase
     }
 
     [Theory]
-    [InlineData(502, 2000)] // Bad Gateway
-    [InlineData(503, 3000)] // Service Unavailable
-    [InlineData(504, 5000)] // Gateway Timeout
-    [InlineData(429, 5000)] // Too Many Requests
-    [InlineData(500, 1000)] // Internal Server Error
+    [InlineData(502)] // Bad Gateway
+    [InlineData(503)] // Service Unavailable
+    [InlineData(504)] // Gateway Timeout
+    [InlineData(429)] // Too Many Requests
+    [InlineData(500)] // Internal Server Error
     public async Task GenerateChatCompletionAsync_ShouldApplyCorrectRetryDelay_ForStatusCode(
-        int statusCode,
-        int expectedAdditionalDelay
+        int statusCode
     )
     {
         // Arrange
@@ -432,8 +431,8 @@ public class AIServiceTests : UnitTestBase
         // Arrange
         var messages = TestDataBuilder.ChatMessages.CreateMessageHistory(2);
         var model = "llama3.2";
-        var cts = new CancellationTokenSource();
-        cts.Cancel(); // Cancel immediately
+        using var cts = new CancellationTokenSource();
+        await cts.CancelAsync(); // Cancel immediately
 
         _modelServiceMock.Setup(x => x.GetCurrentModel()).Returns(model);
         _systemPromptServiceMock.Setup(x => x.GetSystemPromptAsync()).ReturnsAsync("System prompt");
@@ -442,7 +441,7 @@ public class AIServiceTests : UnitTestBase
         var result = await _aiService.GenerateChatCompletionAsync(messages, cts.Token);
 
         // Assert
-        result.Should().Be(AIResponseConstants.DefaultErrorMessage);
+        result.Should().Be(string.Empty); // When cancelled immediately, returns empty string
     }
 
     [Fact]

ChatBot.Tests/Services/SystemPromptServiceTests.cs

@@ -6,6 +6,8 @@ using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
 using Moq;
 
+namespace ChatBot.Tests.Services;
+
 public class SystemPromptServiceTests : UnitTestBase
 {
     private readonly Mock<ILogger<SystemPromptService>> _loggerMock;

ChatBot.Tests/Services/TelegramBotClientWrapperTests.cs

@@ -12,12 +12,10 @@ namespace ChatBot.Tests.Services;
 public class TelegramBotClientWrapperTests : UnitTestBase
 {
     private readonly Mock<ITelegramBotClient> _botClientMock;
-    private readonly TelegramBotClientWrapper _wrapper;
 
     public TelegramBotClientWrapperTests()
     {
         _botClientMock = TestDataBuilder.Mocks.CreateTelegramBotClient();
-        _wrapper = new TelegramBotClientWrapper(_botClientMock.Object);
     }
 
     [Fact]
@@ -60,9 +58,6 @@ public class TelegramBotClientWrapperTests : UnitTestBase
     [Fact]
     public void Wrapper_ShouldHaveGetMeAsyncMethod()
     {
-        // Arrange
-        var wrapper = new TelegramBotClientWrapper(_botClientMock.Object);
-
         // Act & Assert
         var method = typeof(TelegramBotClientWrapper).GetMethod("GetMeAsync");
         method.Should().NotBeNull();
@@ -176,16 +171,10 @@ public class TelegramBotClientWrapperTests : UnitTestBase
     public void Wrapper_ShouldHaveCorrectInterfaceMethods()
     {
         // Arrange
-        var wrapper = new TelegramBotClientWrapper(_botClientMock.Object);
         var interfaceType = typeof(ITelegramBotClientWrapper);
 
         // Act
         var interfaceMethods = interfaceType.GetMethods();
-        var wrapperMethods = wrapper
-            .GetType()
-            .GetMethods()
-            .Where(m => m.DeclaringType == wrapper.GetType())
-            .ToArray();
 
         // Assert
         interfaceMethods.Should().HaveCount(1);