def test_empty_keys(self): handler = ResponseHandler() handler.country_info = CountryInfoPlaceHolder({ "name": "Egypt", "capital": "Cairo" }) response = handler.get('egypt', '') self.assertEqual(response, handler.invalid_info)
def test_get_true_info(self): handler = ResponseHandler() handler.country_info = CountryInfoPlaceHolder({ "name": "Egypt", "capital": "Cairo" }) response = handler.get('egypt', 'name') self.assertEqual(response['name'], 'Egypt')
def test_get_any_info(self): handler = ResponseHandler() handler.country_info = CountryInfoPlaceHolder({ "name": "Egypt", "capital": "Cairo" }) response = handler.get_all_info('egypt') self.assertTrue(len(response) > 0)
def test_get_valid_data(self): handler = ResponseHandler() handler.country_info = CountryInfoPlaceHolder({ "capital": "Cairo", "demonym": "Egyptian" }) response = handler.get('egypt', 'capital,demonym') expected = {"capital": "Cairo", "demonym": "Egyptian"} self.assertEqual(expected, response)
def test_filter_with_invalid_keys(self): handler = ResponseHandler() handler.country_info = CountryInfoPlaceHolder({ "capital": "Cairo", "demonym": "Egyptian" }) with open('offline_country_info.json') as file: info = json.load(file) response = handler.filter('capital,invalid_key', info) self.assertEqual(handler.invalid_info, response)
def test_public_api_error_get(self): handler = ResponseHandler() handler.country_info = CountryInfoPlaceHolder(None) response = handler.get('egypt', 'capital') self.assertEqual(response, handler.public_api_error)