mirror of
https://github.com/aljazceru/recon-pipeline.git
synced 2025-12-20 15:54:25 +01:00
4d1aef2d34
* scans without installed requirements dont tab-complete * added meets_requirements functions to classes * updated tests * added check for None case
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
import pytest
|
|
|
|
from pipeline.recon.helpers import get_ip_address_version, get_scans, is_ip_address
|
|
from pipeline.recon import AmassScan, MasscanScan, FullScan, HTBScan, SearchsploitScan, ThreadedNmapScan
|
|
from pipeline.recon.web import GobusterScan, SubjackScan, TKOSubsScan, AquatoneScan, WaybackurlsScan, WebanalyzeScan
|
|
|
|
|
|
def test_get_scans():
|
|
|
|
scan_names = [
|
|
AmassScan,
|
|
GobusterScan,
|
|
MasscanScan,
|
|
SubjackScan,
|
|
TKOSubsScan,
|
|
AquatoneScan,
|
|
FullScan,
|
|
HTBScan,
|
|
SearchsploitScan,
|
|
ThreadedNmapScan,
|
|
WebanalyzeScan,
|
|
WaybackurlsScan,
|
|
]
|
|
|
|
scans = get_scans()
|
|
|
|
for scan in scan_names:
|
|
if hasattr(scan, "meets_requirements") and scan.meets_requirements():
|
|
assert scan.__name__ in scans.keys()
|
|
else:
|
|
assert scan not in scans.keys()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_input, expected",
|
|
[("127.0.0.1", True), ("::1", True), ("abcd", False), ("", False), (-1, False), (1.0, False)],
|
|
)
|
|
def test_is_ip_address(test_input, expected):
|
|
assert is_ip_address(test_input) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_input, expected", [("127.0.0.1", "4"), ("::1", "6"), ("abcd", None), ("", None), (-1, None), (1.0, None)]
|
|
)
|
|
def test_get_ip_address_version(test_input, expected):
|
|
assert get_ip_address_version(test_input) == expected
|