bugbug/tests/conftest.py

# -*- coding: utf-8 -*-
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
# You can obtain one at http://mozilla.org/MPL/2.0/.

import os
import shutil

import pytest

from bugbug import bugzilla, repository

FIXTURES_DIR = os.path.join(os.path.dirname(__file__), "fixtures")


@pytest.fixture
def mock_data(tmp_path_factory):
    tmp_path = tmp_path_factory.mktemp("")
    os.mkdir(tmp_path / "data")

    DBs = [os.path.basename(bugzilla.BUGS_DB), os.path.basename(repository.COMMITS_DB)]

    for f in DBs:
        shutil.copyfile(os.path.join(FIXTURES_DIR, f), tmp_path / "data" / f)

    os.chdir(tmp_path)


@pytest.fixture
def get_fixture_path():
    def _get_fixture_path(path):
        path = os.path.join(FIXTURES_DIR, path)
        assert os.path.exists(path)
        return path

    return _get_fixture_path