require 'testbase'
require 'rexml/document'
require 'rex/parser/nmap_xml'
require 'spec'

xml = '
<?xml version="1.0" ?>
<?xml-stylesheet href="/usr/share/nmap/nmap.xsl" type="text/xsl"?>
<!-- Nmap 4.76 scan initiated Thu Nov 12 19:54:47 2009 as: nmap -p22,80 -A -oX nmap.xml -T5 192.168.0.1 -->
<nmaprun scanner="nmap" args="nmap -p22,80 -A -oX nmap.xml -T5 192.168.0.1" start="1258080887" startstr="Thu Nov 12 19:54:47 2009" version="4.76" xmloutputversion="1.02">
<scaninfo type="connect"  protocol="tcp" numservices="2" services="22,80" />
<verbose level="0" />
<debugging level="0" />
<host starttime="1258080887" endtime="1258080893"><status state="up" reason="syn-ack"/>
<address addr="192.168.0.1" addrtype="ipv4" />
<hostnames><hostname name="localhost" type="PTR" /></hostnames>
<ports><port protocol="tcp" portid="22"><state state="open" reason="syn-ack" reason_ttl="0"/><service name="ssh" extrainfo="protocol 2.0" servicefp="SF-Port22-TCP:V=4.76%I=7%D=11/12%Time=4AFCCA7D%P=i686-pc-linux-gnu%r(NULL,&#xa;SF:27,&quot;SSH-2\.0-OpenSSH_5\.1p1\x20Debian-5ubuntu1\r\n&quot;);" method="probed" conf="10" /></port>
<port protocol="tcp" portid="80"><state state="open" reason="syn-ack" reason_ttl="0"/><service name="http" product="Apache httpd" version="2.2.11" extrainfo="(Ubuntu) PHP/5.2.6-3ubuntu4.2 with Suhosin-Patch" method="probed" conf="10" /></port>
</ports>
<times srtt="119" rttvar="2882" to="50000" />
</host>
<runstats><finished time="1258080893" timestr="Thu Nov 12 19:54:53 2009"/><hosts up="1" down="0" total="1" />
<!-- Nmap done at Thu Nov 12 19:54:53 2009; 1 IP address (1 host up) scanned in 6.43 seconds -->
</runstats></nmaprun>
'

describe Rex::Parser::NmapXMLStreamParser do
	parser = Rex::Parser::NmapXMLStreamParser.new
	total_hosts = 0
	parser.on_found_host = Proc.new { |host|
		total_hosts += 1
		it "should yield a host" do
			host.should_not be_nil
		end
		it "should populate the host with proper keys" do
			host.should have_key("status")
			host.should have_key("ports")
			host.should have_key("addrs")
			host["ports"].should be_a(Array)
			host["addrs"].should be_a(Hash)
		end
		it "should find the address" do
			host["addrs"].keys.length.should == 1
			host["addrs"].should have_key("ipv4")
			host["addrs"]["ipv4"].should == "192.168.0.1"
		end
	}
	it "should have found exactly one host" do
		total_hosts.should == 1
	end
	REXML::Document.parse_stream(StringIO.new(xml), parser)
end