test: introduce modular test library

Signed-off-by: ISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>

1 files changed, 99 insertions, 0 deletions

diff --git a/test/scenario_test/lib/exabgp.py b/test/scenario_test/lib/exabgp.py
new file mode 100644
index 00000000..91919101
--- /dev/null
+++ b/test/scenario_test/lib/exabgp.py
@@ -0,0 +1,99 @@
+# Copyright (C) 2015 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 base import *
+
+
+class ExaBGPContainer(BGPContainer):
+
+    SHARED_VOLUME = '/root/shared_volume'
+
+    def __init__(self, name, asn, router_id, ctn_image_name='osrg/exabgp',
+                 exabgp_path=''):
+        super(ExaBGPContainer, self).__init__(name, asn, router_id,
+                                              ctn_image_name)
+        self.shared_volumes.append((self.config_dir, self.SHARED_VOLUME))
+        self.exabgp_path = exabgp_path
+
+    def _start_exabgp(self):
+        cmd = CmdBuffer(' ')
+        cmd << 'docker exec -d {0}'.format(self.name)
+        cmd << 'env exabgp.log.destination={0}/exabgpd.log'.format(self.SHARED_VOLUME)
+        cmd << './exabgp/sbin/exabgp {0}/exabgpd.conf'.format(self.SHARED_VOLUME)
+        local(str(cmd), capture=True)
+
+    def _update_exabgp(self):
+        c = CmdBuffer()
+        c << '#!/bin/bash'
+
+        remotepath = '/root/exabgp'
+        localpath = self.exabgp_path
+        if localpath != '':
+            local('cp -r {0} {1}'.format(localpath, self.config_dir))
+            c << 'cp {0}/etc/exabgp/exabgp.env {1}'.format(remotepath, self.SHARED_VOLUME)
+            c << 'sed -i -e \'s/all = false/all = true/g\' {0}/exabgp.env'.format(self.SHARED_VOLUME)
+            c << 'cp -r {0}/exabgp {1}'.format(self.SHARED_VOLUME,
+                                               remotepath[:-1*len('exabgp')])
+            c << 'cp {0}/exabgp.env {1}/etc/exabgp/'.format(self.SHARED_VOLUME, remotepath)
+        cmd = 'echo "{0:s}" > {1}/update.sh'.format(c, self.config_dir)
+        local(cmd, capture=True)
+        cmd = 'chmod 755 {0}/update.sh'.format(self.config_dir)
+        local(cmd, capture=True)
+        cmd = 'docker exec {0} {1}/update.sh'.format(self.name,
+                                                     self.SHARED_VOLUME)
+        local(cmd, capture=True)
+
+    def run(self):
+        super(ExaBGPContainer, self).run()
+        self._update_exabgp()
+        self._start_exabgp()
+        return self.WAIT_FOR_BOOT
+
+    def create_config(self):
+        cmd = CmdBuffer()
+        for peer, info in self.peers.iteritems():
+            cmd << 'neighbor {0} {{'.format(info['neigh_addr'].split('/')[0])
+            cmd << '    router-id {0};'.format(self.router_id)
+
+            local_addr = ''
+            for me, you in itertools.product(self.ip_addrs, peer.ip_addrs):
+                if me[2] == you[2]:
+                    local_addr = me[1]
+            if local_addr == '':
+                raise Exception('local_addr not found')
+            local_addr = local_addr.split('/')[0]
+            cmd << '    local-address {0};'.format(local_addr)
+            cmd << '    local-as {0};'.format(self.asn)
+            cmd << '    peer-as {0};'.format(peer.asn)
+
+            cmd << '    static {'
+            for route, attr in self.routes.iteritems():
+                if attr == '':
+                    cmd << '        route {0} next-hop {1};'.format(route, local_addr)
+                else:
+                    cmd << '        route {0} next-hop {1} attribute {2};'.format(route, local_addr, attr)
+            cmd << '    }'
+            cmd << '}'
+
+        with open('{0}/exabgpd.conf'.format(self.config_dir), 'w') as f:
+            print colors.yellow(str(cmd))
+            f.write(str(cmd))
+
+    def reload_config(self):
+        cmd = 'docker exec {0} /usr/bin/pkill exabgp -SIGALRM'.format(self.name)
+        try:
+            local(cmd, capture=True)
+        except:
+            self._start_exabgp()