# Copyright (C) 2019  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

import re
import os
import pytest
import yaml

from functools import partial

from swh.core.pytest_plugin import get_response_cb
from swh.scheduler.tests.conftest import *  # noqa
from swh.storage.tests.conftest import *  # noqa
from swh.deposit.loader.checker import DepositChecker
from swh.deposit.loader.loader import DepositLoader


@pytest.fixture(scope='session')
def celery_includes():
    return [
        'swh.deposit.loader.tasks',
    ]


@pytest.fixture
def swh_config(tmp_path, swh_storage_postgresql, monkeypatch):
    storage_config = {
        'url': 'https://deposit.softwareheritage.org/',
        'storage': {
            'cls': 'local',
            'args': {
                'db': swh_storage_postgresql.dsn,
                'objstorage': {
                    'cls': 'memory',
                    'args': {}
                },
            },
        },
    }

    conffile = os.path.join(tmp_path, 'deposit.yml')
    with open(conffile, 'w') as f:
        f.write(yaml.dump(storage_config))
    monkeypatch.setenv('SWH_CONFIG_FILENAME', conffile)
    return conffile


@pytest.fixture
def deposit_checker(swh_config):
    return DepositChecker()


@pytest.fixture
def deposit_loader(swh_config):
    return DepositLoader()


@pytest.fixture
def requests_mock_datadir(datadir, requests_mock_datadir):
    """Override default behavior to deal with put method

    """
    cb = partial(get_response_cb, datadir=datadir)
    requests_mock_datadir.put(re.compile('https://'), body=cb)
    return requests_mock_datadir