gradio-fastapi-statics_a / test_main.py
kenken999's picture
test
9d993b2
import json
from fastapi.testclient import TestClient
from mysite.asgi import app
import json
from fastapi.testclient import TestClient
#from main import app
client = TestClient(app)
def load_saved_request(filename: str):
with open(filename, 'r', encoding='utf-8') as f:
data = json.load(f)
headers = data['headers']
body = data['body'].encode('utf-8') # Bodyをバイナリ形式にエンコード
return headers, body
def test_saved_request():
headers, body = load_saved_request('request.json')
# headersの処理
processed_headers = {key.lower(): value for key, value in headers.items()}
response = client.post("/webhook", data=body, headers=processed_headers)
assert response.status_code == 200, f"Unexpected status code: {response.status_code}, {response.json()}"
assert response.json() == {"message": "Webhook received"}
if __name__ == "__main__":
test_saved_request()