[DocumentIntelligence] Enabling live testing (#43455)

Azure · Apr 29, 2024 · f3f1c6a · f3f1c6a
1 parent 151e662
commit f3f1c6a
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/...Azure.AI.DocumentIntelligence/tests/Infrastructure/DocumentIntelligenceTestEnvironment.cs b/...Azure.AI.DocumentIntelligence/tests/Infrastructure/DocumentIntelligenceTestEnvironment.cs
@@ -4,6 +4,7 @@
 using System;
 using System.IO;
 using System.Reflection;
+using System.Threading.Tasks;
 using Azure.Core.TestFramework;
 
 namespace Azure.AI.DocumentIntelligence.Tests
@@ -51,5 +52,23 @@ public static BinaryData CreateBinaryData(string filename)
 
             return BinaryData.FromBytes(bytes);
         }
+
+        protected override async ValueTask<bool> IsEnvironmentReadyAsync()
+        {
+            var endpoint = new Uri(Endpoint);
+            var keyCredential = new AzureKeyCredential(ApiKey);
+            var keyCredentialClient = new DocumentIntelligenceAdministrationClient(endpoint, keyCredential);
+
+            try
+            {
+                await keyCredentialClient.GetResourceInfoAsync();
+            }
+            catch (RequestFailedException e) when (e.Status == 401)
+            {
+                return false;
+            }
+
+            return true;
+        }
     }
 }
diff --git a/sdk/documentintelligence/tests.yml b/sdk/documentintelligence/tests.yml
@@ -4,4 +4,8 @@ extends:
   template: ../../eng/pipelines/templates/stages/archetype-sdk-tests.yml
   parameters:
     ServiceDirectory: documentintelligence
+    CloudConfig:
+      Public:
+        SubscriptionConfiguration: $(sub-config-azure-cloud-test-resources)
+        Location: 'eastus'
     SupportedClouds: 'Public'