lafrite/plesna
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refact: use flux_id instead of name

Browse Source
This commit is contained in:
Bertrand Benjamin
2025-01-19 14:59:47 +01:00
parent d7716a4b8e
commit 9118feb4c6
2 changed files with 24 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/dataplatform/test_dataplateform.py
View File

@@ -84,17 +84,17 @@ def foo_flux(repository: FSDataRepository) -> Flux:
return flux
def test_add_flux(repository: FSDataRepository, copy_flux: Flux):
def test_add_flux(repository: FSDataRepository, copy_flux: Flux, foo_flux: Flux):
dataplatform = DataPlateform()
dataplatform.add_repository(repository)
dataplatform.add_flux(name="copy_flux", flux=copy_flux)
dataplatform.add_flux(flux=copy_flux)
assert dataplatform.fluxes == ["copy_flux"]
dataplatform.add_flux(name="copy_flux_bis", flux=copy_flux)
assert dataplatform.fluxes == ["copy_flux", "copy_flux_bis"]
dataplatform.add_flux(flux=foo_flux)
assert dataplatform.fluxes == ["copy_flux", "foo_flux"]
assert dataplatform.flux("copy_flux") == copy_flux
assert dataplatform.flux("copy_flux_bis") == copy_flux
assert dataplatform.flux("foo_flux") == foo_flux
@pytest.fixture
@@ -107,8 +107,8 @@ def dataplatform(
dp.add_repository(repository)
dp.add_flux("foo", foo_flux)
dp.add_flux("raw_brz_copy_username", copy_flux)
dp.add_flux(foo_flux)
dp.add_flux(copy_flux)
return dp
@@ -269,12 +269,12 @@ def test_content_from_graph_arguments(dataplatform: DataPlateform):
def test_execute_flux(dataplatform: DataPlateform):
meta = dataplatform.execute_flux("foo")
meta = dataplatform.execute_flux("foo_flux")
assert meta.data == {"who": "foo"}
assert dataplatform.repository("test").schema("test-bronze").tables == []
meta = dataplatform.execute_flux("raw_brz_copy_username")
meta = dataplatform.execute_flux("copy_flux")
assert meta.data == {"src_size": 283, "tgt_size": 283}
assert dataplatform.repository("test").schema("test-bronze").tables == ["test-bronze-username"]