wehub-resource-sync
84 lines
2.4 KiB
C#
84 lines
2.4 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Extensions.VectorData;
|
|
using Moq;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.SemanticKernel.Connectors.Qdrant.UnitTests;
|
|
|
|
/// <summary>
|
|
/// Contains tests for the <see cref="QdrantVectorStore"/> class.
|
|
/// </summary>
|
|
public class QdrantVectorStoreTests
|
|
{
|
|
private const string TestCollectionName = "testcollection";
|
|
|
|
private readonly Mock<MockableQdrantClient> _qdrantClientMock;
|
|
|
|
private readonly CancellationToken _testCancellationToken = new(false);
|
|
|
|
public QdrantVectorStoreTests()
|
|
{
|
|
this._qdrantClientMock = new Mock<MockableQdrantClient>(MockBehavior.Strict);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetCollectionReturnsCollection()
|
|
{
|
|
// Arrange.
|
|
using var sut = new QdrantVectorStore(this._qdrantClientMock.Object);
|
|
|
|
// Act.
|
|
var actual = sut.GetCollection<ulong, SinglePropsModel<ulong>>(TestCollectionName);
|
|
|
|
// Assert.
|
|
Assert.NotNull(actual);
|
|
Assert.IsType<QdrantCollection<ulong, SinglePropsModel<ulong>>>(actual);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetCollectionThrowsForInvalidKeyType()
|
|
{
|
|
// Arrange.
|
|
using var sut = new QdrantVectorStore(this._qdrantClientMock.Object);
|
|
|
|
// Act & Assert.
|
|
Assert.Throws<NotSupportedException>(() => sut.GetCollection<string, SinglePropsModel<string>>(TestCollectionName));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ListCollectionNamesCallsSDKAsync()
|
|
{
|
|
// Arrange.
|
|
this._qdrantClientMock
|
|
.Setup(x => x.ListCollectionsAsync(It.IsAny<CancellationToken>()))
|
|
.ReturnsAsync(new[] { "collection1", "collection2" });
|
|
using var sut = new QdrantVectorStore(this._qdrantClientMock.Object);
|
|
|
|
// Act.
|
|
var collectionNames = sut.ListCollectionNamesAsync(this._testCancellationToken);
|
|
|
|
// Assert.
|
|
var collectionNamesList = await collectionNames.ToListAsync();
|
|
Assert.Equal(new[] { "collection1", "collection2" }, collectionNamesList);
|
|
}
|
|
|
|
public sealed class SinglePropsModel<TKey>
|
|
{
|
|
[VectorStoreKey]
|
|
public required TKey Key { get; set; }
|
|
|
|
[VectorStoreData]
|
|
public string Data { get; set; } = string.Empty;
|
|
|
|
[VectorStoreVector(4)]
|
|
public ReadOnlyMemory<float>? Vector { get; set; }
|
|
|
|
public string? NotAnnotated { get; set; }
|
|
}
|
|
}
|