rosdiff / test_owner.py
Chandrakiran-08
Deploy to HF Spaces
8501103
Raw History Blame Contribute Delete
568 Bytes
import sys
import os
import requests
from fastapi.testclient import TestClient
# Make sure we load the env file
from dotenv import load_dotenv
load_dotenv('.env')
# Add to path so we can import api
sys.path.insert(0, os.getcwd())
from api.main import app
client = TestClient(app)
print("Testing with owner key: " + os.environ["OWNER_API_KEY"])
response = client.get("/me/usage", headers={"X-API-Key": os.environ["OWNER_API_KEY"]})
print("\nResponse Status:", response.status_code)
import json
print("\nResponse Body:")
print(json.dumps(response.json(), indent=2))