1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
"""Tests for the 'web' plugin"""
from _common import unittest
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')
|
