Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def test_post_with_image(self):
- """ Post with image is added """
- small_gif = (
- b"\x47\x49\x46\x38\x39\x61\x01\x00\x01\x00\x00\x00\x00\x21\xf9\x04"
- b"\x01\x0a\x00\x01\x00\x2c\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02"
- b"\x02\x4c\x01\x00\x3b"
- )
- img = SimpleUploadedFile(
- "small.gif",
- small_gif,
- content_type="image/gif"
- )
- url = reverse("new_post")
- data = {
- "text": "Abracadabra with image",
- "group": self.testgroup.id,
- "image": img
- }
- response = self.client.post(url, data=data, follow=True)
- self.assertEqual(response.status_code, 200)
- urls = self.generate_urls()
- for url in urls:
- response = self.client.get(url)
- self.assertIn("<img".encode(), response.content)
- def test_non_img_file(self):
- """ Test uploading non-image file """
- wrong_file = (
- b"\x01\x0a\x00\x01\x00\x2c\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02"
- b"\x02\x4c\x01\x00\x3b"
- )
- self.wrong = SimpleUploadedFile(
- "wrong_file.doc",
- wrong_file,
- content_type="doc"
- )
- url = reverse("new_post")
- data = {
- "text": "Abracadabra with non-image",
- "group": self.testgroup.id,
- "image": self.wrong
- }
- response = self.client.post(url, data=data)
- self.assertFormError(
- response,
- form="form",
- field="image",
- errors="Загрузите правильное изображение. Файл, "
- "который вы загрузили, поврежден или не "
- "является изображением.")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement