from IfInfo import IfInfo
from Slaacctl import ShowInterface
import subprocess
import sys
from scapy.all import *
import unittest


rtadv_if = IfInfo(sys.argv[1])
slaac_if = IfInfo(sys.argv[2])
sock = sys.argv[3]

eth = Ether(src=rtadv_if.mac)
ip = IPv6(dst="ff02::1", src=rtadv_if.ll)
ra = ICMPv6ND_RA(prf='Medium (default)', routerlifetime=1800)
pref = ICMPv6NDOptPrefixInfo(prefixlen=64, prefix='2001:db8:1::',
    validlifetime=2592000, preferredlifetime=604800, L=1, A=1)
mtu = ICMPv6NDOptMTU(mtu=1500)
rdnss = ICMPv6NDOptRDNSS(lifetime=86400, dns=['2001:db8:53::a',
    '2001:db8:53::b'])
dnssl = ICMPv6NDOptDNSSL(lifetime=86400, searchlist=['invalid', 'home.invalid'])

p = eth/ip/ra/pref/mtu/rdnss/dnssl

sendp(p, iface=rtadv_if.ifname, verbose=0)

slaac_show_interface = ShowInterface(slaac_if.ifname, sock, debug=0)


class TestRouterAdvertisementParsing(unittest.TestCase):
	def test_number_ras(self):
		self.assertEqual(len(slaac_show_interface.RAs), 1)

	def test_number_addr_proposals(self):
		self.assertEqual(len(slaac_show_interface.addr_proposals), 2)

	def test_number_def_router_proposals(self):
		self.assertEqual(len(
		    slaac_show_interface.def_router_proposals), 1)

if __name__ == '__main__':
	suite = unittest.TestLoader().loadTestsFromTestCase(
	    TestRouterAdvertisementParsing)
	if not unittest.TextTestRunner(verbosity=2).run(suite).wasSuccessful():
		print slaac_show_interface
		sys.exit(1)