Advertisement
matacoder

Untitled

Sep 11th, 2020
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.85 KB | None | 0 0
  1.     def test_post_with_image(self):
  2.         """ Post with image is added """
  3.         small_gif = (
  4.             b"\x47\x49\x46\x38\x39\x61\x01\x00\x01\x00\x00\x00\x00\x21\xf9\x04"
  5.             b"\x01\x0a\x00\x01\x00\x2c\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02"
  6.             b"\x02\x4c\x01\x00\x3b"
  7.         )
  8.         img = SimpleUploadedFile(
  9.             "small.gif",
  10.             small_gif,
  11.             content_type="image/gif"
  12.         )
  13.         url = reverse("new_post")
  14.         data = {
  15.             "text": "Abracadabra with image",
  16.             "group": self.testgroup.id,
  17.             "image": img
  18.         }
  19.         response = self.client.post(url, data=data, follow=True)
  20.         self.assertEqual(response.status_code, 200)
  21.         urls = self.generate_urls()
  22.         for url in urls:
  23.             response = self.client.get(url)
  24.             self.assertIn("<img".encode(), response.content)
  25.  
  26.     def test_non_img_file(self):
  27.         """ Test uploading non-image file """
  28.         wrong_file = (
  29.             b"\x01\x0a\x00\x01\x00\x2c\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02"
  30.             b"\x02\x4c\x01\x00\x3b"
  31.         )
  32.         self.wrong = SimpleUploadedFile(
  33.             "wrong_file.doc",
  34.             wrong_file,
  35.             content_type="doc"
  36.         )
  37.         url = reverse("new_post")
  38.         data = {
  39.             "text": "Abracadabra with non-image",
  40.             "group": self.testgroup.id,
  41.             "image": self.wrong
  42.         }
  43.         response = self.client.post(url, data=data)
  44.         self.assertFormError(
  45.             response,
  46.             form="form",
  47.             field="image",
  48.             errors="Загрузите правильное изображение. Файл, "
  49.                    "который вы загрузили, поврежден или не "
  50.                    "является изображением.")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement