From 12e25c0c5080d3a140e55e85ce580901a3dccfe8 Mon Sep 17 00:00:00 2001
From: lexluo09 <39718951+lexluo09@users.noreply.github.com>
Date: Sat, 6 Apr 2024 13:09:00 +0800
Subject: [PATCH] (improvement)(headless) Fix the issue of failing unit tests.
 (#890)

---
 .../tencent/supersonic/ModelDemoDataLoader.java  | 16 +++++++++-------
 .../src/main/resources/application-local.yaml    |  2 +-
 .../com/tencent/supersonic/headless/TagTest.java |  1 +
 3 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/launchers/standalone/src/main/java/com/tencent/supersonic/ModelDemoDataLoader.java b/launchers/standalone/src/main/java/com/tencent/supersonic/ModelDemoDataLoader.java
index d1ab4765f..9765dcd85 100644
--- a/launchers/standalone/src/main/java/com/tencent/supersonic/ModelDemoDataLoader.java
+++ b/launchers/standalone/src/main/java/com/tencent/supersonic/ModelDemoDataLoader.java
@@ -374,16 +374,17 @@ public class ModelDemoDataLoader {
     }
 
     private void addTags() {
-        addTag(1L);
-        addTag(4L);
-        addTag(5L);
-        addTag(6L);
-        addTag(7L);
+        addTag(1L, TagDefineType.DIMENSION);
+        addTag(4L, TagDefineType.DIMENSION);
+        addTag(5L, TagDefineType.DIMENSION);
+        addTag(6L, TagDefineType.DIMENSION);
+        addTag(7L, TagDefineType.DIMENSION);
+        addTag(5L, TagDefineType.METRIC);
     }
 
-    private void addTag(Long itemId) {
+    private void addTag(Long itemId, TagDefineType tagDefineType) {
         TagReq tagReq = new TagReq();
-        tagReq.setTagDefineType(TagDefineType.DIMENSION);
+        tagReq.setTagDefineType(tagDefineType);
         tagReq.setItemId(itemId);
         tagMetaService.create(tagReq, User.getFakeUser());
     }
@@ -542,6 +543,7 @@ public class ModelDemoDataLoader {
         tagTypeDefaultConfig.setTimeDefaultConfig(tagTimeDefaultConfig);
         DefaultDisplayInfo defaultDisplayInfo = new DefaultDisplayInfo();
         defaultDisplayInfo.setDimensionIds(Lists.newArrayList(4L, 5L, 6L, 7L));
+        defaultDisplayInfo.setMetricIds(Lists.newArrayList(5L));
         tagTypeDefaultConfig.setDefaultDisplayInfo(defaultDisplayInfo);
         MetricTypeDefaultConfig metricTypeDefaultConfig = new MetricTypeDefaultConfig();
         TimeDefaultConfig timeDefaultConfig = new TimeDefaultConfig();
diff --git a/launchers/standalone/src/main/resources/application-local.yaml b/launchers/standalone/src/main/resources/application-local.yaml
index cbe243794..1acba84ae 100644
--- a/launchers/standalone/src/main/resources/application-local.yaml
+++ b/launchers/standalone/src/main/resources/application-local.yaml
@@ -95,7 +95,7 @@ logging:
 
 inMemoryEmbeddingStore:
   persistent:
-    path: d://
+    path: /tmp
 
 query:
   optimizer:
diff --git a/launchers/standalone/src/test/java/com/tencent/supersonic/headless/TagTest.java b/launchers/standalone/src/test/java/com/tencent/supersonic/headless/TagTest.java
index 8e7579b5e..2bdbd615a 100644
--- a/launchers/standalone/src/test/java/com/tencent/supersonic/headless/TagTest.java
+++ b/launchers/standalone/src/test/java/com/tencent/supersonic/headless/TagTest.java
@@ -65,6 +65,7 @@ public class TagTest extends BaseTest {
         tagMetaService.create(tagReq1, user);
         TagFilter tagFilter = new TagFilter();
         tagFilter.setItemIds(Arrays.asList(4L, 5L));
+        tagFilter.setTagDefineType(TagDefineType.DIMENSION);
         List<TagResp> tags = tagMetaService.getTags(tagFilter);
         Assert.assertEquals(2, tags.size());
         TagDeleteReq tagDeleteReq = new TagDeleteReq();