Merge pull request #1152 from hroncok/patch-1

Tests: monkeypatch shutil.get_terminal_size properly
dbcli · Mar 29, 2024 · fa4e3db · fa4e3db
2 parents 11dfb08 + e1ae26d
commit fa4e3db
Showing 1 changed file with 2 additions and 6 deletions.
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()