Page MenuHomeSoftware Heritage
Differential D8566 Diff 31014 swh/loader/package/conda/tests/test_tasks.py

Changeset View

Standalone View

View Options

swh/loader/package/conda/tests/test_tasks.py

  • This file was added.
# Copyright (C) 2022 The Software Heritage developers
# See the AUTHORS file at the top-level directory of this distribution
# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information
def test_tasks_conda_loader(
mocker, swh_scheduler_celery_app, swh_scheduler_celery_worker, swh_config
):
mock_load = mocker.patch("swh.loader.package.conda.loader.CondaLoader.load")
mock_load.return_value = {"status": "eventful"}
res = swh_scheduler_celery_app.send_task(
"swh.loader.package.conda.tasks.LoadConda",
kwargs=dict(
url="some-url/api/packages/some-package",
artifacts=[{"version": "0.0.1", "url": "some-package-0.0.1.crate"}],
),
)
assert res
res.wait()
assert res.successful()
assert mock_load.called
assert res.result == {"status": "eventful"}
anlambertUnsubmitted
Done
Inline Actions

I recently refactored the loading tasks creation tests for all loaders, please replace by this instead:

import uuid

import pytest

from swh.scheduler.model import ListedOrigin, Lister

NAMESPACE = "swh.loader.package.conda"


@pytest.fixture
def conda_lister():
    return Lister(name="conda", instance_name="example", id=uuid.uuid4())


@pytest.fixture
def conda_listed_origin(conda_lister):
    return ListedOrigin(
        lister_id=conda_lister.id,
        url="https://anaconda.org/channel/some-package",
        visit_type="conda",
        extra_loader_arguments={
            "artifacts": [{"version": "0.0.1", "url": "some-package-0.0.1.tar.bz2"}],
        },
    )


def test_conda_loader_task_for_listed_origin(
    loading_task_creation_for_listed_origin_test,
    conda_lister,
    conda_listed_origin,
):
    loading_task_creation_for_listed_origin_test(
        loader_class_name=f"{NAMESPACE}.loader.CondaLoader",
        task_function_name=f"{NAMESPACE}.tasks.LoadConda",
        lister=conda_lister,
        listed_origin=conda_listed_origin,
    )
anlambert: I recently refactored the loading tasks creation tests for all loaders, please replace by this…
About · Developer information · Legal