class IcmpTest(unittest.TestCase): def setUp(self): self.icmp_stack = IcmpStack() self.ip_stack = MockStack() self.icmp_stack.set_ip_stack(self.ip_stack) return def tearDown(self): return def testHandleSouthInco1(self): ''' test handle icmp ping ''' req = dpkt.icmp.ICMP.Echo() req.data = "hello" req.id = 1 req.seq = 1 req_icmp = dpkt.icmp.ICMP() req_icmp.type = dpkt.icmp.ICMP_ECHO req_icmp.code = 0 req_icmp.data = req self.icmp_stack.handle_south_inco(req_icmp) rsp = self.ip_stack.pkg self.assertEqual(rsp.type, dpkt.icmp.ICMP_ECHOREPLY) return
def setUp(self): self.icmp_stack = IcmpStack() self.ip_stack = MockStack() self.icmp_stack.set_ip_stack(self.ip_stack) return