Update of latest tests.
[csit.git] / resources / libraries / python / IPUtil.py
1 # Copyright (c) 2016 Cisco and/or its affiliates.
2 # Licensed under the Apache License, Version 2.0 (the "License");
3 # you may not use this file except in compliance with the License.
4 # You may obtain a copy of the License at:
5 #
6 #     http://www.apache.org/licenses/LICENSE-2.0
7 #
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 # See the License for the specific language governing permissions and
12 # limitations under the License.
13
14 """Common IP utilities library."""
15
16 from ssh import SSH
17 from constants import Constants
18
19
20 class IPUtil(object):
21     """Common IP utilities"""
22
23     def __init__(self):
24         pass
25
26     @staticmethod
27     def vpp_ip_probe(node, interface, addr):
28         """Run ip probe on VPP node.
29
30            Args:
31                node (Dict): VPP node.
32                interface (str): Interface name
33                addr (str): IPv4/IPv6 address
34         """
35         ssh = SSH()
36         ssh.connect(node)
37
38         cmd = "{c}".format(c=Constants.VAT_BIN_NAME)
39         cmd_input = 'exec ip probe {dev} {ip}'.format(dev=interface, ip=addr)
40         (ret_code, _, _) = ssh.exec_command_sudo(cmd, cmd_input)
41         if int(ret_code) != 0:
42             raise Exception('VPP ip probe {dev} {ip} failed on {h}'.format(
43                 dev=interface, ip=addr, h=node['host']))