From eddd8e3588561039985b27edf059db6033bfdfab Mon Sep 17 00:00:00 2001 From: Dave Wallace Date: Wed, 12 May 2021 21:43:59 -0400 Subject: tests: move test source to vpp/test - Generate copyright year and version instead of using hard-coded data Type: refactor Signed-off-by: Dave Wallace Change-Id: I6058f5025323b3aa483f5df4a2c4371e27b5914e --- src/plugins/dhcp/test/vpp_dhcp.py | 131 -------------------------------------- 1 file changed, 131 deletions(-) delete mode 100644 src/plugins/dhcp/test/vpp_dhcp.py (limited to 'src/plugins/dhcp/test/vpp_dhcp.py') diff --git a/src/plugins/dhcp/test/vpp_dhcp.py b/src/plugins/dhcp/test/vpp_dhcp.py deleted file mode 100644 index f8265a26252..00000000000 --- a/src/plugins/dhcp/test/vpp_dhcp.py +++ /dev/null @@ -1,131 +0,0 @@ -from vpp_object import VppObject - - -class VppDHCPProxy(VppObject): - - def __init__( - self, - test, - dhcp_server, - dhcp_src_address, - rx_vrf_id=0, - server_vrf_id=0, - ): - self._test = test - self._rx_vrf_id = rx_vrf_id - self._server_vrf_id = server_vrf_id - self._dhcp_server = dhcp_server - self._dhcp_src_address = dhcp_src_address - - def set_proxy( - self, - dhcp_server, - dhcp_src_address, - rx_vrf_id=0, - server_vrf_id=0): - if self.query_vpp_config(): - raise Exception('Vpp config present') - self._rx_vrf_id = rx_vrf_id - self._server_vrf_id = server_vrf_id - self._dhcp_server = dhcp_server - self._dhcp_src_address = dhcp_src_address - - def add_vpp_config(self): - self._test.vapi.dhcp_proxy_config( - is_add=1, - rx_vrf_id=self._rx_vrf_id, - server_vrf_id=self._server_vrf_id, - dhcp_server=self._dhcp_server, - dhcp_src_address=self._dhcp_src_address) - self._test.registry.register(self, self._test.logger) - - def remove_vpp_config(self): - self._test.vapi.dhcp_proxy_config( - rx_vrf_id=self._rx_vrf_id, - server_vrf_id=self._server_vrf_id, - dhcp_server=self._dhcp_server, - dhcp_src_address=self._dhcp_src_address, - is_add=0) - - def get_vpp_dump(self): - dump = self._test.vapi.dhcp_proxy_dump() - for entry in dump: - if entry.rx_vrf_id == self._rx_vrf_id: - return entry - - def query_vpp_config(self): - dump = self.get_vpp_dump() - return True if dump else False - - def object_id(self): - return "dhcp-proxy-%d" % self._rx_vrf_id - - -class VppDHCPClient(VppObject): - - def __init__( - self, - test, - sw_if_index, - hostname, - id=None, - want_dhcp_event=False, - set_broadcast_flag=True, - dscp=None, - pid=None): - self._test = test - self._sw_if_index = sw_if_index - self._hostname = hostname - self._id = id - self._want_dhcp_event = want_dhcp_event - self._set_broadcast_flag = set_broadcast_flag - self._dscp = dscp - self._pid = pid - - def set_client( - self, - sw_if_index, - hostname, - id=None, - want_dhcp_event=False, - set_broadcast_flag=True, - dscp=None, - pid=None): - if self.query_vpp_config(): - raise Exception('Vpp config present') - self._sw_if_index = sw_if_index - self._hostname = hostname - self._id = id - self._want_dhcp_event = want_dhcp_event - self._set_broadcast_flag = set_broadcast_flag - self._dscp = dscp - self._pid = pid - - def add_vpp_config(self): - id = self._id.encode('ascii') if self._id else None - client = {'sw_if_index': self._sw_if_index, 'hostname': self._hostname, - 'id': id, - 'want_dhcp_event': self._want_dhcp_event, - 'set_broadcast_flag': self._set_broadcast_flag, - 'dscp': self._dscp, 'pid': self._pid} - self._test.vapi.dhcp_client_config(is_add=1, client=client) - self._test.registry.register(self, self._test.logger) - - def remove_vpp_config(self): - client = client = { - 'sw_if_index': self._sw_if_index, - 'hostname': self._hostname} - self._test.vapi.dhcp_client_config(client=client, is_add=0) - - def get_vpp_dump(self): - dump = self._test.vapi.dhcp_client_dump() - for entry in dump: - if entry.client.sw_if_index == self._sw_if_index: - return entry - - def query_vpp_config(self): - dump = self.get_vpp_dump() - return True if dump else False - - def object_id(self): - return "dhcp-client-%s/%d" % (self._hostname, self._sw_if_index) -- cgit 1.2.3-korg