Merge pull request #1153 from parona-source/terminal_size_mocking

Tests: Use monkeypatch.context to avoid side effects on pytest

1 files changed, 4 insertions, 3 deletions

diff --git a/test/test_main.py b/test/test_main.py
index 2a86c13..589d6cd 100644
--- a/test/test_main.py
+++ b/test/test_main.py
@@ -259,9 +259,10 @@ def test_reserved_space_is_integer(monkeypatch):
     def stub_terminal_size():
         return (5, 5)
 
-    monkeypatch.setattr(shutil, 'get_terminal_size', stub_terminal_size)
-    mycli = MyCli()
-    assert isinstance(mycli.get_reserved_space(), int)
+    with monkeypatch.context() as m:
+        m.setattr(shutil, 'get_terminal_size', stub_terminal_size)
+        mycli = MyCli()
+        assert isinstance(mycli.get_reserved_space(), int)
 
 
 def test_list_dsn():