async def test_no_drupal():
respx.get("http://perdu.com/").mock(
return_value=httpx.Response(
200,
text="<html><head><title>Vous Etes Perdu ?</title></head><body><h1>Perdu sur l'Internet ?</h1> \
<h2>Pas de panique, on va vous aider</h2> \
<strong><pre> * <----- vous êtes ici</pre></strong></body></html>"
)
)
respx.get(url__regex=r"http://perdu.com/.*?").mock(return_value=httpx.Response(404))
persister = AsyncMock()
request = Request("http://perdu.com/")
request.path_id = 1
crawler = AsyncCrawler(Request("http://perdu.com/"))
options = {"timeout": 10, "level": 2, "tasks": 20}
module = ModuleDrupalEnum(crawler, persister, options, Event())
await module.attack(request)
assert not persister.add_payload.call_count
await crawler.close()
async def test_multi_versions_detected():
base_dir = os.path.dirname(sys.modules["wapitiCore"].__file__)
test_directory = os.path.join(base_dir, "..", "tests/data/drupal/")
maintainers_file = "MAINTAINERS.txt"
with open(path_join(test_directory, maintainers_file), errors="ignore") as maintainers:
data = maintainers.read()
# Response to tell that Drupal is used
respx.get("http://perdu.com/sites/").mock(return_value=httpx.Response(200))
# Response for maintainers.txt
respx.get("http://perdu.com/core/MAINTAINERS.txt").mock(return_value=httpx.Response(200, text=data))
respx.get(url__regex=r"http://perdu.com/.*?").mock(return_value=httpx.Response(404))
persister = AsyncMock()
request = Request("http://perdu.com/")
request.path_id = 1
crawler = AsyncCrawler(Request("http://perdu.com/"))
options = {"timeout": 10, "level": 2, "tasks": 20}
module = ModuleDrupalEnum(crawler, persister, options, Event())
await module.attack(request)
assert persister.add_payload.call_count == 1
assert persister.add_payload.call_args_list[0][1]["info"] == (
'{"name": "Drupal", "versions": ["8.0.0-beta4", "8.0.0-beta5", "8.0.0-beta6"], "categories": ["CMS Drupal"]}'
)
await crawler.close()
async def test_version_not_detected():
base_dir = os.path.dirname(sys.modules["wapitiCore"].__file__)
test_directory = os.path.join(base_dir, "..", "tests/data/drupal/")
changelog_edited = "CHANGELOG_EDITED.txt"
with open(path_join(test_directory, changelog_edited), errors="ignore") as changelog:
data = changelog.read()
# Response to tell that Drupal is used
respx.get("http://perdu.com/sites/").mock(return_value=httpx.Response(200))
# Response for edited changelog.txt
respx.get("http://perdu.com/CHANGELOG.txt").mock(return_value=httpx.Response(200, text=data))
respx.get(url__regex=r"http://perdu.com/.*?").mock(return_value=httpx.Response(404))
persister = AsyncMock()
request = Request("http://perdu.com/")
request.path_id = 1
crawler = AsyncCrawler(Request("http://perdu.com/"))
options = {"timeout": 10, "level": 2, "tasks": 20}
module = ModuleDrupalEnum(crawler, persister, options, Event())
await module.attack(request)
assert persister.add_payload.call_count == 1
assert persister.add_payload.call_args_list[0][1]["info"] == (
'{"name": "Drupal", "versions": [""], "categories": ["CMS Drupal"]}'
)
await crawler.close()