diff options
Diffstat (limited to 'test/scenario_test')
| -rw-r--r-- | test/scenario_test/graceful_restart_test.py | 3 | ||||
| -rw-r--r-- | test/scenario_test/long_lived_graceful_restart_test.py | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/test/scenario_test/graceful_restart_test.py b/test/scenario_test/graceful_restart_test.py index 5d2c2b9d..537cf96c 100644 --- a/test/scenario_test/graceful_restart_test.py +++ b/test/scenario_test/graceful_restart_test.py @@ -29,6 +29,7 @@ from lib.base import ( BGP_FSM_IDLE, BGP_FSM_ACTIVE, BGP_FSM_ESTABLISHED, + GRACEFUL_RESTART_TIME, ) from lib.gobgp import GoBGPContainer @@ -124,7 +125,7 @@ class GoBGPTestBase(unittest.TestCase): self.assertTrue(len(g3.get_global_rib('10.10.30.0/24')) == 1) def test_06_test_restart_timer_expire(self): - time.sleep(25) + time.sleep(GRACEFUL_RESTART_TIME + 5) g2 = self.bgpds['g2'] self.assertTrue(len(g2.get_global_rib()) == 0) diff --git a/test/scenario_test/long_lived_graceful_restart_test.py b/test/scenario_test/long_lived_graceful_restart_test.py index 0a42afab..44f4a45b 100644 --- a/test/scenario_test/long_lived_graceful_restart_test.py +++ b/test/scenario_test/long_lived_graceful_restart_test.py @@ -29,6 +29,7 @@ from lib import base from lib.base import ( BGP_FSM_ACTIVE, BGP_FSM_ESTABLISHED, + LONG_LIVED_GRACEFUL_RESTART_TIME, ) from lib.gobgp import GoBGPContainer @@ -156,7 +157,7 @@ class GoBGPTestBase(unittest.TestCase): self.assertTrue(g2.asn in rib[0]['paths'][0]['aspath']) def test_05_llgr_restart_timer_expire(self): - time.sleep(35) + time.sleep(LONG_LIVED_GRACEFUL_RESTART_TIME + 5) g3 = self.bgpds['g3'] rib = g3.get_global_rib('10.10.0.0/24') self.assertTrue(len(rib) == 0) |