add unit tests

Author: Aditya Abhishek

src/VirtualClient/VirtualClient.Contracts.UnitTests/KeyVaultDescriptorTests.cs

@@ -0,0 +1,104 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+using NUnit.Framework;
+using VirtualClient.Contracts;
+using VirtualClient.TestExtensions;
+
+namespace VirtualClient
+{
+    [TestFixture]
+    [Category("Unit")]
+    public class KeyVaultDescriptorTests
+    {
+        private KeyVaultDescriptor descriptor;
+        private KeyVaultDescriptor descriptor2;
+
+        [SetUp]
+        public void SetupTest()
+        {
+            this.descriptor = new KeyVaultDescriptor
+            {
+                ObjectType = KeyVaultObjectType.Secret,
+                VaultUri = "https://myvault.vault.azure.net/",
+                Name = "mysecret",
+                Version = "v1",
+                Value = "secret-value",
+                ObjectId = "object-id-1",
+                Policy = "policy-1"
+            };
+
+            this.descriptor2 = new KeyVaultDescriptor
+            {
+                ObjectType = KeyVaultObjectType.Key,
+                VaultUri = "https://othervault.vault.azure.net/",
+                Name = "mykey",
+                Version = "v2",
+                Value = "key-value",
+                ObjectId = "object-id-2",
+                Policy = "policy-2"
+            };
+        }
+
+        [Test]
+        public void KeyVaultDescriptorCorrectlyImplementsEqualitySemantics()
+        {
+            EqualityAssert.CorrectlyImplementsEqualitySemantics<DependencyDescriptor>(() => this.descriptor, () => this.descriptor2);
+        }
+
+        [Test]
+        public void KeyVaultDescriptorEqualitySemanticsAreNotAffectedByNullPropertyValues()
+        {
+            this.descriptor.VaultUri = null;
+            this.descriptor2.VaultUri = null;
+            this.descriptor.Version = null;
+            this.descriptor2.Version = null;
+            this.descriptor.Value = null;
+            this.descriptor2.Value = null;
+
+            EqualityAssert.CorrectlyImplementsEqualitySemantics<DependencyDescriptor>(() => this.descriptor, () => this.descriptor2);
+        }
+
+        [Test]
+        public void KeyVaultDescriptorCorrectlyImplementsHashcodeSemantics()
+        {
+            EqualityAssert.CorrectlyImplementsHashcodeSemantics(() => this.descriptor, () => this.descriptor2);
+        }
+
+        [Test]
+        public void KeyVaultDescriptorHashcodeSemanticsAreNotAffectedByNullPropertyValues()
+        {
+            this.descriptor.VaultUri = null;
+            this.descriptor2.VaultUri = null;
+            this.descriptor.Value = null;
+            this.descriptor2.Value = null;
+            this.descriptor.ObjectId = null;
+            this.descriptor2.ObjectId = null;
+
+            EqualityAssert.CorrectlyImplementsHashcodeSemantics(() => this.descriptor, () => this.descriptor2);
+        }
+
+        [Test]
+        public void KeyVaultDescriptorPropertiesAreSetAndGetCorrectly()
+        {
+            var descriptor = new KeyVaultDescriptor
+            {
+                ObjectType = KeyVaultObjectType.Certificate,
+                VaultUri = "https://vault.uri/",
+                Name = "cert",
+                Version = "v3",
+                Value = "cert-value",
+                ObjectId = "object-id-3",
+                Policy = "policy-3"
+            };
+
+            Assert.AreEqual(KeyVaultObjectType.Certificate, descriptor.ObjectType);
+            Assert.AreEqual("https://vault.uri/", descriptor.VaultUri);
+            Assert.AreEqual("cert", descriptor.Name);
+            Assert.AreEqual("v3", descriptor.Version);
+            Assert.AreEqual("cert-value", descriptor.Value);
+            Assert.AreEqual("object-id-3", descriptor.ObjectId);
+            Assert.AreEqual("policy-3", descriptor.Policy);
+        }
+    }
+}

src/VirtualClient/VirtualClient.Core.UnitTests/DependencyFactoryTests.cs

