summaryrefslogtreecommitdiffhomepage
path: root/test/lib
diff options
context:
space:
mode:
Diffstat (limited to 'test/lib')
-rw-r--r--test/lib/base.py5
-rw-r--r--test/lib/gobgp.py1
2 files changed, 4 insertions, 2 deletions
diff --git a/test/lib/base.py b/test/lib/base.py
index 3d14cf4c..5f73b45d 100644
--- a/test/lib/base.py
+++ b/test/lib/base.py
@@ -280,7 +280,7 @@ class BGPContainer(Container):
is_rr_client=False, cluster_id=None,
flowspec=False, bridge='', reload_config=True, as2=False,
graceful_restart=None, local_as=None, prefix_limit=None,
- v6=False, llgr=None):
+ v6=False, llgr=None, vrf=''):
neigh_addr = ''
local_addr = ''
it = itertools.product(self.ip_addrs, peer.ip_addrs)
@@ -318,7 +318,8 @@ class BGPContainer(Container):
'graceful_restart': graceful_restart,
'local_as': local_as,
'prefix_limit': prefix_limit,
- 'llgr': llgr}
+ 'llgr': llgr,
+ 'vrf': vrf}
if self.is_running and reload_config:
self.create_config()
self.reload_config()
diff --git a/test/lib/gobgp.py b/test/lib/gobgp.py
index 77060893..29a116a8 100644
--- a/test/lib/gobgp.py
+++ b/test/lib/gobgp.py
@@ -277,6 +277,7 @@ class GoBGPContainer(BGPContainer):
{'neighbor-address': info['neigh_addr'].split('/')[0],
'peer-as': peer.asn,
'auth-password': info['passwd'],
+ 'vrf': info['vrf'],
},
'afi-safis': afi_safi_list,
'timers': {'config': {