summaryrefslogtreecommitdiff
path: root/tests/test_mixins.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_mixins.py')
-rw-r--r--tests/test_mixins.py23
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()