Fix proxy in setup that was missed and disable github test

2026-09-08 10:47:40 +02:00 · 2020-02-11 16:24:38 +00:00
parent 981448c9e7
commit fe3d6a332e
5 changed files with 47 additions and 46 deletions
@@ -1,117 +0,0 @@
-from theHarvester.discovery import githubcode
-from theHarvester.discovery.githubcode import RetryResult, ErrorResult, SuccessResult
-from theHarvester.discovery.constants import MissingKey
-from theHarvester.lib.core import Core
-from mock import MagicMock
-from requests import Response
-import pytest
-
-pytestmark = 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):
-        with pytest.raises(MissingKey):
-            Core.github_key = MagicMock(return_value=None)
-            githubcode.SearchGithubCode(word="test", limit=500)
-
-    async def test_fragments_from_response(self):
-        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)
-        assert test_result == ["test1", "test2"]
-
-    async def test_invalid_fragments_from_response(self):
-        Core.github_key = MagicMock(return_value="lol")
-        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
-        test_result = test_class_instance.fragments_from_response(self.MalformedResponse.response)
-        assert test_result == []
-
-    async def test_handle_response_ok(self):
-        Core.github_key = MagicMock(return_value="lol")
-        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
-        test_result = test_class_instance.handle_response(self.OkResponse.response)
-        assert isinstance(test_result, SuccessResult)
-
-    async def test_handle_response_retry(self):
-        Core.github_key = MagicMock(return_value="lol")
-        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
-        test_result = test_class_instance.handle_response(self.RetryResponse.response)
-        assert isinstance(test_result, RetryResult)
-
-    async def test_handle_response_fail(self):
-        Core.github_key = MagicMock(return_value="lol")
-        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
-        test_result = test_class_instance.handle_response(self.FailureResponse.response)
-        assert isinstance(test_result, ErrorResult)
-
-    async def test_next_page(self):
-        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 == test_class_instance.next_page_or_end(test_result))
-
-    async def test_last_page(self):
-        Core.github_key = MagicMock(return_value="lol")
-        test_class_instance = githubcode.SearchGithubCode(word="test", limit=500)
-        test_result = githubcode.SuccessResult(list(), None, None)
-        assert(None is test_class_instance.next_page_or_end(test_result))
-
-    if __name__ == '__main__':
-        pytest.main()