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

Merge pull request #25 from Shchoholiev/feature/SA-197-product–search-suggestions

Browse Source 
SA-197 Update Unit Tests
This commit is contained in:
Serhii Shchoholiev 2023-12-10 17:11:34 -08:00 committed by GitHub
commit 9c53b7c090
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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!"
} }
} }

154
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,47 +46,28 @@ 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", " ?"
}; };
var expectedMessages = new List<string> { " What", " u", " want", " ?", " What", " u", " want", " ?" }; var expectedMessages = new List<string> { " What", " u", " want", " ?", " What", " u", " want", " ?" };
var expectedSuggestion = new List<string> { " USB-C", " Keyboard ultra", " USB-C" }; var expectedSuggestion = new List<string> { "USB-C", "Keyboard ultra", "USB-C" };
// Mock the GetChatCompletionStream method to provide the expected SSE data // Mock the GetChatCompletionStream method to provide the expected SSE data
_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", " ?" };
@ -131,40 +114,25 @@ public class ProductTests
// Mock the GetChatCompletionStream method to provide the expected SSE data // Mock the GetChatCompletionStream method to provide the expected SSE data
_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>()))
.ReturnsAsync(3);
_wishListServiceMock.Setup(w => w.AddMessageToPersonalWishlistAsync(wishlistId, It.IsAny<MessageDto>(), cancellationToken)) _messagesRepositoryMock.Setup(m => m.GetWishlistMessagesAsync(It.IsAny<ObjectId>(), It.IsAny<CancellationToken>()))
.Verifiable(); .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
{ {
Text = "Message 1", new() {
Id = "1", Text = "Message 1",
CreatedById = "User2", Role = "User"
Role = "User" },
}, new Message
new MessageDto {
{ Text = "Message 2",
Text = "Message 2", Role = "User"
Id = "2", },
CreatedById = "User2", new Message
Role = "User" {
}, Text = "Message 3",
new MessageDto Role = "User"
{ },
Text = "Message 3", });
Id = "3",
CreatedById = "User2",
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
{ {
Text = "Message 1", new()
Id = "1", {
CreatedById = "User2", Text = "Message 1",
Role = "User" Role = "User"
}, },
new MessageDto new()
{ {
Text = "Message 2", Text = "Message 2",
Id = "2", Role = "User"
CreatedById = "User2", },
Role = "User" new()
}, {
new MessageDto Text = "Message 3",
{ Role = "User"
Text = "Message 3", },
Id = "3", });
CreatedById = "User2",
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);
} }
} }