Update imports in tests to new module

NeonGeckoCom · Nov 4, 2024 · fbdc143 · fbdc143
1 parent e004b51
commit fbdc143
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/test_databases_sqlite.py b/tests/test_databases_sqlite.py
@@ -6,7 +6,8 @@
 
 from neon_users_service.databases.sqlite import SQLiteUserDatabase
 from neon_users_service.exceptions import UserExistsError, UserNotFoundError
-from neon_users_service.models import User, AccessRoles
+from neon_data_models.models.user import User
+from neon_data_models.enum import AccessRoles
 
 
 class TestSqlite(TestCase):

diff --git a/tests/test_service.py b/tests/test_service.py
@@ -7,7 +7,7 @@
 from neon_users_service.databases.sqlite import SQLiteUserDatabase
 from neon_users_service.exceptions import ConfigurationError, AuthenticationError, UserNotFoundError, \
     UserNotMatchedError
-from neon_users_service.models import User
+from neon_data_models.models.user import User
 from neon_users_service.service import NeonUsersService