add test for orphanKey

vesoft-inc · Jan 27, 2022 · 91af660 · 91af660
1 parent e4f5570
commit 91af660
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 9 deletions.
diff --git a/...park_2.2/src/test/scala/com/vesoft/nebula/exchange/processor/VerticesProcessorSuite.scala b/...park_2.2/src/test/scala/com/vesoft/nebula/exchange/processor/VerticesProcessorSuite.scala
@@ -14,6 +14,7 @@ import com.vesoft.exchange.common.config.{Configs, TagConfigEntry}
 import com.vesoft.exchange.common.utils.NebulaUtils.DEFAULT_EMPTY_VALUE
 import com.vesoft.nebula.meta.{ColumnDef, ColumnTypeDef, Schema, SchemaProp, TagItem}
 import org.apache.commons.codec.binary.Hex
+import org.apache.log4j.Logger
 import org.apache.spark.sql.catalyst.expressions.GenericRowWithSchema
 import org.apache.spark.sql.types.{
   BooleanType,
@@ -32,6 +33,8 @@ import org.scalatest.Assertions.assertThrows
 import scala.collection.JavaConverters._
 
 class VerticesProcessorSuite {
+  private[this] lazy val LOG = Logger.getLogger(this.getClass)
+
   val config: Configs =
     Configs.parse(new File("../exchange-common/src/test/resources/process_application.conf"))
 
@@ -138,11 +141,14 @@ class VerticesProcessorSuite {
     val tagItem = new TagItem(1, "person".getBytes(), -1, schema)
     val map     = getFieldType()
 
-    val (key, value) = processClazz.encodeVertex(row, 10, VidType.STRING, 10, tagItem, map)
+    val (orphanKey, key, value) =
+      processClazz.encodeVertex(row, 10, VidType.STRING, 10, tagItem, map)
 
-    val keyHex   = Hex.encodeHexString(key)
-    val valueHex = Hex.encodeHexString(value)
+    val keyHex       = Hex.encodeHexString(key)
+    val orphanKeyHex = Hex.encodeHexString(orphanKey)
+    val valueHex     = Hex.encodeHexString(value)
     assert(keyHex.equals("010600003100000000000000000001000000"))
+    assert(orphanKeyHex.equals("0706000031000000000000000000"))
   }
 
   private def getRow(): Row = {

diff --git a/...park_2.4/src/test/scala/com/vesoft/nebula/exchange/processor/VerticesProcessorSuite.scala b/...park_2.4/src/test/scala/com/vesoft/nebula/exchange/processor/VerticesProcessorSuite.scala
@@ -139,11 +139,14 @@ class VerticesProcessorSuite {
     val tagItem = new TagItem(1, "person".getBytes(), -1, schema)
     val map     = getFieldType()
 
-    val (key, value) = processClazz.encodeVertex(row, 10, VidType.STRING, 10, tagItem, map)
+    val (orphanKey, key, value) =
+      processClazz.encodeVertex(row, 10, VidType.STRING, 10, tagItem, map)
 
-    val keyHex   = Hex.encodeHexString(key)
-    val valueHex = Hex.encodeHexString(value)
+    val keyHex       = Hex.encodeHexString(key)
+    val orphenKeyHex = Hex.encodeHexString(orphanKey)
+    val valueHex     = Hex.encodeHexString(value)
     assert(keyHex.equals("010600003100000000000000000001000000"))
+    assert(orphenKeyHex.equals("0706000031000000000000000000"))
   }
 
   private def getRow(): Row = {

diff --git a/...park_3.0/src/test/scala/com/vesoft/nebula/exchange/processor/VerticesProcessorSuite.scala b/...park_3.0/src/test/scala/com/vesoft/nebula/exchange/processor/VerticesProcessorSuite.scala
@@ -138,11 +138,14 @@ class VerticesProcessorSuite {
     val tagItem = new TagItem(1, "person".getBytes(), -1, schema)
     val map     = getFieldType()
 
-    val (key, value) = processClazz.encodeVertex(row, 10, VidType.STRING, 10, tagItem, map)
+    val (orphanKey, key, value) =
+      processClazz.encodeVertex(row, 10, VidType.STRING, 10, tagItem, map)
 
-    val keyHex   = Hex.encodeHexString(key)
-    val valueHex = Hex.encodeHexString(value)
+    val keyHex       = Hex.encodeHexString(key)
+    val orphanKeyHex = Hex.encodeHexString(orphanKey)
+    val valueHex     = Hex.encodeHexString(value)
     assert(keyHex.equals("010600003100000000000000000001000000"))
+    assert(orphanKeyHex.equals("0706000031000000000000000000"))
   }
 
   private def getRow(): Row = {