Graph.node interface

1 files changed, 10 insertions, 0 deletions

diff --git a/test/graph/test_graph.py b/test/graph/test_graph.py
index f8e107d..33cf6aa 100644
--- a/test/graph/test_graph.py
+++ b/test/graph/test_graph.py
@@ -74,6 +74,16 @@ class TestGraph(unittest.TestCase):
         self.assertEqual(graph.schema, self.backend.schema)
         self.assertRaises(AttributeError, setattr, graph, 'schema', None)
 
+    def test_node(self):
+        graph = Graph(self.backend, self.user)
+        guid = URI('http://example.com/me/entity#1234')
+        # returns a Nodes instance
+        self.assertEqual(
+            graph.node(ns.bsfs.Entity, guid),
+            Nodes(self.backend, self.user, graph.schema.node(ns.bsfs.Entity), {guid}))
+        # node_type must be in the schema
+        self.assertRaises(KeyError, graph.node, ns.bsfs.Invalid, guid)
+
     def test_nodes(self):
         graph = Graph(self.backend, self.user)
         guids = {URI('http://example.com/me/entity#1234'), URI('http://example.com/me/entity#4321')}