Function test_geojson_find_identifier

tests/test_features.py:177–277

()

def test_geojson_find_identifier():
    def _create(*properties):
        return {
            "type": "FeatureCollection",
            "features": [
                {"type": "Feature", "properties": item} for item in properties
            ],
        }

    def _assert_id_got_added(data):
        _geojson = GeoJson(data)
        assert _geojson.find_identifier() == "feature.id"
        assert _geojson.data["features"][0]["id"] == "0"

    data_with_id = _create(None, None)
    data_with_id["features"][0]["id"] = "this-is-an-id"
    data_with_id["features"][1]["id"] = "this-is-another-id"
    geojson = GeoJson(data_with_id)
    assert geojson.find_identifier() == "feature.id"
    assert geojson.data["features"][0]["id"] == "this-is-an-id"

    data_with_unique_properties = _create(
        {"property-key": "some-value"},
        {"property-key": "another-value"},
    )
    geojson = GeoJson(data_with_unique_properties)
    assert geojson.find_identifier() == "feature.properties.property-key"

    data_with_unique_properties = _create(
        {"property-key": 42},
        {"property-key": 43},
        {"property-key": "or a string"},
    )
    geojson = GeoJson(data_with_unique_properties)
    assert geojson.find_identifier() == "feature.properties.property-key"

    # The test cases below have no id field or unique property,
    # so an id will be added to the data.

    data_with_identical_ids = _create(None, None)
    data_with_identical_ids["features"][0]["id"] = "identical-ids"
    data_with_identical_ids["features"][1]["id"] = "identical-ids"
    _assert_id_got_added(data_with_identical_ids)

    data_with_some_missing_ids = _create(None, None)
    data_with_some_missing_ids["features"][0]["id"] = "this-is-an-id"
    # the second feature doesn't have an id
    _assert_id_got_added(data_with_some_missing_ids)

    data_with_identical_properties = _create(
        {"property-key": "identical-value"},
        {"property-key": "identical-value"},
    )
    _assert_id_got_added(data_with_identical_properties)

    data_bare = _create(None)
    _assert_id_got_added(data_bare)