Exemple #1
0
 def setUp(self):
     self.server = DHCPServerFactory()
     self.dhcp_dp = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip=RELAY_IP)
     self.dhcp_dp_wrong_relay = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip="176.98.33.76")
     self.dhcp_dp_wrong_mac = DHCPDiscoverPacketFactory(hwaddr="00:aa:bb:cc:dd:ee", relay_ip=RELAY_IP)
     self.dhcp_dp_vlan = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip=RELAY_IP, relay_vlan=RELAY_VLAN)
     self.dhcp_dp_wrong_vlan = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip=RELAY_IP, relay_vlan=99)
Exemple #2
0
class DHCPServerTestCase(unittest.TestCase):
    def setUp(self):
        self.server = DHCPServerFactory()
        self.dhcp_dp = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip=RELAY_IP)
        self.dhcp_dp_wrong_relay = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip="176.98.33.76")
        self.dhcp_dp_wrong_mac = DHCPDiscoverPacketFactory(hwaddr="00:aa:bb:cc:dd:ee", relay_ip=RELAY_IP)
        self.dhcp_dp_vlan = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip=RELAY_IP, relay_vlan=RELAY_VLAN)
        self.dhcp_dp_wrong_vlan = DHCPDiscoverPacketFactory(hwaddr=HWADDR, relay_ip=RELAY_IP, relay_vlan=99)

    def test_evaluate_relay(self):
        """Check if packet has correctly relay set and relay properly readed from config"""
        self.assertEqual(self.server._evaluateRelay(self.dhcp_dp), True)
        self.assertEqual(self.server._evaluateRelay(self.dhcp_dp_wrong_relay), False)

    def test_handle_dhcp_discover(self):
        """Check if dhcp server """
        self.server._sendDHCPPacket = MagicMock()
        self.server._handleDHCPDiscover(self.dhcp_dp, SERVER_IP, False)
        self.assertEqual(self.server._sendDHCPPacket.called, True)

    def test_handle_dhcp_discover_wrong_mac(self):
        """Check if dhcp server """
        self.server._sendDHCPPacket = MagicMock()
        self.server._handleDHCPDiscover(self.dhcp_dp_wrong_mac, SERVER_IP, False)
        self.assertEqual(self.server._sendDHCPPacket.called, False)

    def test_handle_dhcp_discover_vlan(self):
        """Check if dhcp server """
        self.server._sendDHCPPacket = MagicMock()
        self.server._handleDHCPDiscover(self.dhcp_dp_vlan, SERVER_IP, False)
        self.assertEqual(self.server._sendDHCPPacket.called, True)

    def test_handle_dhcp_discover_wrong_vlan(self):
        """Check if dhcp server """
        self.server._sendDHCPPacket = MagicMock()
        self.server._handleDHCPDiscover(self.dhcp_dp_wrong_vlan, SERVER_IP, False)
        self.assertEqual(self.server._sendDHCPPacket.called, False)