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()
|
