import pytest
import pytest_mock
import requests
import responses
from geocode.wikidata import APIResponseError, QueryError, api_call, wdqs

max_tries = 5


@responses.activate
def test_api_call_retries_on_failure(mocker: pytest_mock.plugin.MockerFixture) -> None:
    """Test retry for API calls."""
    # Patch 'time.sleep' to instantly return, effectively skipping the sleep
    mocked_sleep = mocker.patch("time.sleep", return_value=None)

    mock_send_mail = mocker.patch("geocode.mail.send_to_admin")

    responses.add(
        responses.GET,
        "https://www.wikidata.org/w/api.php",
        body="bad request",
        status=400,
    )
    with pytest.raises(APIResponseError):
        api_call({"action": "wbgetentities", "ids": "Q42"})
    assert len(responses.calls) == max_tries

    assert mocked_sleep.call_count == max_tries - 1

    mock_send_mail.assert_called()

    send_mail_call = mock_send_mail.call_args_list[0]
    assert send_mail_call[0] == (
        "Geocode error",
        "Error making Wikidata API call\n\nbad request",
    )


def test_api_call_retries_on_connection_error(
    mocker: pytest_mock.plugin.MockerFixture,
) -> None:
    """Test retry for API calls on connection error."""
    # Patch 'time.sleep' to instantly return, effectively skipping the sleep
    mocked_sleep = mocker.patch("time.sleep", return_value=None)

    # Patch 'requests.get' to raise a ConnectionError
    mocker.patch("requests.get", side_effect=requests.exceptions.ConnectionError)
    mocker.patch("geocode.mail.send_to_admin")

    with pytest.raises(requests.exceptions.ConnectionError):
        api_call({"action": "wbgetentities", "ids": "Q42"})

    assert mocked_sleep.call_count == max_tries - 1


def test_wdqs_retry(mocker: pytest_mock.plugin.MockerFixture) -> None:
    """Test retry for WDQS API calls."""
    # Patch 'time.sleep' to instantly return, effectively skipping the sleep
    mocked_sleep = mocker.patch("time.sleep", return_value=None)

    responses.add(
        responses.POST,
        "https://query.wikidata.org/bigdata/namespace/wdq/sparql",
        body="bad request",
        status=400,
    )

    with pytest.raises(QueryError):
        wdqs("test query")

    max_tries = 5
    assert mocked_sleep.call_count == max_tries - 1