test: add addpath_test.py

Signed-off-by: ISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>

1 files changed, 8 insertions, 2 deletions

diff --git a/test/lib/exabgp.py b/test/lib/exabgp.py
index dbf8f023..efc39ac1 100644
--- a/test/lib/exabgp.py
+++ b/test/lib/exabgp.py
@@ -19,6 +19,7 @@ import time
 
 from fabric import colors
 from fabric.api import local
+from itertools import chain
 
 from lib.base import (
     BGPContainer,
@@ -92,7 +93,10 @@ class ExaBGPContainer(BGPContainer):
             if info['passive']:
                 cmd << '    passive;'
 
-            routes = [r for r in self.routes.values() if r['rf'] == 'ipv4' or r['rf'] == 'ipv6']
+            if info['addpath']:
+                cmd << '    add-path send/receive;'
+
+            routes = [r for r in chain.from_iterable(self.routes.itervalues()) if r['rf'] == 'ipv4' or r['rf'] == 'ipv6']
 
             if len(routes) > 0:
                 cmd << '    static {'
@@ -114,11 +118,13 @@ class ExaBGPContainer(BGPContainer):
                         r << 'extended-community [{0}]'.format(route['extended-community'])
                     if route['attr']:
                         r << 'attribute [ {0} ]'.format(route['attr'])
+                    if route['identifier']:
+                        r << 'path-information {0}'.format(route['identifier'])
 
                     cmd << '{0};'.format(str(r))
                 cmd << '    }'
 
-            routes = [r for r in self.routes.itervalues() if 'flowspec' in r['rf']]
+            routes = [r for r in chain.from_iterable(self.routes.itervalues()) if 'flowspec' in r['rf']]
             if len(routes) > 0:
                 cmd << '    flow {'
                 for route in routes: