first round of recordings

Author: m-redding

src/Custom/VectorStores/CreateVectorStoreOperation.cs

@@ -12,6 +12,13 @@ namespace OpenAI.VectorStores;
 [Experimental("OPENAI001")]
 public partial class CreateVectorStoreOperation : OperationResult
 {
+    protected CreateVectorStoreOperation() : base(default!)
+    {
+        _parentClient = default!;
+        _endpoint = default!;
+        _vectorStoreId = default!;
+    }
+
     internal CreateVectorStoreOperation(
         VectorStoreClient parentClient,
         Uri endpoint,

tests/Assistants/Assistants.VectorStoresTests.cs

@@ -1,4 +1,5 @@
 using Microsoft.ClientModel.TestFramework;
+using Microsoft.Extensions.Logging;
 using NUnit.Framework;
 using OpenAI.Files;
 using OpenAI.VectorStores;
@@ -7,6 +8,7 @@
 using System.ClientModel.Primitives;
 using System.Collections.Generic;
 using System.Linq;
+using System.Runtime.CompilerServices;
 using System.Text.Json;
 using System.Threading;
 using System.Threading.Tasks;
@@ -18,7 +20,7 @@ namespace OpenAI.Tests.VectorStores;
 
 [ClientTestFixture]
 [Category("Assistants")]
-public class VectorStoresTests : SyncAsyncTestBase
+public class VectorStoresTests : RecordedTestBase<OpenAITestEnvironment>
 {
     private readonly List<CreateBatchFileJobOperation> _jobsToCancel = [];
     private readonly List<VectorStoreFileAssociation> _associationsToRemove = [];
@@ -27,10 +29,10 @@ public class VectorStoresTests : SyncAsyncTestBase
 
     private static readonly DateTimeOffset s_2024 = new(2024, 1, 1, 0, 0, 0, TimeSpan.Zero);
 
-    private static VectorStoreClient GetTestClient() => GetTestClient<VectorStoreClient>(TestScenario.VectorStores);
+    private VectorStoreClient GetTestClient() => CreateProxyFromClient(GetTestClient<VectorStoreClient>(TestScenario.VectorStores), null);
 
     public VectorStoresTests(bool isAsync)
-        : base(isAsync)
+        : base(isAsync, RecordedTestMode.Record)
     {
     }
 
@@ -669,6 +671,10 @@ protected void Cleanup()
     /// <exception cref="NotImplementedException"> The provided instance type isn't supported. </exception>
     private void Validate<T>(T target)
     {
+        if (target is IProxiedOperationResult result)
+        {
+            target = (T)result.Original;
+        }
         if (target is CreateBatchFileJobOperation job)
         {
             Assert.That(job.BatchId, Is.Not.Null);

tests/OpenAI.Tests.csproj

@@ -15,6 +15,8 @@
   </ItemGroup>
   <ItemGroup>
     <PackageReference Include="Microsoft.Bcl.AsyncInterfaces" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Extensions.Logging" Version="9.0.7" />
+		<PackageReference Include="Microsoft.Extensions.Logging.Console" Version="9.0.7" />
     <PackageReference Include="NUnit" Version="3.13.2" />
     <PackageReference Include="NUnit3TestAdapter" Version="4.4.2" />
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.11.1" />
@@ -30,4 +32,11 @@
 	<ItemGroup>
 		<ProjectReference Include="..\..\azure-sdk-for-net\sdk\core\Microsoft.ClientModel.TestFramework\src\Microsoft.ClientModel.TestFramework.csproj" />
 	</ItemGroup>
+
+	<ItemGroup>
+		<AssemblyAttribute Include="System.Reflection.AssemblyMetadataAttribute">
+			<_Parameter1>SourcePath</_Parameter1>
+			<_Parameter2>$(MSBuildProjectDirectory)</_Parameter2>
+		</AssemblyAttribute>
+	</ItemGroup>
 </Project>

tests/Responses/ResponsesTests.cs

@@ -1,4 +1,5 @@
 using Microsoft.ClientModel.TestFramework;
+using Microsoft.Extensions.Options;
 using Microsoft.VisualStudio.TestPlatform.ObjectModel;
 using NUnit.Framework;
 using OpenAI.Files;
@@ -22,9 +23,9 @@ namespace OpenAI.Tests.Responses;
 [TestFixture(false)]
 [Parallelizable(ParallelScope.Fixtures)]
 [Category("Responses")]
-public partial class ResponsesTests : SyncAsyncTestBase
+public partial class ResponsesTests : OpenAIRecordedTestBase
 {
-    public ResponsesTests(bool isAsync) : base(isAsync)
+    public ResponsesTests(bool isAsync) : base(isAsync, RecordedTestMode.Record)
     {
     }
 
@@ -75,7 +76,8 @@ private void Validate<T>(T input) where T : class
     [Test]
     public async Task FileSearch()
     {
-        OpenAIFileClient fileClient = GetTestClient<OpenAIFileClient>(TestScenario.Files);
+        OpenAIFileClient fileClient = GetProxiedOpenAIClient<OpenAIFileClient>(TestScenario.Files);
+
         OpenAIFile testFile = await fileClient.UploadFileAsync(
             BinaryData.FromString("""
                     Travis's favorite food is pizza.
@@ -84,7 +86,8 @@ public async Task FileSearch()
             FileUploadPurpose.UserData);
         Validate(testFile);
 
-        VectorStoreClient vscClient = GetTestClient<VectorStoreClient>(TestScenario.VectorStores);
+        VectorStoreClient vscClient = GetProxiedOpenAIClient<VectorStoreClient>(TestScenario.VectorStores);
+
         CreateVectorStoreOperation createStoreOp = await vscClient.CreateVectorStoreAsync(
             waitUntilCompleted: true,
             new VectorStoreCreationOptions()
@@ -530,7 +533,7 @@ public async Task FileInputFromIdWorks()
             BinaryData.FromBytes(File.ReadAllBytes(filePath)),
             "test_favorite_foods.pdf",
             FileUploadPurpose.UserData);
-                Validate(newFileToUse);
+        Validate(newFileToUse);
 
         ResponseItem messageItem = ResponseItem.CreateUserMessageItem(
             [
@@ -904,5 +907,5 @@ public async Task CanCancelBackgroundResponses()
                 """),
             false);
 
-    private static OpenAIResponseClient GetTestClient(string overrideModel = null) => GetTestClient<OpenAIResponseClient>(TestScenario.Responses, overrideModel);
+    private OpenAIResponseClient GetTestClient(string overrideModel = null, OpenAIClientOptions options = null) => GetProxiedOpenAIClient<OpenAIResponseClient>(TestScenario.Responses, overrideModel, options);
 }