def test_mark_push_complete(self, mark_push_event, mock_context):
event = {
'task_name': 'mark_push_complete',
'task_kwargs': mark_push_event
}
results = handler(event, mock_context)
assert results
def test_mark_push_complete(self, mark_push_event, mock_context):
with patch('toll_booth.tasks.mark_push.boto3.resource') as mock_dynamo:
mock_resource = MagicMock()
mock_table = MagicMock()
mock_resource.Table = mock_table
mock_dynamo.return_value = mock_resource
event = {'task_name': 'mark_push_complete', 'task_kwargs': mark_push_event}
results = handler(event, mock_context)
assert results is None
assert mock_table.update_item.called
def test_generate_source_vertex(self, source_vertex_task_integration_event, mock_context, mocks):
results = handler(source_vertex_task_integration_event, mock_context)
assert results
parsed_results = ajson.loads(results)
expected_keys = ['source_vertex', 'schema', 'schema_entry', 'extracted_data']
for key_value in expected_keys:
assert key_value in parsed_results
generated_vertex_data = parsed_results['source_vertex']
assert generated_vertex_data.vertex_properties
assert mocks['bullhorn'].called
assert mocks['gql'].called
def test_task(self, mock_context):
event = {
"task_name": "get_encounters",
"task_kwargs": {
"id_source": "PSI"
}
}
event_string = rapidjson.dumps(event)
message_object = {'Message': event_string}
body_object = {'body': rapidjson.dumps(message_object)}
event = {'Records': [body_object]}
results = handler(event, mock_context)
assert results
def test_task(self, task_event, mock_context):
os.environ['ALGERNON_BUCKET_NAME'] = 'algernonsolutions-leech-prod'
os.environ[
'RDS_HOST'] = 'algernon-1.cluster-cnd32dx4xing.us-east-1.rds.amazonaws.com'
os.environ['RDS_DB_NAME'] = 'algernon'
os.environ['INDEX_TABLE_NAME'] = 'Indexes'
os.environ[
'RDS_SECRET_ARN'] = 'arn:aws:secretsmanager:us-east-1:322652498512:secret:rds-xudEmz'
os.environ[
'RDS_CLUSTER_ARN'] = 'arn:aws:rds:us-east-1:322652498512:cluster:algernon'
os.environ[
'FIRE_HOSE_NAME'] = 'arn:aws:rds:us-east-1:322652498512:cluster:algernon'
results = handler(task_event, mock_context)
assert results
def test_parser(self, mock_context):
os.environ['ALGERNON_BUCKET_NAME'] = 'algernonsolutions-leech-dev'
encounter_internal_ids = [
'4b234ef92e04789c211e0e43eea3cd45',
'01e63570d0b834b35f757f9ed5a0ae7f',
'67e449e989722799dbebde18124991e3',
"cebdc99f8c8a80289c6f60d97e9f2de5"
]
for encounter_internal_id in encounter_internal_ids:
event = {
'parse_type': 'distill',
'parse_kwargs': {
'encounter_internal_id': encounter_internal_id
}
}
try:
results = handler(event, mock_context)
except EmptyParserResponseException:
continue
assert results
def test_graph_cluster(self, graph_cluster_event, mock_context, mocks):
results = handler(graph_cluster_event, mock_context)
assert results
def test_query_edges_filtered(self, test_event, mock_context, mocks):
event = test_event('get_edge_filtered')
results = handler(event, mock_context)
assert results
def test_paginated_query(self, test_event, mock_context, mocks):
event = test_event('maltego_query_vertex')
results = handler(event, mock_context)
assert results
def test_task(self, test_task_event, mock_context, mock_bullhorn):
results = handler(test_task_event, mock_context)
assert results
assert mock_bullhorn.called
assert mock_bullhorn.call_count == 1
def test_graph_pusher(self, push_event, mock_context):
push_event['push_type'] = 'graph'
results = toll_booth.handler(push_event, mock_context)
assert results
def test_generate_potential_connections(self, potential_connections_unit_event, mock_context, mocks):
results = handler(potential_connections_unit_event, mock_context)
assert results
assert mocks['bullhorn'].called
def test_handler(self, mock_context):
event = {}
results = handler(event, mock_context)
assert results
def test_handler(self, aio_event, mock_context):
event = {'task_name': 'leech', 'task_kwargs': aio_event}
results = handler(event, mock_context)
assert results
def test_graph_object(self, graph_vertex_event, mock_context, mocks):
event = graph_vertex_event('vertex_connected_edges')
results = handler(event, mock_context)
assert results
def test_query_edge_connection(self, test_event, mock_context, mocks):
event = test_event('edge_connection_edges')
results = handler(event, mock_context)
assert results
def test_check_for_existing_vertexes(self, find_existing_vertexes, mock_context, mocks):
results = handler(find_existing_vertexes, mock_context)
assert results
assert mocks['bullhorn'].called
def test_query_vertex_connected_edges_i(self, test_event, mock_context):
event = test_event('vertex_connected_edges')
results = handler(event, mock_context)
assert results
def test_generate_potential_edge(self, generate_edge_integration_event, mock_context):
results = handler(generate_edge_integration_event, mock_context)
assert results