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