from unittest.mock import MagicMock

import pytest
from _pytest.mark.structures import MarkDecorator
from requests import Response

from theHarvester.discovery import githubcode
from theHarvester.discovery.constants import MissingKey
from theHarvester.lib.core import Core

pytestmark: MarkDecorator = pytest.mark.asyncio


class TestSearchGithubCode:
    class OkResponse:
        response = Response()
        json = {
            "items": [
                {"text_matches": [{"fragment": "test1"}]},
                {"text_matches": [{"fragment": "test2"}]},
            ]
        }
        response.status_code = 200
        response.json = MagicMock(return_value=json)

    class FailureResponse:
        response = Response()
        response.json = MagicMock(return_value={})
        response.status_code = 401

    class RetryResponse:
        response = Response()
        response.json = MagicMock(return_value={})
        response.status_code = 403

    class MalformedResponse:
        response = Response()
        json = {
            "items": [
                {"fail": True},
                {"text_matches": []},
                {"text_matches": [{"weird": "result"}]},
            ]
        }
        response.json = MagicMock(return_value=json)
        response.status_code = 200

    async def test_missing_key(self) -> None:
        with pytest.raises(MissingKey):
            Core.github_key = MagicMock(return_value=None)
            githubcode.SearchGithubCode(word="test", limit=500)

    async def test_fragments_from_response(self) -> None:
        Core.github_key = MagicMock(return_value="lol")
        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
        test_result = await test_class_instance.fragments_from_response(
            self.OkResponse.response.json()
        )
        print("test_result: ", test_result)
        assert test_result == ["test1", "test2"]

    async def test_invalid_fragments_from_response(self) -> None:
        Core.github_key = MagicMock(return_value="lol")
        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
        test_result = await test_class_instance.fragments_from_response(
            self.MalformedResponse.response.json()
        )
        assert test_result == []

    async def test_next_page(self) -> None:
        Core.github_key = MagicMock(return_value="lol")
        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
        test_result = githubcode.SuccessResult(list(), next_page=2, last_page=4)
        assert 2 == await test_class_instance.next_page_or_end(test_result)

    async def test_last_page(self) -> None:
        Core.github_key = MagicMock(return_value="lol")
        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
        test_result = githubcode.SuccessResult(list(), None, None)
        assert await test_class_instance.next_page_or_end(test_result) is None
    if __name__ == "__main__":
        pytest.main()