darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

101
tests/app/dao/test_api_key_dao.py
View File

@@ -15,10 +15,14 @@ from app.models import KEY_TYPE_NORMAL, ApiKey
def test_save_api_key_should_create_new_api_key_and_history(sample_service):
api_key = ApiKey(**{'service': sample_service,
'name': sample_service.name,
'created_by': sample_service.created_by,
'key_type': KEY_TYPE_NORMAL})
api_key = ApiKey(
**{
"service": sample_service,
"name": sample_service.name,
"created_by": sample_service.created_by,
"key_type": KEY_TYPE_NORMAL,
}
)
save_model_api_key(api_key)
all_api_keys = get_model_api_keys(service_id=sample_service.id)
@@ -32,8 +36,9 @@ def test_save_api_key_should_create_new_api_key_and_history(sample_service):
assert all_history[0].version == api_key.version
def test_expire_api_key_should_update_the_api_key_and_create_history_record(notify_api,
sample_api_key):
def test_expire_api_key_should_update_the_api_key_and_create_history_record(
notify_api, sample_api_key
):
expire_api_key(service_id=sample_api_key.service_id, api_key_id=sample_api_key.id)
all_api_keys = get_model_api_keys(service_id=sample_api_key.service_id)
assert len(all_api_keys) == 1
@@ -51,13 +56,19 @@ def test_expire_api_key_should_update_the_api_key_and_create_history_record(noti
sorted_all_history[1].version = 2
def test_get_api_key_should_raise_exception_when_api_key_does_not_exist(sample_service, fake_uuid):
def test_get_api_key_should_raise_exception_when_api_key_does_not_exist(
sample_service, fake_uuid
):
with pytest.raises(NoResultFound):
get_model_api_keys(sample_service.id, id=fake_uuid)
def test_should_return_api_key_for_service(notify_api, notify_db_session, sample_api_key):
api_key = get_model_api_keys(service_id=sample_api_key.service_id, id=sample_api_key.id)
def test_should_return_api_key_for_service(
notify_api, notify_db_session, sample_api_key
):
api_key = get_model_api_keys(
service_id=sample_api_key.service_id, id=sample_api_key.id
)
assert api_key == sample_api_key
@@ -75,26 +86,38 @@ def test_get_unsigned_secret_returns_key(sample_api_key):
def test_should_not_allow_duplicate_key_names_per_service(sample_api_key, fake_uuid):
api_key = ApiKey(**{'id': fake_uuid,
'service': sample_api_key.service,
'name': sample_api_key.name,
'created_by': sample_api_key.created_by,
'key_type': KEY_TYPE_NORMAL})
api_key = ApiKey(
**{
"id": fake_uuid,
"service": sample_api_key.service,
"name": sample_api_key.name,
"created_by": sample_api_key.created_by,
"key_type": KEY_TYPE_NORMAL,
}
)
with pytest.raises(IntegrityError):
save_model_api_key(api_key)
def test_save_api_key_can_create_key_with_same_name_if_other_is_expired(sample_service):
expired_api_key = ApiKey(**{'service': sample_service,
'name': "normal api key",
'created_by': sample_service.created_by,
'key_type': KEY_TYPE_NORMAL,
'expiry_date': datetime.utcnow()})
expired_api_key = ApiKey(
**{
"service": sample_service,
"name": "normal api key",
"created_by": sample_service.created_by,
"key_type": KEY_TYPE_NORMAL,
"expiry_date": datetime.utcnow(),
}
)
save_model_api_key(expired_api_key)
api_key = ApiKey(**{'service': sample_service,
'name': "normal api key",
'created_by': sample_service.created_by,
'key_type': KEY_TYPE_NORMAL})
api_key = ApiKey(
**{
"service": sample_service,
"name": "normal api key",
"created_by": sample_service.created_by,
"key_type": KEY_TYPE_NORMAL,
}
)
save_model_api_key(api_key)
keys = ApiKey.query.all()
assert len(keys) == 2
@@ -106,26 +129,32 @@ def test_save_api_key_should_not_create_new_service_history(sample_service):
assert Service.query.count() == 1
assert Service.get_history_model().query.count() == 1
api_key = ApiKey(**{'service': sample_service,
'name': sample_service.name,
'created_by': sample_service.created_by,
'key_type': KEY_TYPE_NORMAL})
api_key = ApiKey(
**{
"service": sample_service,
"name": sample_service.name,
"created_by": sample_service.created_by,
"key_type": KEY_TYPE_NORMAL,
}
)
save_model_api_key(api_key)
assert Service.get_history_model().query.count() == 1
@pytest.mark.parametrize('days_old, expected_length', [(5, 1), (8, 0)])
@pytest.mark.parametrize("days_old, expected_length", [(5, 1), (8, 0)])
def test_should_not_return_revoked_api_keys_older_than_7_days(
sample_service,
days_old,
expected_length
sample_service, days_old, expected_length
):
expired_api_key = ApiKey(**{'service': sample_service,
'name': sample_service.name,
'created_by': sample_service.created_by,
'key_type': KEY_TYPE_NORMAL,
'expiry_date': datetime.utcnow() - timedelta(days=days_old)})
expired_api_key = ApiKey(
**{
"service": sample_service,
"name": sample_service.name,
"created_by": sample_service.created_by,
"key_type": KEY_TYPE_NORMAL,
"expiry_date": datetime.utcnow() - timedelta(days=days_old),
}
)
save_model_api_key(expired_api_key)
all_api_keys = get_model_api_keys(service_id=sample_service.id)