[python] Flask를 이용한 API 테스트 자동화 방법

Flask는 Python으로 웹 애플리케이션을 개발할 수 있도록 하는 경량화된 웹 프레임워크입니다. 이번 블로그에서는 Flask를 이용하여 API 테스트를 자동화하는 방법에 대해 알아보겠습니다.

1. Flask 개발 및 API 구현

먼저 Flask를 사용하여 간단한 API를 개발합니다. 아래는 “Hello, World!”를 반환하는 간단한 예시입니다.

from flask import Flask

app = Flask(__name__)

@app.route('/')
def hello_world():
    return 'Hello, World!'

if __name__ == '__main__':
    app.run()

2. API 테스트 자동화를 위한 라이브러리 선택

API를 테스트하기 위해 requests 라이브러리를 사용할 것을 권장합니다.

pip install requests

3. API 테스트 자동화 구현

이제 API에 대한 간단한 테스트를 자동으로 실행하는 코드를 작성해봅시다.

import requests

def test_api():
    url = 'http://localhost:5000'
    response = requests.get(url)
    
    assert response.status_code == 200
    assert response.text == 'Hello, World!'

4. 테스트 실행 및 결과 확인

API 테스트를 실행하고 결과를 확인합니다.

test_api()

요약

Flask를 이용한 API 테스트 자동화는 간단하게 구현할 수 있습니다. requests 라이브러리를 사용하여 API를 테스트하는 것은 간편하고 효율적입니다.

위의 예시를 통해 Flask를 이용한 API 테스트 자동화의 기본적인 방법을 익힐 수 있었습니다.

참고 문헌: Flask 공식 문서, Requests: HTTP for Humans