Fix references to QueueMessageTypes

bcgov · May 6, 2024 · 20e1377 · 20e1377
1 parent d7e16c5
commit 20e1377
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/auth-api/tests/unit/services/test_affiliation_invitation.py b/auth-api/tests/unit/services/test_affiliation_invitation.py
@@ -21,6 +21,7 @@
 import mock
 import pytest
 from freezegun import freeze_time
+from sbc_common_components.utils.enums import QueueMessageTypes
 
 import auth_api.services.authorization as auth
 import auth_api.utils.account_mailer
@@ -38,7 +39,7 @@
 from auth_api.services import Org as OrgService
 from auth_api.services import User
 from auth_api.utils import roles
-from auth_api.utils.enums import InvitationStatus, QueueMessageTypes
+from auth_api.utils.enums import InvitationStatus
 from tests.utilities.factory_scenarios import TestContactInfo, TestEntityInfo, TestJwtClaims, TestOrgInfo, TestUserInfo
 from tests.utilities.factory_utils import (
     factory_affiliation_invitation, factory_entity_model, factory_membership_model, factory_user_model,

diff --git a/auth-api/tests/unit/services/test_queues.py b/auth-api/tests/unit/services/test_queues.py
@@ -5,7 +5,7 @@
 from auth_api import create_app
 from auth_api.services.activity_log_publisher import Activity, ActivityLogPublisher
 from auth_api.utils.account_mailer import publish_to_mailer
-from auth_api.utils.enums import QueueMessageTypes
+from sbc_common_components.utils.enums import QueueMessageTypes
 
 
 @pytest.mark.skip(reason='ADHOC only test.')