mirror of
https://github.com/beetbox/beets.git
synced 2025-12-17 14:13:41 +01:00
90b388b775
Include import of __future__ features division, absolute_imports and print_function everywhere. Don't add unicode_literals yet for it is harder to convert. Goal is smoothing the transition to python 3.
110 lines
3.9 KiB
Python
110 lines
3.9 KiB
Python
"""Tests for the 'web' plugin"""
|
|
|
|
from __future__ import division, absolute_import, print_function
|
|
|
|
from test._common import unittest
|
|
from test import _common
|
|
import json
|
|
import beetsplug
|
|
from beets.library import Item, Album
|
|
beetsplug.__path__ = ['./beetsplug', '../beetsplug']
|
|
from beetsplug import web
|
|
|
|
|
|
class WebPluginTest(_common.LibTestCase):
|
|
|
|
def setUp(self):
|
|
super(WebPluginTest, self).setUp()
|
|
|
|
# Add fixtures
|
|
for track in self.lib.items():
|
|
track.remove()
|
|
self.lib.add(Item(title='title', path='', id=1))
|
|
self.lib.add(Item(title='another title', path='', id=2))
|
|
self.lib.add(Album(album='album', id=3))
|
|
self.lib.add(Album(album='another album', id=4))
|
|
|
|
web.app.config['TESTING'] = True
|
|
web.app.config['lib'] = self.lib
|
|
self.client = web.app.test_client()
|
|
|
|
def test_get_all_items(self):
|
|
response = self.client.get('/item/')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(len(response.json['items']), 2)
|
|
|
|
def test_get_single_item_by_id(self):
|
|
response = self.client.get('/item/1')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json['id'], 1)
|
|
self.assertEqual(response.json['title'], 'title')
|
|
|
|
def test_get_multiple_items_by_id(self):
|
|
response = self.client.get('/item/1,2')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(len(response.json['items']), 2)
|
|
response_titles = [item['title'] for item in response.json['items']]
|
|
self.assertItemsEqual(response_titles, ['title', 'another title'])
|
|
|
|
def test_get_single_item_not_found(self):
|
|
response = self.client.get('/item/3')
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_get_item_empty_query(self):
|
|
response = self.client.get('/item/query/')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(len(response.json['items']), 2)
|
|
|
|
def test_get_simple_item_query(self):
|
|
response = self.client.get('/item/query/another')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(len(response.json['results']), 1)
|
|
self.assertEqual(response.json['results'][0]['title'], 'another title')
|
|
|
|
def test_get_all_albums(self):
|
|
response = self.client.get('/album/')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
response_albums = [album['album'] for album in response.json['albums']]
|
|
self.assertItemsEqual(response_albums, ['album', 'another album'])
|
|
|
|
def test_get_single_album_by_id(self):
|
|
response = self.client.get('/album/2')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json['id'], 2)
|
|
self.assertEqual(response.json['album'], 'another album')
|
|
|
|
def test_get_multiple_albums_by_id(self):
|
|
response = self.client.get('/album/1,2')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
response_albums = [album['album'] for album in response.json['albums']]
|
|
self.assertItemsEqual(response_albums, ['album', 'another album'])
|
|
|
|
def test_get_album_empty_query(self):
|
|
response = self.client.get('/album/query/')
|
|
response.json = json.loads(response.data)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(len(response.json['albums']), 2)
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='suite')
|