diff options
Diffstat (limited to 'test/scenario_test/lib/quagga.py')
| -rw-r--r-- | test/scenario_test/lib/quagga.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/test/scenario_test/lib/quagga.py b/test/scenario_test/lib/quagga.py index 1099866a..e4de3ef1 100644 --- a/test/scenario_test/lib/quagga.py +++ b/test/scenario_test/lib/quagga.py @@ -161,14 +161,16 @@ class QuaggaBGPContainer(BGPContainer): c << 'neighbor {0} activate'.format(n_addr) c << 'exit-address-family' - for route in self.routes.iterkeys(): - version = netaddr.IPNetwork(route).version - if version == 4: - c << 'network {0}'.format(route) - elif version == 6: + for route in self.routes.itervalues(): + if route['rf'] == 'ipv4': + c << 'network {0}'.format(route['prefix']) + elif route['rf'] == 'ipv6': c << 'address-family ipv6 unicast' - c << 'network {0}'.format(route) + c << 'network {0}'.format(route['prefix']) c << 'exit-address-family' + else: + raise Exception('unsupported route faily: {0}'.format(route['rf'])) + for name, policy in self.policies.iteritems(): c << 'access-list {0} {1} {2}'.format(name, policy['type'], |