scenario_test: add ipv6 policy test

1 files changed, 7 insertions, 2 deletions

diff --git a/test/scenario_test/docker_control.py b/test/scenario_test/docker_control.py
index 3db00712..ec2d8397 100644
--- a/test/scenario_test/docker_control.py
+++ b/test/scenario_test/docker_control.py
@@ -340,7 +340,7 @@ def change_exabgp_version():
 def reload_config():
     cmd = "docker exec gobgp /usr/bin/pkill gobgpd -SIGHUP"
     local(cmd, capture=True)
-    print "complete append docker container."
+    print "gobgp config reloaded."
 
 
 def init_test_env_executor(quagga_num, use_local, go_path, log_debug=False, is_route_server=True):
@@ -390,7 +390,7 @@ def init_test_env_executor(quagga_num, use_local, go_path, log_debug=False, is_r
     print "complete initialization of test environment."
 
 
-def init_policy_test_env_executor(quagga_num, use_local, go_path, log_debug=False, policy=""):
+def init_policy_test_env_executor(quagga_num, use_local, go_path, log_debug=False, policy="", use_ipv6=False):
     print "start initialization of test environment."
 
     if docker_container_check() or bridge_setting_check():
@@ -400,6 +400,11 @@ def init_policy_test_env_executor(quagga_num, use_local, go_path, log_debug=Fals
 
     print "make gobgp policy test environment."
     create_config_dir()
+
+    if use_ipv6:
+        global IP_VERSION
+        IP_VERSION = IPv6
+
     bridge_setting_for_docker_connection(BRIDGES)
     make_config_with_policy(quagga_num, go_path, BRIDGE_0, policy_pattern=policy)