Fix empty query of graph issue.

coleifer · Oct 6, 2022 · 0969ed9 · 0969ed9
1 parent bb3ab5e
commit 0969ed9
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/walrus/graph.py b/walrus/graph.py
@@ -130,6 +130,8 @@ def keys_for_query(self, s=None, p=None, o=None):
             parts.extend(('pos', p))
         elif o:
             parts.extend(('osp', o))
+        else:
+            parts.extend(('spo',))
         return key(parts + ['']), key(parts + ['\xff'])
 
     def query(self, s=None, p=None, o=None):

diff --git a/walrus/tests/graph.py b/walrus/tests/graph.py
@@ -85,6 +85,22 @@ def assertTriples(self, result, expected):
 
     def test_query(self):
         self.create_graph_data()
+        res = self.g.query()
+        self.assertTriples(res, (
+            ('charlie', 'is', 'human'),
+            ('charlie', 'likes', 'huey'),
+            ('charlie', 'likes', 'mickey'),
+            ('charlie', 'likes', 'zaizee'),
+            ('connor', 'likes', 'huey'),
+            ('connor', 'likes', 'mickey'),
+            ('huey', 'eats', 'catfood'),
+            ('huey', 'is', 'cat'),
+            ('mickey', 'eats', 'anything'),
+            ('mickey', 'is', 'dog'),
+            ('zaizee', 'eats', 'catfood'),
+            ('zaizee', 'is', 'cat'),
+        ))
+
         res = self.g.query('charlie', 'likes')
         self.assertTriples(res, (
             ('charlie', 'likes', 'huey'),