chore: import upstream snapshot with attribution

2026-07-13 13:21:23 +08:00
commit b957a53def
5423 changed files with 863745 additions and 0 deletions
@@ -0,0 +1,106 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System;
+using System.Linq;
+using System.Threading.Tasks;
+using Microsoft.Extensions.VectorData;
+using Moq;
+using StackExchange.Redis;
+using Xunit;
+
+namespace Microsoft.SemanticKernel.Connectors.Redis.UnitTests;
+
+/// <summary>
+/// Contains tests for the <see cref="RedisVectorStore"/> class.
+/// </summary>
+public class RedisVectorStoreTests
+{
+    private const string TestCollectionName = "testcollection";
+
+    private readonly Mock<IDatabase> _redisDatabaseMock;
+
+    public RedisVectorStoreTests()
+    {
+        this._redisDatabaseMock = new Mock<IDatabase>(MockBehavior.Strict);
+        this._redisDatabaseMock.Setup(l => l.Database).Returns(0);
+
+        var batchMock = new Mock<IBatch>();
+        this._redisDatabaseMock.Setup(x => x.CreateBatch(It.IsAny<object>())).Returns(batchMock.Object);
+    }
+
+    [Fact]
+    public void GetCollectionReturnsJsonCollection()
+    {
+        // Arrange.
+        using var sut = new RedisVectorStore(this._redisDatabaseMock.Object);
+
+        // Act.
+        var actual = sut.GetCollection<string, SinglePropsModel<string>>(TestCollectionName);
+
+        // Assert.
+        Assert.NotNull(actual);
+        Assert.IsType<RedisJsonCollection<string, SinglePropsModel<string>>>(actual);
+    }
+
+    [Fact]
+    public void GetCollectionReturnsHashSetCollection()
+    {
+        // Arrange.
+        using var sut = new RedisVectorStore(this._redisDatabaseMock.Object, new() { StorageType = RedisStorageType.HashSet });
+
+        // Act.
+        var actual = sut.GetCollection<string, SinglePropsModel<string>>(TestCollectionName);
+
+        // Assert.
+        Assert.NotNull(actual);
+        Assert.IsType<RedisHashSetCollection<string, SinglePropsModel<string>>>(actual);
+    }
+
+    [Fact]
+    public void GetCollectionThrowsForInvalidKeyType()
+    {
+        // Arrange.
+        using var sut = new RedisVectorStore(this._redisDatabaseMock.Object);
+
+        // Act & Assert.
+        Assert.Throws<NotSupportedException>(() => sut.GetCollection<int, SinglePropsModel<int>>(TestCollectionName));
+    }
+
+    [Fact]
+    public async Task ListCollectionNamesCallsSDKAsync()
+    {
+        // Arrange.
+        var redisResultStrings = new string[] { "collection1", "collection2" };
+        var results = redisResultStrings
+            .Select(x => RedisResult.Create(new RedisValue(x)))
+            .ToArray();
+        this._redisDatabaseMock
+            .Setup(
+                x => x.ExecuteAsync(
+                    It.IsAny<string>(),
+                It.IsAny<object[]>()))
+            .ReturnsAsync(RedisResult.Create(results));
+        using var sut = new RedisVectorStore(this._redisDatabaseMock.Object);
+
+        // Act.
+        var collectionNames = sut.ListCollectionNamesAsync();
+
+        // 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; }
+    }
+}