def test_get_app_server_successfully(client):
with patch(
'auth_server.decorators.admin_user_required_decorator.is_request_from_admin_user'
) as mock:
mock.return_value = True
with patch.object(
AppServerPersistence,
'get_app_server_by_token') as mock_app_server_persistence:
app_server_mocked = AppServer()
mock_app_server_persistence.return_value = app_server_mocked
hed = {'authorization': 'FAKETOKEN'}
response = client.get('/api/app_servers/' +
app_server_mocked.get_token(),
headers=hed,
follow_redirects=False)
value_expected = app_server_mocked.get_token()
assert mock.called
assert mock_app_server_persistence.called
assert json.loads(
response.data)['App server']['token'] == value_expected
def _regiter_app_server():
try:
app_servers_persistence = AppServerPersistence(current_app.db)
app_server = AppServer()
app_servers_persistence.save(app_server)
logger.debug("Successfully registered app server")
return {'App server token': app_server.get_token()}, HTTPStatus.CREATED
except Exception as e:
logger.error("Error:" + str(e))
return {
'Error': 'Couldnt register app server'
}, HTTPStatus.INTERNAL_SERVER_ERROR
def test_delete_app_server_successfully(postgresql_db):
session = postgresql_db.session
create_appservers_table(session)
assert query_first_app_server(session) is None
app_server_to_save = AppServer()
sut = AppServerPersistence(postgresql_db)
sut.save(app_server_to_save)
row = query_first_app_server(session)
assert row is not None
assert row[1].month == datetime.datetime.now().month
sut.delete(app_server_to_save.get_token())
assert query_first_app_server(session) is None
def test_get_all_app_servers_successfully(postgresql_db):
session = postgresql_db.session
create_appservers_table(session)
assert query_first_app_server(session) is None
app_server_to_save = AppServer()
sut = AppServerPersistence(postgresql_db)
sut.save(app_server_to_save)
row = query_first_app_server(session)
assert row is not None
assert row[1].month == datetime.datetime.now().month
app_server_to_save_2 = AppServer()
sut = AppServerPersistence(postgresql_db)
sut.save(app_server_to_save_2)
row = query_first_app_server(session)
assert row is not None
app_servers_received = sut.get_all_app_servers()
assert len(app_servers_received) == 2
def test_app_server_returns_token_successfully():
app_server = AppServer()
assert app_server.get_token() is not None
def test_create_app_server_successfully():
app_server = AppServer()
assert app_server is not None
def test_serialize_app_server_successfully():
app_server = AppServer()
app_server_serialized = app_server.serialize()
app_server_serialized['token'] = app_server.get_token