diff --git a/src/dso_api/dynamic_api/views/doc.py b/src/dso_api/dynamic_api/views/doc.py index e405f5f56..24d6c61c7 100644 --- a/src/dso_api/dynamic_api/views/doc.py +++ b/src/dso_api/dynamic_api/views/doc.py @@ -289,7 +289,7 @@ def _table_context(ds: Dataset, table: DatasetTableSchema): filters = _get_filters(table.fields) exports = [] # if dataset_name in settings.EXPORTED_DATASETS.split(","): - if ds.enable_export: + if ds.tables.get(name=table_name).enable_export: export_info = [] for type_, extension in ( ("csv", "csv"), diff --git a/src/tests/test_dynamic_api/test_doc.py b/src/tests/test_dynamic_api/test_doc.py index da8477aab..b90ca1be5 100644 --- a/src/tests/test_dynamic_api/test_doc.py +++ b/src/tests/test_dynamic_api/test_doc.py @@ -59,10 +59,11 @@ def test_dataset(api_client, filled_router, gebieden_dataset): @pytest.mark.django_db -def test_dataset_for_export_links(api_client, filled_router, gebieden_dataset): +def test_table_for_export_links(api_client, filled_router, gebieden_dataset): """Tests documentation for a single dataset.""" - gebieden_dataset.enable_export = True - gebieden_dataset.save() + table = gebieden_dataset.tables.get(name="bouwblokken") + table.enable_export = True + table.save() gebieden_doc = reverse( "dynamic_api:doc-gebieden" ) # Gebieden has relationships between its tables. @@ -73,9 +74,13 @@ def test_dataset_for_export_links(api_client, filled_router, gebieden_dataset): content = response.rendered_content # Extensions for exported format followed by ".zip" # are signalling links to the generated exports. - assert "gpkg.zip" in content - assert "jsonl.zip" in content - assert "csv.zip" in content + assert "gebieden_bouwblokken.gpkg.zip" in content + assert "gebieden_bouwblokken.jsonl.zip" in content + assert "gebieden_bouwblokken.csv.zip" in content + + assert "gebieden_buurten.csv.zip" not in content + assert "gebieden_buurten.jsonl.zip" not in content + assert "gebieden_buurten.gpkg.zip" not in content @pytest.mark.django_db