def test_edge_creation_and_getting(self):
node = Node({'props': {'a': 1, 'b': 2, 'c': 3}})
other_node = Node({'props': {'useless': True}})
edge_props = Node({
'props': {
'an_int': 2,
'unicode': 'салəм',
'float': 1.25, # 1.25 represents well in binary
'bool_true': True,
'bool_false': False,
'char_z': 'z',
'text_hello': 'hello'
}
})
nid = self.storage.create_node(node)
other_nid = self.storage.create_node(other_node)
edge_nid = self.storage.create_node(edge_props)
edge_id = self.storage.create_edge(nid, other_nid, edge_nid)
edge = self.storage.get_edge(edge_id)
self.assertEqual(nid, edge['fnid'])
self.assertEqual(other_nid, edge['tnid'])
self.assertEqual(edge_props['props'], edge['props'])
self.assertIn(edge_id, self.storage.get_edge_ids())
def test_node_updating_with_different_length(self):
node = Node({'props': {'a': 1, 'b': 2, 'c': 3}})
other_node = Node({'props': {'useless': True}})
nid = self.storage.create_node(node)
other_nid = self.storage.create_node(other_node)
node['b'] = 'text is longer than int' # other length
self.storage.update_node(nid, node)
updated = self.storage.get_node(nid)
other_fresh = self.storage.get_node(other_nid)
self.assertEqual(node['props'], updated['props'])
self.assertEqual(other_node['props'], other_fresh['props'])
def test_node_updating_with_same_length(self):
node = Node({'props': {'a': 1, 'b': 2, 'c': 3}})
other_node = Node({'props': {'useless': True}})
nid = self.storage.create_node(node)
other_nid = self.storage.create_node(other_node)
node['b'] = 123 # any other int
self.storage.update_node(nid, node)
updated = self.storage.get_node(nid)
other_fresh = self.storage.get_node(other_nid)
self.assertEqual(node['props'], updated['props'])
self.assertEqual(other_node['props'], other_fresh['props'])
def test_node_deletion(self):
node = Node({'props': {}})
nid = self.storage.create_node(node)
from_storage = self.storage.get_node(nid)
self.assertEqual(from_storage['props'], node['props'])
self.storage.delete_node(nid)
from_storage_after = self.storage.get_node(nid)
self.assertIsNone(from_storage_after)
def test_deleting_edge(self):
node = Node({'props': {'a': 1, 'b': 2, 'c': 3}})
other_node = Node({'props': {'useless': True}})
edge_props_1 = Node({'props': {'text_hello': 'hello'}})
edge_props_2 = Node({'props': {'an_int': 2}})
edge_props_3 = Node({'props': {'unicode': 'салəм'}})
nid = self.storage.create_node(node)
other_nid = self.storage.create_node(other_node)
edge_nid = self.storage.create_node(edge_props_1)
edge_1 = self.storage.create_edge(nid, other_nid, edge_nid)
edge_nid = self.storage.create_node(edge_props_2)
edge_2 = self.storage.create_edge(nid, other_nid, edge_nid)
edge_nid = self.storage.create_node(edge_props_3)
edge_3 = self.storage.create_edge(nid, other_nid, edge_nid)
self.assertIn(edge_2, list(self.storage.get_edge_ids()))
self.assertEqual(len(list(self.storage.edges_from(nid))), 3)
self.storage.remove_edge(edge_2)
self.assertEqual(len(list(self.storage.edges_from(nid))), 2)
self.assertNotIn(edge_2, list(self.storage.get_edge_ids()))
def test_node_creation_and_getting(self):
node = Node({
'props': {
'an_int': 2,
'unicode': 'салəм',
'float': 1.25, # 1.25 represents well in binary
'bool_true': True,
'bool_false': False,
'char_z': 'z',
'text_hello': 'hello'
}
})
nid = self.storage.create_node(node)
node2 = self.storage.get_node(nid)
self.assertEqual(node['props'], node2['props'])
self.assertIn(nid, self.storage.get_node_ids())
def test_loop_edge(self):
node = Node({'props': {}})
nid1 = self.storage.create_node(node)
nid2 = self.storage.create_node(node)
edge_props = self.storage.create_node(node)
eid1 = self.storage.create_edge(nid1, nid1, edge_props)
eid2 = self.storage.create_edge(nid2, nid1, edge_props)
to_nid1 = list(self.storage.edges_to(nid1))
from_nid1 = list(self.storage.edges_from(nid1))
self.assertIn(eid1, to_nid1)
self.assertIn(eid1, from_nid1)
self.assertIn(eid2, to_nid1)
self.assertNotIn(eid2, from_nid1)
self.storage.remove_edge(eid1)
to_nid1 = list(self.storage.edges_to(nid1))
from_nid1 = list(self.storage.edges_from(nid1))
self.assertNotIn(eid1, to_nid1)
self.assertNotIn(eid1, from_nid1)
self.assertIn(eid2, to_nid1)