Răsfoiți Sursa

fix class names in unittests

oz123 11 ani în urmă
părinte
comite
efdc83948b
2 a modificat fișierele cu 3 adăugiri și 5 ștergeri
  1. 2 3
      pwman/tests/db_tests.py
  2. 1 2
      pwman/tests/test_pwman.py

+ 2 - 3
pwman/tests/db_tests.py

@@ -15,7 +15,7 @@ elif 'win' in sys.platform:
     from pwman.ui.win import PwmanCliWinNew as PwmanCliNew
     OSX = False
 else:
-    from pwman.ui.cli import PwmanCli
+    from pwman.ui.cli import PwmanCliOld
     from pwman.ui.cli import PwmanCliNew
     OSX = False
 
@@ -76,8 +76,7 @@ class DBTests(unittest.TestCase):
         new_node = self.db.getnodes([idx_created])[0]
         for key, attr in {'password': password, 'username': username,
                           'url': url, 'notes': notes}.iteritems():
-            self.assertEquals(attr, eval('new_node.'+key))
-
+            self.assertEquals(attr, eval('new_node.' + key))
         self.db.close()
 
 

+ 1 - 2
pwman/tests/test_pwman.py

@@ -27,6 +27,7 @@ sys.path.insert(0, os.getcwd())
 _saveconfig = True
 
 import sys
+
 from pwman.util.crypto import CryptoEngine
 
 if 'darwin' in sys.platform:
@@ -38,14 +39,12 @@ elif 'win' in sys.platform:
     from pwman.ui.win import PwmanCliWinNew as PwmanCliNew
     OSX = False
 else:
-    from pwman.ui.cli import PwmanCli
     from pwman.ui.cli import PwmanCliNew
     OSX = False
 
 
 import pwman.util.config as config
 import pwman.data.factory
-from pwman.data.convertdb import PwmanConvertDB
 
 
 def which(cmd):