mirror of
https://github.com/spl0k/supysonic.git
synced 2024-12-23 01:16:18 +00:00
151 lines
5.1 KiB
Python
151 lines
5.1 KiB
Python
# This file is part of Supysonic.
|
|
# Supysonic is a Python implementation of the Subsonic server API.
|
|
#
|
|
# Copyright (C) 2017-2022 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 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(f"folder add {name} {shlex.quote(path)}", expect_fail)
|
|
|
|
def test_folder_add(self):
|
|
with tempfile.TemporaryDirectory() as d:
|
|
self.__add_folder("tmpfolder", d)
|
|
|
|
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)
|
|
|
|
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")
|
|
|
|
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)
|
|
|
|
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)
|
|
|
|
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)
|
|
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")
|
|
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")
|
|
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")
|
|
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)
|
|
self.assertEqual(User.select().first().name, "alice")
|
|
|
|
self.__invoke("user rename alice bob")
|
|
self.assertEqual(User.select().first().name, "bob")
|
|
|
|
self.__invoke("user add -p Ch4rl3s charles")
|
|
self.__invoke("user rename bob charles", True)
|
|
self.assertEqual(User.select().where(User.name == "bob").count(), 1)
|
|
self.assertEqual(User.select().where(User.name == "charles").count(), 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|