supysonic/tests/frontend/test_playlist.py

#!/usr/bin/env python
# coding: utf-8
#
# This file is part of Supysonic.
# Supysonic is a Python implementation of the Subsonic server API.
#
# Copyright (C) 2017-2018 Alban 'spl0k' Féron
#
# Distributed under terms of the GNU AGPLv3 license.

import uuid

from pony.orm import db_session

from supysonic.db import Folder, Artist, Album, Track, Playlist, User

from .frontendtestbase import FrontendTestBase

class PlaylistTestCase(FrontendTestBase):
    def setUp(self):
        super(PlaylistTestCase, self).setUp()

        with db_session:
            folder = Folder(name = 'Root', path = 'tests/assets', root = True)
            artist = Artist(name = 'Artist!')
            album = Album(name = 'Album!', artist = artist)

            track = Track(
                path = 'tests/assets/23bytes',
                title = '23bytes',
                artist = artist,
                album = album,
                folder = folder,
                root_folder = folder,
                duration = 2,
                disc = 1,
                number = 1,
                bitrate = 320,
                last_modification = 0
            )

            playlist = Playlist(
                name = 'Playlist!',
                user = User.get(name = 'alice')
            )
            for _ in range(4):
                playlist.add(track)

        self.playlistid = playlist.id

    def test_index(self):
        self._login('alice', 'Alic3')
        rv = self.client.get('/playlist')
        self.assertIn('My playlists', rv.data)

    def test_details(self):
        self._login('alice', 'Alic3')
        rv = self.client.get('/playlist/string', follow_redirects = True)
        self.assertIn('Invalid', rv.data)
        rv = self.client.get('/playlist/' + str(uuid.uuid4()), follow_redirects = True)
        self.assertIn('Unknown', rv.data)
        rv = self.client.get('/playlist/' + str(self.playlistid))
        self.assertIn('Playlist!', rv.data)
        self.assertIn('23bytes', rv.data)
        self.assertIn('Artist!', rv.data)
        self.assertIn('Album!', rv.data)

    def test_update(self):
        self._login('bob', 'B0b')
        rv = self.client.post('/playlist/string', follow_redirects = True)
        self.assertIn('Invalid', rv.data)
        rv = self.client.post('/playlist/' + str(uuid.uuid4()), follow_redirects = True)
        self.assertIn('Unknown', rv.data)
        rv = self.client.post('/playlist/' + str(self.playlistid), follow_redirects = True)
        self.assertNotIn('updated', rv.data)
        self.assertIn('not allowed', rv.data)
        self._logout()

        self._login('alice', 'Alic3')
        rv = self.client.post('/playlist/' + str(self.playlistid), follow_redirects = True)
        self.assertNotIn('updated', rv.data)
        self.assertIn('Missing', rv.data)
        with db_session:
            self.assertEqual(Playlist[self.playlistid].name, 'Playlist!')

        rv = self.client.post('/playlist/' + str(self.playlistid), data = { 'name': 'abc', 'public': True }, follow_redirects = True)
        self.assertIn('updated', rv.data)
        self.assertNotIn('not allowed', rv.data)
        with db_session:
            playlist = Playlist[self.playlistid]
            self.assertEqual(playlist.name, 'abc')
            self.assertTrue(playlist.public)

    def test_delete(self):
        self._login('bob', 'B0b')
        rv = self.client.get('/playlist/del/string', follow_redirects = True)
        self.assertIn('Invalid', rv.data)
        rv = self.client.get('/playlist/del/' + str(uuid.uuid4()), follow_redirects = True)
        self.assertIn('Unknown', rv.data)
        rv = self.client.get('/playlist/del/' + str(self.playlistid), follow_redirects = True)
        self.assertIn('not allowed', rv.data)
        with db_session:
            self.assertEqual(Playlist.select().count(), 1)
        self._logout()

        self._login('alice', 'Alic3')
        rv = self.client.get('/playlist/del/' + str(self.playlistid), follow_redirects = True)
        self.assertIn('deleted', rv.data)
        with db_session:
            self.assertEqual(Playlist.select().count(), 0)

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