Skip to content

Conversation

@gitstart-app
Copy link

@gitstart-app gitstart-app bot commented Jan 30, 2025

What is in the PR?
Add implementation to secure coverage and crashmanager API endpoints with rate-limiting.

This uses Throttling from Django rest framework and Redis cache db to store user request rate data.

Rate Limit Testing Script

A simple bash script to test API rate limiting. The script sends consecutive requests to an endpoint until it receives a rate limit response (HTTP 429) or reaches the maximum request count.

Setup

1. Create the script file:

touch test_rate_limit.sh

2. Copy the script content into test_rate_limit.sh

#!/bin/bash

# Configuration
ENDPOINT="http://localhost:8000/crashmanager/rest/crashes/"  # Adjust URL as needed

# Set default MAX_REQUESTS if not provided as argument
MAX_REQUESTS=${1:-110}  # Use first argument if provided, otherwise default to 110

# Set token from second argument or use hardcoded default
HARDCODED_TOKEN="your-default-token-here"  # Replace with your actual default token
TOKEN=${2:-$HARDCODED_TOKEN}

echo "Starting rate limit test..."
echo "Endpoint: $ENDPOINT"
echo "Maximum requests: $MAX_REQUESTS"

for i in $(seq 1 $MAX_REQUESTS); do
    # Make request and capture both HTTP status code and response body
    response=$(curl -s -w "\n%{http_code}" \
        -H "Authorization: Token $TOKEN" \
        -H "Content-Type: application/json" \
        "$ENDPOINT")

    # Extract status code from last line
    status_code=$(echo "$response" | tail -n1)
    # Extract response body (everything except last line)
    body=$(echo "$response" | sed \$d)

    echo "Request $i - Status Code: $status_code"

    # Check if we hit the rate limit
    if [ "$status_code" -eq 429 ]; then
        echo "Rate limit hit after $i requests!"
        echo "Error response:"
        echo "$body"
        exit 0
    fi
done

echo "Completed $MAX_REQUESTS requests without hitting rate limit." 

3. Make the script executable:

chmod +x test_rate_limit.sh

Usage

Basic Run (110 requests)

./test_rate_limit.sh

Auth setup:

You can update HARDCODED_TOKEN variable inside the script, or pass it as an argument when running the script.

To run it as part of the script command.

./test_rate_limit.sh 110 <token>

Custom Request Count

./test_rate_limit.sh 150 # Will send up to 150 requests

What to expect:

1. It will show the endpoint and maximum requests

2. For each request, you'll see the status code

3. If rate limit is hit (HTTP 429):

  • Shows which request triggered the limit

  • Displays the server's error response

4. If no rate limit is hit, shows completion message


SPEC DOC LINK:

https://docs.google.com/document/d/1l9r-t6mcwirwnxkjxdZ2HF3QWPEzanZ46Azi7F0B2nQ/edit?tab=t.0

DEMO VIDEO:

https://www.loom.com/share/d8cc5c2c5c5a4dc58df78de843cd07d5?sid=aa540f7c-9c38-4523-9085-301bdec5871e

@gitstart-app
Copy link
Author

gitstart-app bot commented Jan 30, 2025

This PR is estimated to cost 30 credits.
🟡 By merging this PR you agree to this estimate. If you disagree, click here.

@community-tc-integration
Copy link

No Taskcluster jobs started for this pull request

The allowPullRequests configuration for this repository (in .taskcluster.yml on the default branch) does not allow starting tasks for this pull request.

@gitstart-app
Copy link
Author

gitstart-app bot commented Feb 21, 2025

This PR is estimated to cost 30 credits.
🟡 By merging this PR you agree to this estimate. If you disagree, click here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants