Updated certspotter search and changed github code to start with test.

2026-10-01 14:04:55 +02:00 · 2020-02-13 18:45:52 -05:00
parent 42f0142aa9
commit d90f940668
2 changed files with 2 additions and 4 deletions
@@ -0,0 +1,99 @@
+from theHarvester.discovery import githubcode
+from theHarvester.discovery.constants import MissingKey
+from theHarvester.lib.core import Core
+from unittest.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.json())
+        print('test_result: ', test_result)
+        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 = await test_class_instance.fragments_from_response(self.MalformedResponse.response.json())
+        assert test_result == []
+
+    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 == await 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 await test_class_instance.next_page_or_end(test_result))
+
+    if __name__ == '__main__':
+        pytest.main()