Use monkeypatch in test_dagascii.py

This commit is contained in:
Tymoteusz Jankowski
2019-12-06 21:58:56 +01:00
parent 6cfc831879
commit ed4e951f22
+3 -3
View File
@@ -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