Azure OpenAI test management update (Azure#33821)

* Initial update

* More progress; token?

* Token-based recordings

* PR feedback and concurrency fix

* Reverting opportunistic changes to client to proactively address merge conflicts with !33834

* PR feedback: concurrency/deployment initialization optimization

* Merged files that weren't merged -- oops!

Author: trrwilson

sdk/openai/Azure.AI.OpenAI/tests/OpenAIInferenceTests.cs

@@ -38,18 +38,35 @@ public async Task CompletionTest()
             completionsRequest.Prompt.Add("Hello world");
             completionsRequest.Prompt.Add("running over the same old ground");
             Assert.That(completionsRequest, Is.InstanceOf<CompletionsOptions>());
-            var response = await client.GetCompletionsAsync(DeploymentId, completionsRequest);
+            var response = await client.GetCompletionsAsync(CompletionsDeploymentId, completionsRequest);
             Assert.That(response, Is.InstanceOf<Response<Completions>>());
         }
 
+        /// <summary>
+        /// Test Completions using a TokenCredential.
+        /// </summary>
+        [RecordedTest]
+        public async Task CompletionTestWithTokenCredential()
+        {
+            OpenAIClient client = GetClientWithCredential();
+            CompletionsOptions requestOptions = new CompletionsOptions();
+            requestOptions.Prompt.Add("Hello, world!");
+            requestOptions.Prompt.Add("I can have multiple prompts");
+            Assert.That(requestOptions, Is.InstanceOf<CompletionsOptions>());
+            Response<Completions> response = await client.GetCompletionsAsync(CompletionsDeploymentId, requestOptions);
+            Assert.That(response, Is.InstanceOf<Response<Completions>>());
+            Assert.That(response.Value.Choices, Is.Not.Null.Or.Empty);
+            Assert.That(response.Value.Choices.Count, Is.EqualTo(2));
+        }
+
         /// <summary>
         /// Test Simplified Completion API.
         /// </summary>
         [RecordedTest]
         public async Task SimpleCompletionTest()
         {
-            var client = GetClientWithCompletionsDeploymentId();
-            var response = await client.GetCompletionsAsync(DeploymentId, "Hello World!");
+            var client = GetClientWithCredential();
+            var response = await client.GetCompletionsAsync(CompletionsDeploymentId, "Hello World!");
             Assert.That(response, Is.InstanceOf<Response<Completions>>());
         }
 
@@ -76,7 +93,7 @@ public void RequestFailedExceptionTest()
             CompletionsOptions completionsRequest = new CompletionsOptions();
             completionsRequest.Prompt.Add("Hello world");
             var exception = Assert.ThrowsAsync<RequestFailedException>(async () => { await client.GetCompletionsAsync("BAD_DEPLOYMENT_ID", completionsRequest); });
-            Assert.AreEqual(401, exception.Status);
+            Assert.AreEqual(404, exception.Status);
         }
     }
 }

sdk/openai/Azure.AI.OpenAI/tests/OpenAITestBase.cs

@@ -1,11 +1,35 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
+using System;
+using System.Linq;
 using Azure.Core.TestFramework;
 
 namespace Azure.AI.OpenAI.Tests
 {
     public class OpenAITestEnvironment : TestEnvironment
     {
+        public void ThrowIfCannotDeploy()
+        {
+            string[] requiredVariableNames = new string[]
+            {
+                "TENANT_ID",
+                "AUTHORITY_HOST",
+                "SUBSCRIPTION_ID",
+                "CLIENT_ID",
+                "CLIENT_SECRET",
+            };
+
+            foreach (string requiredVariableName in requiredVariableNames)
+            {
+                string optionalVariableValue = GetOptionalVariable(requiredVariableName);
+                if (string.IsNullOrEmpty(optionalVariableValue))
+                {
+                    throw new InvalidOperationException($"Resource deployment is required and no environment value was found for required variable '{requiredVariableName}'.\n"
+                        + $"'{Mode}' mode requires full environment specification of an Azure application (service principal) "
+                        + "that manages test resources. This includes: Azure tenant ID; Azure authority host; managed subscription ID; and application ID + secret.");
+                }
+            }
+        }
     }
 }