def setUp(self) -> None:
super(TestTableOwner, self).setUp()
self.table_owner = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners="user1@1, user2@2 ")
def test_create_nodes_with_owners_list(self) -> None:
self.table_owner_list = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', ' user2@2 '])
nodes = self.table_owner_list.create_nodes()
self.assertEqual(len(nodes), 2)
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
actual_nodes = [
neo4_serializer.serialize_node(node)
for node in nodes
]
self.assertTrue(expected_node1 in actual_nodes)
self.assertTrue(expected_node2 in actual_nodes)
def test_create_nodes_with_owners_list(self) -> None:
self.table_owner_list = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', ' user2@2 '])
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
expected = [expected_node1, expected_node2]
actual = []
node = self.table_owner_list.create_next_node()
while node:
serialized_node = neo4_serializer.serialize_node(node)
actual.append(serialized_node)
node = self.table_owner_list.create_next_node()
self.assertEqual(actual, expected)
def setUp(self):
# type: () -> None
super(TestTableOwner, self).setUp()
self.table_owner = TableOwner(db_name='hive',
schema_name=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners="user1@1, UsER2@2 ")
def transform(self, record: Any) -> Iterable[Any]:
yield record
if isinstance(record, TableMetadata):
yield TableOwner(
db_name=record.database,
schema=record.schema,
table_name=record.name,
cluster=record.cluster,
owners=["foo", "bar"],
)
class TestTableOwner(unittest.TestCase):
def setUp(self):
# type: () -> None
super(TestTableOwner, self).setUp()
self.table_owner = TableOwner(db_name='hive',
schema_name=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', 'user2@2'])
def test_get_owner_model_key(self):
# type: () -> None
owner = self.table_owner.get_owner_model_key('user1@1')
self.assertEquals(owner, 'user1@1')
def test_get_metadata_model_key(self):
# type: () -> None
metadata = self.table_owner.get_metadata_model_key()
self.assertEquals(metadata, 'hive://default.base/test')
def test_create_nodes(self):
# type: () -> None
nodes = self.table_owner.create_nodes()
self.assertEquals(len(nodes), 2)
def test_create_relation(self):
# type: () -> None
relations = self.table_owner.create_relation()
self.assertEquals(len(relations), 2)
relation = {
RELATION_START_KEY: 'user1@1',
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
self.assertTrue(relation in relations)
def test_create_nodes_with_owners_list(self) -> None:
self.table_owner_list = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', ' UsER2@2 '])
nodes = self.table_owner_list.create_nodes()
self.assertEquals(len(nodes), 2)
node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
self.assertTrue(node1 in nodes)
self.assertTrue(node2 in nodes)
class TestTableOwner(unittest.TestCase):
def setUp(self) -> None:
super(TestTableOwner, self).setUp()
self.table_owner = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners="user1@1, user2@2 ")
def test_get_owner_model_key(self) -> None:
owner = self.table_owner.get_owner_model_key(owner1)
self.assertEqual(owner, owner1)
def test_get_metadata_model_key(self) -> None:
metadata = self.table_owner.get_metadata_model_key()
self.assertEqual(metadata, 'hive://DEFAULT.BASE/TEST')
def test_create_nodes(self) -> None:
nodes = self.table_owner.create_nodes()
self.assertEqual(len(nodes), 2)
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
actual_nodes = [neo4_serializer.serialize_node(node) for node in nodes]
self.assertTrue(expected_node1 in actual_nodes)
self.assertTrue(expected_node2 in actual_nodes)
def test_create_relation(self) -> None:
relations = self.table_owner.create_relation()
self.assertEqual(len(relations), 2)
expected_relation1 = {
RELATION_START_KEY: owner1,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
expected_relation2 = {
RELATION_START_KEY: owner2,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
actual_relations = [
neo4_serializer.serialize_relationship(relation)
for relation in relations
]
self.assertTrue(expected_relation1 in actual_relations)
self.assertTrue(expected_relation2 in actual_relations)
def test_create_nodes_with_owners_list(self) -> None:
self.table_owner_list = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', ' user2@2 '])
nodes = self.table_owner_list.create_nodes()
self.assertEqual(len(nodes), 2)
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
actual_nodes = [neo4_serializer.serialize_node(node) for node in nodes]
self.assertTrue(expected_node1 in actual_nodes)
self.assertTrue(expected_node2 in actual_nodes)
class TestTableOwner(unittest.TestCase):
def setUp(self):
# type: () -> None
super(TestTableOwner, self).setUp()
self.table_owner = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners="user1@1, UsER2@2 ")
def test_get_owner_model_key(self):
# type: () -> None
owner = self.table_owner.get_owner_model_key(owner1)
self.assertEquals(owner, owner1)
def test_get_metadata_model_key(self):
# type: () -> None
metadata = self.table_owner.get_metadata_model_key()
self.assertEquals(metadata, 'hive://default.base/test')
def test_create_nodes(self):
# type: () -> None
nodes = self.table_owner.create_nodes()
self.assertEquals(len(nodes), 2)
node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
self.assertTrue(node1 in nodes)
self.assertTrue(node2 in nodes)
def test_create_relation(self):
# type: () -> None
relations = self.table_owner.create_relation()
self.assertEquals(len(relations), 2)
relation1 = {
RELATION_START_KEY: owner1,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
relation2 = {
RELATION_START_KEY: owner2,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
self.assertTrue(relation1 in relations)
self.assertTrue(relation2 in relations)
def test_create_nodes_with_owners_list(self):
# type: () -> None
self.table_owner_list = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', ' UsER2@2 '])
nodes = self.table_owner_list.create_nodes()
self.assertEquals(len(nodes), 2)
node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
self.assertTrue(node1 in nodes)
self.assertTrue(node2 in nodes)
class TestTableOwner(unittest.TestCase):
def setUp(self) -> None:
super(TestTableOwner, self).setUp()
self.table_owner = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners="user1@1, user2@2 ")
def test_get_owner_model_key(self) -> None:
owner = self.table_owner.get_owner_model_key(owner1)
self.assertEqual(owner, owner1)
def test_get_metadata_model_key(self) -> None:
metadata = self.table_owner.get_metadata_model_key()
self.assertEqual(metadata, 'hive://DEFAULT.BASE/TEST')
def test_create_nodes(self) -> None:
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
expected = [expected_node1, expected_node2]
actual = []
node = self.table_owner.create_next_node()
while node:
serialized_node = neo4_serializer.serialize_node(node)
actual.append(serialized_node)
node = self.table_owner.create_next_node()
self.assertEqual(actual, expected)
def test_create_nodes_neptune(self) -> None:
expected_node1 = {
NEPTUNE_HEADER_ID:
"User:"******":String(single)":
owner1
}
expected_node2 = {
NEPTUNE_HEADER_ID:
"User:"******":String(single)":
owner2
}
expected = [expected_node1, expected_node2]
actual = []
node = self.table_owner.create_next_node()
while node:
serialized_node = neptune_serializer.convert_node(node)
actual.append(serialized_node)
node = self.table_owner.create_next_node()
self.assertEqual(actual, expected)
def test_create_relation(self) -> None:
expected_relation1 = {
RELATION_START_KEY: owner1,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
expected_relation2 = {
RELATION_START_KEY: owner2,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
expected = [expected_relation1, expected_relation2]
actual = []
relation = self.table_owner.create_next_relation()
while relation:
serialized_relation = neo4_serializer.serialize_relationship(
relation)
actual.append(serialized_relation)
relation = self.table_owner.create_next_relation()
self.assertEqual(actual, expected)
def test_create_relation_neptune(self) -> None:
expected = [[{
NEPTUNE_HEADER_ID:
"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="User:"******"Table:" +
self.table_owner.get_metadata_model_key(),
label=TableOwner.OWNER_TABLE_RELATION_TYPE),
METADATA_KEY_PROPERTY_NAME_BULK_LOADER_FORMAT:
"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="User:"******"Table:" +
self.table_owner.get_metadata_model_key(),
label=TableOwner.OWNER_TABLE_RELATION_TYPE),
NEPTUNE_RELATIONSHIP_HEADER_FROM:
"User:"******"Table:" + self.table_owner.get_metadata_model_key(),
NEPTUNE_HEADER_LABEL:
TableOwner.OWNER_TABLE_RELATION_TYPE,
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT:
ANY,
NEPTUNE_CREATION_TYPE_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT:
NEPTUNE_CREATION_TYPE_JOB
}, {
NEPTUNE_HEADER_ID:
"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="Table:" +
self.table_owner.get_metadata_model_key(),
to_vertex_id="User:"******"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="Table:" +
self.table_owner.get_metadata_model_key(),
to_vertex_id="User:"******"Table:" + self.table_owner.get_metadata_model_key(),
NEPTUNE_RELATIONSHIP_HEADER_TO:
"User:"******"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="User:"******"Table:" +
self.table_owner.get_metadata_model_key(),
label=TableOwner.OWNER_TABLE_RELATION_TYPE),
METADATA_KEY_PROPERTY_NAME_BULK_LOADER_FORMAT:
"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="User:"******"Table:" +
self.table_owner.get_metadata_model_key(),
label=TableOwner.OWNER_TABLE_RELATION_TYPE),
NEPTUNE_RELATIONSHIP_HEADER_FROM:
"User:"******"Table:" + self.table_owner.get_metadata_model_key(),
NEPTUNE_HEADER_LABEL:
TableOwner.OWNER_TABLE_RELATION_TYPE,
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT:
ANY,
NEPTUNE_CREATION_TYPE_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT:
NEPTUNE_CREATION_TYPE_JOB
}, {
NEPTUNE_HEADER_ID:
"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="Table:" +
self.table_owner.get_metadata_model_key(),
to_vertex_id="User:"******"{label}:{from_vertex_id}_{to_vertex_id}".format(
from_vertex_id="Table:" +
self.table_owner.get_metadata_model_key(),
to_vertex_id="User:"******"Table:" + self.table_owner.get_metadata_model_key(),
NEPTUNE_RELATIONSHIP_HEADER_TO:
"User:" + owner2,
NEPTUNE_HEADER_LABEL:
TableOwner.TABLE_OWNER_RELATION_TYPE,
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT:
ANY,
NEPTUNE_CREATION_TYPE_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT:
NEPTUNE_CREATION_TYPE_JOB
}]]
actual = []
relation = self.table_owner.create_next_relation()
while relation:
serialized_relation = neptune_serializer.convert_relationship(
relation)
actual.append(serialized_relation)
relation = self.table_owner.create_next_relation()
self.assertEqual(expected, actual)
def test_create_records(self) -> None:
expected = [{
'rk': User.USER_NODE_KEY_FORMAT.format(email=owner1),
'email': owner1
}, {
'table_rk': self.table_owner.get_metadata_model_key(),
'user_rk': owner1
}, {
'rk': User.USER_NODE_KEY_FORMAT.format(email=owner2),
'email': owner2
}, {
'table_rk': self.table_owner.get_metadata_model_key(),
'user_rk': owner2
}]
actual = []
record = self.table_owner.create_next_record()
while record:
serialized_record = mysql_serializer.serialize_record(record)
actual.append(serialized_record)
record = self.table_owner.create_next_record()
self.assertEqual(actual, expected)
def test_create_nodes_with_owners_list(self) -> None:
self.table_owner_list = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', ' user2@2 '])
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
expected = [expected_node1, expected_node2]
actual = []
node = self.table_owner_list.create_next_node()
while node:
serialized_node = neo4_serializer.serialize_node(node)
actual.append(serialized_node)
node = self.table_owner_list.create_next_node()
self.assertEqual(actual, expected)
schema="schema1",
name="table1",
cluster="prod",
description=""),
TableMetadata(database="db2",
schema="schema2",
name="table2",
cluster="prod",
description=""),
]
EXPECTED_OWNERS = [
TableOwner(
db_name="db1",
cluster="prod",
schema="schema1",
table_name="table1",
owners=["foo", "bar"],
),
TableOwner(
db_name="db2",
cluster="prod",
schema="schema2",
table_name="table2",
owners=["foo", "bar"],
),
]
class TestChainedTransformerTask(unittest.TestCase):
def test_multi_yield_task(self) -> None:
class TestTableOwner(unittest.TestCase):
def setUp(self) -> None:
super(TestTableOwner, self).setUp()
self.table_owner = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners="user1@1, user2@2 ")
def test_get_owner_model_key(self) -> None:
owner = self.table_owner.get_owner_model_key(owner1)
self.assertEqual(owner, owner1)
def test_get_metadata_model_key(self) -> None:
metadata = self.table_owner.get_metadata_model_key()
self.assertEqual(metadata, 'hive://DEFAULT.BASE/TEST')
def test_create_nodes(self) -> None:
nodes = self.table_owner.create_nodes()
self.assertEqual(len(nodes), 2)
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
actual_nodes = [
neo4_serializer.serialize_node(node)
for node in nodes
]
self.assertTrue(expected_node1 in actual_nodes)
self.assertTrue(expected_node2 in actual_nodes)
def test_create_nodes_neptune(self) -> None:
nodes = self.table_owner.create_nodes()
expected_node1 = {
NEPTUNE_HEADER_ID: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NEPTUNE_HEADER_LABEL: User.USER_NODE_LABEL,
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: ANY,
NEPTUNE_CREATION_TYPE_NODE_PROPERTY_NAME_BULK_LOADER_FORMAT: NEPTUNE_CREATION_TYPE_JOB,
User.USER_NODE_EMAIL + ":String(single)": owner1
}
expected_node2 = {
NEPTUNE_HEADER_ID: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NEPTUNE_HEADER_LABEL: User.USER_NODE_LABEL,
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: ANY,
NEPTUNE_CREATION_TYPE_NODE_PROPERTY_NAME_BULK_LOADER_FORMAT: NEPTUNE_CREATION_TYPE_JOB,
User.USER_NODE_EMAIL + ":String(single)": owner2
}
actual_nodes = [
neptune_serializer.convert_node(node)
for node in nodes
]
self.assertTrue(expected_node1 in actual_nodes)
self.assertTrue(expected_node2 in actual_nodes)
def test_create_relation(self) -> None:
relations = self.table_owner.create_relation()
self.assertEqual(len(relations), 2)
expected_relation1 = {
RELATION_START_KEY: owner1,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
expected_relation2 = {
RELATION_START_KEY: owner2,
RELATION_START_LABEL: User.USER_NODE_LABEL,
RELATION_END_KEY: self.table_owner.get_metadata_model_key(),
RELATION_END_LABEL: 'Table',
RELATION_TYPE: TableOwner.OWNER_TABLE_RELATION_TYPE,
RELATION_REVERSE_TYPE: TableOwner.TABLE_OWNER_RELATION_TYPE
}
actual_relations = [
neo4_serializer.serialize_relationship(relation)
for relation in relations
]
self.assertTrue(expected_relation1 in actual_relations)
self.assertTrue(expected_relation2 in actual_relations)
def test_create_relation_neptune(self) -> None:
relations = self.table_owner.create_relation()
expected = [
{
NEPTUNE_HEADER_ID: "{from_vertex_id}_{to_vertex_id}_{label}".format(
from_vertex_id=owner1,
to_vertex_id=self.table_owner.get_metadata_model_key(),
label=TableOwner.OWNER_TABLE_RELATION_TYPE
),
NEPTUNE_RELATIONSHIP_HEADER_FROM: owner1,
NEPTUNE_RELATIONSHIP_HEADER_TO: self.table_owner.get_metadata_model_key(),
NEPTUNE_HEADER_LABEL: TableOwner.OWNER_TABLE_RELATION_TYPE,
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: ANY,
NEPTUNE_CREATION_TYPE_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: NEPTUNE_CREATION_TYPE_JOB
},
{
NEPTUNE_HEADER_ID: "{from_vertex_id}_{to_vertex_id}_{label}".format(
from_vertex_id=self.table_owner.get_metadata_model_key(),
to_vertex_id=owner1,
label=TableOwner.TABLE_OWNER_RELATION_TYPE
),
NEPTUNE_RELATIONSHIP_HEADER_FROM: self.table_owner.get_metadata_model_key(),
NEPTUNE_RELATIONSHIP_HEADER_TO: owner1,
NEPTUNE_HEADER_LABEL: TableOwner.TABLE_OWNER_RELATION_TYPE,
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: ANY,
NEPTUNE_CREATION_TYPE_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: NEPTUNE_CREATION_TYPE_JOB
}
]
actual_relations = [
neptune_serializer.convert_relationship(relation)
for relation in relations
]
self.assertTrue(expected in actual_relations)
def test_create_nodes_with_owners_list(self) -> None:
self.table_owner_list = TableOwner(db_name='hive',
schema=SCHEMA,
table_name=TABLE,
cluster=CLUSTER,
owners=['user1@1', ' user2@2 '])
nodes = self.table_owner_list.create_nodes()
self.assertEqual(len(nodes), 2)
expected_node1 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner1),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner1
}
expected_node2 = {
NODE_KEY: User.USER_NODE_KEY_FORMAT.format(email=owner2),
NODE_LABEL: User.USER_NODE_LABEL,
User.USER_NODE_EMAIL: owner2
}
actual_nodes = [
neo4_serializer.serialize_node(node)
for node in nodes
]
self.assertTrue(expected_node1 in actual_nodes)
self.assertTrue(expected_node2 in actual_nodes)