def test_dashboard_owner_relations(self) -> None:
dashboard_usage = DashboardUsage(
dashboard_group_id='dashboard_group_id',
dashboard_id='dashboard_id',
email='*****@*****.**',
view_count=123,
cluster='cluster_id',
product='product_id')
actual = dashboard_usage.create_next_relation()
actual_serialized = neo4_serializer.serialize_relationship(actual)
expected: Dict[str, Any] = {
'read_count:UNQUOTED': 123,
RELATION_END_KEY: '*****@*****.**',
RELATION_START_LABEL: 'Dashboard',
RELATION_END_LABEL: 'User',
RELATION_START_KEY:
'product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
RELATION_TYPE: 'READ_BY',
RELATION_REVERSE_TYPE: 'READ'
}
assert actual is not None
self.assertDictEqual(expected, actual_serialized)
self.assertIsNone(dashboard_usage.create_next_relation())
def test_dashboard_usage_no_user_nodes(self) -> None:
dashboard_usage = DashboardUsage(dashboard_group_id='dashboard_group_id', dashboard_id='dashboard_id',
email='*****@*****.**', view_count=123,
should_create_user_node=False, cluster='cluster_id',
product='product_id')
self.assertIsNone(dashboard_usage.create_next_node())
def test_dashboard_usage_user_nodes(self) -> None:
dashboard_usage = DashboardUsage(
dashboard_group_id='dashboard_group_id',
dashboard_id='dashboard_id',
email='*****@*****.**',
view_count=123,
cluster='cluster_id',
product='product_id',
should_create_user_node=True)
actual = dashboard_usage.create_next_node()
actual_serialized = neo4_serializer.serialize_node(actual)
expected: Dict[str, Any] = {
'is_active:UNQUOTED': True,
'last_name': '',
'full_name': '',
'employee_type': '',
'first_name': '',
'updated_at:UNQUOTED': 0,
'LABEL': 'User',
'slack_id': '',
'KEY': '*****@*****.**',
'github_username': '',
'team_name': '',
'email': '*****@*****.**',
'role_name': ''
}
assert actual is not None
self.assertDictEqual(expected, actual_serialized)
self.assertIsNone(dashboard_usage.create_next_node())
def test_dashboard_usage_user_records(self) -> None:
dashboard_usage = DashboardUsage(
dashboard_group_id='dashboard_group_id',
dashboard_id='dashboard_id',
email='*****@*****.**',
view_count=123,
cluster='cluster_id',
product='product_id',
should_create_user_node=True)
actual1 = dashboard_usage.create_next_record()
actual1_serialized = mysql_serializer.serialize_record(actual1)
expected1 = {
'rk': '*****@*****.**',
'email': '*****@*****.**',
}
assert actual1 is not None
self.assertDictEqual(expected1, actual1_serialized)
actual2 = dashboard_usage.create_next_record()
actual2_serialized = mysql_serializer.serialize_record(actual2)
expected2 = {
'user_rk': '*****@*****.**',
'dashboard_rk':
'product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
'read_count': 123
}
assert actual2 is not None
self.assertDictEqual(expected2, actual2_serialized)
self.assertIsNone(dashboard_usage.create_next_record())
def test_dashboard_usage_user_records(self) -> None:
dashboard_usage = DashboardUsage(
dashboard_group_id='dashboard_group_id',
dashboard_id='dashboard_id',
email='*****@*****.**',
view_count=123,
cluster='cluster_id',
product='product_id',
should_create_user_node=True)
actual1 = dashboard_usage.create_next_record()
actual1_serialized = mysql_serializer.serialize_record(actual1)
expected1 = {
'rk': '*****@*****.**',
'email': '*****@*****.**',
'is_active': True,
'first_name': '',
'last_name': '',
'full_name': '',
'github_username': '',
'team_name': '',
'employee_type': '',
'slack_id': '',
'role_name': '',
'updated_at': 0
}
assert actual1 is not None
self.assertDictEqual(expected1, actual1_serialized)
actual2 = dashboard_usage.create_next_record()
actual2_serialized = mysql_serializer.serialize_record(actual2)
expected2 = {
'user_rk': '*****@*****.**',
'dashboard_rk':
'product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
'read_count': 123
}
assert actual2 is not None
self.assertDictEqual(expected2, actual2_serialized)
self.assertIsNone(dashboard_usage.create_next_record())
def test_dashboard_usage_user_nodes(self) -> None:
dashboard_usage = DashboardUsage(
dashboard_group_id='dashboard_group_id',
dashboard_id='dashboard_id',
email='*****@*****.**',
view_count=123,
cluster='cluster_id',
product='product_id',
should_create_user_node=True)
actual = dashboard_usage.create_next_node()
actual_serialized = neo4_serializer.serialize_node(actual)
expected: Dict[str, Any] = {
'LABEL': 'User',
'KEY': '*****@*****.**',
'email': '*****@*****.**',
}
assert actual is not None
self.assertDictEqual(expected, actual_serialized)
self.assertIsNone(dashboard_usage.create_next_node())
def test_dashboard_owner_relations_neptune(self) -> None:
dashboard_usage = DashboardUsage(dashboard_group_id='dashboard_group_id', dashboard_id='dashboard_id',
email='*****@*****.**', view_count=123, cluster='cluster_id',
product='product_id')
actual = dashboard_usage.create_next_relation()
actual_serialized = neptune_serializer.convert_relationship(actual)
neptune_forward_expected = {
NEPTUNE_HEADER_ID: "{from_vertex_id}_{to_vertex_id}_{label}".format(
from_vertex_id='product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
to_vertex_id='*****@*****.**',
label='READ_BY'
),
NEPTUNE_RELATIONSHIP_HEADER_FROM: 'product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
NEPTUNE_RELATIONSHIP_HEADER_TO: '*****@*****.**',
NEPTUNE_HEADER_LABEL: 'READ_BY',
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: ANY,
NEPTUNE_CREATION_TYPE_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: NEPTUNE_CREATION_TYPE_JOB,
'read_count:Long(single)': 123
}
neptune_reversed_expected = {
NEPTUNE_HEADER_ID: "{from_vertex_id}_{to_vertex_id}_{label}".format(
from_vertex_id='*****@*****.**',
to_vertex_id='product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
label='READ'
),
NEPTUNE_RELATIONSHIP_HEADER_FROM: '*****@*****.**',
NEPTUNE_RELATIONSHIP_HEADER_TO: 'product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
NEPTUNE_HEADER_LABEL: 'READ',
NEPTUNE_LAST_EXTRACTED_AT_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: ANY,
NEPTUNE_CREATION_TYPE_RELATIONSHIP_PROPERTY_NAME_BULK_LOADER_FORMAT: NEPTUNE_CREATION_TYPE_JOB,
'read_count:Long(single)': 123,
}
assert actual is not None
self.assertDictEqual(neptune_forward_expected, actual_serialized[0])
self.assertDictEqual(neptune_reversed_expected, actual_serialized[1])
self.assertIsNone(dashboard_usage.create_next_relation())
def test_dashboard_owner_relations(self):
# type: () -> None
dashboard_usage = DashboardUsage(
dashboard_group_id='dashboard_group_id',
dashboard_id='dashboard_id',
email='*****@*****.**',
view_count=123,
cluster='cluster_id',
product='product_id')
actual = dashboard_usage.create_next_relation()
expected = {
'read_count:UNQUOTED': 123,
RELATION_END_KEY: '*****@*****.**',
RELATION_START_LABEL: 'Dashboard',
RELATION_END_LABEL: 'User',
RELATION_START_KEY:
'product_id_dashboard://cluster_id.dashboard_group_id/dashboard_id',
RELATION_TYPE: 'READ_BY',
RELATION_REVERSE_TYPE: 'READ'
}
self.assertDictEqual(expected, actual)
self.assertIsNone(dashboard_usage.create_next_relation())
