def setup(self):
self.dummy_object = MockModel()
self.dummy_object.save()
self.dummy_related_object = RelatedMockModel(
parent=self.dummy_object)
self.dummy_related_object2 = RelatedMockModel(
parent=self.dummy_object)
self.dummy_related_object3 = RelatedMockModel(
parent=self.dummy_object)
self.dummy_related_object3.save()
self.dummy_related_object2.save()
self.dummy_related_object.save()
class TestItemLimitInlineMixin:
def setup(self):
self.dummy_object = MockModel()
self.dummy_object.save()
self.dummy_related_object = RelatedMockModel(
parent=self.dummy_object)
self.dummy_related_object2 = RelatedMockModel(
parent=self.dummy_object)
self.dummy_related_object3 = RelatedMockModel(
parent=self.dummy_object)
self.dummy_related_object3.save()
self.dummy_related_object2.save()
self.dummy_related_object.save()
def test_min_forms_valid(self):
formset = inlineformset_factory(
MockModel, RelatedMockModel, formset=MinInlineFormSet, fields='__all__')
formset_instance = formset(instance=self.dummy_object, data={
'related_mock-TOTAL_FORMS': '1',
'related_mock-INITIAL_FORMS': '0',
'related_mock-MAX_NUM_FORMS': '',
'related_mock-0-parent': self.dummy_object.pk,
'related_mock-0-id': self.dummy_related_object.pk,
})
assert formset_instance.is_valid() is True
def test_min_forms_invalid(self):
formset = inlineformset_factory(
MockModel, RelatedMockModel, formset=MinInlineFormSet, fields='__all__')
formset_instance = formset(instance=self.dummy_object)
with pytest.raises(ValidationError) as exc:
formset_instance.clean()
assert exc.value.messages == [
'Please provide at least 1 related mock model.']
def test_max_forms_valid(self):
formset = inlineformset_factory(
MockModel, RelatedMockModel, formset=MaxInlineFormSet, extra=3, fields='__all__')
formset_instance = formset(instance=self.dummy_object, data={
'related_mock-TOTAL_FORMS': '2',
'related_mock-INITIAL_FORMS': '0',
'related_mock-MAX_NUM_FORMS': '',
'related_mock-0-parent': self.dummy_object.pk,
'related_mock-0-id': self.dummy_related_object.pk,
'related_mock-1-parent': self.dummy_object.pk,
'related_mock-1-id': self.dummy_related_object2.pk,
})
assert formset_instance.is_valid() is True
def test_max_forms_invalid(self):
formset = inlineformset_factory(
MockModel, RelatedMockModel, formset=MaxInlineFormSet, extra=3, fields='__all__')
formset_instance = formset(instance=self.dummy_object, data={
'related_mock-TOTAL_FORMS': '3',
'related_mock-INITIAL_FORMS': '0',
'related_mock-MAX_NUM_FORMS': '',
'related_mock-0-parent': self.dummy_object.pk,
'related_mock-0-id': self.dummy_related_object.pk,
'related_mock-1-parent': self.dummy_object.pk,
'related_mock-1-id': self.dummy_related_object2.pk,
'related_mock-2-parent': self.dummy_object.pk,
'related_mock-2-id': self.dummy_related_object3.pk,
})
with pytest.raises(ValidationError) as exc:
formset_instance.clean()
assert exc.value.messages == [
'Please provide at most 2 related mock models.']