Changeset - b639ce7e4309
[Not reviewed]
0 1 0
Lance Edgar (lance) - 3 years ago 2021-12-06 21:15:45
lance@edbob.org
Fix some tests
1 file changed with 2 insertions and 2 deletions:
0 comments (0 inline, 0 general)
rattail/tests/commands/test_core.py
Show inline comments
 
@@ -69,49 +69,49 @@ class TestCommand(TestCase):
 
        command = core.Command()
 
        stdout = StringIO()
 
        command.stdout = stdout
 
        command.print_help()
 
        output = stdout.getvalue()
 
        stdout.close()
 
        self.assertTrue('Usage:' in output)
 
        self.assertTrue('Options:' in output)
 

	
 
    def test_run_with_no_args_prints_help(self):
 
        command = core.Command()
 
        with patch.object(command, 'print_help') as print_help:
 
            command.run()
 
            self.assertRaises(SystemExit, command.run)
 
            print_help.assert_called_once_with()
 

	
 
    def test_run_with_single_help_arg_prints_help(self):
 
        command = core.Command()
 
        with patch.object(command, 'print_help') as print_help:
 
            command.run('help')
 
            print_help.assert_called_once_with()
 

	
 
    def test_run_with_help_and_unknown_subcommand_args_prints_help(self):
 
        command = core.Command()
 
        with patch.object(command, 'print_help') as print_help:
 
            command.run('help', 'invalid-subcommand-name')
 
            print_help.assert_called_once_with()
 

	
 
    def test_run_with_help_and_subcommand_args_prints_subcommand_help(self):
 
        command = core.Command()
 
        fake = command.subcommands['fake'] = Mock()
 
        command.run('help', 'fake')
 
        fake.return_value.parser.print_help.assert_called_once_with()
 

	
 
    def test_run_with_unknown_subcommand_arg_prints_help(self):
 
        command = core.Command()
 
        with patch.object(command, 'print_help') as print_help:
 
            command.run('invalid-command-name')
 
            self.assertRaises(SystemExit, command.run, 'invalid-command-name')
 
            print_help.assert_called_once_with()
 

	
 
    def test_stdout_may_be_redirected(self):
 
        class Fake(core.Subcommand):
 
            def run(self, args):
 
                self.stdout.write("standard output stuff")
 
                self.stdout.flush()
 
        command = core.Command()
 
        fake = command.subcommands['fake'] = Fake
 
        tempdir = tempfile.mkdtemp()
 
        config_path = os.path.join(tempdir, 'test.ini')
 
        with open(config_path, 'wt') as f:
0 comments (0 inline, 0 general)