mirror of
https://github.com/laramies/theHarvester.git
synced 2024-09-20 15:26:31 +08:00
22e1eb9bdb
* add pytest.ini file and fix test warning in qwantsearch * ignore unsed imports * Remove n45ht as the api is down * Remove modules that use google as google blocks you after one go * Remove missed code from when removing modules that use google * Remove missed code from when removing modules that use google * Add new vt-py dep and update pytest * WIP virustotal migration to api * Remove test that fails due to how the api returns not found entries * Remove entries from myparser re google * update fastapi and starlette * update version * Update dockerfile to use dev ubuntu version * remove spyse module * remove spyse dep * fix tests by removing un-needed tests and indentation fix * fix some mypy errors * remove spyse and remove color output and fix mypy error * flake8 fixes * bump version and set it a dev version
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
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()
|