From 453722cfade7fb20a6408b8e8daf24eeb92173a6 Mon Sep 17 00:00:00 2001
From: Meghna Verma <meghnav17@gmail.com>
Date: Wed, 8 Oct 2025 13:56:00 -0700
Subject: [PATCH 1/2] Eventhub update init

---
 .../EventHubTests.cs                          | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/EventHubTests.cs b/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/EventHubTests.cs
index b824a3a63..d5339ab3d 100644
--- a/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/EventHubTests.cs
+++ b/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/EventHubTests.cs
@@ -1,6 +1,9 @@
 
 using Azure;
 using Azure.Core;
+using Azure.Identity;
+using Azure.Messaging.EventHubs;
+using Azure.Messaging.EventHubs.Producer;
 using Azure.ResourceManager.EventHubs;
 using Azure.ResourceManager.EventHubs.Models;
 using Azure.Security.KeyVault.Secrets;
@@ -80,6 +83,37 @@ public async Task RotateConnectionStringSecretTest()
             Assert.That(connectionStringSecret.Value.Value, Is.EqualTo(accessKeysRotated.First()));
         }
 
+        [Test]
+        public async Task ManagedIdentityConnectionTest()
+        {
+            // Test direct managed identity connection to the same Event Hub
+            var credential = new DefaultAzureCredential();
+            string fullyQualifiedNamespace = $"{Namespace}.servicebus.windows.net";
+            
+            var producer = new EventHubProducerClient(fullyQualifiedNamespace, Name, credential);
+            
+            try
+            {
+                // Test that we can connect and get partition information
+                var partitions = await producer.GetPartitionIdsAsync();
+                Assert.That(partitions, Is.Not.Empty, "Should be able to get partition IDs using managed identity");
+                
+                // Test that we can send a test event
+                var testEvent = new EventData("MI test message");
+                await producer.SendAsync(new[] { testEvent });
+                
+                Assert.Pass("Managed identity authentication successful");
+            }
+            catch (Exception ex)
+            {
+                Assert.Fail($"Managed identity authentication failed: {ex.Message}");
+            }
+            finally
+            {
+                await producer.DisposeAsync();
+            }
+        }
+
         [OneTimeTearDown]
         public async Task Cleanup()
         {

From 90ec37fdb447aaf1cf60f6b7eaa5be2e267fd6eb Mon Sep 17 00:00:00 2001
From: Meghna Verma <meghnav17@gmail.com>
Date: Thu, 30 Oct 2025 00:12:46 -0700
Subject: [PATCH 2/2] Add package

---
 Directory.Packages.props                                         | 1 +
 .../Microsoft.DncEng.SecretManager.ScenarioTests.csproj          | 1 +
 2 files changed, 2 insertions(+)

diff --git a/Directory.Packages.props b/Directory.Packages.props
index 91ecbc6d5..a4eaf7220 100644
--- a/Directory.Packages.props
+++ b/Directory.Packages.props
@@ -17,6 +17,7 @@
     <PackageVersion Include="Azure.Extensions.AspNetCore.DataProtection.Blobs" Version="1.3.4" />
     <PackageVersion Include="Azure.Extensions.AspNetCore.DataProtection.Keys" Version="1.2.3" />
     <PackageVersion Include="Azure.Identity" Version="1.12.0" />
+    <PackageVersion Include="Azure.Messaging.EventHubs" Version="5.11.5" />
     <PackageVersion Include="Azure.ResourceManager.EventHubs" Version="1.1.0" />
     <PackageVersion Include="Azure.ResourceManager.ServiceBus" Version="1.1.0-beta.7" />
     <PackageVersion Include="Azure.ResourceManager.Storage" Version="1.3.0" />
diff --git a/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/Microsoft.DncEng.SecretManager.ScenarioTests.csproj b/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/Microsoft.DncEng.SecretManager.ScenarioTests.csproj
index b042cddfb..09b9e85db 100644
--- a/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/Microsoft.DncEng.SecretManager.ScenarioTests.csproj
+++ b/src/SecretManager/Microsoft.DncEng.SecretManager.ScenarioTests/Microsoft.DncEng.SecretManager.ScenarioTests.csproj
@@ -8,6 +8,7 @@
   </PropertyGroup>
 
   <ItemGroup>
+    <PackageReference Include="Azure.Messaging.EventHubs" />
     <PackageReference Include="Microsoft.DotNet.Internal.DependencyInjection.Testing" />
   </ItemGroup>