diff options
Diffstat (limited to 'resources/libraries')
-rw-r--r-- | resources/libraries/python/Dhcp.py | 48 | ||||
-rw-r--r-- | resources/libraries/robot/dhcp_client.robot | 49 |
2 files changed, 97 insertions, 0 deletions
diff --git a/resources/libraries/python/Dhcp.py b/resources/libraries/python/Dhcp.py new file mode 100644 index 0000000000..8b78f457c0 --- /dev/null +++ b/resources/libraries/python/Dhcp.py @@ -0,0 +1,48 @@ +# Copyright (c) 2016 Cisco and/or its affiliates. +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at: +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +"""DHCP utilities for VPP.""" + + +from resources.libraries.python.VatExecutor import VatExecutor +from resources.libraries.python.topology import Topology + + +class DhcpClient(object): + """DHCP Client utilities.""" + + @staticmethod + def set_dhcp_client_on_interface(vpp_node, interface, hostname=None): + """Set DHCP client on interface. + + :param vpp_node: VPP node to set DHCP client on. + :param interface: Interface name to set DHCP client on. + :param hostname: Hostname used in DHCP DISCOVER. + :type vpp_node: dict + :type interface: str + :type hostname: str + :raises RuntimeError: If unable to set DHCP client on interface. + """ + sw_if_index = Topology.get_interface_sw_index(vpp_node, interface) + interface = 'sw_if_index {}'.format(sw_if_index) + hostname = 'hostname {}'.format(hostname) if hostname else '' + output = VatExecutor.cmd_from_template(vpp_node, + "dhcp_client.vat", + interface=interface, + hostname=hostname) + output = output[0] + + if output["retval"] != 0: + raise RuntimeError('Unable to set DHCP client on node {} and' + ' interface {}.' + .format(vpp_node, interface)) diff --git a/resources/libraries/robot/dhcp_client.robot b/resources/libraries/robot/dhcp_client.robot new file mode 100644 index 0000000000..115646d42e --- /dev/null +++ b/resources/libraries/robot/dhcp_client.robot @@ -0,0 +1,49 @@ +# Copyright (c) 2016 Cisco and/or its affiliates. +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at: +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +*** Settings *** +| Library | Collections +| Resource | resources/libraries/robot/default.robot +| Library | resources.libraries.python.Dhcp.DhcpClient +| Library | resources.libraries.python.TrafficScriptExecutor +| Documentation | DHCP Client specific keywords. + +*** Keywords *** +| Check DHCP DISCOVER header +| | [Documentation] | Check if DHCP message contains all required fields. +| | ... +| | ... | *Arguments:* +| | ... | - tg_node - TG node. Type: dictionary +| | ... | - interface - TGs interface where listen for DHCP DISCOVER message. +| | ... | Type: string +| | ... | - src_mac - DHCP clients MAC address. Type: string +| | ... | - hostname - DHCP clients hostname (Optional, Default="", if not +| | ... | specified, the hostneme is not configured). Type: string +| | ... +| | ... | *Return:* +| | ... | - No value returned +| | ... +| | ... | *Example:* +| | ... +| | ... | \| Check DHCP DISCOVER header \| ${nodes['TG']} \ +| | ... | \| eth2 \| 08:00:27:66:b8:57 \| +| | ... | \| Check DHCP DISCOVER header \| ${nodes['TG']} \ +| | ... | \| eth2 \| 08:00:27:66:b8:57 \| client-hostname \| +| | ... +| | [Arguments] | ${tg_node} | ${interface} | ${src_mac} | ${hostname}=${EMPTY} +| | ${args}= | Run Keyword If | "${hostname}" == "" | Catenate +| | | ... | --rx_if | ${interface} | --rx_src_mac | ${src_mac} +| | ... | ELSE | Catenate | --rx_if | ${interface} | --rx_src_mac +| | | ... | ${src_mac} | --hostname | ${hostname} +| | Run Traffic Script On Node | dhcp/check_dhcp_discover.py +| | ... | ${tg_node} | ${args} |