diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_mixins.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tests/test_mixins.py b/tests/test_mixins.py index 1e2f91d..44ae165 100644 --- a/tests/test_mixins.py +++ b/tests/test_mixins.py @@ -1,3 +1,4 @@ +import tempfile import unittest import hazwaz @@ -33,5 +34,27 @@ class testEditorMixin(unittest.TestCase): ) +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() |