def test_client_echo_valid_success(self, generate_key_method):
generate_key_method.return_value = "12345GENKEY"
with requests_mock.Mocker() as mocker:
mocker.register_uri("GET", "http://localhost/endpoint", json={"echo": "12345GENKEY"}, status_code=client.OK)
self.assertTrue(client_echo_valid("http://localhost/endpoint"))
def test_client_echo_valid_fail_wrong_status(self):
with requests_mock.Mocker() as mocker:
mocker.register_uri('GET', 'http://localhost/endpoint',
json={'echo': '12345GENKEY'},
status_code=client.INTERNAL_SERVER_ERROR)
self.assertFalse(client_echo_valid('http://localhost/endpoint'))
def test_client_echo_valid_success(self, generate_key_method):
generate_key_method.return_value = '12345GENKEY'
with requests_mock.Mocker() as mocker:
mocker.register_uri('GET', 'http://localhost/endpoint',
json={'echo': '12345GENKEY'},
status_code=client.OK)
self.assertTrue(client_echo_valid('http://localhost/endpoint'))
def test_client_echo_valid_fail_wrong_status(self):
with requests_mock.Mocker() as mocker:
mocker.register_uri(
"GET",
"http://localhost/endpoint",
json={"echo": "12345GENKEY"},
status_code=client.INTERNAL_SERVER_ERROR,
)
self.assertFalse(client_echo_valid("http://localhost/endpoint"))
def post(self):
"""
Creates a new account
"""
json_data = request.get_json()
if not client_echo_valid(json_data['endpoint']):
return make_response(jsonify({'Error': 'Echo response failed'}),
client.BAD_REQUEST)
return insert_account(DEFAULT_ACCOUNTS_TABLE,
**{'username': json_data['username'],
'endpoint': json_data['endpoint'],
'is_admin': False,
'failed_count': 0,
'api_key': generate_key(),
'secret_key': generate_key()})