From 01016563fccf31ddee6e5b8b4b13c065149dd814 Mon Sep 17 00:00:00 2001
From: Thomas Nixon <tom@tomn.co.uk>
Date: Mon, 19 Feb 2018 22:31:38 +0000
Subject: Move screen stop/start logic into context manager.

As well as reducing duplication and adding screen size detection to
:pipeto and :pyshell, this ensures that the screen is always restarted,
resulting in cleaner error handling if an error occurs while the screen
is stopped.
---
 tests/utilities.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'tests/utilities.py')

diff --git a/tests/utilities.py b/tests/utilities.py
index ea4e700c..85f3d3eb 100644
--- a/tests/utilities.py
+++ b/tests/utilities.py
@@ -175,6 +175,7 @@ def make_key(revoked=False, expired=False, invalid=False, can_encrypt=True,
 
 def make_ui(**kwargs):
     ui = mock.Mock(**kwargs)
+    ui.paused.return_value = mock.MagicMock()
 
     return ui
 
-- 
cgit v1.2.3