Add more testing for baseui

pwman/tests/test_base_ui.py

@@ -101,13 +101,27 @@ class TestBaseUI(unittest.TestCase):
         ce = CryptoEngine.get()
         self.assertIsNone(ce._cipher)
 
-    def test_6_do_print(self):
+    def test_5_do_print(self):
         v = StringIO()
         sys.stdout = v
         self.tester.cli.do_print('1')
         self.assertIn('\x1b[31mUsername:\x1b[0m alice', v.getvalue())
+        self.tester.cli.do_print('a')
+        self.assertIn("print accepts only a single ID ...", v.getvalue())
 
-    def test_get_ids(self):
+        sys.stdout = sys.__stdout__
+
+    def test_6_do_tags(self):
+        v = StringIO()
+        sys.stdout = v
+        self.tester.cli.do_tags('1')
+        v = v.getvalue()
+        for t in ['foo', 'bar', 'baz']:
+            t in v
+        sys.stdout = sys.__stdout__
+        print(v)
+
+    def test_7_get_ids(self):
         # used by do_cp or do_open,
         # this spits many time could not understand your input
         self.assertEqual([1], self.tester.cli._get_ids('1'))
@@ -117,7 +131,7 @@ class TestBaseUI(unittest.TestCase):
         self.assertListEqual([], self.tester.cli._get_ids('5x'))
         self.assertListEqual([], self.tester.cli._get_ids('5\\'))
 
-    def test_5_do_delete(self):
+    def test_8_do_delete(self):
         self.assertIsNone(self.tester.cli._do_rm('x'))
         sys.stdin = StringIO("y\n")
         self.tester.cli.do_rm('1')