diff --git a/tests/discovery/__init__.py b/tests/discovery/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/tests/discovery/test_githubcode.py b/tests/discovery/test_githubcode.py new file mode 100644 index 00000000..275b9100 --- /dev/null +++ b/tests/discovery/test_githubcode.py @@ -0,0 +1,44 @@ +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 + + +class TestSearchGithubCode: + + def test_missing_key(self): + with pytest.raises(MissingKey): + githubcode.SearchGithubCode(word="test", limit=500) + + + def test_fragments_from_response(self): + Core.github_key = MagicMock(return_value="lol") + test_class_instance = githubcode.SearchGithubCode(word="test", limit=500) + response = Response() + json = { + "items": [ + { + "text_matches": [ + { + "fragment": "test1" + } + ] + }, + { + "text_matches": [ + { + "fragment": "test2" + } + ] + } + ] + } + response.json = MagicMock(return_value=json) + test_result = test_class_instance.fragments_from_response(response) + assert test_result == ["test1", "test2"] + + if __name__ == '__main__': + pytest.main() +