Teach editor to be monkeypatchable (see #6)

rec · Jan 6, 2024 · 22d2c98 · 22d2c98
1 parent 4592ddb
commit 22d2c98
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/editor/__init__.py b/editor/__init__.py
@@ -59,7 +59,7 @@
 EDITORS = {'Windows': 'notepad'}
 
 
-@xmod.xmod
+@xmod.xmod(mutable=True)
 def editor(
     text: t.Optional[str] = None,
     filename: t.Union[None, Path, str] = None,

diff --git a/test_editor.py b/test_editor.py
@@ -8,6 +8,7 @@
 
 FILENAME = 'a_file.txt'
 EDITOR = editor.default_editor()
+TEST_CONTENT = 'roses are red,\nwater is blue.\n'
 
 
 @mock.patch('editor.runs.call', autospec=True)
@@ -46,3 +47,15 @@ def test_temp2(self, call):
         expected = 'some contents'
         assert actual == expected
         call.assert_called_once()
+
+
+def main():
+    print(editor.editor())
+
+
+def test_main(monkeypatch, capsys):
+    monkeypatch.setattr('editor.editor', lambda: TEST_CONTENT)
+
+    main()
+
+    assert TEST_CONTENT + '\n' == capsys.readouterr().out