def test_multiple_company_id_filter(app): url = 'http://localhost:80/api/v1/get-company-countries-and-sectors-of-interest' with app.test_client() as app_context: utils.assert_api_response( app_context=app_context, api=url, params='service-company-id=2&service-company-id=1', expected_response=( 200, { 'headers': [ 'serviceCompanyId', 'companyMatchId', 'country', 'sector', 'type', 'service', 'source', 'sourceId', 'timestamp', ], 'next': None, 'values': [list(entry_1.values()), list(entry_2.values())], }, ), )
def test_get_sectors(app): with mock.patch('app.sso.token.is_authenticated') as mock_is_authenticated: mock_is_authenticated.return_value = True url = 'http://localhost:80/api/v1/get-data-visualisation-data/sector' with app.test_client() as app_context: utils.assert_api_response( app_context=app_context, api=url, params='data-source=omis', expected_response=( 200, { 'nInterests': [ { 'date': '2009-10-01T00:00:00', 'nInterests': 1, 'nInterestsCumulative': 1, 'sector': 'sector1', 'shareOfInterest': 0.5, 'shareOfInterestCumulative': 0.5, }, { 'date': '2009-10-01T00:00:00', 'nInterests': 1, 'nInterestsCumulative': 1, 'sector': 'sector3', 'shareOfInterest': 0.5, 'shareOfInterestCumulative': 0.5, }, ], 'top': ['sector1', 'sector3'], }, ), )
def test_pagination_next(app): old_pagination_size = app.config['app']['pagination_size'] app.config['app']['pagination_size'] = 1 url = 'http://localhost:80/api/v1/get-company-countries-and-sectors-of-interest' with app.test_client() as app_context: utils.assert_api_response( app_context=app_context, api=url, params='', expected_response=( 200, { 'headers': [ 'serviceCompanyId', 'companyMatchId', 'country', 'sector', 'type', 'service', 'source', 'sourceId', 'timestamp', ], 'next': 'http://localhost/api/v1/' 'get-company-countries-and-sectors-of-interest?' 'orientation=tabular&' 'next-id=2', 'values': [list(entry_1.values())], }, ), ) app.config['app']['pagination_size'] = old_pagination_size
def test_invalid_sector_data_source(app): with mock.patch('app.sso.token.is_authenticated') as mock_is_authenticated: mock_is_authenticated.return_value = True url = 'http://localhost:80/api/v1/get-data-visualisation-data/sector' with app.test_client() as app_context: utils.assert_api_response( app_context=app_context, api=url, params='data-source=interactions', expected_response=(400, None), )