chore: import upstream snapshot with attribution
This commit is contained in:
+84
@@ -0,0 +1,84 @@
|
||||
// Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
using Microsoft.Extensions.AI;
|
||||
using Microsoft.Extensions.Configuration;
|
||||
using Microsoft.Extensions.DependencyInjection;
|
||||
using Microsoft.SemanticKernel.Connectors.SqlServer;
|
||||
using VectorData.ConformanceTests;
|
||||
using Xunit;
|
||||
|
||||
namespace SqlServer.ConformanceTests;
|
||||
|
||||
public class SqlServerDependencyInjectionTests
|
||||
: DependencyInjectionTests<SqlServerVectorStore, SqlServerCollection<string, DependencyInjectionTests<string>.Record>, string, DependencyInjectionTests<string>.Record>
|
||||
{
|
||||
protected const string ConnectionString = "Server=localhost;Database=master;Integrated Security=True;";
|
||||
|
||||
protected override void PopulateConfiguration(ConfigurationManager configuration, object? serviceKey = null)
|
||||
=> configuration.AddInMemoryCollection(
|
||||
[
|
||||
new(CreateConfigKey("SqlServer", serviceKey, "ConnectionString"), ConnectionString),
|
||||
]);
|
||||
|
||||
private static string ConnectionStringProvider(IServiceProvider sp)
|
||||
=> sp.GetRequiredService<IConfiguration>().GetRequiredSection("SqlServer:ConnectionString").Value!;
|
||||
|
||||
private static string ConnectionStringProvider(IServiceProvider sp, object serviceKey)
|
||||
=> sp.GetRequiredService<IConfiguration>().GetRequiredSection(CreateConfigKey("SqlServer", serviceKey, "ConnectionString")).Value!;
|
||||
|
||||
public override IEnumerable<Func<IServiceCollection, object?, string, ServiceLifetime, IServiceCollection>> CollectionDelegates
|
||||
{
|
||||
get
|
||||
{
|
||||
yield return (services, serviceKey, name, lifetime) => serviceKey is null
|
||||
? services.AddSqlServerCollection<string, Record>(
|
||||
name, connectionString: ConnectionString, lifetime: lifetime)
|
||||
: services.AddKeyedSqlServerCollection<string, Record>(
|
||||
serviceKey, name, connectionString: ConnectionString, lifetime: lifetime);
|
||||
|
||||
yield return (services, serviceKey, name, lifetime) => serviceKey is null
|
||||
? services.AddSqlServerCollection<string, Record>(
|
||||
name, ConnectionStringProvider, lifetime: lifetime)
|
||||
: services.AddKeyedSqlServerCollection<string, Record>(
|
||||
serviceKey, name, sp => ConnectionStringProvider(sp, serviceKey), lifetime: lifetime);
|
||||
}
|
||||
}
|
||||
|
||||
public override IEnumerable<Func<IServiceCollection, object?, ServiceLifetime, IServiceCollection>> StoreDelegates
|
||||
{
|
||||
get
|
||||
{
|
||||
yield return (services, serviceKey, lifetime) => serviceKey is null
|
||||
? services.AddSqlServerVectorStore(
|
||||
ConnectionStringProvider, lifetime: lifetime)
|
||||
: services.AddKeyedSqlServerVectorStore(
|
||||
serviceKey, sp => ConnectionStringProvider(sp, serviceKey), lifetime: lifetime);
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void ConnectionStringProviderCantBeNull()
|
||||
{
|
||||
IServiceCollection services = new ServiceCollection();
|
||||
|
||||
Assert.Throws<ArgumentNullException>(() => services.AddSqlServerVectorStore(connectionStringProvider: null!));
|
||||
Assert.Throws<ArgumentNullException>(() => services.AddKeyedSqlServerVectorStore(serviceKey: "notNull", connectionStringProvider: null!));
|
||||
Assert.Throws<ArgumentNullException>(() => services.AddSqlServerCollection<string, Record>(name: "notNull", connectionStringProvider: null!));
|
||||
Assert.Throws<ArgumentNullException>(() => services.AddKeyedSqlServerCollection<string, Record>(serviceKey: "notNull", name: "notNull", connectionStringProvider: null!));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void ConnectionStringCantBeNullOrEmpty()
|
||||
{
|
||||
IServiceCollection services = new ServiceCollection();
|
||||
|
||||
Assert.Throws<ArgumentNullException>(() => services.AddSqlServerCollection<string, Record>(
|
||||
name: "notNull", connectionString: null!));
|
||||
Assert.Throws<ArgumentException>(() => services.AddSqlServerCollection<string, Record>(
|
||||
name: "notNull", connectionString: ""));
|
||||
Assert.Throws<ArgumentNullException>(() => services.AddKeyedSqlServerCollection<string, Record>(
|
||||
serviceKey: "notNull", name: "notNull", connectionString: null!));
|
||||
Assert.Throws<ArgumentException>(() => services.AddKeyedSqlServerCollection<string, Record>(
|
||||
serviceKey: "notNull", name: "notNull", connectionString: ""));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user