cartaid/shopping-assistant-api
1
0
Fork 0
mirror of https://github.com/Shchoholiev/shopping-assistant-api.git synced 2025-04-04 16:49:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

SA-197 Update Unit Tests

Browse Source 
- Add Logger Mock to Unit tests
- Update ProductTests to handle suggestions
This commit is contained in:
shchoholiev 2023-12-11 01:10:46 +00:00
parent 68ab565800
commit 745bee93ee
2 changed files with 57 additions and 102 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ShoppingAssistantApi.UnitTests/OpenAiServiceTests.cs
View File

@ -1,4 +1,5 @@
using System.Net; using System.Net;
using Microsoft.Extensions.Logging;
using Moq; using Moq;
using Moq.Protected; using Moq.Protected;
using ShoppingAssistantApi.Application.IServices; using ShoppingAssistantApi.Application.IServices;
@ -31,7 +32,7 @@ public class OpenAiServiceTests
return client; return client;
}); });
_openAiService = new OpenAiService(_mockHttpClientFactory.Object); _openAiService = new OpenAiService(_mockHttpClientFactory.Object, new Mock<ILogger<OpenAiService>>().Object);
} }
[Fact] [Fact]
@ -78,7 +79,7 @@ public class OpenAiServiceTests
{ {
new OpenAiMessage new OpenAiMessage
{ {
Role = OpenAiRole.User.RequestConvert(), Role = OpenAiRole.User.ToRequestString(),
Content = "Return Hello World!" Content = "Return Hello World!"
} }
} }

104
ShoppingAssistantApi.UnitTests/ProductTests.cs
View File

