본문 바로가기

Django

[TDD] 이미지 테스트




    # 상담 신청서 업로드 test

    def test_photo_add(self):

        photo_file = self.generate_photo_file()

        url = '/abcdefg/'

        data={

            'image_file': photo_file

        }

        response= self.client.post(url,data,format='multipart')

        self.assertEqual(response.status_code,200)




    def generate_photo_file(self):

        file = io.BytesIO()

        image = Image.new('RGBA'size=(100100), color=(15500))

        image.save(file'png')

        file.name = 'test.png'

        file.seek(0)

        return file