Skip to content

Commit

Permalink
:fix: add tests filtersets
Browse files Browse the repository at this point in the history
  • Loading branch information
@LePetitTim
LePetitTim committed Jan 24, 2023
1 parent 64f8349 commit d4eb87a
Showing 1 changed file with 61 additions and 28 deletions.
89 changes: 61 additions & 28 deletions geotrek/maintenance/tests/test_filters.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -304,6 +304,14 @@ def setUpTestData(cls):
geom=GeometryCollection(Point(5, 5), srid=settings.SRID))
cls.intervention_course_in_2 = InterventionFactory.create(target=cls.course_in_2)

course_deleted = CourseFactory.create(geom=GeometryCollection(Point(1, 1), srid=settings.SRID))
cls.intervention_course_deleted = InterventionFactory.create(target=course_deleted)
course_deleted.delete()

site_deleted = SiteFactory.create(geom=GeometryCollection(Point(1, 1), srid=settings.SRID))
cls.intervention_site_deleted = InterventionFactory.create(target=site_deleted)
site_deleted.delete()

cls.blade_in_1 = BladeFactory.create(signage=signage_in_1)
cls.intervention_blade_in_1 = InterventionFactory.create(target=cls.blade_in_1)

Expand All @@ -325,8 +333,10 @@ def test_filter_in_1_city(self):
{self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_2_city(self):
"""
Expand All @@ -339,8 +349,10 @@ def test_filter_in_2_city(self):
{self.intervention_topology_in_2,
self.intervention_site_in_2,
self.intervention_course_in_2,
self.intervention_blade_in_2})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_2,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_1_and_2_city(self):
"""
Expand All @@ -358,16 +370,18 @@ def test_filter_in_1_and_2_city(self):
self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 8)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 10)

def test_filter_out_city(self):
"""
We should not have any interventions
"""
filter = InterventionFilterSet(data={'city': [CityFactory.create(geom=self.geom_out)]})
self.assertTrue(filter.is_valid())
self.assertEqual(len(filter.qs), 0)
self.assertEqual(len(filter.qs), 2)

def test_filter_in_1_district(self):
"""
Expand All @@ -380,8 +394,11 @@ def test_filter_in_1_district(self):
{self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted
})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_2_district(self):
"""
Expand All @@ -394,8 +411,10 @@ def test_filter_in_2_district(self):
{self.intervention_topology_in_2,
self.intervention_site_in_2,
self.intervention_course_in_2,
self.intervention_blade_in_2})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_2,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_1_and_2_district(self):
"""
Expand All @@ -413,16 +432,18 @@ def test_filter_in_1_and_2_district(self):
self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 8)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 10)

def test_filter_out_district(self):
"""
We should not have any interventions
"""
filter = InterventionFilterSet(data={'district': [DistrictFactory.create(geom=self.geom_out)]})
self.assertTrue(filter.is_valid())
self.assertEqual(len(filter.qs), 0)
self.assertEqual(len(filter.qs), 2)

def test_filter_in_1_restricted_area(self):
"""
Expand All @@ -434,8 +455,10 @@ def test_filter_in_1_restricted_area(self):
{self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_2_restricted_area(self):
"""
Expand All @@ -448,8 +471,10 @@ def test_filter_in_2_restricted_area(self):
{self.intervention_topology_in_2,
self.intervention_site_in_2,
self.intervention_course_in_2,
self.intervention_blade_in_2})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_2,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_1_and_2_restricted_area(self):
"""
Expand All @@ -467,16 +492,18 @@ def test_filter_in_1_and_2_restricted_area(self):
self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 8)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 10)

def test_filter_out_restricted_area(self):
"""
We should not have any interventions
"""
filter = InterventionFilterSet(data={'area': [RestrictedAreaFactory.create(geom=self.geom_out)]})
self.assertTrue(filter.is_valid())
self.assertEqual(len(filter.qs), 0)
self.assertEqual(len(filter.qs), 2)

def test_filter_in_1_restricted_area_type(self):
"""
Expand All @@ -491,8 +518,10 @@ def test_filter_in_1_restricted_area_type(self):
{self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_2_restricted_area_type(self):
"""
Expand All @@ -507,8 +536,10 @@ def test_filter_in_2_restricted_area_type(self):
{self.intervention_topology_in_2,
self.intervention_site_in_2,
self.intervention_course_in_2,
self.intervention_blade_in_2})
self.assertEqual(len(filter.qs), 4)
self.intervention_blade_in_2,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 6)

def test_filter_in_1_and_in_2_restricted_area_type(self):
"""
Expand All @@ -531,8 +562,10 @@ def test_filter_in_1_and_in_2_restricted_area_type(self):
self.intervention_topology_in_1,
self.intervention_site_in_1,
self.intervention_course_in_1,
self.intervention_blade_in_1})
self.assertEqual(len(filter.qs), 8)
self.intervention_blade_in_1,
self.intervention_site_deleted,
self.intervention_course_deleted})
self.assertEqual(len(filter.qs), 10)

def test_filter_out_restricted_area_type(self):
"""
Expand All @@ -542,7 +575,7 @@ def test_filter_out_restricted_area_type(self):
RestrictedAreaFactory.create(geom=self.geom_out, area_type=restricted_area_type)
filter = InterventionFilterSet(data={'area_type': [restricted_area_type.pk]})
self.assertTrue(filter.is_valid())
self.assertEqual(len(filter.qs), 0)
self.assertEqual(len(filter.qs), 2)

def test_filter_restricted_area_type_without_restricted_area(self):
"""
Expand Down

0 comments on commit d4eb87a

Please sign in to comment.