esp-idf/components/test/TestCaseScript/WiFiStress/WifiSmartConfig.py

import random
import os
import time
import copy

from TCAction import TCActionBase
from NativeLog import NativeLog
from Utility import Encoding
from Utility import MakeFolder

AP_PROP = ("ssid", "ssid_len", "pwd",
           "pwd_len", "channel", "enc", "apc")

SMART_TYPE = ("esp-touch", "airkiss")

TEST_METHOD = ("ssid_broadcast", "ssid_hidden")

HT = ("ht20", "ht40")

TEST_STAT = ("total count", "fail count", "total time", "longest time")

_TEST_STAT_INIT_DICT = {"total count": 0,
                        "fail count": 0,
                        "total time": 0,
                        "longest time": 0,
                        }

LOG_FOLDER = os.path.join("AT_LOG", "Performance", "SmartConfig")


SSID_LEN_RANGE = (1, 32)  # in bytes
ENC_TYPE = (0, 2, 3, 4)  # do not support WEP for 8266 soft AP
PWD_RANGE = {0: [0, 0],
             1: [5, 5],
             2: [8, 32],
             3: [8, 32],
             4: [8, 32],
             }


class WifiSmartConfig(TCActionBase.CommonTCActionBase):

    def __init__(self, name, test_env, cmd_set, timeout=30, log_path=TCActionBase.LOG_PATH):
        TCActionBase.CommonTCActionBase.__init__(self, name, test_env, cmd_set=cmd_set,
                                                 timeout=timeout, log_path=log_path)
        # default value for optional configurable params
        self.test_method = ["ssid_hidden", "ssid_broadcast"]
        self.bssid = "ff:ff:ff:ff:ff:ff"
        self.ht_ap = dict(zip(HT, [("<ht20_ap_ssid>", "<ht20_ap_password>"),
                                   ("<ht40_ap_ssid>", "<ht40_ap_password>")]))
        self.ap_channel = {"ht20": 1, "ht40": 1}
        self.delay_time = 3  # default 3s, wait for scan done
        # load param from excel
        for i in range(1, len(cmd_set)):
            if cmd_set[i][0] != "dummy" and cmd_set[i][0] != "":
                cmd_string = "self." + cmd_set[i][0]
                exec cmd_string

        folder_path = MakeFolder.make_folder(LOG_FOLDER)
        file_name = "SmartConfig_log_%s.log" % (time.strftime("%m%d%H%M%S", time.localtime()))
        self._performance_log_file = os.path.join(folder_path, file_name)

        # type
        self.target_type = ["SSC" if test_env.get_port_by_name("AT1") is None else "AT"]
        self.target_type.append("SSC" if test_env.get_port_by_name("AT2") is None else "AT")

        # test statistics
        # better ways to create?
        _test_stat = dict.fromkeys(TEST_STAT, 0)
        _test_method = dict.fromkeys(TEST_METHOD)
        _test_ht = dict.fromkeys(HT)
        self.test_stat = dict.fromkeys(SMART_TYPE)
        for i in SMART_TYPE:
            self.test_stat[i] = copy.deepcopy(_test_ht)
            for j in HT:
                self.test_stat[i][j] = copy.deepcopy(_test_method)
                for k in TEST_METHOD:
                    self.test_stat[i][j][k] = copy.deepcopy(_test_stat)

        self.result_cntx = TCActionBase.ResultCheckContext(self, test_env, self.tc_name)
        pass

    def _generate_random_ap_prop(self, ht_type):
        ap_prop = dict.fromkeys(AP_PROP)
        # generate target ap_value
        ap_prop["ssid_len"] = random.randint(SSID_LEN_RANGE[0], SSID_LEN_RANGE[1])
        ap_prop["channel"] = self.ap_channel[ht_type]
        ap_prop["enc"] = random.choice(ENC_TYPE)
        ap_prop["pwd_len"] = random.randint(PWD_RANGE[ap_prop["enc"]][0], PWD_RANGE[ap_prop["enc"]][1])
        ap_prop["ssid"] = Encoding.generate_random_printable_str(ap_prop["ssid_len"])
        ap_prop["pwd"] = Encoding.generate_random_printable_str(ap_prop["pwd_len"])

        return ap_prop

    def _logging_performance(self, time_cost, ssid, password, smart_type, test_method, ht_type):
        # update test statistics
        stat = self.test_stat[smart_type][ht_type][test_method]
        stat["total count"] += 1
        # log performance to performance log file
        with open(self._performance_log_file, "ab+") as f:
            # log time and ssid
            if time_cost is not False:
                time_tmp = float(time_cost)/10
                f.write("\r\n[%s]:\r\n[Succeed] [%.2f]\r\n" %
                        (time.strftime("%m-%d %H:%M:%S", time.localtime()), time_tmp))
                stat["total time"] += time_tmp
                stat["longest time"] = time_tmp if time_tmp > stat["longest time"] else stat["longest time"]
            else:
                f.write("\r\n[%s]:\r\n[Fail]\r\n" %
                        time.strftime("%m-%d %H:%M:%S", time.localtime()))
                stat["fail count"] += 1

            f.write("[%s] [%s] [%s]\r\n" %
                    (smart_type, test_method, ht_type))
            f.write("[ssid] %s \r\n[password] %s\r\n" %
                    (ssid, password))
        pass

    def _generate_performance_report(self):
        with open(self._performance_log_file, "ab+") as f:
            for i in SMART_TYPE:
                for j in HT:
                    for k in TEST_METHOD:
                        stat = self.test_stat[i][j][k]
                        f.write("\r\n[Test report] [%s] [%s] [%s]\r\n" % (i, j, k))
                        if stat["total count"] > 0:
                            f.write("[Total]: %d\r\n" % stat["total count"])
                            f.write("[Failed]: %d\r\n" % stat["fail count"])
                            f.write("[Fail ratio]: %.2f%%\r\n" %
                                    (float(stat["fail count"])/stat["total count"] * 100))
                            f.write("[Longest time cost]: %.2f\r\n" % stat["longest time"])
                            if (stat["total count"] - stat["fail count"]) > 0:
                                f.write("[Average time cost]: %.2f\r\n" %
                                        (stat["total time"]/(stat["total count"]-stat["fail count"])))

    @staticmethod
    def cmd_exception_catcher(e):
        raise e
        pass

    def execute(self):
        TCActionBase.TCActionBase.execute(self)
        self.result_cntx.start()

        # mandatory configurable params
        try:
            test_count = self.test_count
            delay_time = self.delay_time
        except StandardError, e:
            NativeLog.add_trace_critical("Error configuration for WifiJAP script, error is %s" % e)
            raise StandardError("Error configuration")

        # step 0 : set AT1 mode
        fail_string = "Fail to restore init condition"
        if self.target_type[0] == "AT":
            cmd = ["ATS AT1 AT+CWMODE=1"]
            checker_stings = ["R AT1 L OK"]
        else:
            cmd = ["SSC SSC1 op -S -o 1"]
            checker_stings = ["R SSC1 C +MODE:OK"]
        if self.target_type[1] == "AT":
            cmd.append("ATS AT2 AT+CWMODE=2")
            checker_stings.append("R AT2 L OK")
        else:
            cmd.append("SSC SSC2 op -S -o 2")
            checker_stings.append("R SSC2 C +MODE:OK")

        if self.load_and_exe_one_step(checker_stings, cmd,
                                      fail_string, check_time=150) is False:
            NativeLog.add_trace_critical(fail_string)
            return

        for i in xrange(test_count):
            _method = random.choice(self.test_method)
            _ht = random.choice(self.ht)
            _ap_prop = self._generate_random_ap_prop(_ht)
            _smart_type = random.choice(self.smart_type)
            _ht_ap = self.ht_ap[_ht]
            is_hidden = 0 if _method == "ssid_broadcast" else 1
            # get ip and

            # step 1 : restore init condition
            fail_string = "Fail to restore init condition"
            if self.target_type[0] == "AT":
                cmd = ["ATS AT1 AT+CWSTOPSMART", "WIFI <pc_wifi_nic> CONN %s %s <pc_ip_sc>" % (_ht_ap[0], _ht_ap[1])]
                checker_stings = ["P AT1 L OK", "P PC_COM L OK"]
            else:
                cmd = ["SSC SSC1 smart -E", "WIFI <pc_wifi_nic> CONN %s %s <pc_ip_sc>" % (_ht_ap[0], _ht_ap[1])]
                checker_stings = ["P SSC1 C +SC:OK", "P PC_COM L OK"]

            if self.load_and_exe_one_step(checker_stings, cmd,
                                          fail_string, check_time=200) is False:
                NativeLog.add_trace_critical(fail_string)
                continue
            NativeLog.add_prompt_trace("Step1 Done")

            # step 2 : test method is ssid_broadcast, then set AP on target 2
            if _method == "ssid_broadcast":
                fail_string = "Fail to set AP"
                if self.target_type[1] == "AT":
                    cmd = ["ATS AT2 AT+CWSAP=\"%s\",\"%s\",%d,%d" % (_ap_prop["ssid"], _ap_prop["pwd"],
                                                                     _ap_prop["channel"], _ap_prop["enc"])]
                    checker_stings = ["R AT2 L OK"]
                else:
                    cmd = ["SSC SSC2 ap -S -s %s -p %s -n %d -t %d" % (_ap_prop["ssid"], _ap_prop["pwd"],
                                                                       _ap_prop["channel"], _ap_prop["enc"])]
                    checker_stings = ["R SSC2 C +SAP:OK"]

                if self.load_and_exe_one_step(checker_stings, cmd,
                                              fail_string, check_time=50) is False:
                    NativeLog.add_trace_critical(fail_string)
                    continue
            NativeLog.add_prompt_trace("Step2 Done")

            # step 3 : start SMART
            fail_string = "Fail to start smart config"
            if self.target_type[0] == "AT":
                cmd = ["ATS AT1 AT+CWSTARTSMART"]
                checker_stings = ["R AT1 L OK"]
            else:
                cmd = ["SSC SSC1 smart -S -a 0"]
                checker_stings = ["R SSC1 C +SC:OK"]

            if self.load_and_exe_one_step(checker_stings, cmd,
                                          fail_string, check_time=50) is False:
                NativeLog.add_trace_critical(fail_string)
                continue
            # sleep for delay_time seconds to wait scan done or simulate delay config situation
            time.sleep(delay_time)
            NativeLog.add_prompt_trace("Step3 Done")

            # step 4 : do smart config
            fail_string = "Fail in smart config"
            cmd = ["SMART %s <pc_ip_sc> %s %s %s %d"
                   % (_smart_type, _ap_prop["ssid"], _ap_prop["pwd"], self.bssid, is_hidden)]
            if self.target_type[0] == "AT":
                checker_stings = ["P AT1 C Smart%20get%20wifi%20info",
                                  "P LOG1 C %s C %s" % (_ap_prop["ssid"], _ap_prop["pwd"])]
            else:
                checker_stings = ["P SSC1 C %s C %s" % (_ap_prop["ssid"], _ap_prop["pwd"])]

            try:
                time_cost = self.load_and_exe_one_step(checker_stings, cmd,
                                                       fail_string, check_time=400,
                                                       cmd_exception_catcher=self.cmd_exception_catcher)
            except StandardError:
                NativeLog.add_prompt_trace("Exception occurred during executing cmd")
                continue
                pass
            self._logging_performance(time_cost, _ap_prop["ssid"], _ap_prop["pwd"],
                                      _smart_type, _method, _ht)
            if time_cost is False:
                NativeLog.add_prompt_trace(fail_string)
                continue

            # continue to next loop
            NativeLog.add_prompt_trace("[WifiSmartConfig] Test count %d done" % i)

        # generate report and cleanup
        self._generate_performance_report()

        self.result_cntx.set_result("Succeed")

    def result_check(self, port_name, data):
        TCActionBase.CommonTCActionBase.result_check(self, port_name, data)
        self.result_cntx.append_data(port_name, data)


def main():
    pass

if __name__ == '__main__':
    main()