summaryrefslogtreecommitdiffhomepage
path: root/test/scenario_test/docker_control.py
diff options
context:
space:
mode:
authorISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>2015-08-22 17:44:35 +0900
committerISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>2015-08-22 17:48:57 +0900
commit316e795fbb4ff5deceeabcad31667078be0f2376 (patch)
tree1162da0ef464cbc7bd617d7b754a3d4a85872ae4 /test/scenario_test/docker_control.py
parente0c7d1823a0ffa5661f7b2dd54810e0fd92f72fb (diff)
test: remove unused files
now all scenario_test uses scenario_test/lib. remove unused files. Signed-off-by: ISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>
Diffstat (limited to 'test/scenario_test/docker_control.py')
-rw-r--r--test/scenario_test/docker_control.py566
1 files changed, 0 insertions, 566 deletions
diff --git a/test/scenario_test/docker_control.py b/test/scenario_test/docker_control.py
deleted file mode 100644
index 3232f566..00000000
--- a/test/scenario_test/docker_control.py
+++ /dev/null
@@ -1,566 +0,0 @@
-# Copyright (C) 2014 Nippon Telegraph and Telephone Corporation.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
-# implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from fabric.api import local
-import re
-import os
-import time
-from constant import *
-
-def test_user_check():
- root = False
- outbuf = local("echo $USER", capture=True)
- user = outbuf
- if user == "root":
- root = True
-
- return root
-
-
-def install_docker_and_tools():
- print "start install packages of test environment."
- if test_user_check() is False:
- print "you are not root"
- return
-
- local("apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv-keys "
- "36A1D7869245C8950F966E92D8576A8BA88D21E9", capture=True)
- local('sh -c "echo deb https://get.docker.io/ubuntu docker main > /etc/apt/sources.list.d/docker.list"',
- capture=True)
- local("apt-get update", capture=True)
- local("apt-get install -y --force-yes lxc-docker-1.3.2", capture=True)
- local("ln -sf /usr/bin/docker.io /usr/local/bin/docker", capture=True)
- local("gpasswd -a `whoami` docker", capture=True)
- local("apt-get install -y --force-yes emacs23-nox", capture=True)
- local("apt-get install -y --force-yes wireshark", capture=True)
- local("apt-get install -y --force-yes iputils-arping", capture=True)
- local("apt-get install -y --force-yes bridge-utils", capture=True)
- local("apt-get install -y --force-yes tcpdump", capture=True)
- local("apt-get install -y --force-yes lv", capture=True)
- local("wget https://raw.github.com/jpetazzo/pipework/master/pipework -O /usr/local/bin/pipework",
- capture=True)
- local("chmod 755 /usr/local/bin/pipework", capture=True)
- local("docker pull osrg/quagga", capture=True)
- local("docker pull osrg/gobgp", capture=True)
- local("docker pull osrg/exabgp", capture=True)
-
-
-def docker_pkg_check():
- docker_exists = False
- outbuf = local("dpkg -l | grep docker | awk '{print $2}'", capture=True)
- dpkg_list = outbuf.split('\n')
- for dpkg in dpkg_list:
- if "lxc-docker" in dpkg:
- docker_exists = True
- return docker_exists
-
-
-def go_path_check():
- go_path_exist = False
- outbuf = local("echo `which go`", capture=True)
- if "go" in outbuf:
- go_path_exist = True
- return go_path_exist
-
-
-def docker_container_check():
- container_exists = False
- outbuf = local("docker ps -a", capture=True)
- docker_ps = outbuf.split('\n')
- for container in docker_ps:
- container_name = container.split()[-1]
- if (container_name == GOBGP_CONTAINER_NAME) or \
- (container_name == EXABGP_CONTAINER_NAME) or ("q" in container_name):
- container_exists = True
- return container_exists
-
-
-def bridge_setting_check():
- setting_exists = False
- for bridge in BRIDGES:
- sysfs_name = "/sys/class/net/" + bridge["BRIDGE_NAME"]
- if os.path.exists(sysfs_name):
- setting_exists = True
- return setting_exists
- return setting_exists
-
-
-def docker_containers_get():
- containers = []
- cmd = "docker ps -a | awk '{print $NF}'"
- outbuf = local(cmd, capture=True)
- docker_ps = outbuf.split('\n')
- for container in docker_ps:
- if container != "NAMES":
- containers.append(container.split()[-1])
- return containers
-
-
-def docker_container_set_ipaddress(bridge, name, address):
- cmd = "pipework " + bridge["BRIDGE_NAME"] + " -i e" + bridge["BRIDGE_NAME"]\
- + " " + name + " " + address
- local(cmd, capture=True)
-
-
-def docker_container_run_quagga(quagga_num, bridge):
- quagga_name = "q" + str(quagga_num)
- cmd = "docker run --privileged=true -v " + CONFIG_DIR + "/" + quagga_name +\
- ":/etc/quagga --name " + quagga_name + " -id osrg/quagga"
- local(cmd, capture=True)
- quagga_address = BASE_NET[bridge["BRIDGE_NAME"]][IP_VERSION] + str(quagga_num) + BASE_MASK[IP_VERSION]
- docker_container_set_ipaddress(bridge, quagga_name, quagga_address)
- # restart the quagga after the docker container has become IP reachable
- cmd = 'docker kill --signal="HUP" ' + quagga_name
- local(cmd, capture=True)
-
-
-def docker_container_run_gobgp(bridge, image='osrg/gobgp'):
- cmd = "docker run --privileged=true -v " + CONFIG_DIR + ":" + SHARE_VOLUME + " -d --name "\
- + GOBGP_CONTAINER_NAME + " -id " + image
- local(cmd, capture=True)
- docker_container_set_ipaddress(bridge, GOBGP_CONTAINER_NAME, GOBGP_ADDRESS_0[IP_VERSION] + BASE_MASK[IP_VERSION])
-
-
-def docker_container_run_exabgp(bridge):
- pwd = local("pwd", capture=True)
- test_pattern_dir = pwd + "/exabgp_test_conf"
- cmd = "cp -r " + test_pattern_dir + " " + CONFIG_DIRR
- local(cmd, capture=True)
- cmd = "docker run --privileged=true -v " + CONFIG_DIR + ":" + SHARE_VOLUME + " -d --name "\
- + EXABGP_CONTAINER_NAME + " -id osrg/exabgp"
- local(cmd, capture=True)
- docker_container_set_ipaddress(bridge, EXABGP_CONTAINER_NAME, EXABGP_ADDRESS)
-
-
-def change_owner_to_root(target):
- cmd = "chown -R root:root " + target
- local(cmd, capture=True)
-
-
-def create_config_dir():
- cmd = "mkdir " + CONFIG_DIR
- local(cmd, capture=True)
-
-
-def recreate_conf_dir(dirname):
- cmd = "rm -rf " + dirname
- local(cmd, capture=True)
- cmd = "mkdir " + dirname
- local(cmd, capture=True)
-
-
-def make_startup_file(log_opt=""):
- file_buff = '#!/bin/bash' + '\n'
- file_buff += "cd /go/bin" + '\n'
- file_buff += "./gobgpd -f " + SHARE_VOLUME + "/gobgpd.conf " + log_opt + " > " + SHARE_VOLUME + "/gobgpd.log 2>&1 "
-
- cmd = "echo \"" + file_buff + "\" > " + CONFIG_DIR + "/" + STARTUP_FILE_NAME
- local(cmd, capture=True)
- cmd = "chmod 755 " + CONFIG_DIRR + STARTUP_FILE_NAME
- local(cmd, capture=True)
-
-
-def make_install_file():
- file_buff = '#!/bin/bash' + '\n'
- file_buff += 'rm -rf /go/src/github.com/osrg/gobgp' + '\n'
- file_buff += 'cp -r ' + SHARE_VOLUME + '/gobgp /go/src/github.com/osrg/' + '\n'
- file_buff += 'go get github.com/osrg/gobgp/gobgpd' + '\n'
- file_buff += 'go get github.com/osrg/gobgp/gobgp' + '\n'
- cmd = "echo \"" + file_buff + "\" > " + CONFIG_DIR + "/" + INSTALL_FILE_NAME
- local(cmd, capture=True)
- cmd = "chmod 755 " + CONFIG_DIRR + INSTALL_FILE_NAME
- local(cmd, capture=True)
-
-
-def docker_container_stop_quagga(quagga):
- if docker_check_running(quagga):
- cmd = "docker stop " + quagga
- local(cmd, capture=True)
-
- cmd = "docker rm " + quagga
- local(cmd, capture=True)
- cmd = "rm -rf " + CONFIG_DIRR + quagga
- local(cmd, capture=True)
-
-
-def docker_container_stop_gobgp(remove=True):
- if docker_check_running(GOBGP_CONTAINER_NAME):
- cmd = "docker stop --time=0 " + GOBGP_CONTAINER_NAME
- local(cmd, capture=True)
-
- if remove:
- cmd = "docker rm " + GOBGP_CONTAINER_NAME
- local(cmd, capture=True)
-
-
-def docker_container_stop_exabgp():
- if docker_check_running(EXABGP_CONTAINER_NAME):
- cmd = "docker stop --time=0 " + EXABGP_CONTAINER_NAME
- local(cmd, capture=True)
-
- cmd = "docker rm " + EXABGP_CONTAINER_NAME
- local(cmd, capture=True)
-
-
-def docker_containers_destroy(gobgp_remove=True, confdir_remove=True):
- containers = docker_containers_get()
- for container in containers:
- if re.match(r'q[0-9][0-9]*', container) is not None:
- docker_container_stop_quagga(container)
- if container == GOBGP_CONTAINER_NAME:
- docker_container_stop_gobgp(remove=gobgp_remove)
- if container == EXABGP_CONTAINER_NAME:
- docker_container_stop_exabgp()
- bridge_unsetting_for_docker_connection()
- if confdir_remove:
- cmd = "rm -rf " + CONFIG_DIRR
- local(cmd, capture=True)
-
-
-def docker_check_running(cname):
- cmd = "docker ps | awk '{print $NF}'"
- outbuf = local(cmd, capture=True)
- docker_ps = outbuf.split('\n')
- if cname in docker_ps:
- return True
-
-
-def docker_container_quagga_append(quagga_num, bridge):
- print "start append docker container."
- docker_container_run_quagga(quagga_num, bridge)
-
-
-def docker_container_quagga_removed(quagga_num):
- print "start removed docker container."
- quagga = "q" + str(quagga_num)
- docker_container_stop_quagga(quagga)
- print "complete removed docker container."
-
-
-def bridge_setting_for_docker_connection(bridges):
- # bridge_unsetting_for_docker_connection()
- for bridge in bridges:
- cmd = "brctl addbr " + bridge["BRIDGE_NAME"]
- local(cmd, capture=True)
- if IP_VERSION == IPv6:
- cmd = "ifconfig " + bridge["BRIDGE_NAME"] + " " + IF_CONFIG_OPTION[IP_VERSION] +\
- " add " + bridge[IP_VERSION] + BASE_MASK[IP_VERSION]
- else:
- cmd = "ifconfig " + bridge["BRIDGE_NAME"] + " " + bridge[IP_VERSION]
- local(cmd, capture=True)
- cmd = "ifconfig " + bridge["BRIDGE_NAME"] + " up"
- local(cmd, capture=True)
-
-
-def bridge_unsetting_for_docker_connection():
- for bridge in BRIDGES:
- sysfs_name = "/sys/class/net/" + bridge["BRIDGE_NAME"]
- if os.path.exists(sysfs_name):
- cmd = "ifconfig " + bridge["BRIDGE_NAME"] + " down"
- local(cmd, capture=True)
- cmd = "brctl delbr " + bridge["BRIDGE_NAME"]
- local(cmd, capture=True)
-
-
-def build_gobgp():
- cmd = "docker exec gobgp " + INSTALL_FILE
- local(cmd, capture=True)
-
-
-def start_gobgp(build=False):
- if build:
- build_gobgp()
- cmd = "docker exec -d gobgp " + STARTUP_FILE
- local(cmd, capture=True)
-
-
-def start_exabgp(conf_file):
- # run exabgp docker container
- docker_container_run_exabgp(BRIDGE_0)
-
- cmd = "docker exec exabgp cp -f " + SHARE_VOLUME + "/exabgp_test_conf/exabgp.env /root/exabgp/etc/exabgp/exabgp.env"
- local(cmd, capture=True)
- conf_path = EXABGP_CONFDIR + "/" + conf_file
- cmd = "docker exec exabgp /root/exabgp/sbin/exabgp " + conf_path + " > /dev/null 2>&1 &"
- local(cmd, capture=True)
-
-
-def stop_exabgp():
- docker_container_stop_exabgp()
- log_path = CONFIG_DIRR + EXABGP_LOG_FILE
- clean_log(log_path)
-
-
-def clean_log(file_path):
- cmd = ": > " + file_path
- local(cmd, capture=True)
-
-
-def get_notification_from_exabgp_log():
- log_path = CONFIG_DIRR + EXABGP_LOG_FILE
- cmd = "grep notification " + log_path + " | head -1"
- err_mgs = local(cmd, capture=True)
- return err_mgs
-
-
-def make_config(quagga_num, go_path, bridge, peer_opts="", use_compiled=False, ipver=''):
- if go_path != "":
- print "specified go path is [ " + go_path + " ]."
- if os.path.isdir(go_path):
- go_path += "/"
- else:
- print "specified go path do not use."
- pwd = local("pwd", capture=True)
-
- tool = go_path + "go run " + pwd + "/quagga-rsconfig.go "
- if use_compiled:
- tool = pwd + "/quagga-rsconfig "
-
- # I want to avoid a global variable.
- ip_version = ipver if ipver else IP_VERSION
- cmd = tool + " -n " + str(quagga_num) +\
- " -c /tmp/gobgp -v " + ip_version + " -i " + bridge["BRIDGE_NAME"][-1] + " " + peer_opts
- local(cmd, capture=True)
-
-
-def update_policy_config(go_path, neighbor, policy_name, target, isReplace=False, defaultReject=False):
- if go_path != "":
- print "specified go path is [ " + go_path + " ]."
- if os.path.isdir(go_path):
- go_path += "/"
- else:
- print "specified go path is not used."
-
- pwd = local("pwd", capture=True)
- replace = ' -r' if isReplace else ''
- reject = ' -j' if defaultReject else ''
- cmd = pwd + "/policy_generator -d /tmp/gobgp -n " + neighbor + " -t " + target + " -p " + policy_name + replace + reject
- local(cmd, capture=True)
- # reload_config()
-
-
-def make_config_append(quagga_num, go_path, bridge, peer_opts="", use_compiled=False, ipver=''):
- if go_path != "":
- print "specified go path is [ " + go_path + " ]."
- if os.path.isdir(go_path):
- go_path += "/"
- else:
- print "specified go path do not use."
- pwd = local("pwd", capture=True)
- ip_version = ipver if ipver else IP_VERSION
-
- tool = go_path + "go run " + pwd + "/quagga-rsconfig.go "
- if use_compiled:
- tool = pwd + "/quagga-rsconfig "
-
- cmd = tool + " -a " + str(quagga_num) +\
- " -c /tmp/gobgp -v " + ip_version + " -i " + bridge["BRIDGE_NAME"][-1] + " " + peer_opts
- local(cmd, capture=True)
-
-
-def reload_config():
- cmd = "docker exec gobgp /usr/bin/pkill gobgpd -SIGHUP"
- local(cmd, capture=True)
- print "gobgp config reloaded."
-
-
-def build_config_tools(go_path):
- if go_path != "":
- print "specified go path is [ " + go_path + " ]."
- if os.path.isdir(go_path):
- go_path += "/"
- else:
- print "the specified go path is not directory.."
-
- pwd = local("pwd", capture=True)
- cmd = go_path + "go build " + pwd + "/quagga-rsconfig.go"
- local(cmd, capture=True)
- cmd = go_path + "go build " + pwd + "/policy/policy_generator.go"
- local(cmd, capture=True)
-
-
-def prepare_gobgp(log_debug, image='osrg/gobgp'):
-
- # cleanup gobgp container
- containers = docker_containers_get()
- if GOBGP_CONTAINER_NAME in containers:
- docker_container_stop_gobgp(remove=True)
- recreate_conf_dir(CONFIG_DIRR)
- # set log option
- opt = "-l debug" if log_debug else ""
- do_build = False
- make_startup_file(log_opt=opt)
- change_owner_to_root(CONFIG_DIR)
- cmd = "docker run --privileged=true -v " + CONFIG_DIR + ":" + SHARE_VOLUME + " -d --name " \
- + GOBGP_CONTAINER_NAME + " -id " + image
- local(cmd, capture=True)
- if do_build:
- build_gobgp()
-
-
-def init_test_env_executor(quagga_num, image, go_path, log_debug=False, is_route_server=True):
- print "start initialization of test environment."
-
- if docker_container_check() or bridge_setting_check():
- print "gobgp test environment already exists."
- print "so that remake gobgp test environment."
- docker_containers_destroy()
-
- print "make gobgp test environment."
- create_config_dir()
- bridge_setting_for_docker_connection(BRIDGES)
- make_config(quagga_num, go_path, BRIDGE_0, ("" if is_route_server else "--normal-bgp"))
-
- # run gobgp docker container
- docker_container_run_gobgp(BRIDGE_0, image)
-
- # set log option
- opt = "-l debug" if log_debug else ""
- do_build = False
- # execute local gobgp program in the docker container if the input option is local
- make_startup_file(log_opt=opt)
- change_owner_to_root(CONFIG_DIR)
- start_gobgp(build=do_build)
-
- # run quagga docker container
- for num in range(1, quagga_num + 1):
- docker_container_run_quagga(num, BRIDGE_0)
-
- print "complete initialization of test environment."
-
-
-def init_policy_test_env_executor(quagga_num, use_ipv6=False, use_exabgp=False):
- print "start initialization of test environment."
-
- global IP_VERSION
- IP_VERSION = IPv6 if use_ipv6 else IPv4
-
- bridge_setting_for_docker_connection(BRIDGES)
-
- cmd = "docker start %s" % GOBGP_CONTAINER_NAME
- local(cmd, capture=True)
- docker_container_set_ipaddress(BRIDGE_0, GOBGP_CONTAINER_NAME, GOBGP_ADDRESS_0[IP_VERSION] + BASE_MASK[IP_VERSION])
-
- if use_exabgp:
- # run exabgp
- start_exabgp(EXABGP_COMMON_CONF)
-
- start_gobgp()
-
- # run quagga docker container
- for num in range(1, quagga_num + 1):
- docker_container_run_quagga(num, BRIDGE_0)
-
- print "complete initialization of test environment."
-
-
-def init_ipv6_test_env_executor(quagga_num, image, go_path, log_debug=False):
- print "start initialization of test environment."
-
- if docker_container_check() or bridge_setting_check():
- print "gobgp test environment already exists."
- print "so that remake gobgp test environment."
- docker_containers_destroy()
-
- print "make gobgp test environment."
- create_config_dir()
- bridge_setting_for_docker_connection([BRIDGE_0])
- make_config(quagga_num, go_path, BRIDGE_0)
-
- # run gobgp docker container
- docker_container_run_gobgp(BRIDGE_0, image)
-
- # set log option
- opt = "-l debug" if log_debug else ""
- do_build = False
- make_startup_file(log_opt=opt)
- change_owner_to_root(CONFIG_DIR)
- start_gobgp(do_build)
-
- # run quagga docker container
- for num in range(1, quagga_num + 1):
- docker_container_run_quagga(num, BRIDGE_0)
-
- print "complete initialization of test environment."
-
-
-def init_malformed_test_env_executor(image, go_path, exabgp_path, log_debug=False):
- print "start initialization of exabgp test environment."
-
- if docker_container_check() or bridge_setting_check():
- print "gobgp test environment already exists."
- print "so that remake gobgp test environment."
- docker_containers_destroy()
-
- print "make gobgp test environment."
- peer_opts = "--none-peer"
- create_config_dir()
- bridge_setting_for_docker_connection(BRIDGES)
- make_config(1, go_path, BRIDGE_0)
- make_config_append(100, go_path, BRIDGE_0, peer_opts)
-
- # run gobgp docker container
- docker_container_run_gobgp(BRIDGE_0, image)
-
- # set log option
- opt = "-l debug" if log_debug else ""
- do_build = False
- make_startup_file(log_opt=opt)
- change_owner_to_root(CONFIG_DIR)
-
- if exabgp_path != "":
- cmd = "cp -rf %s %s" % (exabgp_path, CONFIG_DIR)
- local(cmd, capture=True)
- cmd = "docker exec exabgp cp -rf " + SHARE_VOLUME + "/exabgp /root/"
- local(cmd, capture=True)
-
- start_gobgp(do_build)
-
- # run quagga docker container
- docker_container_run_quagga(1, BRIDGE_0)
-
-
-def docker_container_quagga_append_executor(quagga_num, go_path, is_route_server=True):
- make_config_append(quagga_num, go_path, BRIDGE_0, ("" if is_route_server else "--normal-bgp"))
- docker_container_quagga_append(quagga_num, BRIDGE_0)
- reload_config()
-
-
-def docker_container_ipv6_quagga_append_executor(quagga_nums, go_path):
- print "append ipv6 quagga container."
- global IP_VERSION
- IP_VERSION = IPv6
- bridge_setting_for_docker_connection([BRIDGE_1])
- docker_container_set_ipaddress(BRIDGE_1, GOBGP_CONTAINER_NAME, GOBGP_ADDRESS_1[IP_VERSION] + BASE_MASK[IP_VERSION])
- for quagga_num in quagga_nums:
- make_config_append(quagga_num, go_path, BRIDGE_1)
- docker_container_quagga_append(quagga_num, BRIDGE_1)
- reload_config()
-
-
-def docker_container_quagga_removed_executor(quagga_num):
- docker_container_quagga_removed(quagga_num)
-
-
-def docker_container_make_bestpath_env_executor(append_quagga_num, go_path, is_route_server=True):
- print "start make bestpath environment"
- make_config_append(append_quagga_num, go_path, BRIDGE_1, ("" if is_route_server else "--normal-bgp"))
- append_quagga_name = "q" + str(append_quagga_num)
- docker_container_quagga_append(append_quagga_num, BRIDGE_1)
- reload_config()
- docker_container_set_ipaddress(BRIDGE_1, "q2", "11.0.0.2/16")
- docker_container_set_ipaddress(BRIDGE_2, append_quagga_name, "12.0.0.20/16")
- docker_container_set_ipaddress(BRIDGE_2, "q3", "12.0.0.3/16")
-