import time
from sat.tester.tests import base_prepost

class SatSuitePrePostSteps(base_prepost.BaseSuitePrePostSteps):
    def SAT_PRE_SUITE(self):
        self.logger.log("**************SAT_PRE_SUITE called")
        self.setDutBoardInfo('board_name', 'boardNameHere')
        self.setDutPlatformInfo('DutPlatformInfoHere')
        rid = {}
        for dev in self.devices:
            device = self.devices[dev]
            if hasattr(self.devices[dev], 'btNetMgr') and device.haveBTNetMgr and device.btNetMgr.haveUiautomator2:
                if self.runtime.INIT_DEVICE_TOOLS:
                    self.addSapi(device.netDeviceMgr.SAPI_INIT_DEVICE_TOOLS())
                self.addSapi(device.wlanNetMgr.WL.SAPI_WL_ENABLE_MEMUSE(device.INTERFACES.STA))
                self.addSapi(device.btNetMgr.bttool.SAPI_WIFI_ON())
                self.addSapi(device.btNetMgr.bttool.SAPI_RESET_CRASH_REPORT())
                self.addSapi(device.btNetMgr.bttool.SAPI_GO_HOME())
                rid[dev] = self.addSapi(device.btNetMgr.bttool.SAPI_GET_BT_INFO())
            if hasattr(self.devices[dev], 'btNetMgr') and device.haveBTNetMgr and device.btNetMgr.haveBluez:
                if self.runtime.INIT_DEVICE_TOOLS:
                    self.addSapi(device.netDeviceMgr.SAPI_INIT_DEVICE_TOOLS())
            if self.runtime.TAKE_SYSTEM_LOGS and device.TYPE == 'STA':
                self.addSapi(device.netDeviceMgr.SAPI_START_SYSTEM_LOG_COLLECTION())
        self.execSapi()
        for dev in self.devices:
            device = self.devices[dev]
            if hasattr(self.devices[dev], 'btNetMgr') and device.haveBTNetMgr and device.btNetMgr.haveUiautomator2:
                device.bt_name = self.getSapiOut(rid[dev], 'bt_name')
                device.bt_addr = self.getSapiOut(rid[dev], 'bt_addr')
                device.build = self.getSapiOut(rid[dev], 'build')
                device.bt_baseline = self.getSapiOut(rid[dev], 'bt_baseline')
                self.setTestEnvironmentInfo(dev + '_bt_name', device.bt_name)
                self.setTestEnvironmentInfo(dev + '_bt_addr', device.bt_addr)
                self.setTestEnvironmentInfo(dev + '_SW_Version', device.build)
                self.setTestEnvironmentInfo(dev + '_bt_baseline', device.bt_baseline)
                self.setTestEnvironmentInfo(dev + '_CRASH_REPORT', 'crash_reporter.txt (PFA in Comments)')

    def SAT_POST_SUITE(self):
        self.logger.log("**************SAT_POST_SUITE called")
        for dev in self.devices:
            device = self.devices[dev]
            if self.runtime.TAKE_SYSTEM_LOGS and device.TYPE == 'STA':
                self.addSapi(device.netDeviceMgr.SAPI_STOP_SYSTEM_LOG_COLLECTION())
        self.execSapi()


class SatTestPrePostSteps(base_prepost.BaseTestPrePostSteps):
    def SAT_PRE_TEST(self):
        self.logger.log("**************SAT_PRE_TEST called")
        self.curTest.testStepsRuntime = list()
        rid = {}
        for dev in self.devices:
            if hasattr(self.devices[dev], 'btNetMgr') and self.devices[dev].haveBTNetMgr and self.devices[dev].btNetMgr.haveBluez:
                rid[dev] = self.addSapi(self.devices[dev].btNetMgr.SAPI_GET_INFO())
        self.execSapi()
        for dev in self.devices:
            if hasattr(self.devices[dev], 'btNetMgr') and self.devices[dev].haveBTNetMgr and self.devices[dev].btNetMgr.haveBluez:
                setattr(self.devices[dev], "bt_addr", self.getSapiOut(rid[dev], 'bt_addr'))
                setattr(self.devices[dev], "bt_name", self.getSapiOut(rid[dev], 'bt_name'))

    def SAT_PRE_STEP(self):
        self.logger.log("**************SAT_PRE_STEP called")
        self.testStepStartTime = time.time()
        for dev in self.devices:
            if hasattr(self.devices[dev], 'btNetMgr') and self.devices[dev].haveBTNetMgr and self.devices[
                dev].btNetMgr.haveUiautomator2:
                self.addSapi(self.devices[dev].btNetMgr.bttool.SAPI_TestInit(
                    '{}_{}_{}_logcat.log'.format(self.testId, self.currentStepIndex, dev)))
        self.execSapi()

    def SAT_PRE_CASE(self):
        self.logger.log("**************SAT_PRE_CASE called")

    def SAT_POST_CASE(self):
        self.logger.log("**************SAT_POST_CASE called")

    def SAT_POST_STEP(self):
        testStepEndTime = time.time()
        self.curTest.testStepsRuntime.append({self.curTest.currentStepNames['{}'.format(self.currentStepIndex)]:
                                                  testStepEndTime - self.testStepStartTime})
        self.logger.info("{} step time - {} seconds".format(self.curTest.currentStepNames['{}'.format(
            self.currentStepIndex)], testStepEndTime - self.testStepStartTime))
        self.logger.log("**************SAT_POST_STEP called")
        try:
            for dev in self.devices:
                if hasattr(self.devices[dev], 'btNetMgr') and self.devices[dev].haveBTNetMgr and \
                        self.devices[dev].btNetMgr.haveUiautomator2:
                    self.addSapi(self.devices[dev].btNetMgr.bttool.SAPI_TestTearDown(
                        cur_test_id='{}_{}_{}_snoop.log'.format(self.testId, self.currentStepIndex, dev)))
            self.execSapi()
        except self.error:
            pass

    def SAT_POST_TEST(self):
        self.logger.log("**************SAT_POST_TEST called")