def test_del_relation(self): io = Io() value = "You spin my head right ground right ground" label = Label(cfg.INV_ID, value) label = io.write_label(label) prop_value = True property = Property(cfg.INV_ID, PropertyType.BOOL, label, prop_value, None) property = io.write_property(property) node = Node(cfg.INV_ID, label, property, None) node2 = Node(cfg.INV_ID, label, property, None) written1 = io.write_node(node) written2 = io.write_node(node2) relation = Relationship(cfg.INV_ID, False, written1, written2, label, property, None, None, None, None) written = io.write_relation(relation) io.del_relation(written.id) retrived = io.get_relations_by_id([written.id]) node1 = io.read_node(written1.id) node2 = io.read_node(written2.id) self.assertEqual(len(retrived), 0) print(node1.next_rel) print(node2.next_rel) self.assertEqual(node1.next_rel is None, True) self.assertEqual(node2.next_rel is None, True)
def test_relation_io(self): io = Io() value = "You spin my head right ground right ground" label = Label(cfg.INV_ID, value) label = io.write_label(label) prop_value = True property = Property(cfg.INV_ID, PropertyType.BOOL, label, prop_value, None) property = io.write_property(property) node = Node(cfg.INV_ID, label, property, None) node2 = Node(cfg.INV_ID, label, property, None) written1 = io.write_node(node) written2 = io.write_node(node2) relation = Relationship(cfg.INV_ID, False, written1, written2, label, property, None, None, None, None) written = io.write_relation(relation) retrieved = io.read_relation(written.id) written1 = io.read_node(written1.id) written2 = io.read_node(written2.id) self.assertEqual(retrieved.label.value, value) self.assertEqual(retrieved.next_prop.value, prop_value) self.assertEqual(retrieved.first_node.id, written1.id) self.assertEqual(retrieved.second_node.id, written2.id) self.assertEqual(written2.next_rel, retrieved.id) self.assertEqual(written1.next_rel, retrieved.id)
def test_label_io(self): io = Io() value = "I LOVE ADB" label = Label(cfg.INV_ID, value) written = io.write_label(label) retrieved = io.read_label(written.id) self.assertEqual(label.id, retrieved.id) self.assertEqual(label.value, retrieved.value)
def test_string_property_io(self): io = Io() value = "I LOVE ADB" label = Label(cfg.INV_ID, value) label = io.write_label(label) property = Property(cfg.INV_ID, PropertyType.STRING, label, value, None) written = io.write_property(property) retrieved = io.read_property(written.id) self.assertEqual(property.id, retrieved.id) self.assertEqual(property.value, retrieved.value) self.assertEqual(property.type.value, retrieved.type.value) self.assertEqual(property.label.value, retrieved.label.value) self.assertEqual(property.next_prop, retrieved.next_prop)
def test_get_nodes_io(self): io = Io() nodes = [] for i in range(0, 10): value = "Macarena" label = Label(cfg.INV_ID, value) label = io.write_label(label) node = Node(cfg.INV_ID, label, None, None) node = io.write_node(node) nodes.append(node.id) node_list = io.get_nodes_by_id(set(nodes)) for id, node in enumerate(node_list): self.assertEqual(node.id, nodes[id]) self.assertEqual(node.label.value, "Macarena")
def test_multirelation_deletion(self): io = Io() value = "You spin my head right ground right ground" label = Label(cfg.INV_ID, value) label = io.write_label(label) prop_value = True property = Property(cfg.INV_ID, PropertyType.BOOL, label, prop_value, None) property = io.write_property(property) node = Node(cfg.INV_ID, label, property, None) node2 = Node(cfg.INV_ID, label, property, None) written_node1 = io.write_node(node) written_node2 = io.write_node(node2) relation1 = Relationship(cfg.INV_ID, False, written_node1, written_node2, label, property, None, None, None, None) relation2 = Relationship(cfg.INV_ID, False, written_node1, written_node2, label, property, None, None, None, None) written_rel1 = io.write_relation(relation1) written_rel2 = io.write_relation(relation2) io.del_node(written_node1.id) retrived = io.get_relations_by_id([written_rel1.id, written_rel2.id]) self.assertEqual(len(retrived), 0)