1 files changed, 77 insertions, 0 deletions
diff --git a/tests/test_graph_utils/test_edge.py b/tests/test_graph_utils/test_edge.py
new file mode 100644
index 0000000..42358de
--- /dev/null
+++ b/tests/test_graph_utils/test_edge.py
@@ -0,0 +1,77 @@
+import pytest
+
+from redis.commands.graph import edge, node
+
+
+@pytest.mark.redismod
+def test_init():
+
+    with pytest.raises(AssertionError):
+        edge.Edge(None, None, None)
+        edge.Edge(node.Node(), None, None)
+        edge.Edge(None, None, node.Node())
+
+    assert isinstance(
+        edge.Edge(node.Node(node_id=1), None, node.Node(node_id=2)), edge.Edge
+    )
+
+
+@pytest.mark.redismod
+def test_toString():
+    props_result = edge.Edge(
+        node.Node(), None, node.Node(), properties={"a": "a", "b": 10}
+    ).toString()
+    assert props_result == '{a:"a",b:10}'
+
+    no_props_result = edge.Edge(
+        node.Node(), None, node.Node(), properties={}
+    ).toString()
+    assert no_props_result == ""
+
+
+@pytest.mark.redismod
+def test_stringify():
+    john = node.Node(
+        alias="a",
+        label="person",
+        properties={"name": "John Doe", "age": 33, "someArray": [1, 2, 3]},
+    )
+    japan = node.Node(alias="b", label="country", properties={"name": "Japan"})
+    edge_with_relation = edge.Edge(
+        john, "visited", japan, properties={"purpose": "pleasure"}
+    )
+    assert (
+        '(a:person{age:33,name:"John Doe",someArray:[1, 2, 3]})'
+        '-[:visited{purpose:"pleasure"}]->'
+        '(b:country{name:"Japan"})' == str(edge_with_relation)
+    )
+
+    edge_no_relation_no_props = edge.Edge(japan, "", john)
+    assert (
+        '(b:country{name:"Japan"})'
+        "-[]->"
+        '(a:person{age:33,name:"John Doe",someArray:[1, 2, 3]})'
+        == str(edge_no_relation_no_props)
+    )
+
+    edge_only_props = edge.Edge(john, "", japan, properties={"a": "b", "c": 3})
+    assert (
+        '(a:person{age:33,name:"John Doe",someArray:[1, 2, 3]})'
+        '-[{a:"b",c:3}]->'
+        '(b:country{name:"Japan"})' == str(edge_only_props)
+    )
+
+
+@pytest.mark.redismod
+def test_comparision():
+    node1 = node.Node(node_id=1)
+    node2 = node.Node(node_id=2)
+    node3 = node.Node(node_id=3)
+
+    edge1 = edge.Edge(node1, None, node2)
+    assert edge1 == edge.Edge(node1, None, node2)
+    assert edge1 != edge.Edge(node1, "bla", node2)
+    assert edge1 != edge.Edge(node1, None, node3)
+    assert edge1 != edge.Edge(node3, None, node2)
+    assert edge1 != edge.Edge(node2, None, node1)
+    assert edge1 != edge.Edge(node1, None, node2, properties={"a": 10})