Updated tests to check container args

CommunityToolkit · Dec 2, 2024 · d1db720 · d1db720
1 parent 27cab49
commit d1db720
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/CommunityToolkit.Aspire.Hosting.Ollama.Tests/AddOllamaTests.cs b/tests/CommunityToolkit.Aspire.Hosting.Ollama.Tests/AddOllamaTests.cs
@@ -549,8 +549,8 @@ public async Task WithGPUSupport(OllamaGpuVendor vendor, string expectedArg, str
 
         var resource = Assert.Single(appModel.Resources.OfType<OllamaResource>());
 
-        Assert.True(resource.TryGetLastAnnotation(out CommandLineArgsCallbackAnnotation? argsAnnotations));
-        CommandLineArgsCallbackContext context = new([]);
+        Assert.True(resource.TryGetLastAnnotation(out ContainerRuntimeArgsCallbackAnnotation? argsAnnotations));
+        ContainerRuntimeArgsCallbackContext context = new([]);
         await argsAnnotations.Callback(context);
 
         Assert.Collection(