python으로 dummy 데이터와 REST API test code 작성하기

API로 만들어달래서 만들어줬더니 눈에 보이지않아서 작업한거 맞냐고 항의하는 고객을 상대하는 나의 심정

ML API를 임의로 만든 dummy 데이터로 테스트 해야하는 경우 간단하게 작성하는 테스트 코드

 

물론 postman에서도 여러 기능이 있고, 특히 fastapi를 사용하면 docs에서 편하게 테스트할 수 있지만

아래 경우를 위해 작성한 code임.

 

- 테스트케이스(데이터)를 여러건 저장해두고 테스트해야하는 경우

- 테스트데이터의 형식이 다양한데 한 요청에서 다 처리해야하는 경우

   (초반에 시간에 쫓겨 api 설계를 대충하면 이런 사태가 벌어짐)

- IDE만 띄워놓고 콘솔에서 바로 확인하기 위해

- 테스트데이터 값을 바꿔가면서 결과를 확인해야하는 경우

- 웹 소스코드를 수정하지못하고 DB에서 트리거될때마다 API 요청을 통해 변경된 값으로 반환, DB에 업데이트하는 프로그램을 python으로 작성하는 경우 (왜 웹에서 요청못하게하냐고 쉬익)

 

import requests
import json

url = "로컬에서 띄운 API 주소"

# dummy data
payload = json.dumps(
{
  "data1": {
    "subdata1": 0,
    "subdata2": 0
  },
  "data2": {
    "subdata1": "id-01",
    "subdata2": 2021,
    "subdata3": 0,
    "subdata4": 2
  }
}
)

headers = {
  'Content-Type': 'application/json'
}

response = requests.request("POST", url, headers=headers, data=payload)

print(response.text)