85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
import json
|
|
import pytest
|
|
|
|
from scrobbles.models import Scrobble
|
|
from rest_framework.authtoken.models import Token
|
|
from django.contrib.auth import get_user_model
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class MopidyRequest:
|
|
name = "Same in the End"
|
|
artist = "Sublime"
|
|
album = "Sublime"
|
|
track_number = 4
|
|
run_time_ticks = 156604
|
|
run_time = "156"
|
|
playback_time_ticks = 15045
|
|
musicbrainz_track_id = "54214d63-5adf-4909-87cd-c65c37a6d558"
|
|
musicbrainz_album_id = "03b864cd-7761-314c-a892-05a89ddff00d"
|
|
musicbrainz_artist_id = "95f5b748-d370-47fe-85bd-0af2dc450bc0"
|
|
mopidy_uri = "local:track:Sublime%20-%20Sublime/Disc%201%20-%2004%20-%20Same%20in%20the%20End.mp3"
|
|
status = "resumed"
|
|
|
|
def __init__(self, **kwargs):
|
|
self.request_data = {
|
|
"name": kwargs.get('name', self.name),
|
|
"artist": kwargs.get("artist", self.artist),
|
|
"album": kwargs.get("album", self.album),
|
|
"track_number": int(kwargs.get("track_number", self.track_number)),
|
|
"run_time_ticks": int(
|
|
kwargs.get("run_time_ticks", self.run_time_ticks)
|
|
),
|
|
"run_time": int(kwargs.get("run_time", self.run_time)),
|
|
"playback_time_ticks": int(
|
|
kwargs.get("playback_time_ticks", self.playback_time_ticks)
|
|
),
|
|
"musicbrainz_track_id": kwargs.get(
|
|
"musicbrainz_track_id", self.musicbrainz_track_id
|
|
),
|
|
"musicbrainz_album_id": kwargs.get(
|
|
"musicbrainz_album_id", self.musicbrainz_album_id
|
|
),
|
|
"musicbrainz_artist_id": kwargs.get(
|
|
"musicbrainz_artist_id", self.musicbrainz_artist_id
|
|
),
|
|
"mopidy_uri": kwargs.get("mopidy_uri", self.mopidy_uri),
|
|
"status": kwargs.get("status", self.status),
|
|
}
|
|
|
|
def __eq__(self, other):
|
|
for key in self.request_data.keys():
|
|
if self.request_data[key] != getattr(self, key):
|
|
return False
|
|
return True
|
|
|
|
@property
|
|
def request_json(self):
|
|
return json.dumps(self.request_data)
|
|
|
|
|
|
@pytest.fixture
|
|
def valid_auth_token():
|
|
user = User.objects.create(email='test@exmaple.com')
|
|
return Token.objects.create(user=user).key
|
|
|
|
|
|
@pytest.fixture
|
|
def mopidy_track_request_data():
|
|
return MopidyRequest().request_json
|
|
|
|
|
|
@pytest.fixture
|
|
def mopidy_track_diff_album_request_data(**kwargs):
|
|
mb_album_id = "0c56c457-afe1-4679-baab-759ba8dd2a58"
|
|
return MopidyRequest(
|
|
album="Gold", musicbrainz_album_id=mb_album_id
|
|
).request_json
|
|
|
|
|
|
@pytest.fixture
|
|
def mopidy_podcast_request_data():
|
|
mopidy_uri = "local:podcast:Up%20First/2022-01-01%20Up%20First.mp3"
|
|
return MopidyRequest(mopidy_uri=mopidy_uri).request_json
|