A quick test interface to run tests with voltha config.
A sample run to test tls and igmp with voltha configuration without changing test config files:
./olt-test.py --test-type=tls:eap_auth_exchange.test_eap_tls,igmp:igmp_exchange.test_igmp_join_verify_traffic --manifest=manifest-ponsim-local.json --voltha-host=172.17.0.1 --olt-type=ponsim_olt --olt-arg=172.17.0.1
Change-Id: I81357adc87309fd31193bdfd4dfae549506262a5
diff --git a/src/test/setup/olt-test.py b/src/test/setup/olt-test.py
new file mode 100755
index 0000000..287d7a3
--- /dev/null
+++ b/src/test/setup/olt-test.py
@@ -0,0 +1,179 @@
+#!/usr/bin/env python
+
+# Copyright 2017-present Open Networking Foundation
+#
+# 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 argparse import ArgumentParser
+import os
+import sys
+utils_dir = os.path.join( os.path.dirname(os.path.realpath(__file__)), '../utils')
+cli_dir = os.path.join( os.path.dirname(os.path.realpath(__file__)), '../cli')
+sys.path.append(utils_dir)
+sys.path.append(cli_dir)
+sys.path.insert(1, '/usr/local/lib/python2.7/dist-packages')
+from CordTestUtils import getstatusoutput
+import time
+import requests
+import httplib
+import json
+import signal
+
+class CordTesterWebClient(object):
+
+ def __init__(self, host = 'localhost', port = 5000):
+ self.host = host
+ self.port = port
+ self.rest = 'http://{}:{}'.format(self.host, self.port)
+
+ def get_config(self, test_case):
+ rest_uri = '{}/get'.format(self.rest)
+ config = { 'test_case' : test_case }
+ resp = requests.get(rest_uri, data = json.dumps(config))
+ if resp.ok and resp.status_code == 200:
+ config = resp.json()
+ return config
+ return None
+
+ def set_config(self, test_case, test_config):
+ rest_uri = '{}/update'.format(self.rest)
+ config = { 'test_case' : test_case, 'config' : test_config }
+ resp = requests.post(rest_uri, data = json.dumps(config))
+ return resp.ok, resp.status_code
+
+ def restore_config(self, test_case):
+ rest_uri = '{}/restore'.format(self.rest)
+ config = { 'test_case' : test_case }
+ resp = requests.post(rest_uri, data = json.dumps(config))
+ return resp.ok, resp.status_code
+
+ def start(self, manifest = 'manifest.json'):
+ rest_uri = '{}/start'.format(self.rest)
+ config = { 'manifest' : manifest }
+ resp = requests.post(rest_uri, data = json.dumps(config))
+ return resp.ok, resp.status_code
+
+ def cleanup(self, manifest = 'manifest.json'):
+ rest_uri = '{}/cleanup'.format(self.rest)
+ config = { 'manifest' : manifest }
+ resp = requests.post(rest_uri, data = json.dumps(config))
+ return resp.ok, resp.status_code
+
+ def test(self, test, manifest = 'manifest.json', test_config = None):
+ rest_uri = '{}/test'.format(self.rest)
+ config = { 'manifest' : manifest, 'test' : test }
+ if test_config:
+ config['config'] = test_config
+ resp = requests.post(rest_uri, data = json.dumps(config))
+ return resp.ok, resp.status_code
+
+class Tester(CordTesterWebClient):
+
+ def __init__(self, host = 'localhost', port = 5000):
+ super(Tester, self).__init__(host = host, port = port)
+
+ def execute(self, test_case, manifest = 'manifest.json', test_config = None):
+ print('Executing test %s' %test_case)
+ _, status = self.start(manifest = manifest)
+ assert status == httplib.OK, 'Test setup failed with status code %d' %status
+ _, status = self.test(test_case, manifest = manifest, test_config = test_config)
+ assert status == httplib.OK, 'Test run for test %s failed with status %d' %(test_case, status)
+ _, status = self.cleanup(manifest = manifest)
+ assert status == httplib.OK, 'Test cleanup failed with status %d' %status
+ print('Test executed successfully')
+
+
+class CordTesterWeb(object):
+
+ def __init__(self, args, start_in = 3):
+ self.args = args
+ self.tester = Tester()
+ self.start_in = start_in
+
+ def run(self):
+ manifest = self.args.manifest
+ olt_type = self.args.olt_type
+ test_type = self.args.test_type
+ test_config = { 'VOLTHA_HOST' : self.args.voltha_host,
+ 'VOLTHA_OLT_TYPE' : self.args.olt_type,
+ }
+ if olt_type.startswith('tibit'):
+ test_config['VOLTHA_OLT_MAC'] = self.args.olt_arg
+ elif olt_type.startswith('maple'):
+ test_config['VOLTHA_OLT_IP'] = self.args.olt_arg
+ elif olt_type.startswith('ponsim'):
+ test_config['VOLTHA_PONSIM_HOST'] = self.args.olt_arg
+ else:
+ print('Unsupported OLT type %s' %olt_type)
+ return 127
+
+ if self.start_in:
+ time.sleep(self.start_in)
+
+ _, status = self.tester.start(manifest = manifest)
+ assert status == httplib.OK, 'Test setup failed with status %d' %status
+
+ for test in test_type.split(','):
+ print('Running test case %s' %(test))
+ _, status = self.tester.test(test, manifest = manifest, test_config = test_config)
+ if status != httplib.OK:
+ print('Test case %s failed with status code %d' %(test, status))
+
+ print('Cleaning up the test')
+ self.tester.cleanup(manifest = manifest)
+ return 0 if status == httplib.OK else 127
+
+class CordTesterWebServer(object):
+
+ server_path = os.path.dirname(os.path.realpath(__file__))
+ server = 'webserver-run.py'
+ pattern = 'pgrep -f "python ./{}"'.format(server)
+
+ def running(self):
+ st, _ = getstatusoutput(self.pattern)
+ return True if st == 0 else False
+
+ def kill(self):
+ st, output = getstatusoutput(self.pattern)
+ if st == 0 and output:
+ pids = output.strip().splitlines()
+ for pid in pids:
+ try:
+ os.kill(int(pid), signal.SIGKILL)
+ except:
+ pass
+
+ def start(self):
+ if self.running() is False:
+ print('Starting CordTester Web Server')
+ cmd = 'cd {} && python ./{} &'.format(self.server_path, self.server)
+ os.system(cmd)
+
+def run_test(args):
+ testWebServer = CordTesterWebServer()
+ testWebServer.start()
+ testWeb = CordTesterWeb(args, start_in = 3)
+ status = testWeb.run()
+ testWebServer.kill()
+ return status
+
+if __name__ == '__main__':
+ parser = ArgumentParser(description = 'Olt tester')
+ parser.add_argument('-test-type', '--test-type', default = 'tls:eap_auth_exchange.test_eap_tls', help = 'Test type to run')
+ parser.add_argument('-manifest', '--manifest', default='manifest-voltha.json', help = 'Manifest file to use')
+ parser.add_argument('-voltha-host', '--voltha-host', default='172.17.0.1', help = 'VOLTHA host ip')
+ parser.add_argument('-olt-type', '--olt-type', default = 'ponsim_olt', help = 'OLT type')
+ parser.add_argument('-olt-arg', '--olt-arg', default = '172.17.0.1', help = 'OLT type argument')
+ parser.set_defaults(func = run_test)
+ args = parser.parse_args()
+ res = args.func(args)
+ sys.exit(res)