from pathlib import Path

import pytest

from plesna.dataplatform import DataPlateform
from plesna.datastore.fs_datacatalogue import FSDataCatalogue

FIXTURE_DIR = Path(__file__).parent / Path("raw_data")


@pytest.fixture
def raw_catalogue(tmp_path):
    raw_path = Path(tmp_path) / "raw"
    raw_path.mkdir()
    return FSDataCatalogue("raw", raw_path)


@pytest.fixture
def bronze_catalogue(tmp_path):
    bronze_path = Path(tmp_path) / "bronze"
    bronze_path.mkdir()
    return FSDataCatalogue("bronze", bronze_path)


@pytest.fixture
def silver_catalogue(tmp_path):
    silver_path = Path(tmp_path) / "silver"
    silver_path.mkdir()
    return FSDataCatalogue("silver", silver_path)


def test_add_catalogue(
    raw_catalogue: FSDataCatalogue,
    bronze_catalogue: FSDataCatalogue,
    silver_catalogue: FSDataCatalogue,
):
    dp = DataPlateform()
    dp.add_datacatalague("raw", raw_catalogue)
    dp.add_datacatalague("bronze", bronze_catalogue)
    dp.add_datacatalague("silver", silver_catalogue)

    assert dp.datacatalogues == ["raw", "bronze", "silver"]
    assert dp.get_datacatalogue("raw") == raw_catalogue