tests: replace spy() with pytest-mock spy

Fixes #3198
This commit is contained in:
Fábio Santos
2020-01-21 14:47:49 +00:00
parent 74bbc3bc04
commit 5dfcda54f9
5 changed files with 40 additions and 31 deletions
-13
View File
@@ -2,22 +2,9 @@ import os
from contextlib import contextmanager
from filecmp import dircmp
from mock import MagicMock
from dvc.scm import Git
def spy(method_to_decorate):
mock = MagicMock()
def wrapper(self, *args, **kwargs):
mock(*args, **kwargs)
return method_to_decorate(self, *args, **kwargs)
wrapper.mock = mock
return wrapper
def get_gitignore_content():
with open(Git.GITIGNORE, "r") as gitignore:
return gitignore.read().splitlines()