summaryrefslogtreecommitdiffhomepage
path: root/test/scenario_test
diff options
context:
space:
mode:
Diffstat (limited to 'test/scenario_test')
-rw-r--r--test/scenario_test/graceful_restart_test.py7
-rw-r--r--test/scenario_test/long_lived_graceful_restart_test.py3
2 files changed, 4 insertions, 6 deletions
diff --git a/test/scenario_test/graceful_restart_test.py b/test/scenario_test/graceful_restart_test.py
index ac4130d2..5d2c2b9d 100644
--- a/test/scenario_test/graceful_restart_test.py
+++ b/test/scenario_test/graceful_restart_test.py
@@ -78,8 +78,7 @@ class GoBGPTestBase(unittest.TestCase):
self.assertTrue(p['stale'])
g1.routes = {}
- g1._start_gobgp(graceful_restart=True)
- time.sleep(3)
+ g1.start_gobgp(graceful_restart=True)
g1.add_route('10.10.20.0/24')
def test_03_neighbor_established(self):
@@ -134,7 +133,7 @@ class GoBGPTestBase(unittest.TestCase):
g2 = self.bgpds['g2']
g3 = self.bgpds['g3']
- g1._start_gobgp()
+ g1.start_gobgp()
g1.del_peer(g2)
g1.del_peer(g3)
@@ -157,7 +156,7 @@ class GoBGPTestBase(unittest.TestCase):
g2.wait_for(expected_state=BGP_FSM_ACTIVE, peer=g1)
g3.wait_for(expected_state=BGP_FSM_ACTIVE, peer=g1)
- g1._start_gobgp(graceful_restart=True)
+ g1.start_gobgp(graceful_restart=True)
count = 0
while (g1.get_neighbor_state(g2) != BGP_FSM_ESTABLISHED
diff --git a/test/scenario_test/long_lived_graceful_restart_test.py b/test/scenario_test/long_lived_graceful_restart_test.py
index a9f6a0ee..0a42afab 100644
--- a/test/scenario_test/long_lived_graceful_restart_test.py
+++ b/test/scenario_test/long_lived_graceful_restart_test.py
@@ -110,8 +110,7 @@ class GoBGPTestBase(unittest.TestCase):
# withdrawn
self.assertTrue(len(g4.get_global_rib('10.0.0.0/24')) == 0)
- g2._start_gobgp(graceful_restart=True)
- time.sleep(2)
+ g2.start_gobgp(graceful_restart=True)
g2.local('gobgp global rib add 10.0.0.0/24')
g2.local('gobgp global rib add 10.10.0.0/24')