test: change from entity to models

klepp0 · Jul 6, 2024 · 729720b · 729720b
1 parent 66e75ac
commit 729720b
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 9 deletions.
diff --git a/tests/integrationtests/spotify/test_session.py b/tests/integrationtests/spotify/test_session.py
@@ -3,7 +3,8 @@
 
 import pytest
 
-from musync.common.entity import Origin, Playlist, Track, User
+# from musync.common.entity import Origin, Playlist, Track, User
+from musync.app.models import Origin, Playlist, Track, User
 from musync.spotify import SpotifySession
 from tests.unittests.common.entity import DATA_DIR
 

diff --git a/tests/integrationtests/test_sync_manager.py b/tests/integrationtests/test_sync_manager.py
@@ -1,6 +1,6 @@
 import pytest
 
-from musync.common.entity import Playlist, Track
+from musync.app.models import Playlist, Track
 from musync.common.sync_manager import SyncManager
 from musync.spotify import SpotifySession
 from musync.tidal import TidalSession

diff --git a/tests/integrationtests/tidal/test_session.py b/tests/integrationtests/tidal/test_session.py
@@ -3,7 +3,8 @@
 
 import pytest
 
-from musync.common.entity import Origin, Playlist, Track, User
+# from musync.common.entity import Origin, Playlist, Track, User
+from musync.app.models import Origin, Playlist, Track, User
 from musync.tidal import TidalSession
 from tests.unittests.common.entity import DATA_DIR
 

diff --git a/tests/unittests/app/models/test_artist.py b/tests/unittests/app/models/test_artist.py
@@ -2,8 +2,7 @@
 
 import pytest
 
-from musync.app.models.artist import Artist
-from musync.common.entity import Origin
+from musync.app.models import Artist, Origin
 from tests.unittests.app.models import DATA_DIR
 
 

diff --git a/tests/unittests/app/models/test_playlist.py b/tests/unittests/app/models/test_playlist.py
@@ -2,8 +2,7 @@
 
 import pytest
 
-from musync.app.models.playlist import Playlist
-from musync.common.entity.origin import Origin
+from musync.app.models import Origin, Playlist
 from tests.unittests.app.models import DATA_DIR
 
 

diff --git a/tests/unittests/app/models/test_track.py b/tests/unittests/app/models/test_track.py
@@ -4,8 +4,7 @@
 import pytest
 import pytz
 
-from musync.app.models.track import Track
-from musync.common.entity.origin import Origin
+from musync.app.models import Origin, Track
 from tests.unittests.app.models import DATA_DIR