TST: 100% coverage

ccdb/locations/tests/test_admin.py

@@ -0,0 +1,38 @@
+from django.test import TestCase
+from django.contrib.admin.sites import AdminSite
+
+from ..models import Site, StudyLocation
+from ..admin import SiteAdmin, StudyLocationAdmin
+from .factories import SiteFactory, StudyLocationFactory
+
+
+class SiteAdminTests(TestCase):
+    def setUp(self):
+        self.site = SiteFactory()
+        self.admin_site = AdminSite()
+
+    def test_list_display(self):
+        admin_obj = SiteAdmin(Site, self.admin_site)
+        self.assertEqual(admin_obj.check(), [])
+
+        region_name_from_callable = admin_obj.region_name(self.site)
+        self.assertEqual(region_name_from_callable,
+                         self.site.region.name)
+
+
+class StudyLocationAdminTests(TestCase):
+    def setUp(self):
+        self.sl = StudyLocationFactory()
+        self.site = AdminSite()
+
+    def test_list_display(self):
+        admin_obj = StudyLocationAdmin(StudyLocation, self.site)
+        self.assertEqual(admin_obj.check(), [])
+
+        site_name_from_callable = admin_obj.site_name(self.sl)
+        self.assertEqual(site_name_from_callable,
+                         self.sl.site.name)
+
+        ml_name_from_callable = admin_obj.ml_name(self.sl)
+        self.assertEqual(ml_name_from_callable,
+                         str(self.sl.municipal_location))