From d1db7200deb2f98d1322128872489af2466cb5f0 Mon Sep 17 00:00:00 2001 From: MatsM16 Date: Tue, 3 Dec 2024 00:01:20 +0100 Subject: [PATCH] Updated tests to check container args --- .../AddOllamaTests.cs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/CommunityToolkit.Aspire.Hosting.Ollama.Tests/AddOllamaTests.cs b/tests/CommunityToolkit.Aspire.Hosting.Ollama.Tests/AddOllamaTests.cs index 983478fe..f90bc2a0 100644 --- 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()); - 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(