TIL(22.12.16) - Django testcode 임시 이미지 파일 생성 후 리뷰 작성하기

# 이미지 업로드
from django.test.client import MULTIPART_CONTENT, encode_multipart, BOUNDARY
from PIL import Image
import tempfile

def get_temporary_image(temp_file):
    size = (200,200)
    color = (255, 0, 0, 0)
    image = Image.new("RGBA", size, color)
    image.save(temp_file, 'png')
    return temp_file
    
# 이미지 포함 리뷰 작성 성공
def test_create_review_with_image(self):
  # 임시 이미지 파일 생성
  temp_file = tempfile.NamedTemporaryFile()
  temp_file.name = "image.png"
  image_file = get_temporary_image(temp_file)
  image_file.seek(0)  # 첫번째 프레임을 받아옴
  self.review_data["review_image_one"] = image_file

  # 전송
  response = self.client.post(
  path=reverse("review_list_view", kwargs={'place_id':1}),
  HTTP_AUTHORIZATION=f"Bearer {self.access_token}",
  content_type=MULTIPART_CONTENT,
  data=encode_multipart(data = self.review_data, boundary=BOUNDARY)
  )
  self.assertEqual(response.status_code, 201)