From 332fd0446f00c68ddf8dbce1038c4f6f553e746d Mon Sep 17 00:00:00 2001
From: spl0k <alban.feron@gmail.com>
Date: Sun, 10 Dec 2017 17:53:02 +0100
Subject: [PATCH] CLI basic tests

---
 tests/base/__init__.py |   2 +
 tests/base/test_cli.py | 132 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 134 insertions(+)
 create mode 100644 tests/base/test_cli.py

diff --git a/tests/base/__init__.py b/tests/base/__init__.py
index 261ddd3..f8734bc 100644
--- a/tests/base/__init__.py
+++ b/tests/base/__init__.py
@@ -13,6 +13,7 @@ import unittest
 from .test_db import DbTestCase
 from .test_scanner import ScannerTestCase
 from .test_watcher import suite as watcher_suite
+from .test_cli import CLITestCase
 
 def suite():
     suite = unittest.TestSuite()
@@ -20,6 +21,7 @@ def suite():
     suite.addTest(unittest.makeSuite(DbTestCase))
     suite.addTest(unittest.makeSuite(ScannerTestCase))
     suite.addTest(watcher_suite())
+    suite.addTest(unittest.makeSuite(CLITestCase))
 
     return suite
 
diff --git a/tests/base/test_cli.py b/tests/base/test_cli.py
new file mode 100644
index 0000000..ea49e7e
--- /dev/null
+++ b/tests/base/test_cli.py
@@ -0,0 +1,132 @@
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+#
+# This file is part of Supysonic.
+# Supysonic is a Python implementation of the Subsonic server API.
+#
+# Copyright (C) 2017 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 StringIO import StringIO
+
+from supysonic.db import Folder, User, get_store
+from supysonic.cli import SupysonicCLI
+
+from ..testbase import TestConfig
+
+class CLITestCase(unittest.TestCase):
+    """ Really basic tests. Some even don't check anything but are juste there for coverage """
+
+    def setUp(self):
+        conf = TestConfig(False, False)
+        self.__dbfile = tempfile.mkstemp()[1]
+        conf.BASE['database_uri'] = 'sqlite:///' + self.__dbfile
+        self.__store = get_store(conf.BASE['database_uri'])
+
+        with io.open('schema/sqlite.sql', 'r') as sql:
+            schema = sql.read()
+            for statement in schema.split(';'):
+                self.__store.execute(statement)
+        self.__store.commit()
+
+        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()
+        self.__store.close()
+        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)
+
+        f = self.__store.find(Folder).one()
+        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')
+
+        self.assertEqual(self.__store.find(Folder).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')
+        self.assertEqual(self.__store.find(Folder).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')
+        self.assertEqual(self.__store.find(User).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')
+        self.assertEqual(self.__store.find(User).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')
+        self.assertTrue(self.__store.find(User, User.name == 'alice').one().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()
+