def test_to_jsont(self, SwitchCls): # Arrange json_dict = { "node1": { "hid": 1, "interfaces": [{ "hw_addr": "11:22:33:44:55:66", "ips": ["127.0.0.1"], "__type__": "sts.entities.hosts.HostInterface", "name": "eth0" }], "__type__": "sts.entities.hosts.Host", "name": "h1" }, "port2": { "hw_addr": "00:00:00:00:00:00", "curr": 0, "name": "", "supported": 0, "__type__": "pox.openflow.libopenflow_01.ofp_phy_port", "state": 0, "advertised": 0, "peer": 0, "config": 0, "port_no": 1 }, "node2": 1, "__type__": "sts.entities.sts_entities.AccessLink", "port1": { "hw_addr": "11:22:33:44:55:66", "ips": ["127.0.0.1"], "__type__": "sts.entities.hosts.HostInterface", "name": "eth0" } } sw1 = SwitchCls() sw1.dpid = 1 sw1.to_json.return_value = 1 # It's really hard to mock this, because of using assert_type p1 = ofp_phy_port(port_no=1) hw_addr_str = "11:22:33:44:55:66" hw_addr = EthAddr(hw_addr_str) ip_str = "127.0.0.1" ip = IPAddr(ip_str) ifname = "eth0" interface = HostInterface(hw_addr, ip, name=ifname) hname = "h1" hid = 1 host = Host(interface, name=hname, hid=hid) # Act link = AccessLink.from_json(json_dict) # Assert self.assertEquals(link.host.to_json(), host.to_json()) self.assertEquals(link.interface.to_json(), interface.to_json()) self.assertEquals(link.switch, 1) self.assertEquals(link.switch_port.to_json(), p1.to_json())
def test_invalid_arp(self): """Receive a ARP packet that isn't desinated to it and ensure there is no reply""" # Arrange iface1 = HostInterface(EthAddr( self.H2_I1_ETH), [IPAddr(self.H2_I1_IP1), IPAddr(self.H2_I1_IP2)]) iface2 = HostInterface(EthAddr(self.H2_I2_ETH), [IPAddr(self.H2_I2_IP)]) interfaces = [iface1, iface2] h = Host(interfaces) arp_req = arp() arp_req.hwsrc = EthAddr(self.H1_I1_ETH) arp_req.hwdst = EthAddr(b"\xff\xff\xff\xff\xff\xff") arp_req.opcode = arp.REQUEST arp_req.protosrc = IPAddr(self.H1_I1_IP) arp_req.protodst = IPAddr(self.H3_I3_IP) ether = ethernet() ether.type = ethernet.ARP_TYPE ether.dst = EthAddr(b"\xff\xff\xff\xff\xff\xff") ether.src = EthAddr(self.H1_I1_ETH) ether.payload = arp_req # Act # Get the action and reply packet reply_packet = h.receive(interfaces[0], ether) # Assert self.assertIsNone(reply_packet)
def test_arp_reply(self): """Receive a valid ARP packet and ensure the correct reply""" # Arrange iface1 = HostInterface(EthAddr( self.H2_I1_ETH), [IPAddr(self.H2_I1_IP1), IPAddr(self.H2_I1_IP2)]) iface2 = HostInterface(EthAddr(self.H2_I2_ETH), [IPAddr(self.H2_I2_IP)]) interfaces = [iface1, iface2] h = Host(interfaces) arp_req = arp() arp_req.hwsrc = EthAddr(self.H1_I1_ETH) arp_req.hwdst = EthAddr(b"\xff\xff\xff\xff\xff\xff") arp_req.opcode = arp.REQUEST arp_req.protosrc = IPAddr(self.H1_I1_IP) arp_req.protodst = IPAddr(self.H2_I1_IP1) ether = ethernet() ether.type = ethernet.ARP_TYPE ether.dst = EthAddr(b"\xff\xff\xff\xff\xff\xff") ether.src = EthAddr(self.H1_I1_ETH) ether.payload = arp_req # Act # Get the action and arp reply packet arp_reply = h.receive(interfaces[0], ether) # Assert self.assertIsNotNone(arp_reply) self.assertEquals(arp_reply.src, EthAddr(self.H2_I1_ETH)) self.assertEquals(arp_reply.dst, EthAddr(self.H1_I1_ETH)) self.assertEquals(arp_reply.type, ethernet.ARP_TYPE) reply_payload = arp_reply.payload self.assertEquals(reply_payload.opcode, arp.REPLY) self.assertEquals(reply_payload.hwsrc, EthAddr(self.H2_I1_ETH)) self.assertEquals(reply_payload.hwdst, EthAddr(self.H1_I1_ETH)) self.assertEquals(reply_payload.protosrc, self.H2_I1_IP1) self.assertEquals(reply_payload.protodst, self.H1_I1_IP)
def test_send(self): # Arrange interfaces = [mock.Mock()] name = "test-host" hid = 123 Host.raiseEvent = mock.Mock(name="mock_interface") pkt = ethernet() # Act host = Host(interfaces, name, hid) host.send(interfaces[0], pkt) # Assert self.assertEquals(Host.raiseEvent.call_count, 1)
def test_from_json(self): # Arrange json_dict = { 'hid': 1, '__type__': 'sts.entities.hosts.Host', 'name': 'h1', 'interfaces': [{ '__type__': 'sts.entities.hosts.HostInterface', 'name': 'eth0', 'hw_addr': '11:22:33:44:55:66', 'ips': ['127.0.0.1'], }], } hw_addr_str = "11:22:33:44:55:66" hw_addr = EthAddr(hw_addr_str) ip_str = "127.0.0.1" ip = IPAddr(ip_str) ifname = "eth0" interface = HostInterface(hw_addr, ip, name=ifname) hname = "h1" hid = 1 # Act host = Host.from_json(json_dict) # Assert self.assertEquals(host.name, hname) self.assertEquals(host.hid, hid) self.assertEquals(len(host.interfaces), 1) self.assertEquals(host.interfaces[0], interface)
def test_to_json(self): # Arrange hw_addr_str = "11:22:33:44:55:66" hw_addr = EthAddr(hw_addr_str) ip_str = "127.0.0.1" ip = IPAddr(ip_str) ifname = "eth0" interface = HostInterface(hw_addr, ip, name=ifname) hname = "h1" hid = 1 host = Host(interface, name=hname, hid=hid) # Act json_dict = host.to_json() # Assert self.assertEquals(json_dict['name'], hname) self.assertEquals(json_dict['hid'], hid) self.assertEquals(len(json_dict['interfaces']), 1) self.assertEquals(json_dict['interfaces'][0], interface.to_json())
def test_none_arp(self): """Receive a non-ARP packet and ensure there is no reply""" # Arrange iface1 = HostInterface(EthAddr( self.H2_I1_ETH), [IPAddr(self.H2_I1_IP1), IPAddr(self.H2_I1_IP2)]) iface2 = HostInterface(EthAddr(self.H2_I2_ETH), [IPAddr(self.H2_I2_IP)]) interfaces = [iface1, iface2] h = Host(interfaces) ether = ethernet() ether.type = ethernet.IP_TYPE ether.dst = EthAddr(self.H2_I1_ETH) ether.src = EthAddr(self.H1_I1_ETH) # Act # Get the action and reply packet reply_packet = h.receive(interfaces[0], ether) # Assert self.assertIsNone(reply_packet)
def test_init(self): # Arrange interfaces = [mock.Mock()] name = "test-host" hid = 123 # Act host = Host(interfaces, name, hid) # Assert self.assertEquals(host.interfaces, interfaces) self.assertEquals(host.name, name) self.assertEquals(host.hid, hid)
def test_to_jsont(self, SwitchCls): # Arrange sw1 = SwitchCls() sw1.dpid = 1 sw1.to_json.return_value = 1 # It's really hard to mock this, because of using assert_type p1 = ofp_phy_port(port_no=1) hw_addr_str = "11:22:33:44:55:66" hw_addr = EthAddr(hw_addr_str) ip_str = "127.0.0.1" ip = IPAddr(ip_str) ifname = "eth0" interface = HostInterface(hw_addr, ip, name=ifname) hname = "h1" hid = 1 host = Host(interface, name=hname, hid=hid) link = AccessLink(host, interface, sw1, p1) # Act json_dict = link.to_json() # Assert self.assertEquals(json_dict['node1'], host.to_json()) self.assertEquals(json_dict['port1'], interface.to_json()) self.assertEquals(json_dict['node2'], 1) self.assertEquals(json_dict['port2'], p1.to_json())
def test_init(self, SwitchCls): # Arrange sw1 = SwitchCls() sw1.dpid = 1 # It's really hard to mock this, because of using assert_type p1 = ofp_phy_port(port_no=1) hw_addr_str = "11:22:33:44:55:66" hw_addr = EthAddr(hw_addr_str) ip_str = "127.0.0.1" ip = IPAddr(ip_str) ifname = "eth0" interface = HostInterface(hw_addr, ip, name=ifname) hname = "h1" hid = 1 host = Host(interface, name=hname, hid=hid) # Act link = AccessLink(host, interface, sw1, p1) # Assert self.assertEquals(link.host, host) self.assertEquals(link.interface, interface)
def test_to_jsont(self, SwitchCls): # Arrange json_dict = { "node1": { "hid": 1, "interfaces": [ { "hw_addr": "11:22:33:44:55:66", "ips": [ "127.0.0.1" ], "__type__": "sts.entities.hosts.HostInterface", "name": "eth0" } ], "__type__": "sts.entities.hosts.Host", "name": "h1" }, "port2": { "hw_addr": "00:00:00:00:00:00", "curr": 0, "name": "", "supported": 0, "__type__": "pox.openflow.libopenflow_01.ofp_phy_port", "state": 0, "advertised": 0, "peer": 0, "config": 0, "port_no": 1 }, "node2": 1, "__type__": "sts.entities.sts_entities.AccessLink", "port1": { "hw_addr": "11:22:33:44:55:66", "ips": [ "127.0.0.1" ], "__type__": "sts.entities.hosts.HostInterface", "name": "eth0" } } sw1 = SwitchCls() sw1.dpid = 1 sw1.to_json.return_value = 1 # It's really hard to mock this, because of using assert_type p1 = ofp_phy_port(port_no=1) hw_addr_str = "11:22:33:44:55:66" hw_addr = EthAddr(hw_addr_str) ip_str = "127.0.0.1" ip = IPAddr(ip_str) ifname = "eth0" interface = HostInterface(hw_addr, ip, name=ifname) hname = "h1" hid = 1 host = Host(interface, name=hname, hid=hid) # Act link = AccessLink.from_json(json_dict) # Assert self.assertEquals(link.host.to_json(), host.to_json()) self.assertEquals(link.interface.to_json(), interface.to_json()) self.assertEquals(link.switch, 1) self.assertEquals(link.switch_port.to_json(), p1.to_json())