mirror of
https://github.com/spl0k/supysonic.git
synced 2024-11-10 04:02:17 +00:00
189 lines
8.3 KiB
Python
189 lines
8.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# vim:fenc=utf-8
|
|
#
|
|
# This file is part of Supysonic.
|
|
# Supysonic is a Python implementation of the Subsonic server API.
|
|
#
|
|
# Copyright (C) 2017 Alban 'spl0k' Féron
|
|
#
|
|
# Distributed under terms of the GNU AGPLv3 license.
|
|
|
|
from lxml import etree
|
|
import time
|
|
import uuid
|
|
|
|
from supysonic.db import Folder, Artist, Album, Track
|
|
|
|
from .apitestbase import ApiTestBase
|
|
|
|
class BrowseTestCase(ApiTestBase):
|
|
def setUp(self):
|
|
super(BrowseTestCase, self).setUp()
|
|
|
|
empty = Folder()
|
|
empty.root = True
|
|
empty.name = 'Empty root'
|
|
empty.path = '/tmp'
|
|
self.store.add(empty)
|
|
|
|
root = Folder()
|
|
root.root = True
|
|
root.name = 'Root folder'
|
|
root.path = 'tests/assets'
|
|
self.store.add(root)
|
|
|
|
for letter in 'ABC':
|
|
folder = Folder()
|
|
folder.name = letter + 'rtist'
|
|
folder.path = 'tests/assets/{}rtist'.format(letter)
|
|
folder.parent = root
|
|
|
|
artist = Artist()
|
|
artist.name = letter + 'rtist'
|
|
|
|
for lether in 'AB':
|
|
afolder = Folder()
|
|
afolder.name = letter + lether + 'lbum'
|
|
afolder.path = 'tests/assets/{0}rtist/{0}{1}lbum'.format(letter, lether)
|
|
afolder.parent = folder
|
|
|
|
album = Album()
|
|
album.name = letter + lether + 'lbum'
|
|
album.artist = artist
|
|
|
|
for num, song in enumerate([ 'One', 'Two', 'Three' ]):
|
|
track = Track()
|
|
track.disc = 1
|
|
track.number = num
|
|
track.title = song
|
|
track.duration = 2
|
|
track.album = album
|
|
track.artist = artist
|
|
track.bitrate = 320
|
|
track.path = 'tests/assets/{0}rtist/{0}{1}lbum/{2}'.format(letter, lether, song)
|
|
track.content_type = 'audio/mpeg'
|
|
track.last_modification = 0
|
|
track.root_folder = root
|
|
track.folder = afolder
|
|
self.store.add(track)
|
|
|
|
self.store.commit()
|
|
|
|
self.assertEqual(self.store.find(Folder).count(), 11)
|
|
self.assertEqual(self.store.find(Folder, Folder.root == True).count(), 2)
|
|
self.assertEqual(self.store.find(Artist).count(), 3)
|
|
self.assertEqual(self.store.find(Album).count(), 6)
|
|
self.assertEqual(self.store.find(Track).count(), 18)
|
|
|
|
def test_get_music_folders(self):
|
|
# Do not validate against the XSD here, this is the only place where the API should return ids as ints
|
|
# all our ids are uuids :/
|
|
rv = self.client.get('/rest/getMusicFolders.view', query_string = { 'u': 'alice', 'p': 'Alic3', 'c': 'tests' })
|
|
self.assertEqual(rv.status_code, 200)
|
|
xml = etree.fromstring(rv.data)
|
|
|
|
NS = '{http://subsonic.org/restapi}'
|
|
self.assertEqual(xml.tag, NS + 'subsonic-response')
|
|
self.assertEqual(xml.get('status'), 'ok')
|
|
self.assertEqual(len(xml), 1)
|
|
self.assertEqual(xml[0].tag, NS + 'musicFolders')
|
|
self.assertEqual(len(xml[0]), 2)
|
|
self.assertSequenceEqual(sorted(self._xpath(xml, './musicFolders/musicFolder/@name')), [ 'Empty root', 'Root folder' ])
|
|
|
|
def test_get_indexes(self):
|
|
self._make_request('getIndexes', { 'musicFolderId': 'abcdef' }, error = 0)
|
|
self._make_request('getIndexes', { 'musicFolderId': str(uuid.uuid4()) }, error = 70)
|
|
self._make_request('getIndexes', { 'ifModifiedSince': 'quoi' }, error = 0)
|
|
|
|
rv, child = self._make_request('getIndexes', { 'ifModifiedSince': int(time.time() * 1000 + 1000) }, tag = 'indexes')
|
|
self.assertEqual(len(child), 0)
|
|
|
|
fid = self.store.find(Folder, Folder.name == 'Empty root').one().id
|
|
rv, child = self._make_request('getIndexes', { 'musicFolderId': str(fid) }, tag = 'indexes')
|
|
self.assertEqual(len(child), 0)
|
|
|
|
rv, child = self._make_request('getIndexes', tag = 'indexes')
|
|
self.assertEqual(len(child), 3)
|
|
for i, letter in enumerate([ 'A', 'B', 'C' ]):
|
|
self.assertEqual(child[i].get('name'), letter)
|
|
self.assertEqual(len(child[i]), 1)
|
|
self.assertEqual(child[i][0].get('name'), letter + 'rtist')
|
|
|
|
def test_get_music_directory(self):
|
|
self._make_request('getMusicDirectory', error = 10)
|
|
self._make_request('getMusicDirectory', { 'id': 'id' }, error = 0)
|
|
self._make_request('getMusicDirectory', { 'id': str(uuid.uuid4()) }, error = 70)
|
|
|
|
# should test with folders with both children folders and tracks. this code would break in that case
|
|
for f in self.store.find(Folder):
|
|
rv, child = self._make_request('getMusicDirectory', { 'id': str(f.id) }, tag = 'directory')
|
|
self.assertEqual(child.get('id'), str(f.id))
|
|
self.assertEqual(child.get('name'), f.name)
|
|
self.assertEqual(len(child), f.children.count() + f.tracks.count())
|
|
for dbc, xmlc in zip(sorted(f.children, key = lambda c: c.name), sorted(child, key = lambda c: c.get('title'))):
|
|
self.assertEqual(dbc.name, xmlc.get('title'))
|
|
self.assertEqual(xmlc.get('artist'), f.name)
|
|
self.assertEqual(xmlc.get('parent'), str(f.id))
|
|
for t, xmlc in zip(sorted(f.tracks, key = lambda t: t.title), sorted(child, key = lambda c: c.get('title'))):
|
|
self.assertEqual(t.title, xmlc.get('title'))
|
|
self.assertEqual(xmlc.get('parent'), str(f.id))
|
|
|
|
def test_get_artists(self):
|
|
# same as getIndexes standard case
|
|
# dataset should be improved to have a different directory structure than /root/Artist/Album/Track
|
|
|
|
rv, child = self._make_request('getArtists', tag = 'artists')
|
|
self.assertEqual(len(child), 3)
|
|
for i, letter in enumerate([ 'A', 'B', 'C' ]):
|
|
self.assertEqual(child[i].get('name'), letter)
|
|
self.assertEqual(len(child[i]), 1)
|
|
self.assertEqual(child[i][0].get('name'), letter + 'rtist')
|
|
|
|
def test_get_artist(self):
|
|
# dataset should be improved to have tracks by a different artist than the album's artist
|
|
self._make_request('getArtist', error = 10)
|
|
self._make_request('getArtist', { 'id': 'artist' }, error = 0)
|
|
self._make_request('getArtist', { 'id': str(uuid.uuid4()) }, error = 70)
|
|
|
|
for ar in self.store.find(Artist):
|
|
rv, child = self._make_request('getArtist', { 'id': str(ar.id) }, tag = 'artist')
|
|
self.assertEqual(child.get('id'), str(ar.id))
|
|
self.assertEqual(child.get('albumCount'), str(len(child)))
|
|
self.assertEqual(len(child), ar.albums.count())
|
|
for dal, xal in zip(sorted(ar.albums, key = lambda a: a.name), sorted(child, key = lambda c: c.get('name'))):
|
|
self.assertEqual(dal.name, xal.get('name'))
|
|
self.assertEqual(xal.get('artist'), ar.name) # could break with a better dataset
|
|
self.assertEqual(xal.get('artistId'), str(ar.id)) # see above
|
|
|
|
def test_get_album(self):
|
|
self._make_request('getAlbum', error = 10)
|
|
self._make_request('getAlbum', { 'id': 'nastynasty' }, error = 0)
|
|
self._make_request('getAlbum', { 'id': str(uuid.uuid4()) }, error = 70)
|
|
|
|
a = self.store.find(Album)[0]
|
|
rv, child = self._make_request('getAlbum', { 'id': str(a.id) }, tag = 'album')
|
|
self.assertEqual(child.get('id'), str(a.id))
|
|
self.assertEqual(child.get('songCount'), str(len(child)))
|
|
|
|
self.assertEqual(len(child), a.tracks.count())
|
|
for dal, xal in zip(sorted(a.tracks, key = lambda t: t.title), sorted(child, key = lambda c: c.get('title'))):
|
|
self.assertEqual(dal.title, xal.get('title'))
|
|
self.assertEqual(xal.get('album'), a.name)
|
|
self.assertEqual(xal.get('albumId'), str(a.id))
|
|
|
|
def test_get_song(self):
|
|
self._make_request('getSong', error = 10)
|
|
self._make_request('getSong', { 'id': 'nastynasty' }, error = 0)
|
|
self._make_request('getSong', { 'id': str(uuid.uuid4()) }, error = 70)
|
|
|
|
s = self.store.find(Track)[0]
|
|
self._make_request('getSong', { 'id': str(s.id) }, tag = 'song')
|
|
|
|
def test_get_videos(self):
|
|
self._make_request('getVideos', error = 0)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|