summaryrefslogtreecommitdiffhomepage
path: root/test/performance_test/test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/performance_test/test.py')
-rw-r--r--test/performance_test/test.py94
1 files changed, 0 insertions, 94 deletions
diff --git a/test/performance_test/test.py b/test/performance_test/test.py
deleted file mode 100644
index 4d3e8cc8..00000000
--- a/test/performance_test/test.py
+++ /dev/null
@@ -1,94 +0,0 @@
-from lib.gobgp import *
-from lib.quagga import *
-from lib.bird import *
-from fabric.api import local
-from optparse import OptionParser
-import sys
-import os
-
-def make_tester_ctn(tag='tester', local_gobgp_path='', from_image='osrg/quagga'):
- if local_gobgp_path == '':
- local_gobgp_path = os.getcwd()
-
- c = CmdBuffer()
- c << 'FROM {0}'.format(from_image)
- c << 'ADD gobgp /go/src/github.com/osrg/gobgp/'
- c << 'RUN go get github.com/osrg/gobgp/test/performance_test'
- c << 'RUN go install github.com/osrg/gobgp/test/performance_test'
-
- rindex = local_gobgp_path.rindex('gobgp')
- if rindex < 0:
- raise Exception('{0} seems not gobgp dir'.format(local_gobgp_path))
-
- workdir = local_gobgp_path[:rindex]
- with lcd(workdir):
- local('echo \'{0}\' > Dockerfile'.format(str(c)))
- local('docker build -t {0} .'.format(tag))
- local('rm Dockerfile')
-
-
-class DummyPeer(object):
- def __init__(self, asn):
- self.asn = asn
-
-if __name__ == '__main__':
- parser = OptionParser()
- parser.add_option("-t", "--target", dest="target_rs", default="gobgp")
- parser.add_option("-n", "--num-peer", dest="num_peer", type="int", default=1000)
- (options, args) = parser.parse_args()
-
- if options.target_rs == "gobgp":
- target = GoBGPContainer("target", 1000, "10.10.0.1", log_level='info')
- elif options.target_rs == "quagga":
- target = QuaggaBGPContainer("target", 1000, "10.10.0.1")
- elif options.target_rs == "bird":
- target = BirdContainer("target", 1000, "10.10.0.1")
- else:
- print 'Unknown target implementation:', options.target_rs
- sys.exit(1)
-
- if options.num_peer < 0 or options.num_peer > 255*255:
- print 'invalid number of peer'
- sys.exit(1)
-
- br = Bridge("br01", with_ip=False)
- tester = Container("tester", "tester")
- tester.shared_volumes.append(('/tmp', '/root/shared_volume'))
- tester.run()
- target.run()
- br.addif(tester)
- br.addif(target)
- target.local("ip a add 10.10.0.1/16 dev eth1")
-
- c = CmdBuffer()
- c << "#!/bin/sh"
- for i in range(options.num_peer):
- p = DummyPeer(i+1001)
- neigh_addr = '10.10.{0}.{1}/16'.format((i+2)/255, (i+2)%255)
- target.peers[p] = {'neigh_addr': neigh_addr,
- 'passwd': None,
- 'evpn': False,
- 'flowspec': False,
- 'is_rs_client': True,
- 'is_rr_client': False,
- 'cluster_id': None,
- 'policies': {},
- 'passive': True,
- 'local_addr': '10.10.0.1'}
- c << "ip a add {0} dev eth1".format(neigh_addr)
- with open('/tmp/setup_tester.sh', 'w') as f:
- f.write(str(c))
- local('chmod +x /tmp/setup_tester.sh')
- tester.local("/root/shared_volume/setup_tester.sh")
-
- c = CmdBuffer()
- c << "#!/bin/sh"
- c << "performance_test -n {0} {1} > /root/shared_volume/test.log".format(options.num_peer, ' '.join(args))
- with open('/tmp/start_test.sh', 'w') as f:
- f.write(str(c))
- local('chmod +x /tmp/start_test.sh')
-
- target.create_config()
- target.reload_config()
-
- tester.local("/root/shared_volume/start_test.sh")