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_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 route_scenario_2(name, keys):
handler = ResponseHandler()
response = handler.get(name, keys)
return 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)