chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,80 @@
|
||||
// Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Net;
|
||||
using System.Net.Http;
|
||||
using System.Text.Json;
|
||||
using System.Threading.Tasks;
|
||||
using Microsoft.SemanticKernel.Connectors.Weaviate;
|
||||
using Xunit;
|
||||
|
||||
namespace SemanticKernel.Connectors.Weaviate.UnitTests;
|
||||
|
||||
/// <summary>
|
||||
/// Unit tests for <see cref="WeaviateVectorStore"/> class.
|
||||
/// </summary>
|
||||
public sealed class WeaviateVectorStoreTests : IDisposable
|
||||
{
|
||||
private readonly HttpMessageHandlerStub _messageHandlerStub = new();
|
||||
private readonly HttpClient _mockHttpClient;
|
||||
|
||||
public WeaviateVectorStoreTests()
|
||||
{
|
||||
this._mockHttpClient = new(this._messageHandlerStub, false) { BaseAddress = new Uri("http://test") };
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void GetCollectionWithNotSupportedKeyThrowsException()
|
||||
{
|
||||
// Arrange
|
||||
using var sut = new WeaviateVectorStore(this._mockHttpClient);
|
||||
|
||||
// Act & Assert
|
||||
Assert.Throws<NotSupportedException>(() => sut.GetCollection<string, WeaviateHotel>("Collection"));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void GetCollectionWithSupportedKeyReturnsCollection()
|
||||
{
|
||||
// Arrange
|
||||
using var sut = new WeaviateVectorStore(this._mockHttpClient);
|
||||
|
||||
// Act
|
||||
var collection = sut.GetCollection<Guid, WeaviateHotel>("Collection1");
|
||||
|
||||
// Assert
|
||||
Assert.NotNull(collection);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ListCollectionNamesReturnsCollectionNamesAsync()
|
||||
{
|
||||
// Arrange
|
||||
var expectedCollectionNames = new List<string> { "Collection1", "Collection2", "Collection3" };
|
||||
var response = new WeaviateGetCollectionsResponse
|
||||
{
|
||||
Collections = expectedCollectionNames.Select(name => new WeaviateCollectionSchema(name)).ToList()
|
||||
};
|
||||
|
||||
this._messageHandlerStub.ResponseToReturn = new HttpResponseMessage(HttpStatusCode.OK)
|
||||
{
|
||||
Content = new StringContent(JsonSerializer.Serialize(response))
|
||||
};
|
||||
|
||||
using var sut = new WeaviateVectorStore(this._mockHttpClient);
|
||||
|
||||
// Act
|
||||
var actualCollectionNames = await sut.ListCollectionNamesAsync().ToListAsync();
|
||||
|
||||
// Assert
|
||||
Assert.Equal(expectedCollectionNames, actualCollectionNames);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
this._mockHttpClient.Dispose();
|
||||
this._messageHandlerStub.Dispose();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user