From 4bd2ce25e1e8e76c232b6a87002b0783b577084e Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Tue, 20 Aug 2019 14:27:20 +0100 Subject: [PATCH] get_files_number test fix --- tests/unit/output/test_local.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/unit/output/test_local.py b/tests/unit/output/test_local.py index 8149e83ff..6e2af1132 100644 --- a/tests/unit/output/test_local.py +++ b/tests/unit/output/test_local.py @@ -2,6 +2,7 @@ from mock import patch from dvc.stage import Stage from dvc.output import OutputLOCAL +from dvc.remote.local import RemoteLOCAL from tests.basic_env import TestDvc @@ -32,8 +33,12 @@ class TestGetFilesNumber(TestDvc): self.assertEqual(0, o.get_files_number()) @patch.object(OutputLOCAL, "checksum", "12345678.dir") - @patch.object(OutputLOCAL, "dir_cache", [{"md5": "asdf"}, {"md5": "qwe"}]) - def test_return_mutiple_for_dir(self): + @patch.object( + RemoteLOCAL, + "get_dir_cache", + return_value=[{"md5": "asdf"}, {"md5": "qwe"}], + ) + def test_return_mutiple_for_dir(self, mock_get_dir_cache): o = self._get_output() self.assertEqual(2, o.get_files_number())