From 54be1ce54f17ba240ffbcbf3082cb3ec2b75d4e2 Mon Sep 17 00:00:00 2001 From: Lauren Qurashi Date: Fri, 20 Sep 2024 16:16:58 +0100 Subject: [PATCH] Add Geo Areas serialize deserialise test --- measures/tests/test_forms.py | 32 ++++++++++++-------------------- 1 file changed, 12 insertions(+), 20 deletions(-) diff --git a/measures/tests/test_forms.py b/measures/tests/test_forms.py index b8b8f00e9..15bc69d44 100644 --- a/measures/tests/test_forms.py +++ b/measures/tests/test_forms.py @@ -2018,34 +2018,26 @@ def test_simple_measure_edit_forms_serialize_deserialize( assert deserialized_form.is_valid() assert deserialized_form.data == form_data -@pytest.mark.parametrize( - "form_data", - [ - ("measure_geo_area_erga_omnes_form_data"), - ("measure_geo_area_erga_omnes_exclusions_form_data"), - ("measure_geo_area_geo_group_form_data"), - ("measure_geo_area_geo_group_exclusions_form_data"), - ], - ids=[ - "erga-omnes", - "erga-omnes-exclusions", - "geo-group", - "geo-group-exclusions", - ], -) -def test_measure_edit_forms_geo_area_exclusions_serialize_deserialize(form_data, request): - form_data = request.getfixturevalue(form_data) + +def test_measure_edit_forms_geo_area_exclusions_serialize_deserialize(): + geo_area1 = factories.GeographicalAreaFactory.create() + geo_area2 = factories.GeographicalAreaFactory.create() + + form_data = { + "form-0-excluded_area": geo_area1, + "form-1-excluded_area": geo_area2 + } with override_current_transaction(Transaction.objects.last()): - form = forms.MeasureGeographicalAreaForm( + form = forms.MeasureGeographicalAreaExclusionsFormSet( form_data, ) assert form.is_valid() serializable_form_data = form.serializable_data() - deserialized_form = forms.MeasureGeographicalAreaForm( + deserialized_form = forms.MeasureGeographicalAreaExclusionsFormSet( data=serializable_form_data, ) assert deserialized_form.is_valid() - assert type(deserialized_form) == forms.MeasureGeographicalAreaForm + assert type(deserialized_form) == forms.MeasureGeographicalAreaExclusionsFormSet assert deserialized_form.data == form.data \ No newline at end of file