#!/usr/bin/perl
use strict;
use blib;
use File::Spec;
use Cwd;
use Test::More tests => 15;
use Nmap::Parser;
use constant TEST_FILE => 'nmap_results.xml';
use constant HOST1     => '127.0.0.1';
use constant HOST2     => '127.0.0.2';
use constant HOST3     => '127.0.0.3';
use constant HOST4     => '127.0.0.4';
use constant TOTAL_ADDRS => 4;
use vars qw($FH $TOTAL @UP_HOSTS @DOWN_HOSTS);
$FH = File::Spec->catfile( cwd(), 't', TEST_FILE );
$FH = File::Spec->catfile( cwd(), TEST_FILE ) unless ( -e $FH );
my $np = new Nmap::Parser;
isa_ok( $np, 'Nmap::Parser' );
ok( $np->callback( \&my_callback ), 'Registering callback function' );
is( $np->callback(), 0, 'Unregistering callback function' );
ok( $np->callback( \&my_callback ), 're-registering callback function' );
$TOTAL = 0;
$np->parsefile($FH);
is( $TOTAL, TOTAL_ADDRS, 'Making sure all hosts were parsed in callback' );
ok( eq_set( [@UP_HOSTS], [ HOST1, HOST3, HOST4 ] ),
    'Testing for correct UP hosts' );
ok( eq_set( [@DOWN_HOSTS], [HOST2] ), 'Testing for correct DOWN hosts' );
for my $host ( HOST1, HOST2, HOST3, HOST4 ) {
    is( $np->get_host($host), undef,
        'Making sure ' . $host . ' does not exists' );
}
sub my_callback {
    my $host = shift;
    my $addr = $host->addr;
    if ( $addr =~ /^127\.0\.0/ ) {
        $TOTAL++;
    }
    isa_ok( $host, 'Nmap::Parser::Host', $host->addr );
    if    ( $host->status eq 'up' )   { push @UP_HOSTS,   $addr; }
    elsif ( $host->status eq 'down' ) { push @DOWN_HOSTS, $addr; }
}