blob: 4fc70141b9162d17c468574b2364e342b2808239 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
import os
import pytest
from generalresearch.models.network.xml_parser import NmapXmlParser
@pytest.fixture
def nmap_xml_str(request) -> str:
fp = os.path.join(request.config.rootpath, "data/nmaprun1.xml")
with open(fp, "r") as f:
data = f.read()
return data
@pytest.fixture
def nmap_xml_str2(request) -> str:
fp = os.path.join(request.config.rootpath, "data/nmaprun2.xml")
with open(fp, "r") as f:
data = f.read()
return data
def test_nmap_xml_parser(nmap_xml_str, nmap_xml_str2):
p = NmapXmlParser()
n = p.parse_xml(nmap_xml_str)
assert n.tcp_open_ports == [61232]
assert len(n.trace.hops) == 18
n = p.parse_xml(nmap_xml_str2)
assert n.tcp_open_ports == [22, 80, 9929, 31337]
assert n.trace is None
|
