feat: add District context for templates

Author: lalver1

pems/districts/views.py

@@ -1,15 +1,24 @@
 from django.views.generic import TemplateView
 
+from .models import District
+
 
 class IndexView(TemplateView):
     template_name = "districts/index.html"
 
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context["district_queryset"] = District.objects.all()
+        return context
+
 
 class DistrictView(TemplateView):
     template_name = "districts/district.html"
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
         district = self.kwargs.get("district")
-        context["district"] = district
+        context["url_district"] = district
+        context["district_queryset"] = District.objects.all()
+        context["district_query"] = District.objects.all().get(number=district)
         return context

tests/pytest/pems/districts/test_views.py

@@ -1,6 +1,26 @@
 import pytest
 
 from pems.districts import views
+from pems.districts.models import District
+
+
+class TestIndexViewView:
+    @pytest.fixture
+    def view(app_request):
+        v = views.IndexView()
+        v.setup(app_request)
+
+        return v
+
+    @pytest.mark.django_db
+    def test_get_context_data(self, view):
+
+        context = view.get_context_data()
+
+        assert set(context["district_queryset"]) == set(District.objects.all())
+
+    def test_template_name(self, view):
+        assert view.template_name == "districts/index.html"
 
 
 class TestDistrictView:
@@ -11,11 +31,13 @@ def view(app_request):
 
         return v
 
+    @pytest.mark.django_db
+    @pytest.mark.usefixtures("model_District")
     def test_get_context_data(self, view):
 
         context = view.get_context_data()
 
-        assert context["district"] == 1
+        assert context["url_district"] == 1
 
     def test_template_name(self, view):
         assert view.template_name == "districts/district.html"