wlan-lanforge-scripts/py-scripts/lf_wifi_capacity_test.py

"""
Note: This is a test file which will run a wifi capacity test.
    ex. on how to run this script:
    ./lf_wifi_capacity_test.py -lfmgr "localhost -port "8080" -test_name "WiFi Capacity" -instance_name
        "test_instance" -config_name "config_file"  -pull_report y -batch_size "1" -loop_iter 1 -protocol "TCP-IPv4" -duration 5000
    if -r flag is set to y. at end of the script you will receive a report dir. from lanforge to your local machine
"""
import sys
import os
import argparse
import time
import json
from os import path

if sys.version_info[0] != 3:
    print("This script requires Python 3")
    exit(1)

if 'py-json' not in sys.path:
    sys.path.append(os.path.join(os.path.abspath('..'), 'py-json'))

from cv_test_manager import cv_test as cvtest
from cv_commands import chamberview as cv
from cv_test_reports import lanforge_reports as lf_rpt

def main():
    parser = argparse.ArgumentParser(description="""
        ./lf_wifi_capacity_test.py -lfmgr "localhost -port "8080" -test_name "WiFi Capacity" -instance_name 
        "test_instance" -config_name "config_file" -pull_report y -batch_size "1" -loop_iter 1 -protocol "TCP-IPv4" -duration 5000""")
    parser.add_argument("-m", "--lfmgr", type=str, default="localhost",
                        help="address of the LANforge GUI machine (localhost is default)")
    parser.add_argument("-o", "--port", type=int, default="8080",
                        help="IP Port the LANforge GUI is listening on (8080 is default)")
    parser.add_argument("-t", "--test_name", type=str, default="WiFi Capacity",
                        help="name of test to be run ex. \"WiFi Capacity\"")
    parser.add_argument("-i", "--instance_name", type=str, required=True,
                        help="name of test instance (by default: test_ref)")
    parser.add_argument("-c", "--config_name", type=str, required=True,
                        help="Test config name (by default: DEFAULT)")
    parser.add_argument("-r", "--pull_report", type=str, required=True,
                        help="pull reports from lanforge (by default: y)")
    parser.add_argument("-b", "--batch_size", type=str, required=True,
                        help="config batch size (by default: 1)")
    parser.add_argument("-l", "--loop_iter", type=str, required=True,
                        help="config loop iter (by default: 1)")
    parser.add_argument("-p", "--protocol", type=str, required=True,
                        help="config protocol (by default: TCP-IPv4)")
    parser.add_argument("-d", "--duration", type=str, required=True,
                        help="config duration (by default: 5000)")

    args = parser.parse_args()
    if args.lfmgr is not None:
        lf_host = args.lfmgr
    if args.port is not None:
        lf_hostport = args.port

    try:
        test_name = args.test_name
        instance_name = args.instance_name
        config_name = args.config_name
        batch_size = args.batch_size
        loop_iter = args.loop_iter
        protocol = args.protocol
        duration = args.duration
        pull_report = args.pull_report
    except:
        print("Wrong arguments entered")
        exit(1)


    # Test related settings
    dict = {"batch_size": "batch_size:" + " " + str(batch_size),
            "loop_iter": "loop_iter:" + " " + str(loop_iter),
            "protocol": "protocol:" + " " + str(protocol),
            "duration": "duration:" + " " + str(duration)}

    run_test = cvtest(lf_host, lf_hostport)
    createCV = cv(lf_host, lf_hostport);  # Create a object

    port_list = []

    response = run_test.check_ports();
    port_size = json.dumps(len(response["interfaces"]))

    for i in range(int(port_size)):
        list_val = json.dumps(response["interfaces"][i])
        list_val_ = json.loads(list_val).keys()
        list_val_ = str(list_val_).replace("dict_keys(['", "")
        list_val_ = str(list_val_).replace("'])", "")
        if (list_val_.__contains__("sta") or list_val_.__contains__("eth1")):
            port_list.append(list_val_)

    for i in range(len(port_list)):
        add_port = "sel_port-" + str(i) + ":" + " " + port_list[i]
        run_test.create_test_config(config_name,"Wifi-Capacity-",add_port)
        time.sleep(0.2)

    for key, value in dict.items():
        run_test.create_test_config(config_name,"Wifi-Capacity-",value)
        time.sleep(0.2)

    run_test.create_test(test_name, instance_name)
    time.sleep(5)
    createCV.sync_cv()
    time.sleep(2)
    run_test.load_test_config(config_name, instance_name)
    time.sleep(2)
    run_test.auto_save_report(instance_name)
    time.sleep(4)
    run_test.start_test(instance_name)

    while (True):
        check = run_test.get_report_location(instance_name)
        location = json.dumps(check[0]["LAST"]["response"])
        print("WiFi Capacity Test Running...")
        if location != "\"Report Location:::\"":
            location = location.replace("Report Location:::", "")
            print(location)
            time.sleep(1)
            run_test.close_instance(instance_name)
            time.sleep(1)
            run_test.cancel_instance(instance_name)
            time.sleep(4)
            location = location.strip("\"")
            report = lf_rpt()
            print(location)
            if (pull_report == "yes" ) or (pull_report == "y") or (pull_report == "Y"):
                report.pull_reports(hostname=lf_host, username="lanforge", password="lanforge",
                                    report_location=location)
            break


if __name__ == "__main__":
    main()