initial

tests/__init__.py

@@ -1,7 +1,6 @@
 import uuid
 
 from flask import current_app
-from notifications_python_client.authentication import create_jwt_token
 from sqlalchemy import select
 
 from app import db
@@ -9,6 +8,7 @@ from app.dao.api_key_dao import save_model_api_key
 from app.dao.services_dao import dao_fetch_service_by_id
 from app.enums import KeyType
 from app.models import ApiKey
+from notifications_python_client.authentication import create_jwt_token
 
 
 def create_service_authorization_header(service_id, key_type=KeyType.NORMAL):

tests/app/authentication/test_authentication.py

@@ -4,7 +4,6 @@ import uuid
 import jwt
 import pytest
 from flask import g, request
-from notifications_python_client.authentication import create_jwt_token
 
 from app import db
 from app.authentication.auth import (
@@ -18,6 +17,7 @@ from app.authentication.auth import (
 )
 from app.dao.api_key_dao import expire_api_key, get_model_api_keys, get_unsigned_secrets
 from app.dao.services_dao import dao_fetch_service_by_id
+from notifications_python_client.authentication import create_jwt_token
 from tests import create_admin_authorization_header, create_service_authorization_header
 from tests.conftest import set_config_values
 

tests/app/notifications/test_rest.py

@@ -3,13 +3,13 @@ import uuid
 import pytest
 from flask import current_app, json
 from freezegun import freeze_time
-from notifications_python_client.authentication import create_jwt_token
 
 from app.dao.api_key_dao import save_model_api_key
 from app.dao.notifications_dao import dao_update_notification
 from app.dao.templates_dao import dao_update_template
 from app.enums import KeyType, NotificationStatus, NotificationType, TemplateType
 from app.models import ApiKey
+from notifications_python_client.authentication import create_jwt_token
 from tests import create_service_authorization_header
 from tests.app.db import create_api_key, create_notification
 

tests/app/service/send_notification/test_send_notification.py

@@ -4,7 +4,6 @@ import string
 import pytest
 from flask import current_app, json
 from freezegun import freeze_time
-from notifications_python_client.authentication import create_jwt_token
 from sqlalchemy import func, select
 
 import app
@@ -17,6 +16,7 @@ from app.enums import KeyType, NotificationType, TemplateType
 from app.errors import InvalidRequest
 from app.models import ApiKey, Notification, NotificationHistory, Template
 from app.service.send_notification import send_one_off_notification
+from notifications_python_client.authentication import create_jwt_token
 from notifications_utils import SMS_CHAR_COUNT_LIMIT
 from tests import create_service_authorization_header
 from tests.app.db import (