mirror of
https://github.com/spl0k/supysonic.git
synced 2024-09-20 03:11:04 +00:00
142 lines
4.4 KiB
Python
142 lines
4.4 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8
|
|
#
|
|
# This file is part of Supysonic.
|
|
# Supysonic is a Python implementation of the Subsonic server API.
|
|
#
|
|
# Copyright (C) 2017-2018 Alban 'spl0k' Féron
|
|
#
|
|
# Distributed under terms of the GNU AGPLv3 license.
|
|
|
|
import io
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
from contextlib import contextmanager
|
|
from pony.orm import db_session
|
|
|
|
try: # Don't use io.StringIO on py2, it only accepts unicode and the CLI spits strs
|
|
from StringIO import StringIO
|
|
except ImportError:
|
|
from io import StringIO
|
|
|
|
from supysonic.db import Folder, User, init_database, release_database
|
|
from supysonic.cli import SupysonicCLI
|
|
|
|
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):
|
|
conf = TestConfig(False, False)
|
|
self.__dbfile = tempfile.mkstemp()[1]
|
|
conf.BASE["database_uri"] = "sqlite:///" + self.__dbfile
|
|
init_database(conf.BASE["database_uri"])
|
|
|
|
self.__stdout = StringIO()
|
|
self.__stderr = StringIO()
|
|
self.__cli = SupysonicCLI(conf, stdout=self.__stdout, stderr=self.__stderr)
|
|
|
|
def tearDown(self):
|
|
self.__stdout.close()
|
|
self.__stderr.close()
|
|
release_database()
|
|
os.unlink(self.__dbfile)
|
|
|
|
@contextmanager
|
|
def _tempdir(self):
|
|
d = tempfile.mkdtemp()
|
|
try:
|
|
yield d
|
|
finally:
|
|
shutil.rmtree(d)
|
|
|
|
def test_folder_add(self):
|
|
with self._tempdir() as d:
|
|
self.__cli.onecmd("folder add tmpfolder " + d)
|
|
|
|
with db_session:
|
|
f = Folder.select().first()
|
|
self.assertIsNotNone(f)
|
|
self.assertEqual(f.path, d)
|
|
|
|
def test_folder_add_errors(self):
|
|
with self._tempdir() as d:
|
|
self.__cli.onecmd("folder add f1 " + d)
|
|
self.__cli.onecmd("folder add f2 " + d)
|
|
with self._tempdir() as d:
|
|
self.__cli.onecmd("folder add f1 " + d)
|
|
self.__cli.onecmd("folder add f3 /invalid/path")
|
|
|
|
with db_session:
|
|
self.assertEqual(Folder.select().count(), 1)
|
|
|
|
def test_folder_delete(self):
|
|
with self._tempdir() as d:
|
|
self.__cli.onecmd("folder add tmpfolder " + d)
|
|
self.__cli.onecmd("folder delete randomfolder")
|
|
self.__cli.onecmd("folder delete tmpfolder")
|
|
|
|
with db_session:
|
|
self.assertEqual(Folder.select().count(), 0)
|
|
|
|
def test_folder_list(self):
|
|
with self._tempdir() as d:
|
|
self.__cli.onecmd("folder add tmpfolder " + d)
|
|
self.__cli.onecmd("folder list")
|
|
self.assertIn("tmpfolder", self.__stdout.getvalue())
|
|
self.assertIn(d, self.__stdout.getvalue())
|
|
|
|
def test_folder_scan(self):
|
|
with self._tempdir() as d:
|
|
self.__cli.onecmd("folder add tmpfolder " + d)
|
|
with tempfile.NamedTemporaryFile(dir=d):
|
|
self.__cli.onecmd("folder scan")
|
|
self.__cli.onecmd("folder scan tmpfolder nonexistent")
|
|
|
|
def test_user_add(self):
|
|
self.__cli.onecmd("user add -p Alic3 alice")
|
|
self.__cli.onecmd("user add -p alice alice")
|
|
|
|
with db_session:
|
|
self.assertEqual(User.select().count(), 1)
|
|
|
|
def test_user_delete(self):
|
|
self.__cli.onecmd("user add -p Alic3 alice")
|
|
self.__cli.onecmd("user delete alice")
|
|
self.__cli.onecmd("user delete bob")
|
|
|
|
with db_session:
|
|
self.assertEqual(User.select().count(), 0)
|
|
|
|
def test_user_list(self):
|
|
self.__cli.onecmd("user add -p Alic3 alice")
|
|
self.__cli.onecmd("user list")
|
|
self.assertIn("alice", self.__stdout.getvalue())
|
|
|
|
def test_user_setadmin(self):
|
|
self.__cli.onecmd("user add -p Alic3 alice")
|
|
self.__cli.onecmd("user setadmin alice")
|
|
self.__cli.onecmd("user setadmin bob")
|
|
with db_session:
|
|
self.assertTrue(User.get(name="alice").admin)
|
|
|
|
def test_user_changepass(self):
|
|
self.__cli.onecmd("user add -p Alic3 alice")
|
|
self.__cli.onecmd("user changepass alice newpass")
|
|
self.__cli.onecmd("user changepass bob B0b")
|
|
|
|
def test_other(self):
|
|
self.assertTrue(self.__cli.do_EOF(""))
|
|
self.__cli.onecmd("unknown command")
|
|
self.__cli.postloop()
|
|
self.__cli.completedefault("user", "user", 4, 4)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|