scenario_test: Change to be able to test the local gobgp program

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>

5 files changed, 128 insertions, 17 deletions

diff --git a/test/scenario_test/README.md b/test/scenario_test/README.md
index c491e10d..1308de36 100644
--- a/test/scenario_test/README.md
+++ b/test/scenario_test/README.md
@@ -40,19 +40,23 @@ Please make sure following packages are installed properly inside the VM.
 
 Start
 -----
-Please run the test script using nosetests as root.
+Please run the test script as root.
 
 route_server_test.py is scenario test script.
 ```
-# nosetests -v route_server_test.py
+# python route_server_test.py -v [ --use-local ]
 
 ```
 
+
 If you want to do malformed packet test, please run route_server_malformed_test.py
 ```
-# nosetests -v route_server_malformed_test.py
+# python route_server_malformed_test.py -v [ --use-local ]
 
 ```
 
 After the test, test results will be shown.
 
+Notes
+-----
+ use [ --use-local ] option when execute gobgp program of local system
\ No newline at end of file
diff --git a/test/scenario_test/docker_control.py b/test/scenario_test/docker_control.py
index 8af4ae69..2d93a134 100644
--- a/test/scenario_test/docker_control.py
+++ b/test/scenario_test/docker_control.py
@@ -16,6 +16,7 @@
 from fabric.api import local
 import re
 import os
+import sys
 
 GOBGP_CONTAINER_NAME = "gobgp"
 GOBGP_ADDRESS = "10.0.255.1/16"
@@ -34,6 +35,7 @@ BRIDGE_0 = {"BRIDGE_NAME": "br0", "BRIDGE_ADDRESS": "10.0.255.2"}
 BRIDGE_1 = {"BRIDGE_NAME": "br1", "BRIDGE_ADDRESS": "11.0.255.2"}
 BRIDGE_2 = {"BRIDGE_NAME": "br2", "BRIDGE_ADDRESS": "12.0.255.2"}
 BRIDGES = [BRIDGE_0, BRIDGE_1, BRIDGE_2]
+A_PART_OF_CURRENT_DIR = "/test/scenario_test"
 
 
 def test_user_check():
@@ -155,6 +157,20 @@ def make_startup_file():
     local(cmd, capture=True)
 
 
+def make_startup_file_use_local_gobgp():
+    file_buff = '#!/bin/bash' + '\n'
+    file_buff += 'rm -rf  /go/src/github.com/osrg/gobgp' + '\n'
+    file_buff += 'cp -r /mnt/gobgp /go/src/github.com/osrg/' + '\n'
+    file_buff += 'cd /go/src/github.com/osrg/gobgp' + '\n'
+    file_buff += 'go get -v' + '\n'
+    file_buff += 'go build' + '\n'
+    file_buff += './gobgp -f /mnt/gobgpd.conf > /mnt/gobgpd.log'
+    cmd = "echo \"" + file_buff + "\" > " + CONFIG_DIRR + STARTUP_FILE_NAME
+    local(cmd, capture=True)
+    cmd = "chmod 755 " + CONFIG_DIRR + STARTUP_FILE_NAME
+    local(cmd, capture=True)
+
+
 def docker_container_stop_quagga(quagga):
     cmd = "docker rm -f " + quagga
     local(cmd, capture=True)
@@ -189,7 +205,8 @@ def docker_containers_destroy():
 def docker_container_quagga_append(quagga_num, bridge):
     print "start append docker container."
     pwd = local("pwd", capture=True)
-    cmd = "$GOROOT/bin/go run " + pwd + "/quagga-rsconfig.go -a " + str(quagga_num) + " -c /usr/local/gobgp"
+    # cmd = "$GOROOT/bin/go run " + pwd + "/quagga-rsconfig.go -a " + str(quagga_num) + " -c /usr/local/gobgp"
+    cmd = "go run " + pwd + "/quagga-rsconfig.go -a " + str(quagga_num) + " -c /usr/local/gobgp"
     local(cmd, capture=True)
     docker_container_run_quagga(quagga_num, bridge)
     cmd = "docker exec gobgp /usr/bin/pkill gobgp -SIGHUP"
@@ -244,7 +261,7 @@ def get_notification_from_exabgp_log():
     return err_mgs
 
 
