Exemple #1
0
 def test_22_stdout_output(self):
     """Use stdout output decorator with default options."""
     wrapper_f = decor.stdout_output()(_func11)
     stdout = log.capture_output(wrapper_f)
     self.assertEqual(stdout, '123\n')
     stdout = log.capture_output(wrapper_f,
                                 args=(10, '20'),
                                 kwargs={'a': 30})
     self.assertEqual(stdout, '123\n')
Exemple #2
0
 def test_20_stdout_input(self):
     """Use stdout input decorator with default options."""
     wrapper_f = decor.stdout_input()(_func10)
     stdout = log.capture_output(wrapper_f)
     # \n is added by print statement.
     self.assertEqual(stdout, '_func10()\n')
     stdout = log.capture_output(wrapper_f,
                                 args=(10, '20'),
                                 kwargs={'a': 30})
     # \n is added by print statement.
     self.assertEqual(stdout, "_func10(10, '20', a=30)\n")
Exemple #3
0
 def test_06_time_it(self):
     """Run decorator and capture its print output."""
     d = decor.time_it(uom='h')(_dummy)
     res = log.capture_output(d)
     self.assertEqual(res, "'_dummy' took 0.00 hour(s)\n")
     d = decor.time_it(
         uom='h',
         msg_fmt="'{_f_name}' took {_dur:.2f} {_uom_name} {arg1} {c}")(
             _dummy2)
     res = log.capture_output(d, args=(0, 1), kwargs={'c': 2})
     self.assertEqual(res, "'_dummy2' took 0.00 hour(s) 1 2\n")
Exemple #4
0
 def test_23_stdout_output(self):
     """Use stdout output decorator with custom options."""
     wrapper_f = decor.stdout_output(options={
         'stdout_writer': print,
         'prefix': 'custom ',
     })(_func11)
     stdout = log.capture_output(wrapper_f)
     self.assertEqual(stdout, 'custom 123\n')
     wrapper_f = decor.stdout_output(
         options={
             'stdout_getter': lambda *args, **kwargs: print,
         })(_func11)
     stdout = log.capture_output(wrapper_f,
                                 args=(10, '20'),
                                 kwargs={'a': 30})
     self.assertEqual(stdout, "123\n")
Exemple #5
0
 def test_02_command(self):
     """Execute commands in lifo order."""
     invoker = self.DequeInvoker()
     invoker.add_command(self.mc1)
     invoker.add_command(self.mc2)
     invoker.add_command(self.mc3)
     res = log.capture_output(invoker.run, kwargs={'priority': 'lifo'})
     self.assertEqual(res, 'cba')
Exemple #6
0
 def test_01_command(self):
     """Execute commands in fifo order."""
     invoker = self.DequeInvoker()
     invoker.add_command(self.mc1)
     invoker.add_command(self.mc2)
     invoker.add_command(self.mc3)
     res = log.capture_output(invoker.run)
     self.assertEqual(res, 'abc')
Exemple #7
0
 def test_21_stdout_input(self):
     """Use stdout input decorator with custom options."""
     wrapper_f = decor.stdout_input(options={
         'stdout_writer': print,
         'no_first_arg': True,
         'prefix': 'custom ',
     })(_func10)
     stdout = log.capture_output(wrapper_f)
     self.assertEqual(stdout, 'custom _func10()\n')
     wrapper_f = decor.stdout_input(
         options={
             'stdout_getter': lambda *args, **kwargs: print,
             'command': True
         })(_func10)
     stdout = log.capture_output(wrapper_f,
                                 args=(10, '20'),
                                 kwargs={'a': 30})
     self.assertEqual(stdout, "_func10 10 20 a=30\n")
Exemple #8
0
 def test_capture_output_1(self):
     """Capture output from print."""
     res = log.capture_output(print, args=('test', ))
     self.assertEqual(res, 'test\n')