Merge pull request dbcli#1153 from parona-source/terminal_size_mocking

amjith · web-flow · commit eabe1327ab45 · 2024-03-31T10:42:32.000-07:00
Tests: Use monkeypatch.context to avoid side effects on pytest
diff --git 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():
