darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-02 20:58:38 -04:00
Code Issues Packages Projects Releases Wiki Activity

more tests

Browse Source
This commit is contained in:
Kenneth Kehl
2025-07-07 10:15:48 -07:00
parent 34193e0e54
commit fe6903c6cc
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/commands.py
View File

@@ -752,12 +752,13 @@ def create_admin_jwt():
@notify_command(name="create-user-jwt")
@click.option("-t", "--token", required=True, prompt=False)
def create_user_jwt(token):
if getenv("NOTIFY_ENVIRONMENT", "") != "development":
if getenv("NOTIFY_ENVIRONMENT", "") not in ["development", "test"]:
current_app.logger.error("Can only be run in development")
return
service_id = token[-73:-37]
api_key = token[-36:]
current_app.logger.info(create_jwt_token(api_key, service_id))
token = create_jwt_token(api_key, service_id)
current_app.logger.info(token)
def _update_template(id, name, template_type, content, subject):

10
tests/app/test_commands.py
View File

@@ -13,6 +13,7 @@ from app.commands import (
bulk_invite_user_to_service,
create_new_service,
create_test_user,
create_user_jwt,
download_csv_file_by_name,
dump_sms_senders,
dump_user_info,
@@ -761,3 +762,12 @@ def test_get_service_sender_phones(mock_execute, notify_api):
result = runner.invoke(get_service_sender_phones, ["-s", "service-id"])
assert result.exit_code == 0
mock_execute.assert_called_once()
@patch("app.commands.current_app.logger.info")
def test_create_user_jwt(mock_logger, notify_api, sample_api_key, sample_service):
token = f"{sample_service.id}{sample_api_key}"
runner = notify_api.test_cli_runner()
result = runner.invoke(create_user_jwt, ["-t", token])
assert result.exit_code == 0
mock_logger.assert_called_once()