fix test for config

vesoft-inc · Sep 17, 2021 · d607eaa · d607eaa
1 parent f7ca581
commit d607eaa
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",