def test_be_able_to_get_attachments_for_model_instance(self): second = Second.objects.create(third_field="xyz") Attachment.objects.create(file_name="x.doc", attach_to=second, attachment="first bunch of something") Attachment.objects.create(file_name="x.doc", attach_to=second, attachment="second bunch of something") first = First.objects.create(first_field="asdf", second_field="xyz") Attachment.objects.create(file_name="x.doc", attach_to=first, attachment_type=Attachment.DOCUMENT, attachment="Hi there, i'm a bunch of bytes") for attachment in Attachment.get_attachments_for(second): self.assertEqual(attachment.attach_to, second) self.assertEqual(Attachment.get_attachments_for(second).count(), 2)
def test_store_attachment_description(self): second = Second.objects.create(third_field="xyz") Attachment.objects.create(file_name="x.doc", attach_to=second, attachment="xxx", attachment_type=Attachment.DOCUMENT, description="Three x's") self.assertEqual(Attachment.get_attachments_for(second).count(), 1)