diff options
Diffstat (limited to 'tests/models/network/rdns.py')
| -rw-r--r-- | tests/models/network/rdns.py | 78 |
1 files changed, 33 insertions, 45 deletions
diff --git a/tests/models/network/rdns.py b/tests/models/network/rdns.py index 64e8351..e56c494 100644 --- a/tests/models/network/rdns.py +++ b/tests/models/network/rdns.py @@ -1,45 +1,33 @@ -# from generalresearch.models.network.rdns import run_rdns -# import faker -# -# fake = faker.Faker() -# -# -# def test_dig_rdns(): -# # Actually runs dig -x. Idk how stable this is -# ip = "45.33.32.156" -# rdns_result = run_rdns(ip) -# assert rdns_result.primary_hostname == "scanme.nmap.org" -# assert rdns_result.primary_org == "nmap" -# -# ip = "65.19.129.53" -# rdns_result = run_rdns(ip) -# assert rdns_result.primary_hostname == "in1-smtp.grlengine.com" -# assert rdns_result.primary_org == "grlengine" -# -# ip = fake.ipv6() -# rdns_result = run_rdns(ip) -# assert rdns_result.primary_hostname is None -# assert rdns_result.primary_org is None -# print(rdns_result.model_dump_postgres()) - - -# -# -# def test_run_dig(toolrun_manager): -# reverse_dns_run = run_dig(ip="65.19.129.53") -# -# toolrun_manager.create_rdns_run(reverse_dns_run) -# -# run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id) -# -# assert reverse_dns_run == run_out -# -# -# def test_run_dig_empty(toolrun_manager): -# reverse_dns_run = run_dig(ip=fake.ipv6()) -# -# toolrun_manager.create_rdns_run(reverse_dns_run) -# -# run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id) -# -# assert reverse_dns_run == run_out
\ No newline at end of file +from generalresearch.models.network.rdns.execute import execute_rdns +import faker + +from generalresearch.models.network.tool_run import ToolName, ToolClass + +fake = faker.Faker() + + +def test_execute_rdns_grl(toolrun_manager): + ip = "65.19.129.53" + run = execute_rdns(ip=ip) + assert run.tool_name == ToolName.DIG + assert run.tool_class == ToolClass.RDNS + assert run.ip == ip + result = run.parsed + assert result.primary_hostname == "in1-smtp.grlengine.com" + assert result.primary_domain == "grlengine.com" + assert result.hostname_count == 1 + + toolrun_manager.create_rdns_run(run) + + +def test_execute_rdns_none(toolrun_manager): + ip = fake.ipv6() + run = execute_rdns(ip) + result = run.parsed + + assert result.primary_hostname is None + assert result.primary_domain is None + assert result.hostname_count == 0 + assert result.hostnames == [] + + toolrun_manager.create_rdns_run(run) |