Browse Source

test is_image and is_file flags on attachment model

Rafał Pitoń 8 years ago
parent
commit
8dd766175a
1 changed files with 6 additions and 0 deletions
  1. 6 0
      misago/threads/tests/test_attachments_api.py

+ 6 - 0
misago/threads/tests/test_attachments_api.py

@@ -193,6 +193,9 @@ class AttachmentsApiTestCase(AuthenticatedUserTestCase):
         attachment = Attachment.objects.get(id=response_json['id'])
 
         self.assertEqual(attachment.filename, 'document.pdf')
+        self.assertTrue(attachment.is_file)
+        self.assertFalse(attachment.is_image)
+
         self.assertIsNotNone(attachment.file)
         self.assertTrue(not attachment.image)
         self.assertTrue(not attachment.thumbnail)
@@ -219,6 +222,9 @@ class AttachmentsApiTestCase(AuthenticatedUserTestCase):
         attachment = Attachment.objects.get(id=response_json['id'])
 
         self.assertEqual(attachment.filename, 'small.jpg')
+        self.assertFalse(attachment.is_file)
+        self.assertTrue(attachment.is_image)
+
         self.assertTrue(not attachment.file)
         self.assertIsNotNone(attachment.image)
         self.assertIsNotNone(attachment.thumbnail)