Unverified Commit cf2d28e0 authored by mvdbeek's avatar mvdbeek
Browse files

Fix unit tests

parent 7986110c
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -207,6 +207,9 @@ class MockObjectStore:
    def exists(self, *args, **kwargs):
        return True

    def construct_path(self, *args, **kwds):
        return self.working_directory

    def get_filename(self, *args, **kwds):
        if kwds.get("base_dir", "") == "job_work":
            return self.working_directory
+4 −1
Original line number Diff line number Diff line
@@ -248,7 +248,10 @@ def test_dataset_false_extra_files_path():
    new_path = "/new/path/dataset_123.dat"
    dataset_path = DatasetPath(123, MOCK_DATASET_PATH, false_path=new_path)
    wrapper = DatasetFilenameWrapper(
        dataset, compute_environment=cast(ComputeEnvironment, MockComputeEnvironment(dataset_path))
        dataset,
        compute_environment=cast(
            ComputeEnvironment, MockComputeEnvironment(dataset_path, MOCK_DATASET_EXTRA_FILES_PATH)
        ),
    )
    # Setting false_path is not enough to override
    assert wrapper.extra_files_path == MOCK_DATASET_EXTRA_FILES_PATH
+3 −0
Original line number Diff line number Diff line
@@ -1039,6 +1039,9 @@ class MockObjectStore:
    def get_filename(self, *args, **kwds):
        return "mock_dataset_14.dat"

    def construct_path(self, *args, **kwds):
        return "mock_dataset_14.dat"

    def get_store_by(self, *args, **kwds):
        return "id"