"""Functional tests for fetching an mkt priority of product.""" import json from tests.testutils import db @db.test_schema def test_get_mkt_priority_success_with_data(client): """Check that mkt priority works.""" db.insert_mkt_priority_data() get_response = client.get( '/product/{}/mkt_priority'.format('92949')) assert get_response.status_code == 200 response_body = json.loads(get_response.data.decode()) assert response_body == { 'mkt_priority': [ { 'id': 1, 'upc': 829410910552, 'priority': 'a', 'country_id': 0, 'release_id': 92949 }, { 'id': 2, 'upc': 829410750950, 'priority': 'a', 'country_id': 0, 'release_id': 92949 } ] } @db.test_schema def test_get_mkt_priority_success_without_data(client): """Check that mkt priority works.""" db.insert_mkt_priority_data() get_response = client.get( '/product/{}/mkt_priority'.format('123')) assert get_response.status_code == 200 response_body = json.loads(get_response.data.decode()) assert response_body == { 'mkt_priority': [] } @db.test_schema def test_get_mkt_priority_failure(client): """Check that mkt priority works.""" db.insert_mkt_priority_data() get_response = client.get( '/product/{}/mkt_priority'.format('123aaa')) assert get_response.status_code == 404