def test_custom(self):
with self.settings(NOTORHOT_SETTINGS={
'WRITE_IN_MODEL': '_tests.SimpleWriteIn'
}):
self.assertEqual(get_write_in_model_name(), '_tests.SimpleWriteIn')
self.assertEqual(get_write_in_model(), SimpleWriteIn)
from django.contrib import admin
from django.conf import settings
# Register your models here.
from notorhot.contrib.write_in.models import DefaultWriteIn
from notorhot.contrib.write_in.utils import get_write_in_model_name, \
get_write_in_model
def use_default_admin():
notorhot_settings = getattr(settings, 'NOTORHOT_SETTINGS', {})
return notorhot_settings.get('USE_DEFAULT_ADMIN', True)
class DefaultWriteInAdmin(admin.ModelAdmin):
list_display = ['__unicode__', 'category', 'status', 'date_submitted',
'date_processed',]
list_filter = ['category', 'status',]
search_fields = ['candidate_name', 'submitter_name', 'submitter_email',]
date_hierarchy = 'date_submitted'
readonly_fields = ['date_submitted',]
if use_default_admin():
admin.site.register(get_write_in_model(), DefaultWriteInAdmin)
def test_defaults(self):
with self.settings(NOTORHOT_SETTINGS={}):
self.assertEqual(get_write_in_model_name(), 'write_in.DefaultWriteIn')
self.assertEqual(get_write_in_model(), DefaultWriteIn)