| #!/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) |