def test_import_with_invalid_url(self):
# conditions
get = Mock()
# execution
cmd = 'hello_world'
rtn = import_character(cmd, self.db, get)
# expected
self.assertEqual(rtn, 'hello_world is not a valid URL')
def test_valid_import(self):
# conditions
response = Mock()
response.status_code = 200
response.json = Mock(return_value=CHARACTER_JSON)
get = Mock(return_value=response)
# execution
cmd = 'http://example.com/my/character'
rtn = import_character(cmd, self.db, get)
# expected
self.db.save_character_info.assert_called_with('123456',
CHARACTER_JSON)
self.assertEqual(rtn, 'Character "John Wick" imported successfully!')
def test_import_with_request_error(self):
# conditions
response = Mock()
response.status_code = 500
get = Mock(return_value=response)
# execution
cmd = 'http://example.com/my/character'
rtn = import_character(cmd, self.db, get)
# expected
self.assertEqual(
rtn,
'Error fetching http://example.com/my/character (status_code: 500)'
)