supysonic/tests/issue129.py

# This file is part of Supysonic.
# Supysonic is a Python implementation of the Subsonic server API.
#
# Copyright (C) 2018-2022 Alban 'spl0k' Féron
#
# Distributed under terms of the GNU AGPLv3 license.

import os.path
import unittest

from supysonic.db import User, Track, StarredTrack, RatingTrack
from supysonic.managers.folder import FolderManager
from supysonic.scanner import Scanner

from .testbase import TestBase


class Issue129TestCase(TestBase):
    def setUp(self):
        super().setUp()

        FolderManager.add("folder", os.path.abspath("tests/assets/folder"))
        scanner = Scanner()
        scanner.queue_folder("folder")
        scanner.run()

        self.trackid = Track.select().first().id
        self.userid = User.get(name="alice").id

    def test_last_play(self):
        user = User[self.userid]
        user.last_play = Track[self.trackid]
        user.save()
        FolderManager.delete_by_name("folder")

    def test_starred(self):
        StarredTrack.create(user=self.userid, starred=self.trackid)
        FolderManager.delete_by_name("folder")

    def test_rating(self):
        RatingTrack.create(user=self.userid, rated=self.trackid, rating=5)
        FolderManager.delete_by_name("folder")


if __name__ == "__main__":
    unittest.main()