vrobbler/tests/scrobbles_tests/conftest.py

import json

import pytest
from django.contrib.auth import get_user_model
from rest_framework.authtoken.models import Token

User = get_user_model()


class MopidyRequest:
    name = "Same in the End"
    artist = "Sublime"
    album = "Sublime"
    track_number = 4
    run_time_ticks = 156604
    run_time = 60
    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"  # noqa
    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():
    return MopidyRequest()


@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():
    mopidy_uri = "local:podcast:Up%20First/2022-01-01%20Up%20First.mp3"
    return MopidyRequest(mopidy_uri=mopidy_uri)


class JellyfinTrackRequest:
    name = "Emotion"
    artist = "Carly Rae Jepsen"
    album = "Emotion"
    track_number = 1
    item_type = "Audio"
    timestamp = "2024-01-14 12:00:19"
    run_time_ticks = 156604
    run_time = "00:00:60"
    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"
    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),
            "TrackNumber": int(kwargs.get("track_number", self.track_number)),
            "RunTime": kwargs.get("run_time", self.run_time),
            "ItemType": kwargs.get("item_type", self.item_type),
            "UtcTimestamp": kwargs.get("timestamp", self.timestamp),
            "PlaybackPositionTicks": int(
                kwargs.get("playback_time_ticks", self.playback_time_ticks)
            ),
            "Provider_musicbrainztrack": kwargs.get(
                "musicbrainz_track_id", self.musicbrainz_track_id
            ),
            "Provider_musicbrainzalbum": kwargs.get(
                "musicbrainz_album_id", self.musicbrainz_album_id
            ),
            "Provider_musicbrainzartist": kwargs.get(
                "musicbrainz_artist_id", self.musicbrainz_artist_id
            ),
            "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 jellyfin_track():
    return JellyfinTrackRequest()