From e985dd7ccf6901d325e51d54ccc49f028ac6db4c Mon Sep 17 00:00:00 2001
From: Rick <1450685+LinuxSuRen@users.noreply.github.com>
Date: Wed, 2 Aug 2023 10:08:05 +0800
Subject: [PATCH] fix: the API request from UI missed the header (#155)

---
 console/atest-ui/src/views/APIInput.vue           | 2 +-
 console/atest-ui/src/views/TestCase.vue           | 7 +++++--
 console/atest-ui/src/views/TestSuite.vue          | 4 ++--
 console/atest-ui/src/views/__test__/types.spec.ts | 6 +++---
 console/atest-ui/src/views/types.ts               | 9 ++++++---
 5 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/console/atest-ui/src/views/APIInput.vue b/console/atest-ui/src/views/APIInput.vue
index e59fd7e..8ac2d05 100644
--- a/console/atest-ui/src/views/APIInput.vue
+++ b/console/atest-ui/src/views/APIInput.vue
@@ -9,7 +9,7 @@ const props = defineProps({
   operationId: String
 })
 // const emit = defineEmits(['update:address', 'update:method', 'update:operationId'])
-const querySuggestedAPIs = NewSuggestedAPIsQuery(props.suite!)
+const querySuggestedAPIs = NewSuggestedAPIsQuery('', props.suite!)
 
 const handleSelect = (item: TestCase) => {
   // emit('update:address', item.request.api)
diff --git a/console/atest-ui/src/views/TestCase.vue b/console/atest-ui/src/views/TestCase.vue
index 6d25826..6efcc2c 100644
--- a/console/atest-ui/src/views/TestCase.vue
+++ b/console/atest-ui/src/views/TestCase.vue
@@ -14,7 +14,7 @@ const props = defineProps({
 })
 const emit = defineEmits(['updated'])
 
-let querySuggestedAPIs = NewSuggestedAPIsQuery(props.suite!)
+let querySuggestedAPIs = NewSuggestedAPIsQuery(props.store!, props.suite!)
 const testResultActiveTab = ref('output')
 const requestLoading = ref(false)
 const testResult = ref({ header: [] as Pair[] } as TestResult)
@@ -65,6 +65,9 @@ function generateCode() {
   const suite = props.suite
   const requestOptions = {
     method: 'POST',
+    headers: {
+      'X-Store-Name': props.store
+    },
     body: JSON.stringify({
       TestSuite: suite,
       TestCase: name,
@@ -424,7 +427,7 @@ const queryPupularHeaders = (queryString: string, cb: (arg: any) => void) => {
           v-model="testCaseWithSuite.data.request.api"
           :fetch-suggestions="querySuggestedAPIs"
           placeholder="API Address"
-          style="width: 70%; margin-left: 5px; margin-right: 5px"
+          style="width: 50%; margin-left: 5px; margin-right: 5px"
         >
           <template #default="{ item }">
             <div class="value">{{ item.request.method }}</div>
diff --git a/console/atest-ui/src/views/TestSuite.vue b/console/atest-ui/src/views/TestSuite.vue
index a96741a..4450e52 100644
--- a/console/atest-ui/src/views/TestSuite.vue
+++ b/console/atest-ui/src/views/TestSuite.vue
@@ -11,7 +11,7 @@ const props = defineProps({
   store: String,
 })
 const emit = defineEmits(['updated'])
-let querySuggestedAPIs = NewSuggestedAPIsQuery(props.name!)
+let querySuggestedAPIs = NewSuggestedAPIsQuery(props.store!, props.name!)
 
 const suite = ref({
   name: '',
@@ -87,7 +87,7 @@ const rules = reactive<FormRules<Suite>>({
 
 function openNewTestCaseDialog() {
   dialogVisible.value = true
-  querySuggestedAPIs = NewSuggestedAPIsQuery(props.name!)
+  querySuggestedAPIs = NewSuggestedAPIsQuery(props.store!, props.name!)
 }
 
 const submitForm = async (formEl: FormInstance | undefined) => {
diff --git a/console/atest-ui/src/views/__test__/types.spec.ts b/console/atest-ui/src/views/__test__/types.spec.ts
index 1558719..e877c4c 100644
--- a/console/atest-ui/src/views/__test__/types.spec.ts
+++ b/console/atest-ui/src/views/__test__/types.spec.ts
@@ -20,7 +20,7 @@ global.fetch = jest.fn((key: string) =>
 
 describe('NewSuggestedAPIsQuery', () => {
   test('empty data', () => {
-    const func = NewSuggestedAPIsQuery('')
+    const func = NewSuggestedAPIsQuery('', '')
     expect(func).not.toBeNull()
 
     func('xxx', function(e) {
@@ -31,7 +31,7 @@ describe('NewSuggestedAPIsQuery', () => {
   test('have data', () => {
     matchFake('/server.Runner/GetSuggestedAPIs', `{"data":[{"request":{"api":"xxx"}}]}`)
 
-    const func = NewSuggestedAPIsQuery('suite')
+    const func = NewSuggestedAPIsQuery('', 'suite')
     expect(func).not.toBeNull()
 
     func('xxx', function(e) {
@@ -42,7 +42,7 @@ describe('NewSuggestedAPIsQuery', () => {
       expect(e.length).toBe(0)
     })
 
-    const anotherFunc = NewSuggestedAPIsQuery('')
+    const anotherFunc = NewSuggestedAPIsQuery('', '')
     expect(anotherFunc).not.toBeNull()
   })
 })
diff --git a/console/atest-ui/src/views/types.ts b/console/atest-ui/src/views/types.ts
index 05798c2..a54ba15 100644
--- a/console/atest-ui/src/views/types.ts
+++ b/console/atest-ui/src/views/types.ts
@@ -56,9 +56,9 @@ export interface TestCaseResponse {
 
 // Suggested APIs query
 const localCache = ref({} as TestCaseWithValue[])
-export function NewSuggestedAPIsQuery(suite: string) {
+export function NewSuggestedAPIsQuery(store: string, suite: string) {
   return function (queryString: string, cb: (arg: any) => void) {
-    loadCache(suite, function () {
+    loadCache(store, suite, function () {
       const results = queryString
         ? localCache.value.filter(CreateFilter(queryString))
         : localCache.value
@@ -72,7 +72,7 @@ export function CreateFilter(queryString: string) {
     return v.value.toLowerCase().indexOf(queryString.toLowerCase()) !== -1
   }
 }
-function loadCache(suite: string, callback: Function) {
+function loadCache(store: string, suite: string, callback: Function) {
   if (localCache.value.length > 0) {
     callback()
     return
@@ -84,6 +84,9 @@ function loadCache(suite: string, callback: Function) {
 
   const requestOptions = {
     method: 'POST',
+    headers: {
+      'X-Store-Name': store
+    },
     body: JSON.stringify({
       name: suite
     })