Move factory related test to own module

pwman/tests/test_factory.py

@@ -0,0 +1,101 @@
+# ============================================================================
+# This file is part of Pwman3.
+#
+# Pwman3 is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2
+# as published by the Free Software Foundation;
+#
+# Pwman3 is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Pwman3; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+# ============================================================================
+# Copyright (C) 2013-2014 Oz Nahum Tiram <nahumoz@gmail.com>
+# ============================================================================
+
+from pwman.data.nodes import NewNode
+from pwman.data.tags import TagNew
+from pwman.data import factory
+from pwman.data.drivers.sqlite import DatabaseException, SQLiteDatabaseNewForm
+from pwman.ui import get_ui_platform
+from pwman.data.database import __DB_FORMAT__
+import sys
+import unittest
+import os
+import os.path
+
+dummyfile = """
+[Encryption]
+
+[Readline]
+
+[Global]
+xsel = /usr/bin/xsel
+colors = yes
+umask = 0100
+cls_timeout = 5
+
+[Database]
+"""
+
+
+def node_factory(username, password, url, notes, tags=None):
+    node = NewNode()
+    node.username = username
+    node.password = password
+    node.url = url
+    node.notes = notes
+    tags = [TagNew(tn) for tn in tags]
+    node.tags = tags
+
+    return node
+
+_saveconfig = False
+
+PwmanCliNew, OSX = get_ui_platform(sys.platform)
+
+
+from .test_tools import (SetupTester)
+
+testdb = os.path.join(os.path.dirname(__file__), "test.pwman.db")
+
+
+class TestFactory(unittest.TestCase):
+
+    def setUp(self):
+        "test that the right db instance was created"
+        self.dbtype = 'SQLite'
+        self.db = factory.create(self.dbtype, __DB_FORMAT__, testdb)
+        self.tester = SetupTester(__DB_FORMAT__, testdb)
+        self.tester.create()
+
+    def test_factory_check_db_ver(self):
+        self.assertEqual(factory.check_db_version('SQLite', testdb), 0.6)
+
+    def test_factory_check_db_file(self):
+        factory.create('SQLite', version='0.3', filename='baz.db')
+        self.assertEqual(factory.check_db_version('SQLite', 'baz.db'), 0.3)
+        os.unlink('baz.db')
+
+    def test_factory_create(self):
+        db = factory.create('SQLite', filename='foo.db')
+        db._open()
+        self.assertTrue(os.path.exists('foo.db'))
+        db.close()
+        os.unlink('foo.db')
+        self.assertIsInstance(db, SQLiteDatabaseNewForm)
+        self.assertRaises(DatabaseException, factory.create, 'UNKNOWN')
+
+
+if __name__ == '__main__':
+    # make sure we use local pwman
+    sys.path.insert(0, os.getcwd())
+    # check if old DB exists, if so remove it.
+    # excuted only once when invoked upon import or
+    # upon run
+    SetupTester().clean()
+    unittest.main(verbosity=1, failfast=True)

pwman/tests/test_pwman.py

@@ -21,16 +21,19 @@
 import os
 import sys
 import unittest
-from .db_tests import (DBTests, SetupTester, CLITests,
+from .db_tests import (SetupTester)
+                       #, CLITests,
+                       #DBTests,
                        # ConfigTest,
                        # TestDBFalseConfig,
-                       FactoryTest)
+                       #FactoryTest)
 
 #from .crypto_tests import CryptoTest
 from .test_crypto_engine import CryptoEngineTest
 from .test_config import TestConfig
 from .test_sqlite import TestSQLite
 from .test_importer import TestImporter
+from .test_factory import TestFactory
 
 if 'win' not in sys.platform:
     from .test_complete_ui import (Ferrum, NEW_DB_PATH)
@@ -49,16 +52,17 @@ SetupTester().clean()
 def suite():
     loader = unittest.TestLoader()
     suite = unittest.TestSuite()
-    suite.addTest(loader.loadTestsFromTestCase(DBTests))
+    #suite.addTest(loader.loadTestsFromTestCase(DBTests))
     #suite.addTest(loader.loadTestsFromTestCase(CryptoTest))
-    suite.addTest(loader.loadTestsFromTestCase(CLITests))
+    #suite.addTest(loader.loadTestsFromTestCase(CLITests))
     #suite.addTest(loader.loadTestsFromTestCase(ConfigTest))
-    suite.addTest(loader.loadTestsFromTestCase(FactoryTest))
+    #suite.addTest(loader.loadTestsFromTestCase(FactoryTest))
     #suite.addTest(loader.loadTestsFromTestCase(TestDBFalseConfig))
     suite.addTest(loader.loadTestsFromTestCase(CryptoEngineTest))
     suite.addTest(loader.loadTestsFromTestCase(TestConfig))
     suite.addTest(loader.loadTestsFromTestCase(TestSQLite))
     suite.addTest(loader.loadTestsFromTestCase(TestImporter))
+    suite.addTest(loader.loadTestsFromTestCase(TestFactory))
     #if 'win' not in sys.platform:
     #    suite.addTest(loader.loadTestsFromTestCase(Ferrum))
     return suite