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

added some fix for tests

Browse Source
This commit is contained in:
stasex 2023-10-24 23:24:23 +03:00
parent 3372a0910b
commit f5d9c3e80e
3 changed files with 76 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

147
ShoppingAssistantApi.Infrastructure/Services/ProductService.cs
View File

@ -1,4 +1,5 @@
using ShoppingAssistantApi.Application.IServices;
using System.Diagnostics;
using ShoppingAssistantApi.Application.IServices;
using ShoppingAssistantApi.Application.Models.CreateDtos;
using ShoppingAssistantApi.Application.Models.Dtos;
using ShoppingAssistantApi.Application.Models.OpenAi;
@ -24,7 +25,6 @@ public class ProductService : IProductService
public async IAsyncEnumerable<ServerSentEvent> SearchProductAsync(string wishlistId, MessageCreateDto message, CancellationToken cancellationToken)
{
bool checker = false;
var isFirstMessage = _wishlistsService
.GetMessagesPageFromPersonalWishlistAsync(wishlistId, 1, 1, cancellationToken).Result;
@ -78,11 +78,9 @@ public class ProductService : IProductService
Event = SearchEventType.Suggestion,
Data = "Laptop"
};
checker = true;
}
if(isFirstMessage!=null && checker==false)
if(isFirstMessage!=null)
{
var previousMessages = _wishlistsService
.GetMessagesPageFromPersonalWishlistAsync(wishlistId, 1, 1, cancellationToken).Result.Items.ToList();
@ -105,91 +103,91 @@ public class ProductService : IProductService
};
var suggestionBuffer = new Suggestion();
var messageBuffer = new MessagePart();
var productBuffer = new ProductName();
var currentDataType = SearchEventType.Wishlist;
var dataTypeHolder = string.Empty;
var messageBuffer = new MessagePart();
var productBuffer = new ProductName();
var currentDataType = SearchEventType.Wishlist;
var dataTypeHolder = string.Empty;
await foreach (var data in _openAiService.GetChatCompletionStream(chatRequest, cancellationToken))
{
if (data.Contains("["))
await foreach (var data in _openAiService.GetChatCompletionStream(chatRequest, cancellationToken))
{
if (dataTypeHolder=="[Message]" && messageBuffer.Text!=null)
if (data.Contains("["))
{
_wishlistsService.AddMessageToPersonalWishlistAsync(wishlistId, new MessageCreateDto()
if (dataTypeHolder=="[Message]" && messageBuffer.Text!=null)
{
Text = messageBuffer.Text,
}, cancellationToken);
}
if (dataTypeHolder=="[Products]" && productBuffer.Name!=null)
{
_wishlistsService.AddProductToPersonalWishlistAsync(wishlistId, new ProductCreateDto()
_wishlistsService.AddMessageToPersonalWishlistAsync(wishlistId, new MessageCreateDto()
{
Text = messageBuffer.Text,
}, cancellationToken);
}
if (dataTypeHolder=="[Products]" && productBuffer.Name!=null)
{
Url = "",
Name = productBuffer.Name,
Rating = 0,
Description = "",
ImagesUrls = new []{"", ""},
WasOpened = false
}, cancellationToken);
_wishlistsService.AddProductToPersonalWishlistAsync(wishlistId, new ProductCreateDto()
{
Url = "",
Name = productBuffer.Name,
Rating = 0,
Description = "",
ImagesUrls = new []{"", ""},
WasOpened = false
}, cancellationToken);
}
dataTypeHolder = string.Empty;
dataTypeHolder += data;
}
dataTypeHolder = string.Empty;
dataTypeHolder += data;
}
else if (data.Contains("]"))
{
dataTypeHolder += data;
currentDataType = DetermineDataType(dataTypeHolder);
}
else if (dataTypeHolder=="[" && !data.Contains("["))
{
dataTypeHolder += data;
}
else
{
switch (currentDataType)
else if (data.Contains("]"))
{
case SearchEventType.Message:
yield return new ServerSentEvent
{
Event = SearchEventType.Message,
Data = data
};
messageBuffer.Text += data;
break;
dataTypeHolder += data;
currentDataType = DetermineDataType(dataTypeHolder);
}
case SearchEventType.Suggestion:
suggestionBuffer.Text += data;
if (data.Contains(";"))
{
else if (dataTypeHolder=="[" && !data.Contains("["))
{
dataTypeHolder += data;
}
else
{
switch (currentDataType)
{
case SearchEventType.Message:
yield return new ServerSentEvent
{
Event = SearchEventType.Suggestion,
Data = suggestionBuffer.Text
Event = SearchEventType.Message,
Data = data
};
suggestionBuffer.Text = string.Empty;
}
break;
case SearchEventType.Product:
productBuffer.Name += data;
if (data.Contains(";"))
{
yield return new ServerSentEvent
messageBuffer.Text += data;
break;
case SearchEventType.Suggestion:
suggestionBuffer.Text += data;
if (data.Contains(";"))
{
Event = SearchEventType.Product,
Data = productBuffer.Name
};
productBuffer.Name = string.Empty;
}
break;
yield return new ServerSentEvent
{
Event = SearchEventType.Suggestion,
Data = suggestionBuffer.Text
};
suggestionBuffer.Text = string.Empty;
}
break;
case SearchEventType.Product:
productBuffer.Name += data;
if (data.Contains(";"))
{
yield return new ServerSentEvent
{
Event = SearchEventType.Product,
Data = productBuffer.Name
};
productBuffer.Name = string.Empty;
}
break;
}
}
}
}
}
}
private SearchEventType DetermineDataType(string dataTypeHolder)
@ -215,4 +213,5 @@ public class ProductService : IProductService
return SearchEventType.Wishlist;
}
}
}

6
ShoppingAssistantApi.Tests/Tests/ProductsTests.cs
View File

@ -19,15 +19,13 @@ public class ProductsTests : TestsBase
public async Task StreamDataToClient_ReturnsExpectedResponse()
{
// Arrange
var wishlistId = "your_wishlist_id";
var wishlistId = "ab79cde6f69abcd3efab65cd";
var message = new MessageCreateDto { Text = "Your message text" };
// Act
var response = await _httpClient.PostAsJsonAsync($"http://localhost:5183/api/products/search/{"ab79cde6f69abcd3efab65cd"}", message);
var response = await _httpClient.PostAsJsonAsync($"http://127.0.0.1:5183//api/products/search/{wishlistId}", message);
// Assert
Assert.Equal(HttpStatusCode.OK, response.StatusCode);
// Додайте додаткові перевірки на відповідь, якщо необхідно
}
}

1
ShoppingAssistantApi.UnitTests/ProductTests.cs
View File

@ -168,5 +168,6 @@ public class ProductTests
// Assert
Assert.NotNull(actualSseEvents);
Assert.Equal(3, actualSseEvents.Count);
}
}