-
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #308 from AmericaSCORESBayArea/sandbox
Sandbox to Production [Dec 5, 2024]
- Loading branch information
Showing
6 changed files
with
941 additions
and
26 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,95 @@ | ||
| import asyncio | ||
| import aiohttp | ||
| import json | ||
| import time | ||
|
|
||
| # ============================================================== | ||
| # HOW TO USE THIS SCRIPT FOR TESTING API ENDPOINTS CONCURRENTLY | ||
| # ============================================================== | ||
| # | ||
| # 1. **Setting Up the API URL**: | ||
| # - Replace the `url` variable with the actual endpoint URL you want to test. | ||
| # - Example: | ||
| # url = "https://api.example.com/enrollments" | ||
| # | ||
| # 2. **Headers**: | ||
| # - Ensure the `headers` dictionary contains any necessary authentication/authorization headers required by the API. | ||
| # - Common headers include: | ||
| # - `'client_id'`: Your client ID or application key. | ||
| # - `'client_secret'`: Your client secret or API key. | ||
| # - `'Content-Type'`: Typically set to `'application/json'` if sending JSON data. | ||
| # | ||
| # Example: | ||
| # headers = { | ||
| # 'client_id': 'your-client-id', | ||
| # 'client_secret': 'your-client-secret', | ||
| # 'Content-Type': 'application/json' | ||
| # } | ||
| # | ||
| # 3. **Request Payload**: | ||
| # - Modify the `payload` variable to match the structure and data your API expects. | ||
| # - For example, if testing an enrollment API, ensure fields like `TeamSeasonId`, `StudentId`, `StartDate`, and `EndDate` are populated with valid test data. | ||
| # | ||
| # Example: | ||
| # payload = json.dumps({ | ||
| # "TeamSeasonId": "12345", | ||
| # "StudentId": "67890", | ||
| # "StartDate": "2024-01-01", | ||
| # "EndDate": "2024-12-31" | ||
| # }) | ||
| # | ||
| # 4. **Number of Requests**: | ||
| # - The script sends 20 concurrent requests to the API (default). | ||
| # - If you need to change the number of requests to test different loads, simply adjust the `number_of_requests` variable. | ||
| # | ||
| # Example: | ||
| # number_of_requests = 50 # Send 50 concurrent requests | ||
| # | ||
| # 5. **Running the Script**: | ||
| # - Run the script to test the API. It will send requests concurrently and print out: | ||
| # - The HTTP status code for each request. | ||
| # - The response time for each request. | ||
| # - The response body returned by the API. | ||
| # | ||
| # ================================================ | ||
|
|
||
| number_of_requests = 20 | ||
| url = "https://production-salesforce-data-api.us-e2.cloudhub.io/api/enrollments" | ||
| headers = { | ||
| 'client_id': '', | ||
| 'client_secret': '', | ||
| 'Content-Type': 'application/json' | ||
| } | ||
| payload = json.dumps({ | ||
| "TeamSeasonId": "", | ||
| "StudentId": "", | ||
| "StartDate": "", | ||
| "EndDate": "" | ||
| }) | ||
|
|
||
| async def send_post_request(session, url, headers, payload, idx): | ||
| start_time = time.time() | ||
| print(f"Starting request {idx + 1} to {url}") | ||
| async with session.post(url, headers=headers, data=payload) as response: | ||
| elapsed_time = time.time() - start_time | ||
| print(f"Completed request {idx + 1} to {url}") | ||
| return await response.text(), response.status, elapsed_time | ||
|
|
||
|
|
||
| async def main(): | ||
| async with aiohttp.ClientSession() as session: | ||
| tasks = [send_post_request(session, url, headers, payload, idx) for idx in range(number_of_requests)] | ||
| responses = await asyncio.gather(*tasks) | ||
|
|
||
| for idx, (response_text, status_code, elapsed_time) in enumerate(responses): | ||
| print(f"Response from Request {idx + 1}:") | ||
| print(f"Status Code: {status_code}") | ||
| print(f"Time Taken: {elapsed_time:.2f} seconds") | ||
| print(response_text) | ||
|
|
||
| start_total_time = time.time() | ||
|
|
||
| asyncio.run(main()) | ||
|
|
||
| end_total_time = time.time() | ||
| print(f"\nTotal time for all requests to complete: {end_total_time - start_total_time:.2f} seconds") |
Oops, something went wrong.