import json

from tests import constants


RATELIMIT_REACHED_RESPONSE_BODY = json.dumps({"exception": True, "code": 9, "detail": "Test Exception"})


ONE_POST_SUCCESS = [
    {"method": "POST", "url": constants.URL_PATTERN, "body": "{}", "headers": {"abc": "123"}, "status": 200},
]


THREE_POST_SUCCESS = [
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": "{}",
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 123),
        },
        "status": 200,
    },
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": "{}",
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 456),
        },
        "status": 200,
    },
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": "{}",
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 789),
        },
        "status": 200,
    },
]

THREE_POST_RATELIMIT_REACHED = [
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": RATELIMIT_REACHED_RESPONSE_BODY,
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 123),
        },
        "status": 429,
    },
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": RATELIMIT_REACHED_RESPONSE_BODY,
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 456),
        },
        "status": 429,
    },
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": RATELIMIT_REACHED_RESPONSE_BODY,
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 789),
        },
        "status": 429,
    },
]

TWO_POST_RATELIMIT_REACHED_ONE_SUCCESS = [
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": RATELIMIT_REACHED_RESPONSE_BODY,
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 123),
        },
        "status": 429,
    },
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": RATELIMIT_REACHED_RESPONSE_BODY,
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 456),
        },
        "status": 429,
    },
    {
        "method": "POST",
        "url": constants.URL_PATTERN,
        "body": "{}",
        "headers": {
            constants.RATELIMIT_LIMIT_HEADER: "1000",
            constants.RATELIMIT_REMAINING_HEADER: "0",
            constants.RATELIMIT_RESET_HEADER: str(constants.TIMESTAMP + 789),
        },
        "status": 200,
    },
]