def test_non_select2_lookups(self): class TestForm(forms.Form): form_field = forms.ModelChoiceField(TestModel.objects) value = self.searchable_models[0] form = TestForm(initial={'form_field': value.pk}) field = form['form_field'] field = select2_lookups(field) choices = [choice for choice in field.field.choices] self.assertEqual(len(choices), len(self.models) + 1)
def test_select2_lookups_posted(self): class TestForm(forms.Form): form_field = forms.ModelChoiceField(SearchableTestModel.objects) value = self.searchable_models[0] form = TestForm(data={'form_field': value.pk}) field = form['form_field'] field = select2_lookups(field) choices = [choice for choice in field.field.choices] self.assertEqual(len(choices), 1) self.assertEqual(choices[0][0], value.pk)