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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
# This file is part of Supysonic.
# Supysonic is a Python implementation of the Subsonic server API.
#
# Copyright (C) 2017-2021 Alban 'spl0k' Féron
#
# Distributed under terms of the GNU AGPLv3 license.
import os
import tempfile
import shlex
import unittest
from click.testing import CliRunner
from pony.orm import db_session
from supysonic.db import Folder, User, init_database, release_database
from supysonic.cli import cli
from ..testbase import TestConfig
class CLITestCase(unittest.TestCase):
"""Really basic tests. Some even don't check anything but are just there for coverage"""
def setUp(self):
self.__conf = TestConfig(False, False)
self.__db = tempfile.mkstemp()
self.__conf.BASE["database_uri"] = "sqlite:///" + self.__db[1]
init_database(self.__conf.BASE["database_uri"])
self.__runner = CliRunner()
def tearDown(self):
release_database()
os.close(self.__db[0])
os.remove(self.__db[1])
def __invoke(self, cmd, expect_fail=False):
rv = self.__runner.invoke(cli, shlex.split(cmd), obj=self.__conf)
func = self.assertNotEqual if expect_fail else self.assertEqual
func(rv.exit_code, 0)
return rv
def __add_folder(self, name, path, expect_fail=False):
self.__invoke("folder add {} {}".format(name, shlex.quote(path)), expect_fail)
def test_folder_add(self):
with tempfile.TemporaryDirectory() as d:
self.__add_folder("tmpfolder", d)
with db_session:
f = Folder.select().first()
self.assertIsNotNone(f)
self.assertEqual(f.path, d)
def test_folder_add_errors(self):
with tempfile.TemporaryDirectory() as d:
self.__add_folder("f1", d)
self.__add_folder("f2", d, True)
with tempfile.TemporaryDirectory() as d:
self.__add_folder("f1", d, True)
self.__invoke("folder add f3 /invalid/path", True)
with db_session:
self.assertEqual(Folder.select().count(), 1)
def test_folder_delete(self):
with tempfile.TemporaryDirectory() as d:
self.__add_folder("tmpfolder", d)
self.__invoke("folder delete randomfolder", True)
self.__invoke("folder delete tmpfolder")
with db_session:
self.assertEqual(Folder.select().count(), 0)
def test_folder_list(self):
with tempfile.TemporaryDirectory() as d:
self.__add_folder("tmpfolder", d)
rv = self.__invoke("folder list")
self.assertIn("tmpfolder", rv.output)
self.assertIn(d, rv.output)
def test_folder_scan(self):
with tempfile.TemporaryDirectory() as d:
self.__add_folder("tmpfolder", d)
with tempfile.NamedTemporaryFile(dir=d):
self.__invoke("folder scan")
self.__invoke("folder scan tmpfolder nonexistent")
def test_user_add(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user add -p alice alice", True)
with db_session:
self.assertEqual(User.select().count(), 1)
def test_user_delete(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user delete alice")
self.__invoke("user delete bob", True)
with db_session:
self.assertEqual(User.select().count(), 0)
def test_user_list(self):
self.__invoke("user add -p Alic3 alice")
rv = self.__invoke("user list")
self.assertIn("alice", rv.output)
def test_user_setadmin(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user setroles -A alice")
self.__invoke("user setroles -A bob", True)
with db_session:
self.assertTrue(User.get(name="alice").admin)
def test_user_unsetadmin(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user setroles -A alice")
self.__invoke("user setroles -a alice")
with db_session:
self.assertFalse(User.get(name="alice").admin)
def test_user_setjukebox(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user setroles -J alice")
with db_session:
self.assertTrue(User.get(name="alice").jukebox)
def test_user_unsetjukebox(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user setroles -J alice")
self.__invoke("user setroles -j alice")
with db_session:
self.assertFalse(User.get(name="alice").jukebox)
def test_user_changepass(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user changepass alice -p newpass")
self.__invoke("user changepass bob -p B0b", True)
def test_user_rename(self):
self.__invoke("user add -p Alic3 alice")
self.__invoke("user rename alice alice")
self.__invoke("user rename bob charles", True)
self.__invoke("user rename alice ''", True)
with db_session:
self.assertEqual(User.select().first().name, "alice")
self.__invoke("user rename alice bob")
with db_session:
self.assertEqual(User.select().first().name, "bob")
self.__invoke("user add -p Ch4rl3s charles")
self.__invoke("user rename bob charles", True)
with db_session:
self.assertEqual(User.select(lambda u: u.name == "bob").count(), 1)
self.assertEqual(User.select(lambda u: u.name == "charles").count(), 1)
if __name__ == "__main__":
unittest.main()
|