@ -1,4 +1,6 @@
using Moq; using Microsoft.Extensions.Logging;
using MongoDB.Bson;
using Moq;
using ShoppingAssistantApi.Application.IRepositories; using ShoppingAssistantApi.Application.IRepositories;
using ShoppingAssistantApi.Application.IServices; using ShoppingAssistantApi.Application.IServices;
using ShoppingAssistantApi.Application.Models.CreateDtos; using ShoppingAssistantApi.Application.Models.CreateDtos;
@ -27,14 +29,14 @@ public class ProductTests
_messagesRepositoryMock = new Mock<IMessagesRepository>(); _messagesRepositoryMock = new Mock<IMessagesRepository>();
_openAiServiceMock = new Mock<IOpenAiService>(); _openAiServiceMock = new Mock<IOpenAiService>();
_wishListServiceMock = new Mock<IWishlistsService>(); _wishListServiceMock = new Mock<IWishlistsService>();
_productService = new ProductService(_openAiServiceMock.Object, _wishListServiceMock.Object, _messagesRepositoryMock.Object); _productService = new ProductService(_openAiServiceMock.Object, _wishListServiceMock.Object, _messagesRepositoryMock.Object, new Mock<ILogger<ProductService>>().Object);
} }
[Fact] [Fact]
public async Task SearchProductAsync_WhenWishlistsWithoutMessages_ReturnsExpectedEvents() public async Task SearchProductAsync_WhenWishlistsWithoutMessages_ReturnsExpectedEvents()
{ {
// Arrange // Arrange
string wishlistId = "existingWishlistId"; string wishlistId = "657657677c13ae4bc95e2f41";
var message = new MessageCreateDto var message = new MessageCreateDto
{ {
Text = "Your message text here" Text = "Your message text here"
@ -44,8 +46,8 @@ public class ProductTests
// Define your expected SSE data for the test // Define your expected SSE data for the test
var expectedSseData = new List<string> var expectedSseData = new List<string>
{ {
"[", "Message", "]", " What", " u", " want", " ?", "[", "Options", "]", " USB-C", " ;", " Keyboard", " ultra", "[", "Message", "]", " What", " u", " want", " ?", "[", "Suggestions", "]", " USB-C", " ;", " Keyboard", " ultra",
" ;", "[", "Options", "]", " USB", "-C", " ;", "[", "Products", "]", " GTX", " 3090", " ;", " GTX", " ;", "[", "Suggestions", "]", " USB", "-C", " ;", "[", "Products", "]", " GTX", " 3090", " ;", " GTX",
" 3070TI", " ;", " GTX", " 4070TI", " ;", " ?", "[", "Message", "]", " What", " u", " want", " ?" " 3070TI", " ;", " GTX", " 4070TI", " ;", " ?", "[", "Message", "]", " What", " u", " want", " ?"
}; };
@ -56,34 +58,15 @@ public class ProductTests
_openAiServiceMock.Setup(x => x.GetChatCompletionStream(It.IsAny<ChatCompletionRequest>(), cancellationToken)) _openAiServiceMock.Setup(x => x.GetChatCompletionStream(It.IsAny<ChatCompletionRequest>(), cancellationToken))
.Returns(expectedSseData.ToAsyncEnumerable()); .Returns(expectedSseData.ToAsyncEnumerable());
_messagesRepositoryMock.Setup(m => m.GetCountAsync(It.IsAny<Expression<Func<Message, bool>>>(), It.IsAny<CancellationToken>())) _messagesRepositoryMock.Setup(m => m.GetWishlistMessagesAsync(It.IsAny<ObjectId>(), It.IsAny<CancellationToken>()))
.ReturnsAsync(1); .ReturnsAsync(new List<Message>
_wishListServiceMock.Setup(w => w.AddMessageToPersonalWishlistAsync(wishlistId, It.IsAny<MessageDto>(), cancellationToken))
.Verifiable();
_wishListServiceMock
.Setup(m => m.GetMessagesPageFromPersonalWishlistAsync(
It.IsAny<string>(),
It.IsAny<int>(),
It.IsAny<int>(),
It.IsAny<CancellationToken>())
)
.ReturnsAsync(new PagedList<MessageDto>(
new List<MessageDto>
{ {
new MessageDto new()
{ {
Text = "What are you looking for?", Text = "What are you looking for?",
Id = "3",
CreatedById = "User2",
Role = "User" Role = "User"
}, },
}, });
1,
1,
1
));
// Act // Act
var resultStream = _productService.SearchProductAsync(wishlistId, message, cancellationToken); var resultStream = _productService.SearchProductAsync(wishlistId, message, cancellationToken);
@ -113,7 +96,7 @@ public class ProductTests
public async void SearchProductAsync_WithExistingMessageInWishlist_ReturnsExpectedEvents() public async void SearchProductAsync_WithExistingMessageInWishlist_ReturnsExpectedEvents()
{ {
// Arrange // Arrange
var wishlistId = "your_wishlist_id"; var wishlistId = "657657677c13ae4bc95e2f41";
var message = new MessageCreateDto { Text = "Your message text" }; var message = new MessageCreateDto { Text = "Your message text" };
var cancellationToken = new CancellationToken(); var cancellationToken = new CancellationToken();
@ -121,8 +104,8 @@ public class ProductTests
var expectedSseData = new List<string> var expectedSseData = new List<string>
{ {
"[", "Message", "]", " What", " u", " want", " ?", "[", "Options", "]", "USB-C", " ;", "Keyboard", " ultra", "[", "Message", "]", " What", " u", " want", " ?", "[", "Suggestions", "]", "USB-C", " ;", "Keyboard", " ultra",
" ;", "[", "Options", "]", "USB", "-C", " ;" " ;", "[", "Suggestions", "]", "USB", "-C", " ;"
}; };
var expectedMessages = new List<string> { " What", " u", " want", " ?" }; var expectedMessages = new List<string> { " What", " u", " want", " ?" };
@ -132,39 +115,24 @@ public class ProductTests
_openAiServiceMock.Setup(x => x.GetChatCompletionStream(It.IsAny<ChatCompletionRequest>(), cancellationToken)) _openAiServiceMock.Setup(x => x.GetChatCompletionStream(It.IsAny<ChatCompletionRequest>(), cancellationToken))
.Returns(expectedSseData.ToAsyncEnumerable()); .Returns(expectedSseData.ToAsyncEnumerable());
_messagesRepositoryMock.Setup(m => m.GetCountAsync(It.IsAny<Expression<Func<Message, bool>>>(), It.IsAny<CancellationToken>())) _messagesRepositoryMock.Setup(m => m.GetWishlistMessagesAsync(It.IsAny<ObjectId>(), It.IsAny<CancellationToken>()))
.ReturnsAsync(3); .ReturnsAsync(new List<Message>
_wishListServiceMock.Setup(w => w.AddMessageToPersonalWishlistAsync(wishlistId, It.IsAny<MessageDto>(), cancellationToken))
.Verifiable();
_wishListServiceMock
.Setup(w => w.GetMessagesPageFromPersonalWishlistAsync(
It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>(), It.IsAny<CancellationToken>()))
.ReturnsAsync(new PagedList<MessageDto>(new List<MessageDto>
{
new MessageDto
{ {
new() {
Text = "Message 1", Text = "Message 1",
Id = "1",
CreatedById = "User2",
Role = "User" Role = "User"
}, },
new MessageDto new Message
{ {
Text = "Message 2", Text = "Message 2",
Id = "2",
CreatedById = "User2",
Role = "User" Role = "User"
}, },
new MessageDto new Message
{ {
Text = "Message 3", Text = "Message 3",
Id = "3",
CreatedById = "User2",
Role = "User" Role = "User"
}, },
}, 1, 3, 3)); });
// Act // Act
var resultStream = _productService.SearchProductAsync(wishlistId, message, cancellationToken); var resultStream = _productService.SearchProductAsync(wishlistId, message, cancellationToken);
@ -186,7 +154,6 @@ public class ProductTests
Assert.NotNull(actualSseEvents); Assert.NotNull(actualSseEvents);
Assert.Equal(expectedMessages, receivedMessages); Assert.Equal(expectedMessages, receivedMessages);
Assert.Equal(expectedSuggestions, receivedSuggestions); Assert.Equal(expectedSuggestions, receivedSuggestions);
_wishListServiceMock.Verify(w => w.AddMessageToPersonalWishlistAsync(wishlistId, It.IsAny<MessageDto>(), cancellationToken), Times.Once);
} }
@ -194,7 +161,7 @@ public class ProductTests
public async void SearchProductAsync_WithExistingMessageInWishlistAndAddProduct_ReturnsExpectedEvents() public async void SearchProductAsync_WithExistingMessageInWishlistAndAddProduct_ReturnsExpectedEvents()
{ {
// Arrange // Arrange
var wishlistId = "your_wishlist_id"; var wishlistId = "657657677c13ae4bc95e2f41";
var message = new MessageCreateDto { Text = "Your message text" }; var message = new MessageCreateDto { Text = "Your message text" };
var cancellationToken = new CancellationToken(); var cancellationToken = new CancellationToken();
@ -202,8 +169,8 @@ public class ProductTests
var expectedSseData = new List<string> var expectedSseData = new List<string>
{ {
"[", "Message", "]", " What", " u", " want", " ?", "[", "Options", "]", "USB-C", " ;", "Keyboard", " ultra", "[", "Message", "]", " What", " u", " want", " ?", "[", "Suggestions", "]", "USB-C", " ;", "Keyboard", " ultra",
" ;", "[", "Options", "]", "USB", "-C", " ;", "[", "Products", "]", " GTX", " 3090", " ;", " GTX", " ;", "[", "Suggestions", "]", "USB", "-C", " ;", "[", "Products", "]", " GTX", " 3090", " ;", " GTX",
" 3070TI", " ;", " GTX", " 4070TI", " ;", " ?" " 3070TI", " ;", " GTX", " 4070TI", " ;", " ?"
}; };
@ -214,36 +181,25 @@ public class ProductTests
_openAiServiceMock.Setup(x => x.GetChatCompletionStream(It.IsAny<ChatCompletionRequest>(), cancellationToken)) _openAiServiceMock.Setup(x => x.GetChatCompletionStream(It.IsAny<ChatCompletionRequest>(), cancellationToken))
.Returns(expectedSseData.ToAsyncEnumerable()); .Returns(expectedSseData.ToAsyncEnumerable());
_messagesRepositoryMock.Setup(m => m.GetCountAsync(It.IsAny<Expression<Func<Message, bool>>>(), It.IsAny<CancellationToken>())) _messagesRepositoryMock.Setup(m => m.GetWishlistMessagesAsync(It.IsAny<ObjectId>(), It.IsAny<CancellationToken>()))
.ReturnsAsync(3); .ReturnsAsync(new List<Message>
_wishListServiceMock
.Setup(w => w.GetMessagesPageFromPersonalWishlistAsync(
It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>(), It.IsAny<CancellationToken>()))
.ReturnsAsync(new PagedList<MessageDto>(new List<MessageDto>
{ {
new MessageDto new()
{ {
Text = "Message 1", Text = "Message 1",
Id = "1",
CreatedById = "User2",
Role = "User" Role = "User"
}, },
new MessageDto new()
{ {
Text = "Message 2", Text = "Message 2",
Id = "2",
CreatedById = "User2",
Role = "User" Role = "User"
}, },
new MessageDto new()
{ {
Text = "Message 3", Text = "Message 3",
Id = "3",
CreatedById = "User2",
Role = "User" Role = "User"
}, },
}, 1, 3, 3)); });
// Act // Act
var resultStream = _productService.SearchProductAsync(wishlistId, message, cancellationToken); var resultStream = _productService.SearchProductAsync(wishlistId, message, cancellationToken);
@ -265,7 +221,5 @@ public class ProductTests
Assert.NotNull(actualSseEvents); Assert.NotNull(actualSseEvents);
Assert.Equal(expectedMessages, receivedMessages); Assert.Equal(expectedMessages, receivedMessages);
Assert.Equal(expectedSuggestions, receivedSuggestions); Assert.Equal(expectedSuggestions, receivedSuggestions);
_wishListServiceMock.Verify(w => w.AddMessageToPersonalWishlistAsync(
wishlistId, It.IsAny<MessageDto>(), cancellationToken), Times.Once);
} }
} }