summaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/test_main.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/test/test_main.py b/test/test_main.py
index 00fdc1b..7731603 100644
--- a/test/test_main.py
+++ b/test/test_main.py
@@ -1,4 +1,5 @@
import os
+import shutil
import click
from click.testing import CliRunner
@@ -258,13 +259,13 @@ def test_reserved_space_is_integer():
def stub_terminal_size():
return (5, 5)
- old_func = click.get_terminal_size
+ old_func = shutil.get_terminal_size
- click.get_terminal_size = stub_terminal_size
+ shutil.get_terminal_size = stub_terminal_size
mycli = MyCli()
assert isinstance(mycli.get_reserved_space(), int)
- click.get_terminal_size = old_func
+ shutil.get_terminal_size = old_func
def test_list_dsn():