use strict;
use warnings;

use Test::More tests => 3;

use NetPacket::TCP;
use NetPacket::UDP;

my $ip = { 
	src_ip => '127.0.0.1',
	dest_ip => '192.168.0.1',
};

bless $ip, 'NetPacket::IP';

my $tcp = {
	dest_port => 22,
	src_port => 13,
	seqnum => 1,
	acknum => 2,
	winsize => 32,
	urg => 0,
	hlen => 5,
	data => "DEADBEEF",
};

bless $tcp, 'NetPacket::TCP';

is NetPacket::TCP::checksum( $tcp, $ip ) => 25303;

$tcp->{data} = "DEADBEEF\x01";
my $odd_checksum = NetPacket::TCP::checksum( $tcp, $ip );

is $odd_checksum => 25046, 'TCP padding done correctly';

my $udp = {
	src_port => 13,
	dest_port => 14,
	len => 15,
	data => "foo",
};

bless $udp, 'NetPacket::UDP';

is NetPacket::UDP::checksum( $udp, $ip ) => 49177, 'UDP padding';