38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
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))
|