Fixed test bug

plume-oss · May 23, 2024 · 34ac98c · 34ac98c
1 parent 18c544a
commit 34ac98c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/drivers/neo4j/src/main/scala/com/github/plume/oss/drivers/Neo4jDriver.scala b/drivers/neo4j/src/main/scala/com/github/plume/oss/drivers/Neo4jDriver.scala
@@ -148,7 +148,7 @@ final class Neo4jDriver(
           .map { case (key: String, value: Any, node: StoredNode) =>
             val newV = value match {
               case x: String => "\"" + x + "\""
-              case Seq()     => IDriver.STRING_DEFAULT
+              case Seq()     => SchemaBuilder.STRING_DEFAULT
               case xs: Seq[_] =>
                 "[" + xs.map { x => Seq("\"", x, "\"").mkString }.mkString(",") + "]"
               case x: Number => x.toString
@@ -249,15 +249,15 @@ final class Neo4jDriver(
             (keys :+ "id").flatMap { k =>
               val v = record.get(k)
               if (v.hasType(typeSystem.NULL())) {
-                Some(k -> IDriver.getPropertyDefault(k))
+                Some(k -> SchemaBuilder.getPropertyDefault(k))
               } else if (k == "id") {
-                Some(k -> v.asLong(IDriver.LONG_DEFAULT))
+                Some(k -> v.asLong(SchemaBuilder.LONG_DEFAULT))
               } else if (v.hasType(typeSystem.INTEGER())) {
-                Some(k -> v.asInt(IDriver.INT_DEFAULT))
+                Some(k -> v.asInt(SchemaBuilder.INT_DEFAULT))
               } else if (v.hasType(typeSystem.BOOLEAN())) {
-                Some(k -> v.asBoolean(IDriver.BOOL_DEFAULT))
+                Some(k -> v.asBoolean(SchemaBuilder.BOOL_DEFAULT))
               } else if (v.hasType(typeSystem.STRING())) {
-                Some(k -> v.asString(IDriver.STRING_DEFAULT))
+                Some(k -> v.asString(SchemaBuilder.STRING_DEFAULT))
               } else if (v.hasType(typeSystem.LIST())) {
                 Some(k -> v.asList())
               } else {