diff options
Diffstat (limited to 'test/scenario_test/vrf_neighbor_test.py')
| -rw-r--r-- | test/scenario_test/vrf_neighbor_test.py | 24 |
1 files changed, 14 insertions, 10 deletions
diff --git a/test/scenario_test/vrf_neighbor_test.py b/test/scenario_test/vrf_neighbor_test.py index de2b0e41..78125169 100644 --- a/test/scenario_test/vrf_neighbor_test.py +++ b/test/scenario_test/vrf_neighbor_test.py @@ -13,17 +13,21 @@ # See the License for the specific language governing permissions and # limitations under the License. -import unittest -from fabric.api import local -from lib import base -from lib.gobgp import * -from lib.quagga import * +from __future__ import absolute_import + import sys -import os import time +import unittest + +from fabric.api import local import nose + from noseplugin import OptionParser, parser_option +from lib import base +from lib.base import BGP_FSM_ESTABLISHED +from lib.gobgp import GoBGPContainer + class GoBGPTestBase(unittest.TestCase): @@ -126,13 +130,13 @@ class GoBGPTestBase(unittest.TestCase): self.assertTrue(len(dst) == 1) self.assertTrue(len(dst[0]['paths']) == 1) path = dst[0]['paths'][0] - self.assertTrue([self.g1.asn] == path['aspath']) + self.assertTrue([self.g1.asn] == path['aspath']) dst = self.g6.get_global_rib('10.0.0.0/24') self.assertTrue(len(dst) == 1) self.assertTrue(len(dst[0]['paths']) == 1) path = dst[0]['paths'][0] - self.assertTrue([self.g5.asn, self.g4.asn, self.g1.asn] == path['aspath']) + self.assertTrue([self.g5.asn, self.g4.asn, self.g1.asn] == path['aspath']) dst = self.g7.get_global_rib('10.0.0.0/24') self.assertTrue(len(dst) == 0) @@ -155,13 +159,13 @@ class GoBGPTestBase(unittest.TestCase): self.assertTrue(len(dst) == 1) self.assertTrue(len(dst[0]['paths']) == 1) path = dst[0]['paths'][0] - self.assertTrue([self.g5.asn, self.g4.asn, self.g1.asn] == path['aspath']) + self.assertTrue([self.g5.asn, self.g4.asn, self.g1.asn] == path['aspath']) dst = self.g7.get_global_rib('10.0.0.0/24') self.assertTrue(len(dst) == 1) self.assertTrue(len(dst[0]['paths']) == 1) path = dst[0]['paths'][0] - self.assertTrue([self.g5.asn, self.g4.asn, self.g3.asn] == path['aspath']) + self.assertTrue([self.g5.asn, self.g4.asn, self.g3.asn] == path['aspath']) if __name__ == '__main__': |