aboutsummaryrefslogtreecommitdiff
path: root/tests/test_mixins.py
blob: 0133f43d67197b02cdc3a9557a6ae72513e73201 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import tempfile
import unittest

import hazwaz
import hazwaz.unittest


class testEditorMixin(unittest.TestCase):
    def test_open_with_cat_existing_file(self):
        subcmd = hazwaz.mixins.ExternalEditorMixin()
        subcmd.editors = [("cat", "cat")]
        # TODO: suppress this output in the tests (we can't use
        # contextlib.redirect_stdout because that doesn't redirect the
        # stdout used by subprocess.
        res = subcmd.edit_file_in_external_editor("/bin/fgrep")
        self.assertTrue(res)

    def test_open_with_cat_missing_file(self):
        subcmd = hazwaz.mixins.ExternalEditorMixin()
        subcmd.editors = [("cat", "cat")]
        # TODO: suppress this output in the tests (we can't use
        # contextlib.redirect_stderr because that doesn't redirect the
        # stderr used by subprocess.
        res = subcmd.edit_file_in_external_editor("no_such_file")
        self.assertFalse(res)

    def test_open_with_non_existing_editor(self):
        subcmd = hazwaz.mixins.ExternalEditorMixin()
        subcmd.editors = [("no_such_command", "no_such_command")]
        with self.assertLogs() as cm:
            subcmd.edit_file_in_external_editor("no_such_file")
        self.assertIn(
            "Could not open file no_such_file with no_such_command",
            cm.output[0]
        )


class MyCommand(hazwaz.MainCommand, hazwaz.mixins.ExternalEditorMixin):
    """
    A command that edits a file
    """

    def main(self):
        my_file = tempfile.NamedTemporaryFile()
        self.edit_file_in_external_editor(my_file.name)
        my_file.close()


class testCommandWithMixin(hazwaz.unittest.HazwazTestCase):
    def test_run(self):
        cmd = MyCommand()
        cmd.editors = [("true", "true")]
        stream = self.run_with_argv(cmd, [
            "mycommand",
        ])
        self.assertEqual(stream["stdout"].getvalue(), "")
        self.assertEqual(stream["stderr"].getvalue(), "")


if __name__ == '__main__':
    unittest.main()