Add more test
This commit is contained in:
Leonid Pershin
380
ChatBot.Tests/Services/Interfaces/IOllamaClientTests.cs
Normal file
380
ChatBot.Tests/Services/Interfaces/IOllamaClientTests.cs
Normal file
@@ -0,0 +1,380 @@
|
||||
using ChatBot.Services;
|
||||
using ChatBot.Services.Interfaces;
|
||||
using ChatBot.Tests.TestUtilities;
|
||||
using FluentAssertions;
|
||||
using Moq;
|
||||
using OllamaSharp.Models;
|
||||
using OllamaSharp.Models.Chat;
|
||||
|
||||
namespace ChatBot.Tests.Services.Interfaces;
|
||||
|
||||
public class IOllamaClientTests : UnitTestBase
|
||||
{
|
||||
[Fact]
|
||||
public void IOllamaClient_ShouldHaveCorrectMethodSignatures()
|
||||
{
|
||||
// Arrange & Act
|
||||
var interfaceType = typeof(IOllamaClient);
|
||||
var methods = interfaceType.GetMethods();
|
||||
var properties = interfaceType.GetProperties();
|
||||
|
||||
// Assert
|
||||
methods.Should().HaveCount(4); // ChatAsync, ListLocalModelsAsync, get_SelectedModel, set_SelectedModel
|
||||
properties.Should().HaveCount(1);
|
||||
|
||||
// SelectedModel property
|
||||
var selectedModelProperty = properties.FirstOrDefault(p => p.Name == "SelectedModel");
|
||||
selectedModelProperty.Should().NotBeNull();
|
||||
selectedModelProperty!.PropertyType.Should().Be(typeof(string));
|
||||
selectedModelProperty.CanRead.Should().BeTrue();
|
||||
selectedModelProperty.CanWrite.Should().BeTrue();
|
||||
|
||||
// ChatAsync method
|
||||
var chatAsyncMethod = methods.FirstOrDefault(m => m.Name == "ChatAsync");
|
||||
chatAsyncMethod.Should().NotBeNull();
|
||||
chatAsyncMethod!.ReturnType.Should().Be(typeof(IAsyncEnumerable<ChatResponseStream?>));
|
||||
chatAsyncMethod.GetParameters().Should().HaveCount(1);
|
||||
chatAsyncMethod.GetParameters()[0].ParameterType.Should().Be(typeof(ChatRequest));
|
||||
|
||||
// ListLocalModelsAsync method
|
||||
var listLocalModelsAsyncMethod = methods.FirstOrDefault(m =>
|
||||
m.Name == "ListLocalModelsAsync"
|
||||
);
|
||||
listLocalModelsAsyncMethod.Should().NotBeNull();
|
||||
listLocalModelsAsyncMethod!.ReturnType.Should().Be(typeof(Task<IEnumerable<Model>>));
|
||||
listLocalModelsAsyncMethod.GetParameters().Should().BeEmpty();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IOllamaClient_ShouldBeImplementedByOllamaClientAdapter()
|
||||
{
|
||||
// Arrange & Act
|
||||
var ollamaClientAdapterType = typeof(OllamaClientAdapter);
|
||||
var interfaceType = typeof(IOllamaClient);
|
||||
|
||||
// Assert
|
||||
interfaceType.IsAssignableFrom(ollamaClientAdapterType).Should().BeTrue();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IOllamaClient_SelectedModel_ShouldBeReadableAndWritable()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var expectedModel = "llama2:7b";
|
||||
|
||||
mock.SetupProperty(x => x.SelectedModel, "default-model");
|
||||
|
||||
// Act
|
||||
mock.Object.SelectedModel = expectedModel;
|
||||
var result = mock.Object.SelectedModel;
|
||||
|
||||
// Assert
|
||||
result.Should().Be(expectedModel);
|
||||
mock.VerifySet(x => x.SelectedModel = expectedModel, Times.Once);
|
||||
mock.VerifyGet(x => x.SelectedModel, Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IOllamaClient_ChatAsync_ShouldReturnAsyncEnumerable()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var request = new ChatRequest
|
||||
{
|
||||
Model = "llama2:7b",
|
||||
Messages = new List<Message>
|
||||
{
|
||||
new() { Role = "user", Content = "Hello" },
|
||||
},
|
||||
};
|
||||
|
||||
var expectedResponse = new List<ChatResponseStream?>
|
||||
{
|
||||
new()
|
||||
{
|
||||
Message = new Message
|
||||
{
|
||||
Role = "assistant",
|
||||
Content = "Hello! How can I help you?",
|
||||
},
|
||||
},
|
||||
new() { Done = true },
|
||||
};
|
||||
|
||||
mock.Setup(x => x.ChatAsync(It.IsAny<ChatRequest>()))
|
||||
.Returns(CreateAsyncEnumerable(expectedResponse));
|
||||
|
||||
// Act
|
||||
var result = mock.Object.ChatAsync(request);
|
||||
var responses = new List<ChatResponseStream?>();
|
||||
await foreach (var response in result)
|
||||
{
|
||||
responses.Add(response);
|
||||
}
|
||||
|
||||
// Assert
|
||||
responses.Should().HaveCount(2);
|
||||
responses[0]?.Message?.Content.Should().Be("Hello! How can I help you?");
|
||||
responses[1]?.Done.Should().BeTrue();
|
||||
mock.Verify(x => x.ChatAsync(request), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IOllamaClient_ChatAsync_ShouldHandleEmptyResponse()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var request = new ChatRequest { Model = "llama2:7b", Messages = new List<Message>() };
|
||||
|
||||
var expectedResponse = new List<ChatResponseStream?>();
|
||||
|
||||
mock.Setup(x => x.ChatAsync(It.IsAny<ChatRequest>()))
|
||||
.Returns(CreateAsyncEnumerable(expectedResponse));
|
||||
|
||||
// Act
|
||||
var result = mock.Object.ChatAsync(request);
|
||||
var responses = new List<ChatResponseStream?>();
|
||||
await foreach (var response in result)
|
||||
{
|
||||
responses.Add(response);
|
||||
}
|
||||
|
||||
// Assert
|
||||
responses.Should().BeEmpty();
|
||||
mock.Verify(x => x.ChatAsync(request), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IOllamaClient_ChatAsync_ShouldHandleNullResponse()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var request = new ChatRequest
|
||||
{
|
||||
Model = "llama2:7b",
|
||||
Messages = new List<Message>
|
||||
{
|
||||
new() { Role = "user", Content = "Test" },
|
||||
},
|
||||
};
|
||||
|
||||
var expectedResponse = new List<ChatResponseStream?>
|
||||
{
|
||||
null,
|
||||
new()
|
||||
{
|
||||
Message = new Message { Role = "assistant", Content = "Response" },
|
||||
},
|
||||
};
|
||||
|
||||
mock.Setup(x => x.ChatAsync(It.IsAny<ChatRequest>()))
|
||||
.Returns(CreateAsyncEnumerable(expectedResponse));
|
||||
|
||||
// Act
|
||||
var result = mock.Object.ChatAsync(request);
|
||||
var responses = new List<ChatResponseStream?>();
|
||||
await foreach (var response in result)
|
||||
{
|
||||
responses.Add(response);
|
||||
}
|
||||
|
||||
// Assert
|
||||
responses.Should().HaveCount(2);
|
||||
responses[0].Should().BeNull();
|
||||
responses[1]?.Message?.Content.Should().Be("Response");
|
||||
mock.Verify(x => x.ChatAsync(request), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IOllamaClient_ListLocalModelsAsync_ShouldReturnModels()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var expectedModels = new List<Model>
|
||||
{
|
||||
new()
|
||||
{
|
||||
Name = "llama2:7b",
|
||||
Size = 3825819519,
|
||||
ModifiedAt = DateTime.UtcNow,
|
||||
},
|
||||
new()
|
||||
{
|
||||
Name = "codellama:7b",
|
||||
Size = 3825819519,
|
||||
ModifiedAt = DateTime.UtcNow,
|
||||
},
|
||||
};
|
||||
|
||||
mock.Setup(x => x.ListLocalModelsAsync()).ReturnsAsync(expectedModels);
|
||||
|
||||
// Act
|
||||
var result = await mock.Object.ListLocalModelsAsync();
|
||||
|
||||
// Assert
|
||||
result.Should().BeEquivalentTo(expectedModels);
|
||||
mock.Verify(x => x.ListLocalModelsAsync(), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IOllamaClient_ListLocalModelsAsync_ShouldReturnEmptyList()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var expectedModels = new List<Model>();
|
||||
|
||||
mock.Setup(x => x.ListLocalModelsAsync()).ReturnsAsync(expectedModels);
|
||||
|
||||
// Act
|
||||
var result = await mock.Object.ListLocalModelsAsync();
|
||||
|
||||
// Assert
|
||||
result.Should().BeEmpty();
|
||||
mock.Verify(x => x.ListLocalModelsAsync(), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IOllamaClient_ListLocalModelsAsync_ShouldHandleNullModels()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
IEnumerable<Model>? expectedModels = null;
|
||||
|
||||
mock.Setup(x => x.ListLocalModelsAsync()).ReturnsAsync(expectedModels!);
|
||||
|
||||
// Act
|
||||
var result = await mock.Object.ListLocalModelsAsync();
|
||||
|
||||
// Assert
|
||||
result.Should().BeNull();
|
||||
mock.Verify(x => x.ListLocalModelsAsync(), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IOllamaClient_SelectedModel_ShouldHandleNullValue()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
string? expectedModel = null;
|
||||
|
||||
mock.SetupProperty(x => x.SelectedModel, "default-model");
|
||||
|
||||
// Act
|
||||
mock.Object.SelectedModel = expectedModel!;
|
||||
var result = mock.Object.SelectedModel;
|
||||
|
||||
// Assert
|
||||
result.Should().BeNull();
|
||||
mock.VerifySet(x => x.SelectedModel = expectedModel!, Times.Once);
|
||||
mock.VerifyGet(x => x.SelectedModel, Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IOllamaClient_SelectedModel_ShouldHandleEmptyString()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var expectedModel = "";
|
||||
|
||||
mock.SetupProperty(x => x.SelectedModel, "default-model");
|
||||
|
||||
// Act
|
||||
mock.Object.SelectedModel = expectedModel;
|
||||
var result = mock.Object.SelectedModel;
|
||||
|
||||
// Assert
|
||||
result.Should().Be(expectedModel);
|
||||
mock.VerifySet(x => x.SelectedModel = expectedModel, Times.Once);
|
||||
mock.VerifyGet(x => x.SelectedModel, Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IOllamaClient_ShouldBePublicInterface()
|
||||
{
|
||||
// Arrange & Act
|
||||
var interfaceType = typeof(IOllamaClient);
|
||||
|
||||
// Assert
|
||||
interfaceType.IsPublic.Should().BeTrue();
|
||||
interfaceType.IsInterface.Should().BeTrue();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IOllamaClient_ShouldHaveCorrectNamespace()
|
||||
{
|
||||
// Arrange & Act
|
||||
var interfaceType = typeof(IOllamaClient);
|
||||
|
||||
// Assert
|
||||
interfaceType.Namespace.Should().Be("ChatBot.Services.Interfaces");
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void IOllamaClient_ShouldHaveCorrectGenericConstraints()
|
||||
{
|
||||
// Arrange & Act
|
||||
var interfaceType = typeof(IOllamaClient);
|
||||
var methods = interfaceType.GetMethods();
|
||||
var properties = interfaceType.GetProperties();
|
||||
|
||||
// Assert
|
||||
// All methods should be public
|
||||
methods.All(m => m.IsPublic).Should().BeTrue();
|
||||
|
||||
// All properties should be public
|
||||
properties.All(p => p.GetGetMethod()?.IsPublic == true).Should().BeTrue();
|
||||
properties.All(p => p.GetSetMethod()?.IsPublic == true).Should().BeTrue();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IOllamaClient_ChatAsync_ShouldHandleLargeRequest()
|
||||
{
|
||||
// Arrange
|
||||
var mock = new Mock<IOllamaClient>();
|
||||
var messages = new List<Message>();
|
||||
// Add many messages
|
||||
for (int i = 0; i < 100; i++)
|
||||
{
|
||||
messages.Add(
|
||||
new Message { Role = i % 2 == 0 ? "user" : "assistant", Content = $"Message {i}" }
|
||||
);
|
||||
}
|
||||
|
||||
var request = new ChatRequest { Model = "llama2:7b", Messages = messages };
|
||||
|
||||
var expectedResponse = new List<ChatResponseStream?>
|
||||
{
|
||||
new()
|
||||
{
|
||||
Message = new Message { Role = "assistant", Content = "Large response" },
|
||||
},
|
||||
};
|
||||
|
||||
mock.Setup(x => x.ChatAsync(It.IsAny<ChatRequest>()))
|
||||
.Returns(CreateAsyncEnumerable(expectedResponse));
|
||||
|
||||
// Act
|
||||
var result = mock.Object.ChatAsync(request);
|
||||
var responses = new List<ChatResponseStream?>();
|
||||
await foreach (var response in result)
|
||||
{
|
||||
responses.Add(response);
|
||||
}
|
||||
|
||||
// Assert
|
||||
responses.Should().HaveCount(1);
|
||||
responses[0]?.Message?.Content.Should().Be("Large response");
|
||||
mock.Verify(x => x.ChatAsync(request), Times.Once);
|
||||
}
|
||||
|
||||
private static async IAsyncEnumerable<ChatResponseStream?> CreateAsyncEnumerable(
|
||||
List<ChatResponseStream?> items
|
||||
)
|
||||
{
|
||||
foreach (var item in items)
|
||||
{
|
||||
yield return item;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user