oz123 10 лет назад
Родитель
Сommit
084e803d46
2 измененных файлов с 12 добавлено и 3 удалено
  1. 9 0
      pwman/tests/test_base_ui.py
  2. 3 3
      pwman/ui/baseui.py

+ 9 - 0
pwman/tests/test_base_ui.py

@@ -92,6 +92,15 @@ class TestBaseUI(unittest.TestCase):
             l = f.readlines()
 
         self.assertIn('alice;example.com;secret;some notes;foo,bar,baz', l[1])
+        self.tester.cli.do_export("f")
+        with open('pwman-export.csv') as f:
+            l = f.readlines()
+        self.assertIn('alice;example.com;secret;some notes;foo,bar,baz', l[1])
+
+    def test_4_do_forget(self):
+        self.tester.cli.do_forget('')
+        ce = CryptoEngine.get()
+        self.assertIsNone(ce._cipher)
 
 if __name__ == '__main__':
 

+ 3 - 3
pwman/ui/baseui.py

@@ -40,7 +40,7 @@ class BaseCommands(HelpUIMixin, AliasesMixin):
         if self.hasxsel:
             return True
 
-    def error(self, exception):
+    def error(self, exception):  # pragma: no cover
         if (isinstance(exception, KeyboardInterrupt)):
             print('')
         else:
@@ -129,7 +129,7 @@ class BaseCommands(HelpUIMixin, AliasesMixin):
         enc = CryptoEngine.get()
         enc.forget()
 
-    def do_passwd(self, args):
+    def do_passwd(self, args):  # pragma: no cover
         """change the master password of the database"""
         pass
 
@@ -161,7 +161,7 @@ class BaseCommands(HelpUIMixin, AliasesMixin):
         self.do_cls('')
         if sys.platform != 'win32':
             rows, cols = tools.gettermsize()
-        else:
+        else:  # pragma: no cover
             rows, cols = 18, 80  # fix this !
 
         cols -= 8