def test_infringements_by_product_id_and_pirate_source_id_request_success(
pg_connect_mock, db_config,
infringements_by_product_id_and_pirate_source_id_request):
mock_pg_fetchall(pg_connect_mock, [])
response = lambda_handler(
infringements_by_product_id_and_pirate_source_id_request, None)
assert response["statusCode"] == 200
def test_query_partner_request_success(pg_connect_mock, db_config,
query_partner_request):
mock_pg_fetchall(pg_connect_mock, [])
response = lambda_handler(query_partner_request, None)
assert response["statusCode"] == 200
def test_single_pirate_source_request_success(
pg_connect_mock, db_config, single_pirate_source_request
):
mock_pg_fetchall(pg_connect_mock, [])
response = lambda_handler(single_pirate_source_request, None)
assert response["statusCode"] == 200
def test_all_products_request_success(
pg_connect_mock, db_config, all_pirate_sources_for_partner_request
):
mock_pg_fetchall(pg_connect_mock, [])
response = lambda_handler(all_pirate_sources_for_partner_request, None)
assert response["statusCode"] == 200