From ed4e951f2259de76d1560aac8c34ef3589b31e6b Mon Sep 17 00:00:00 2001
From: Tymoteusz Jankowski <tymoteusz.jankowski@gmail.com>
Date: Fri, 6 Dec 2019 21:57:45 +0100
Subject: [PATCH] Use monkeypatch in test_dagascii.py

---
 tests/unit/test_dagascii.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/unit/test_dagascii.py b/tests/unit/test_dagascii.py
index 7ef20bd85..6e8ad4a10 100644
--- a/tests/unit/test_dagascii.py
+++ b/tests/unit/test_dagascii.py
@@ -1,4 +1,3 @@
-import mock
 import os
 
 from dvc import dagascii
@@ -21,8 +20,9 @@ def test_plainpager_returned_when_less_missing(mocker):
     assert pager.__name__ == "plainpager"
 
 
-def test_tempfilepager_returned_when_var_defined(mocker):
-    mocker.patch.dict(os.environ, {"DVC_PAGER": dagascii.DEFAULT_PAGER})
+def test_tempfilepager_returned_when_var_defined(monkeypatch):
+    monkeypatch.setenv(DVC_PAGER, dagascii.DEFAULT_PAGER)
+
     pager = dagascii.find_pager()
 
     assert pager.cmd == dagascii.DEFAULT_PAGER