Merge pull request dbcli#1152 from hroncok/patch-1

amjith · web-flow · commit fa4e3dbfcfe5 · 2024-03-28T22:14:03.000-07:00
Tests: monkeypatch shutil.get_terminal_size properly
diff --git a/test/test_main.py b/test/test_main.py
@@ -254,19 +254,15 @@ def test_conditional_pager(monkeypatch):
     SPECIAL_COMMANDS['pager'].handler('')
 
 
-def test_reserved_space_is_integer():
+def test_reserved_space_is_integer(monkeypatch):
     """Make sure that reserved space is returned as an integer."""
     def stub_terminal_size():
         return (5, 5)
 
-    old_func = shutil.get_terminal_size
-
-    shutil.get_terminal_size = stub_terminal_size
+    monkeypatch.setattr(shutil, 'get_terminal_size', stub_terminal_size)
     mycli = MyCli()
     assert isinstance(mycli.get_reserved_space(), int)
 
-    shutil.get_terminal_size = old_func
-
 
 def test_list_dsn():
     runner = CliRunner()
