wehub-resource-sync
89 lines
3.0 KiB
C#
89 lines
3.0 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using Microsoft.Extensions.VectorData;
|
|
using Microsoft.SemanticKernel.Connectors.MongoDB;
|
|
using MongoDB.Bson;
|
|
using Xunit;
|
|
|
|
namespace SemanticKernel.Connectors.MongoDB.UnitTests;
|
|
|
|
/// <summary>
|
|
/// Unit tests for <see cref="MongoMapper{TRecord}"/> class.
|
|
/// </summary>
|
|
public sealed class MongoMapperTests
|
|
{
|
|
private readonly MongoMapper<MongoHotelModel> _sut;
|
|
|
|
public MongoMapperTests()
|
|
{
|
|
var keyProperty = new VectorStoreKeyProperty("HotelId", typeof(string));
|
|
|
|
var definition = new VectorStoreCollectionDefinition
|
|
{
|
|
Properties =
|
|
[
|
|
keyProperty,
|
|
new VectorStoreDataProperty("HotelName", typeof(string)),
|
|
new VectorStoreDataProperty("Tags", typeof(List<string>)),
|
|
new VectorStoreDataProperty("ParkingIncluded", typeof(bool)),
|
|
new VectorStoreVectorProperty("DescriptionEmbedding", typeof(ReadOnlyMemory<float>?), 10)
|
|
]
|
|
};
|
|
|
|
this._sut = new(new MongoModelBuilder().Build(typeof(MongoHotelModel), typeof(string), definition, defaultEmbeddingGenerator: null));
|
|
}
|
|
|
|
[Fact]
|
|
public void MapFromDataToStorageModelReturnsValidObject()
|
|
{
|
|
// Arrange
|
|
var hotel = new MongoHotelModel("key")
|
|
{
|
|
HotelName = "Test Name",
|
|
Tags = ["tag1", "tag2"],
|
|
ParkingIncluded = true,
|
|
DescriptionEmbedding = new ReadOnlyMemory<float>([1f, 2f, 3f])
|
|
};
|
|
|
|
// Act
|
|
var document = this._sut.MapFromDataToStorageModel(hotel, recordIndex: 0, generatedEmbeddings: null);
|
|
|
|
// Assert
|
|
Assert.NotNull(document);
|
|
|
|
Assert.Equal("key", document["_id"]);
|
|
Assert.Equal("Test Name", document["HotelName"]);
|
|
Assert.Equal(["tag1", "tag2"], document["Tags"].AsBsonArray);
|
|
Assert.True(document["parking_is_included"].AsBoolean);
|
|
Assert.Equal([1f, 2f, 3f], document["DescriptionEmbedding"].AsBsonArray);
|
|
}
|
|
|
|
[Fact]
|
|
public void MapFromStorageToDataModelReturnsValidObject()
|
|
{
|
|
// Arrange
|
|
var document = new BsonDocument
|
|
{
|
|
["_id"] = "key",
|
|
["HotelName"] = "Test Name",
|
|
["Tags"] = BsonArray.Create(new List<string> { "tag1", "tag2" }),
|
|
["parking_is_included"] = BsonValue.Create(true),
|
|
["DescriptionEmbedding"] = BsonArray.Create(new List<float> { 1f, 2f, 3f })
|
|
};
|
|
|
|
// Act
|
|
var hotel = this._sut.MapFromStorageToDataModel(document, includeVectors: true);
|
|
|
|
// Assert
|
|
Assert.NotNull(hotel);
|
|
|
|
Assert.Equal("key", hotel.HotelId);
|
|
Assert.Equal("Test Name", hotel.HotelName);
|
|
Assert.Equal(["tag1", "tag2"], hotel.Tags);
|
|
Assert.True(hotel.ParkingIncluded);
|
|
Assert.True(new ReadOnlyMemory<float>([1f, 2f, 3f]).Span.SequenceEqual(hotel.DescriptionEmbedding!.Value.Span));
|
|
}
|
|
}
|