-def init_test_env_executor(quagga_num):
+def init_test_env_executor(quagga_num, gobgp_local):
     print "start initialization of test environment."
     if test_user_check() is False:
         print "you are not root"
@@ -259,19 +276,39 @@ def init_test_env_executor(quagga_num):
         print "make gobgp test environment."
         bridge_setting_for_docker_connection()
         pwd = local("pwd", capture=True)
-        cmd = "$GOROOT/bin/go run " + pwd + "/quagga-rsconfig.go -n " + str(quagga_num) + " -c /usr/local/gobgp"
+        # cmd = "$GOROOT/bin/go run " + pwd + "/quagga-rsconfig.go -n " + str(quagga_num) + " -c /usr/local/gobgp"
+        cmd = "go run " + pwd + "/quagga-rsconfig.go -n " + str(quagga_num) + " -c /usr/local/gobgp"
         local(cmd, capture=True)
+
+        # run each docker container
         for num in range(1, quagga_num + 1):
             docker_container_run_quagga(num, BRIDGE_0)
         docker_container_run_gobgp(BRIDGE_0)
-        make_startup_file()
+
+        # execute local gobgp program in the docker container if the input option is local
+        if gobgp_local:
+            print "use local-gobgp program in this test."
+            pwd = local("pwd", capture=True)
+            if A_PART_OF_CURRENT_DIR in pwd:
+                gobgp_path = re.sub(A_PART_OF_CURRENT_DIR, "", pwd)
+                cmd = "cp -r " + gobgp_path + " " + CONFIG_DIRR
+                local(cmd, capture=True)
+                make_startup_file_use_local_gobgp()
+            else:
+                print "current directory is not."
+                print "use gobgp in docker container."
+                make_startup_file()
+
+        else:
+            print "use docker-container-gobgp program in this test."
+            make_startup_file()
 
         start_gobgp()
 
         print "complete initialization of test environment."
 
 
-def init_malformed_test_env_executor(conf_file):
+def init_malformed_test_env_executor(conf_file, gobgp_local):
     print "start initialization of exabgp test environment."
     if test_user_check() is False:
         print "you are not root"
@@ -290,8 +327,25 @@ def init_malformed_test_env_executor(conf_file):
         cmd = "cp " + gobgp_file + " " + CONFIG_DIRR
         local(cmd, capture=True)
         docker_container_run_gobgp(BRIDGE_0)
-        make_startup_file()
         docker_container_run_exabgp(BRIDGE_0)
+
+        # execute local gobgp program in the docker container if the input option is local
+        if gobgp_local:
+            print "use local-gobgp program in this test."
+            pwd = local("pwd", capture=True)
+            if A_PART_OF_CURRENT_DIR in pwd:
+                gobgp_path = re.sub(A_PART_OF_CURRENT_DIR, "", pwd)
+                cmd = "cp -r " + gobgp_path + " " + CONFIG_DIRR
+                local(cmd, capture=True)
+                make_startup_file_use_local_gobgp()
+            else:
+                print "current directory is not."
+                print "use gobgp in docker container."
+                make_startup_file()
+
+        else:
+            print "use docker-container-gobgp program in this test."
+            make_startup_file()
         
         start_gobgp()
         start_exabgp(conf_file)
diff --git a/test/scenario_test/noseplugin.py b/test/scenario_test/noseplugin.py
new file mode 100644
index 00000000..8247d2fd
--- /dev/null
+++ b/test/scenario_test/noseplugin.py
@@ -0,0 +1,22 @@
+import os
+from nose.plugins import Plugin
+
+parser_option = None
+
+
+class OptionParser(Plugin):
+
+    def options(self, parser, env=os.environ):
+        super(OptionParser, self).options(parser, env=env)
+        parser.add_option('--use-local', action="store_true", dest="use_local", default=False)
+
+    def configure(self, options, conf):
+        super(OptionParser, self).configure(options, conf)
+        global parser_option
+        parser_option = options
+
+        if not self.enabled:
+            return
+
+    def finalize(self, result):
+        pass
\ No newline at end of file
diff --git a/test/scenario_test/route_server_malformed_test.py b/test/scenario_test/route_server_malformed_test.py
index bfaf874c..e6a5975a 100644
--- a/test/scenario_test/route_server_malformed_test.py
+++ b/test/scenario_test/route_server_malformed_test.py
@@ -17,8 +17,12 @@ import os
 import time
 import re
 import sys
