From f75c678280522298b5e7127d4d9eefdfabaa45c5 Mon Sep 17 00:00:00 2001
From: niuran <ran.niu@opencsg.com>
Date: Fri, 24 Jan 2025 16:42:10 +0800
Subject: [PATCH] fix ut

---
 .../components/__tests__/endpoints/EndpointPage.spec.js   | 3 ++-
 .../__tests__/endpoints/EndpointPlayground.spec.js        | 8 ++++----
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/frontend/src/components/__tests__/endpoints/EndpointPage.spec.js b/frontend/src/components/__tests__/endpoints/EndpointPage.spec.js
index 24594c3d5..824f8dd6d 100644
--- a/frontend/src/components/__tests__/endpoints/EndpointPage.spec.js
+++ b/frontend/src/components/__tests__/endpoints/EndpointPage.spec.js
@@ -58,6 +58,7 @@ describe('EndpointPage.vue', () => {
     sku: 'mockSku',
     modelId: 'mockModelId',
     private: 'true',
+    task: 'text-generation',
     replicaList: [
       { name: 'Replica1', status: 'Running' },
       { name: 'Replica2', status: 'Stopped' }
@@ -76,7 +77,7 @@ describe('EndpointPage.vue', () => {
     })
   })
 
-  it('renders EndpointPlayground when appStatus is Running and appEndpoint is provided', () => {
+  it('renders EndpointPlayground when appStatus is Running and appEndpoint is provided and task is text-generation', () => {
     expect(wrapper.findComponent(EndpointPlayground).exists()).toBe(true)
   })
 
diff --git a/frontend/src/components/__tests__/endpoints/EndpointPlayground.spec.js b/frontend/src/components/__tests__/endpoints/EndpointPlayground.spec.js
index 0e8f6bbc3..cabdcda6e 100644
--- a/frontend/src/components/__tests__/endpoints/EndpointPlayground.spec.js
+++ b/frontend/src/components/__tests__/endpoints/EndpointPlayground.spec.js
@@ -59,13 +59,13 @@ describe("EndpointPlayground", () => {
     expect(wrapper.vm.playgroundMode).toBe('test');
   });
 
-  it("renders TestEndpoint component in 'test' mode", () => {
-    const wrapper = createWrapper();
+  it("renders TestEndpoint component in 'test' mode when task is 'text-generation'", () => {
+    const wrapper = createWrapper({ task: 'text-generation' });
     expect(wrapper.findComponent(TestEndpoint).exists()).toBe(true);
   });
 
-  it("renders ApiExample component in 'api' mode", async () => {
-    const wrapper = createWrapper();
+  it("renders ApiExample component in 'api' mode when task is 'text-generation'", async () => {
+    const wrapper = createWrapper({ task: 'text-generation' });
     await wrapper.find('.cursor-pointer').trigger('click');
     expect(wrapper.findComponent(ApiExample).exists()).toBe(true);
     expect(wrapper.findComponent(TestEndpoint).exists()).toBe(true);