Test against eom:governor

rate_test.py

# rate_test.py
# usage: python test_rate.py <count> <url> <auth_token>
# reqs: pip install grequests

from __future__ import print_function
import grequests
import itertools
import requests
import sys


if __name__ == '__main__':

    if len(sys.argv) < 4:
        print('usage: {0} count url auth_token'.format(sys.argv[0]))
        quit()

    count, url, auth_token = sys.argv[1:]
    count = int(count)

    headers = {'x-auth-token': auth_token}
    responses = []
    print(requests.get(url, headers=headers))
    responses = [r.status_code for r in grequests.map(
        (grequests.get(u, headers=headers)
         for u in itertools.repeat(url, count))
    )]

    codes = {
        '429': sum([1 for r in responses if r == 429]),
        '200': sum([1 for r in responses if r == 200]),
        '204': sum([1 for r in responses if r == 204]),
        'other': sum([1 for r in responses if r not in [200, 204, 429]]) 
    }

    print(codes)