Skip to content

Commit

Permalink
Add ut test
Browse files Browse the repository at this point in the history
  • Loading branch information
qinguoyi committed Nov 10, 2024
1 parent c56f680 commit 69ddacf
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/e2e/playground_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ var _ = ginkgo.Describe("playground e2e tests", func() {
})

ginkgo.It("Deploy a ollama model with ollama", func() {
backendRuntime := wrapper.MakeBackendRuntime("ollama").
backendRuntime := wrapper.MakeBackendRuntime("llmaz-ollama").
Image("alpine/ollama").Version("latest").
Command([]string{"sh", "-c"}).
Arg("default", []string{"ollama serve & while true;do output=$(ollama list 2>&1);if ! echo $output | grep -q 'could not connect to ollama app' && echo $output | grep -q 'NAME';then echo 'ollama is running';break; else echo 'Waiting for the ollama to be running...';sleep 1;fi;done;ollama run {{.ModelName}};while true;do sleep 60;done"}).
Expand All @@ -60,7 +60,7 @@ var _ = ginkgo.Describe("playground e2e tests", func() {
defer func() {
gomega.Expect(k8sClient.Delete(ctx, model)).To(gomega.Succeed())
}()
playground := wrapper.MakePlayground("qwen2-0--5b", ns.Name).ModelClaim("qwen2-0--5b").BackendRuntime("ollama").Replicas(1).Obj()
playground := wrapper.MakePlayground("qwen2-0--5b", ns.Name).ModelClaim("qwen2-0--5b").BackendRuntime("llmaz-ollama").Replicas(1).Obj()
gomega.Expect(k8sClient.Create(ctx, playground)).To(gomega.Succeed())
validation.ValidatePlayground(ctx, k8sClient, playground)
validation.ValidatePlaygroundStatusEqualTo(ctx, k8sClient, playground, inferenceapi.PlaygroundAvailable, "PlaygroundReady", metav1.ConditionTrue)
Expand Down

0 comments on commit 69ddacf

Please sign in to comment.