mirror of
https://github.com/spl0k/supysonic.git
synced 2024-11-10 04:02:17 +00:00
141 lines
4.4 KiB
Python
141 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()
|
|
|