tests: replace pycodestyle with black
[vpp.git] / test / vpp_ipip_tun_interface.py
1 from vpp_tunnel_interface import VppTunnelInterface
2 from ipaddress import ip_address
3 from vpp_papi import VppEnum
4
5
6 class VppIpIpTunInterface(VppTunnelInterface):
7     """
8     VPP IP-IP Tunnel interface
9     """
10
11     def __init__(
12         self, test, parent_if, src, dst, table_id=0, dscp=0x0, flags=0, mode=None
13     ):
14         super(VppIpIpTunInterface, self).__init__(test, parent_if)
15         self.src = src
16         self.dst = dst
17         self.table_id = table_id
18         self.dscp = dscp
19         self.flags = flags
20         self.mode = mode
21         if not self.mode:
22             self.mode = VppEnum.vl_api_tunnel_mode_t.TUNNEL_API_MODE_P2P
23
24     def add_vpp_config(self):
25         r = self.test.vapi.ipip_add_tunnel(
26             tunnel={
27                 "src": self.src,
28                 "dst": self.dst,
29                 "table_id": self.table_id,
30                 "flags": self.flags,
31                 "dscp": self.dscp,
32                 "instance": 0xFFFFFFFF,
33                 "mode": self.mode,
34             }
35         )
36         self.set_sw_if_index(r.sw_if_index)
37         r = self.test.vapi.ipip_tunnel_dump(sw_if_index=self.sw_if_index)
38         self.instance = r[0].tunnel.instance
39         self.test.registry.register(self, self.test.logger)
40         return self
41
42     def remove_vpp_config(self):
43         self.test.vapi.ipip_del_tunnel(sw_if_index=self._sw_if_index)
44
45     def query_vpp_config(self):
46         ts = self.test.vapi.ipip_tunnel_dump(sw_if_index=0xFFFFFFFF)
47         for t in ts:
48             if t.tunnel.sw_if_index == self._sw_if_index:
49                 return True
50         return False
51
52     def __str__(self):
53         return self.object_id()
54
55     def object_id(self):
56         return "ipip%d" % self.instance
57
58     @property
59     def remote_ip(self):
60         return self.dst
61
62     @property
63     def local_ip(self):
64         return self.src