44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
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
|