Merge pull request #149 from Nicole00/test

fix test for config
vesoft-inc · Sep 23, 2021 · ea96629 · ea96629
2 parents b886279 + 1c991dd
commit ea96629
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nebula-exchange/src/test/scala/com/vesoft/nebula/exchange/utils/NebulaUtilsSuite.scala b/nebula-exchange/src/test/scala/com/vesoft/nebula/exchange/utils/NebulaUtilsSuite.scala
@@ -11,7 +11,7 @@ import com.vesoft.nebula.client.graph.NebulaPoolConfig
 import com.vesoft.nebula.client.graph.data.HostAddress
 import com.vesoft.nebula.client.graph.net.NebulaPool
 import com.vesoft.nebula.client.storage.StorageClient
-import com.vesoft.nebula.exchange.config.TagConfigEntry
+import com.vesoft.nebula.exchange.config.{NebulaSinkConfigEntry, SinkCategory, TagConfigEntry}
 import com.vesoft.nebula.exchange.utils.NebulaUtils
 import com.vesoft.nebula.exchange.{KeyPolicy, MetaProvider, VidType}
 import com.vesoft.nebula.meta.PropertyType
@@ -70,10 +70,11 @@ class NebulaUtilsSuite {
                             "col10",
                             "col11",
                             "col12")
-    val label = "person"
+    val label               = "person"
+    val dataSinkConfigEntry = NebulaSinkConfigEntry(SinkCategory.SST, List("127.0.0.1:9669"))
     val sourceConfig = TagConfigEntry(label,
                                       null,
-                                      null,
+                                      dataSinkConfigEntry,
                                       sourceFields,
                                       nebulaFields,
                                       "id",