Merge pull request #167 from SmedbergM/fixUpdateWith

[Issue #166] Make ScalaElement.updateWith(x) remove properties which …
mpollmeier · Aug 20, 2016 · d3fb15f · d3fb15f
2 parents 30a8b63 + c87b542
commit d3fb15f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gremlin-scala/src/main/scala/gremlin/scala/ScalaElement.scala b/gremlin-scala/src/main/scala/gremlin/scala/ScalaElement.scala
@@ -56,7 +56,11 @@ trait ScalaElement[ElementType <: Element] {
 
   def updateWith[CC <: Product: Marshallable](update: CC): ElementType = {
     val propMap = implicitly[Marshallable[CC]].fromCC(update).valueMap
-    propMap foreach {case (prop, value) => element.property(prop, value)}
+    this.valueMap.keySet.diff(propMap.keySet) foreach { key =>
+      val prop = element.property(key)
+      if (prop.isPresent) prop.remove()
+    }
+    propMap foreach {case (key, value) => element.property(key, value)}
 
     element
   }