+import nose
 import collections
 import docker_control as fab
+from noseplugin import OptionParser
+from noseplugin import parser_option
+
 
 sleep_time = 20
 
@@ -42,14 +46,20 @@ def test_malformed_packet():
     pwd = os.getcwd()
     pattern = check_pattern()
     if len(pattern) <= 0:
-        print "read test patten is faild."
-        print "csv element is " + str(len(pattern))
+        print "read test pattern is faild."
+        print "pattern element is " + str(len(pattern))
         sys.exit(1)
 
+    gobgp_local = parser_option.use_local
+    if gobgp_local:
+        print "execute gobgp program in local"
+    else:
+        print "execute gobgp program in gobgp container"
+
     for pkey in pattern:
         conf_file = pwd + "/exabgp_test_conf/" + pkey
         if os.path.isfile(conf_file) is True:
-            fab.init_malformed_test_env_executor(pkey)
+            fab.init_malformed_test_env_executor(pkey, gobgp_local)
             print "please wait"
             time.sleep(sleep_time)
             yield check_em, pkey, pattern[pkey]
@@ -69,4 +79,7 @@ def check_em(exabgp_conf, result):
     print " >>> " + notification_src
     notification = notification_src[1:-1]
 
-    assert notification == result
\ No newline at end of file
+    assert notification == result
+
+if __name__ == '__main__':
+    nose.main(argv=sys.argv, addplugins=[OptionParser()], defaultTest=sys.argv[0])
\ No newline at end of file
diff --git a/test/scenario_test/route_server_test.py b/test/scenario_test/route_server_test.py
index f1864122..24fccc65 100644
--- a/test/scenario_test/route_server_test.py
+++ b/test/scenario_test/route_server_test.py
@@ -19,9 +19,13 @@ import json
 import toml
 import os
 import time
+import sys
+import nose
 import quagga_access as qaccess
 from ciscoconfparse import CiscoConfParse
 import docker_control as fab
+from noseplugin import OptionParser
+from noseplugin import parser_option
 
 
 class GoBGPTest(unittest.TestCase):
@@ -35,10 +39,7 @@ class GoBGPTest(unittest.TestCase):
     append_quagga = 10
     remove_quagga = 10
     append_quagga_best = 20
-    fab.init_test_env_executor(quagga_num)
-    print "please wait"
     sleep_time = 20
-    time.sleep(sleep_time)
 
     def __init__(self, *args, **kwargs):
         super(GoBGPTest, self).__init__(*args, **kwargs)
@@ -51,8 +52,21 @@ class GoBGPTest(unittest.TestCase):
     # test each neighbor state is turned establish
     def test_01_neighbor_established(self):
         print "test_neighbor_established"
+        gobgp_local = parser_option.use_local
+
+        if gobgp_local:
+            print "execute gobgp program in local"
+        else:
+            print "execute gobgp program in gobgp container"
+        fab.init_test_env_executor(self.quagga_num, gobgp_local)
+
+        print "please wait"
+        time.sleep(self.sleep_time)
         if self.check_load_config() is False:
             return
+        self.load_gobgp_config()
+        self.load_quagga_config()
+
         addresses = self.get_neighbor_address(self.gobgp_config)
 
         for address in addresses:
@@ -373,6 +387,8 @@ class GoBGPTest(unittest.TestCase):
         try:
             content = os.listdir(self.base_dir)
             for item in content:
+                if "q" != item[0]:
+                    continue
                 if os.path.isdir(os.path.join(self.base_dir, item)):
                     dirs.append(item)
         except OSError, (errno, strerror):
@@ -415,6 +431,7 @@ class GoBGPTest(unittest.TestCase):
         neighbors_config = config['NeighborList']
         for neighbor_config in neighbors_config:
             neighbor_ip = neighbor_config['NeighborAddress']
+            print neighbor_ip
             address.append(neighbor_ip)
         return address
 
@@ -451,4 +468,5 @@ class Path:
         self.as_path = []
         self.metric = None
 
-
+if __name__ == '__main__':
+    nose.main(argv=sys.argv, addplugins=[OptionParser()], defaultTest=sys.argv[0])