summaryrefslogtreecommitdiffhomepage
path: root/test/scenario_test/graceful_restart_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/scenario_test/graceful_restart_test.py')
-rw-r--r--test/scenario_test/graceful_restart_test.py22
1 files changed, 15 insertions, 7 deletions
diff --git a/test/scenario_test/graceful_restart_test.py b/test/scenario_test/graceful_restart_test.py
index 5c8bba65..926bcbd0 100644
--- a/test/scenario_test/graceful_restart_test.py
+++ b/test/scenario_test/graceful_restart_test.py
@@ -13,17 +13,25 @@
# 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_ACTIVE,
+ BGP_FSM_ESTABLISHED,
+)
+from lib.gobgp import GoBGPContainer
+
+
class GoBGPTestBase(unittest.TestCase):
@classmethod
@@ -86,7 +94,7 @@ class GoBGPTestBase(unittest.TestCase):
def test_04_add_non_graceful_restart_enabled_peer(self):
g1 = self.bgpds['g1']
- g2 = self.bgpds['g2']
+ # g2 = self.bgpds['g2']
gobgp_ctn_image_name = parser_option.gobgp_image
g3 = GoBGPContainer(name='g3', asn=65002, router_id='192.168.0.3',
ctn_image_name=gobgp_ctn_image_name,