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)