class TestTableLastUpdated(unittest.TestCase): def setUp(self) -> None: super(TestTableLastUpdated, self).setUp() self.tableLastUpdated = TableLastUpdated( table_name='test_table', last_updated_time_epoch=25195665, schema='default') self.expected_node_result = { NODE_KEY: 'hive://gold.default/test_table/timestamp', NODE_LABEL: 'Timestamp', 'last_updated_timestamp:UNQUOTED': 25195665, timestamp_constants.TIMESTAMP_PROPERTY + ":UNQUOTED": 25195665, 'name': 'last_updated_timestamp' } self.expected_relation_result = { RELATION_START_KEY: 'hive://gold.default/test_table', RELATION_START_LABEL: 'Table', RELATION_END_KEY: 'hive://gold.default/test_table/timestamp', RELATION_END_LABEL: 'Timestamp', RELATION_TYPE: 'LAST_UPDATED_AT', RELATION_REVERSE_TYPE: 'LAST_UPDATED_TIME_OF' } def test_create_next_node(self) -> None: next_node = self.tableLastUpdated.create_next_node() next_node_serialized = neo4_serializer.serialize_node(next_node) self.assertEqual(next_node_serialized, self.expected_node_result) def test_create_next_relation(self) -> None: next_relation = self.tableLastUpdated.create_next_relation() next_relation_serialized = neo4_serializer.serialize_relationship( next_relation) self.assertEqual(next_relation_serialized, self.expected_relation_result) def test_get_table_model_key(self) -> None: table = self.tableLastUpdated.get_table_model_key() self.assertEqual(table, 'hive://gold.default/test_table') def test_get_last_updated_model_key(self) -> None: last_updated = self.tableLastUpdated.get_last_updated_model_key() self.assertEqual(last_updated, 'hive://gold.default/test_table/timestamp') def test_create_nodes(self) -> None: nodes = self.tableLastUpdated.create_nodes() self.assertEquals(len(nodes), 1) serialize_node = neo4_serializer.serialize_node(nodes[0]) self.assertEquals(serialize_node, self.expected_node_result) def test_create_relation(self) -> None: relation = self.tableLastUpdated.create_relation() self.assertEquals(len(relation), 1) serialized_relation = neo4_serializer.serialize_relationship( relation[0]) self.assertEquals(serialized_relation, self.expected_relation_result)
class TestTableLastUpdated(unittest.TestCase): def setUp(self): # type: () -> None super(TestTableLastUpdated, self).setUp() self.tableLastUpdated = TableLastUpdated( table_name='test_table', last_updated_time_epoch=25195665, schema='default') self.expected_node_result = { NODE_KEY: 'hive://gold.default/test_table/timestamp', NODE_LABEL: 'Timestamp', 'last_updated_timestamp': 25195665, 'name': 'last_updated_timestamp' } self.expected_relation_result = { RELATION_START_KEY: 'hive://gold.default/test_table', RELATION_START_LABEL: 'Table', RELATION_END_KEY: 'hive://gold.default/test_table/timestamp', RELATION_END_LABEL: 'Timestamp', RELATION_TYPE: 'LAST_UPDATED_AT', RELATION_REVERSE_TYPE: 'LAST_UPDATED_TIME_OF' } def test_create_next_node(self): # type: () -> None next_node = self.tableLastUpdated.create_next_node() self.assertEquals(next_node, self.expected_node_result) def test_create_next_relation(self): # type: () -> None next_relation = self.tableLastUpdated.create_next_relation() self.assertEquals(next_relation, self.expected_relation_result) def test_get_table_model_key(self): # type: () -> None table = self.tableLastUpdated.get_table_model_key() self.assertEquals(table, 'hive://gold.default/test_table') def test_get_last_updated_model_key(self): # type: () -> None last_updated = self.tableLastUpdated.get_last_updated_model_key() self.assertEquals(last_updated, 'hive://gold.default/test_table/timestamp') def test_create_nodes(self): # type: () -> None nodes = self.tableLastUpdated.create_nodes() self.assertEquals(len(nodes), 1) self.assertEquals(nodes[0], self.expected_node_result) def test_create_relation(self): # type: () -> None relation = self.tableLastUpdated.create_relation() self.assertEquals(len(relation), 1) self.assertEquals(relation[0], self.expected_relation_result)