From cfc685bcf9eadb5bf3c9dfaaf19f8a3d48813a1b Mon Sep 17 00:00:00 2001
From: Jeffrey Lovitz <jeffrey.lovitz@gmail.com>
Date: Mon, 13 Jul 2020 16:35:50 -0400
Subject: [PATCH] Add unit test for entity updates

---
 spec/redisgraph_spec.rb | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/spec/redisgraph_spec.rb b/spec/redisgraph_spec.rb
index 1824d51..08da1ec 100644
--- a/spec/redisgraph_spec.rb
+++ b/spec/redisgraph_spec.rb
@@ -76,4 +76,20 @@ def create_graph()
       expect(res.resultset).to eq([["src1", [{"name"=>"dest1"}, {"color"=>"magenta"}], [{"weight"=>7.8}]]])
     end
   end
+
+  context "update" do
+    it "should support adding new properties" do
+      q = """MATCH (a {name: 'src1'}) SET a.newval = true"""
+      plan = @r.explain(q)
+      expect(plan.detect { |row| row.include?("Update") }).to_not be_nil
+      res = @r.query(q)
+      expect(res.stats[:properties_set]).to eq(1)
+    end
+
+    it "should print property strings correctly after updates" do
+      q = """MATCH (a {name: 'src1'}) RETURN a"""
+      res = @r.query(q)
+      expect(res.resultset).to eq([[[{"name"=>"src1"}, {"color"=>"cyan"}, {"newval"=>TRUE}]]])
+    end
+  end
 end