@@ -0,0 +1,47 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+using System;
+using NUnit.Framework;
+using VirtualClient.Contracts;
+using VirtualClient;
+
+namespace VirtualClient.Core.UnitTests
+{
+    [TestFixture]
+    [Category("Unit")]
+    public class DependencyFactoryTests
+    {
+        [Test]
+        public void CreateKeyVaultManager_ThrowsIfDependencyStoreIsNotKeyVaultStore()
+        {
+            // Arrange: Use a base DependencyBlobStore, not a DependencyKeyVaultStore
+            var store = new DependencyBlobStore("KeyVault", new Uri("https://myblob.azure.net/"));
+
+            // Act & Assert
+            var ex = Assert.Throws<DependencyException>(() => DependencyFactory.CreateKeyVaultManager(store));
+            StringAssert.Contains("Required Key Vault information not provided", ex.Message);
+        }
+
+        [Test]
+        public void CreateKeyVaultManager_ReturnsKeyVaultManagerForValidStore()
+        {
+            // Arrange: Use a valid DependencyKeyVaultStore
+            var keyVaultStore = new DependencyKeyVaultStore("KeyVault", new Uri("https://myvault.vault.azure.net/"));
+
+            // Act
+            var manager = DependencyFactory.CreateKeyVaultManager(keyVaultStore);
+
+            // Assert
+            Assert.IsNotNull(manager);
+            Assert.IsInstanceOf<IKeyVaultManager>(manager);
+        }
+
+        [Test]
+        public void CreateKeyVaultManager_ThrowsIfNullPassed()
+        {
+            // Act & Assert
+            Assert.Throws<DependencyException>(() => DependencyFactory.CreateKeyVaultManager(null));
+        }
+    }
+}

src/VirtualClient/VirtualClient.Core.UnitTests/EndpointUtilityTests.cs

@@ -10,8 +10,9 @@ namespace VirtualClient
     using AutoFixture;
     using Azure.Identity;
     using Moq;
-    using NUnit.Framework;
+    using NUnit.Framework;    
     using VirtualClient.Contracts;
+    using VirtualClient.Identity;
     using VirtualClient.TestExtensions;
 
     [TestFixture]
@@ -795,5 +796,63 @@ public void EndpointUtilityThrowsWhenCreatingAProfileReferenceIfTheValueProvided
                 invalidEndpoint,
                 this.mockFixture.CertificateManager.Object));
         }
+
+        [Test]
+        [TestCase(
+            "https://my-keyvault.vault.azure.net/?cid=985bbc17-e3a5-4fec-b0cb-40dbb8bc5959&tid=307591a4-abb2-4559-af59-b47177d140cf&crtt=1234567",
+            "https://my-keyvault.vault.azure.net/")]
+        [TestCase(
+            "Endpoint=https://my-keyvault.vault.azure.net/;CertificateThumbprint=1234567;ClientId=985bbc17;TenantId=307591a4",
+            "https://my-keyvault.vault.azure.net/")]
+        public void EndpointUtility_CreateKeyVaultStoreReference_Entra(string connectionString, string expectedUri)
+        {
+            // Setup: A matching certificate is found in the local store.
+            this.mockFixture.CertificateManager.Setup(mgr => mgr.GetCertificateFromStoreAsync("1234567", It.IsAny<IEnumerable<StoreLocation>>(), It.IsAny<StoreName>()))
+                .ReturnsAsync(this.mockFixture.Create<X509Certificate2>());
+
+            var store = EndpointUtility.CreateKeyVaultStoreReference(
+                DependencyStore.KeyVault,
+                connectionString,
+                this.mockFixture.CertificateManager.Object);
+
+            Assert.IsNotNull(store);
+            Assert.AreEqual(DependencyStore.KeyVault, store.StoreName);
+            Assert.AreEqual(DependencyStore.StoreTypeAzureKeyVault, store.StoreType);
+            Assert.AreEqual(new Uri(expectedUri).ToString(), store.EndpointUri.ToString());
+            Assert.IsNotNull(store.Credentials);
+            Assert.IsInstanceOf<ClientCertificateCredential>(store.Credentials);
+        }
+
+        [Test]
+        [TestCase(
+            "Endpoint=https://my-keyvault.vault.azure.net/;ManagedIdentityId=307591a4-abb2-4559-af59-b47177d140cf",
+            "https://my-keyvault.vault.azure.net/")]
+        [TestCase(
+            "https://my-keyvault.vault.azure.net/?miid=307591a4-abb2-4559-af59-b47177d140cf",
+            "https://my-keyvault.vault.azure.net/")]
+        public void EndpointUtility_CreateKeyVaultStoreReference_Miid(string connectionString, string expectedUri)
+        {
+            var store = EndpointUtility.CreateKeyVaultStoreReference(
+                DependencyStore.KeyVault,
+                connectionString,
+                this.mockFixture.CertificateManager.Object);
+
+            Assert.IsNotNull(store);
+            Assert.AreEqual(DependencyStore.KeyVault, store.StoreName);
+            Assert.AreEqual(DependencyStore.StoreTypeAzureKeyVault, store.StoreType);
+            Assert.AreEqual(new Uri(expectedUri).ToString(), store.EndpointUri.ToString());
+            Assert.IsNotNull(store.Credentials);
+            Assert.IsInstanceOf<ManagedIdentityCredential>(store.Credentials);
+        }
+
+        [Test]
+        public void CreateKeyVaultStoreReference_ConnectionString_ThrowsOnInvalid()
+        {
+            Assert.Throws<SchemaException>(() =>
+                EndpointUtility.CreateKeyVaultStoreReference(
+                    DependencyStore.KeyVault,
+                    "InvalidConnectionString",
+                    this.mockFixture.CertificateManager.Object));
+        }
     }
 }