def __init__(self, *args, **kwargs):
super(WriteInDefaultView, self).__init__(*args, **kwargs)
# If we're using the default model, we can specify our fields explicitly
# if they haven't been explicitly overridden
if getattr(self, 'fields', []) == [] and \
get_write_in_model_name() == 'write_in.DefaultWriteIn':
self.fields = self.default_fields
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)
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)
