Переглянути джерело

Add more testing for baseui

oz123 10 роки тому
батько
коміт
a10da5f8b0
2 змінених файлів з 12 додано та 3 видалено
  1. 11 2
      pwman/tests/test_base_ui.py
  2. 1 1
      pwman/ui/baseui.py

+ 11 - 2
pwman/tests/test_base_ui.py

@@ -78,12 +78,11 @@ class TestBaseUI(unittest.TestCase):
 
     def test_2_do_list(self):
         self.output = StringIO()
-        self.saved_stdout = sys.stdout
         sys.stdout = self.output
         self.tester.cli.do_list('')
         self.tester.cli.do_list('foo')
         self.tester.cli.do_list('bar')
-        sys.stdout = self.saved_stdout
+        sys.stdout = sys.__stdout__
         self.output.getvalue()
 
     def test_3_do_export(self):
@@ -102,6 +101,16 @@ class TestBaseUI(unittest.TestCase):
         ce = CryptoEngine.get()
         self.assertIsNone(ce._cipher)
 
+    def test_5_do_delete(self):
+        self.assertIsNone(self.tester.cli._do_rm('x'))
+        sys.stdin = StringIO("y\n")
+        self.tester.cli.do_rm('1')
+        sys.stdin = sys.__stdin__
+        sys.stdout = StringIO()
+        self.tester.cli.do_ls('')
+        self.assertNotIn('alice', sys.stdout.getvalue())
+        sys.stdout = sys.__stdout__
+
 if __name__ == '__main__':
 
     ce = CryptoEngine.get()

+ 1 - 1
pwman/ui/baseui.py

@@ -36,7 +36,7 @@ if sys.version_info.major > 2:  # pragma: no cover
 class BaseCommands(HelpUIMixin, AliasesMixin):
 
     @property
-    def _xsel(self):
+    def _xsel(self):  # pragma: no cover
         if self.hasxsel:
             return True