mirror of
https://github.com/aljazceru/recon-pipeline.git
synced 2025-12-18 23:04:21 +01:00
7a24d85db4
* recon.targets tests added * restructured tests logically * fixed yaml error * fixed job names * recon.__init__ tests added * recon.config tests added * recon.amass.ParseAmassScan tests added * fixed test destined to fail on CI pipeline * testing amass partially complete * Changed the dir layout (#6) and fixed paths (#8) this commit closes #6 and #8 updated existing tests to utilize new paths * tests of current codebase complete * added is_kali check to searchsploit test * added test_web action to pipeline
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import pickle
|
|
from pathlib import Path
|
|
from recon.masscan import ParseMasscanOutput, MasscanScan
|
|
|
|
tfp = "../data/bitdiscovery"
|
|
tf = Path(tfp).stem
|
|
el = "../data/blacklist"
|
|
rd = "../data/recon-results"
|
|
|
|
test_dict = {
|
|
"104.20.60.51": {"tcp": {"8443", "443"}},
|
|
"104.20.61.51": {"tcp": {"8080", "80", "443"}},
|
|
"13.225.54.100": {"tcp": {"443"}},
|
|
"13.225.54.22": {"tcp": {"80"}},
|
|
"52.53.92.161": {"tcp": {"443", "80"}},
|
|
"52.9.23.177": {"tcp": {"80"}},
|
|
}
|
|
|
|
|
|
def test_massscan_output_location(tmp_path):
|
|
asc = MasscanScan(target_file=tf, exempt_list=el, results_dir=str(tmp_path), top_ports=100)
|
|
|
|
assert asc.output().path == str(Path(tmp_path) / "masscan-results" / "masscan.json")
|
|
|
|
|
|
def test_parsemassscan_output_location(tmp_path):
|
|
pmo = ParseMasscanOutput(target_file=tf, exempt_list=el, results_dir=str(tmp_path), top_ports=100)
|
|
|
|
assert pmo.output().path == str(Path(tmp_path) / "masscan-results" / "masscan.parsed.pickle")
|
|
|
|
|
|
def test_parsemassscan_output_dictionary(tmp_path):
|
|
ip_dict = pickle.load(
|
|
(Path(__file__).parent.parent / "data" / "recon-results" / "masscan-results" / "masscan.parsed.pickle").open(
|
|
"rb"
|
|
)
|
|
)
|
|
|
|
for ip, proto_dict in test_dict.items():
|
|
for proto, ports in proto_dict.items():
|
|
assert not ip_dict.get(ip).get(proto).difference(ports)
|