Adjust tests

microsoft · Nov 12, 2024 · 96e6c63 · 96e6c63
1 parent 32d3793
commit 96e6c63
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/extensions/AzureAISearch/AzureAISearch.FunctionalTests/DefaultTests.cs b/extensions/AzureAISearch/AzureAISearch.FunctionalTests/DefaultTests.cs
@@ -14,14 +14,14 @@ public class DefaultTests : BaseFunctionalTestCase
     public DefaultTests(IConfiguration cfg, ITestOutputHelper output) : base(cfg, output)
     {
         Assert.False(string.IsNullOrEmpty(this.AzureAiSearchConfig.Endpoint));
-        Assert.False(string.IsNullOrEmpty(this.AzureAiSearchConfig.APIKey));
+        Assert.False(this.AzureAiSearchConfig.Auth == AzureAISearchConfig.AuthTypes.APIKey && string.IsNullOrEmpty(this.AzureAiSearchConfig.APIKey));
         Assert.False(string.IsNullOrEmpty(this.OpenAiConfig.APIKey));
 
         this._memory = new KernelMemoryBuilder()
             .With(new KernelMemoryConfig { DefaultIndexName = "default4tests" })
             .WithSearchClientConfig(new SearchClientConfig { EmptyAnswer = NotFound })
             .WithOpenAI(this.OpenAiConfig)
-            .WithAzureAISearchMemoryDb(this.AzureAiSearchConfig.Endpoint, this.AzureAiSearchConfig.APIKey)
+            .WithAzureAISearchMemoryDb(this.AzureAiSearchConfig)
             .Build<MemoryServerless>();
     }
 

diff --git a/service/tests/Abstractions.UnitTests/Diagnostics/SensitiveDataLoggerTests.cs b/service/tests/Abstractions.UnitTests/Diagnostics/SensitiveDataLoggerTests.cs
@@ -10,6 +10,7 @@ public sealed class SensitiveDataLoggerTests : IDisposable
     private const string DotNetEnvVar = "DOTNET_ENVIRONMENT";
 
     [Fact]
+    [Trait("Category", "UnitTest")]
     public void ItIsDisabledByDefault()
     {
         // Assert