def main():
    scan_filter = dict()

    bleAdapter.init()
    agent = securityAgent.createSecurityAgent()

    scan_filter.update({"UUIDs": [runTest.DUT_UUID_128]})
    bleAdapter.setDiscoveryFilter(scan_filter)
    # Discovery test
    bleAdapter.startDiscovery(runTest.discoveryEventCb)
    runTest.mainloop.run()
    isTestSuccessFull = True
    runTest.submitTestResult(isTestSuccessFull, runTest.advertisement)
    bleAdapter.stopDiscovery()

    # Simple Connection test
    testDevice = runTest.getTestDevice()
    isTestSuccessFull = bleAdapter.connect(testDevice)
    runTest.submitTestResult(isTestSuccessFull, runTest.simpleConnection)
    time.sleep(2)  # wait for connection parameters update

    # Discover all primary services
    isTestSuccessFull = runTest.discoverPrimaryServices()
    runTest.submitTestResult(isTestSuccessFull,
                             runTest.discoverPrimaryServices)

    bleAdapter.gatt.updateLocalAttributeTable()

    # Check device not present. After discovery of services, advertisement
    # should have stopped.
    runTest.stopAdvertisement(scan_filter)

    # Change MTU size
    # bleAdapter.readLocalMTU()

    # Check attribute table UUIDs
    bleAdapter.gatt.updateLocalAttributeTable()
    isTestSuccessFull = runTest.checkUUIDs(bleAdapter.gatt)
    runTest.submitTestResult(isTestSuccessFull, runTest.checkUUIDs)

    # Check attribute table properties
    isTestSuccessFull = runTest.checkProperties(bleAdapter.gatt)
    runTest.submitTestResult(isTestSuccessFull, runTest.checkProperties)

    # CHeck long write
    isTestSuccessFull = runTest.writeLongCharacteristic()
    runTest.submitTestResult(isTestSuccessFull,
                             runTest.writeLongCharacteristic)

    # Check read/write, simple connection
    isTestSuccessFull = runTest.readWriteSimpleConnection()
    runTest.submitTestResult(isTestSuccessFull,
                             runTest.readWriteSimpleConnection)

    # check write without response
    isTestSuccessFull = runTest.writeWithoutResponse()
    runTest.submitTestResult(isTestSuccessFull, runTest.writeWithoutResponse)

    # Enable and receive notification and indication then disable.
    bleAdapter.setNotificationCallBack(runTest.notificationCb)
    bleAdapter.subscribeForNotification(
        runTest.DUT_NOTIFY_CHAR_UUID)  # subscribe for next test
    bleAdapter.subscribeForNotification(
        runTest.DUT_INDICATE_CHAR_UUID)  # subscribe for next test
    isTestSuccessFull = True
    runTest.mainloop.run()
    runTest.submitTestResult(isTestSuccessFull, runTest.notification)

    bleAdapter.setNotificationCallBack(runTest.indicationCb)
    isTestSuccessFull = True
    runTest.mainloop.run()
    runTest.submitTestResult(isTestSuccessFull, runTest.indication)

    isTestSuccessFull = bleAdapter.subscribeForNotification(
        runTest.DUT_NOTIFY_CHAR_UUID, subscribe=False)  # unsubscribe
    isTestSuccessFull = True
    runTest.submitTestResult(isTestSuccessFull, runTest.removeNotification)

    isTestSuccessFull = bleAdapter.subscribeForNotification(
        runTest.DUT_INDICATE_CHAR_UUID, subscribe=False)  # unsubscribe
    isTestSuccessFull = True
    runTest.submitTestResult(isTestSuccessFull, runTest.removeIndication)

    # Check writing to protected characteristic triggers pairing
    isTestSuccessFull = runTest.pairing()
    runTest.submitTestResult(isTestSuccessFull, runTest.pairing)
    bleAdapter.bondToRemoteDevice()

    # Check writing to protected characteristic after successfull pairing
    # succeed
    time.sleep(2)  # wait before starting next test
    isTestSuccessFull = runTest.readWriteProtectedAttributesWhilePaired()
    runTest.submitTestResult(isTestSuccessFull,
                             runTest.readWriteProtectedAttributesWhilePaired)

    # disconnect, Note it is not a test happening on bluez, the DUT is waiting
    # for a disconnect Cb
    runTest.disconnect()

    # reconnect! Since devices bonded, it should not ask for pairing again.
    # Security agent can be destroyed
    # remove security agent so as not to trigger auto pairing.
    securityAgent.removeSecurityAgent()
    bleAdapter.setDiscoveryFilter(scan_filter)
    # wait for DUT to start advertising
    bleAdapter.startDiscovery(runTest.discoveryStartedCb)
    runTest.mainloop.run()
    bleAdapter.stopDiscovery()
    runTest.reconnectWhileBonded()

    # reconnect while not bonded. Pairing should fail since Just works is not
    # accepted
    bleAdapter.disconnect()
    bleAdapter.removeBondedDevices()
    time.sleep(2)  # wait for bonded devices to be deleted
    bleAdapter.setDiscoveryFilter(scan_filter)
    bleAdapter.startDiscovery(runTest.discoveryEventCb)
    runTest.mainloop.run()
    bleAdapter.stopDiscovery()

    agent = securityAgent.createSecurityAgent("NoInputNoOutput", agent)
    runTest.reconnectWhileNotBonded()

    time.sleep(2)
    runTest.printTestsSummary()
def main():
    agent = None
    STRESS_TIER = 1

    if (STRESS_TIER == 0):
        STRESS_TEST_INIT_REPETITION = 1
        STRESS_TEST_ENABLE_REPETITION = 1
        STRESS_TEST_MODULE_REPETITION = 1
    elif (STRESS_TIER == 1):
        STRESS_TEST_INIT_REPETITION = 1
        STRESS_TEST_ENABLE_REPETITION = 1
        STRESS_TEST_MODULE_REPETITION = 10
    elif (STRESS_TIER == 2):
        STRESS_TEST_INIT_REPETITION = 2
        STRESS_TEST_ENABLE_REPETITION = 10
        STRESS_TEST_MODULE_REPETITION = 100
    elif (STRESS_TIER == 3):
        STRESS_TEST_INIT_REPETITION = 3
        STRESS_TEST_ENABLE_REPETITION = 100
        STRESS_TEST_MODULE_REPETITION = 1000

    isTestSuccessFull = True

    scan_filter = dict()

    bleAdapter.init()
    agent = securityAgent.createSecurityAgent(agent=agent)

    scan_filter.update({"UUIDs": [runTest.DUT_UUID_128]})

    for i in range(STRESS_TEST_INIT_REPETITION *
                   STRESS_TEST_ENABLE_REPETITION):
        bleAdapter.setDiscoveryFilter(scan_filter)
        for j in range(STRESS_TEST_MODULE_REPETITION):
            if i == 0 and j == 0:
                bleAdapter.startDiscovery(runTest.discoveryEventCb)
            else:
                bleAdapter.startDiscovery(runTest.discoveryStartedCb)
            runTest.mainloop.run()
            bleAdapter.stopDiscovery()

            testDevice = runTest.getTestDevice()
            isConnectSuccessFull = bleAdapter.connect(testDevice)

            isTestSuccessFull &= isConnectSuccessFull
            runTest.stopAdvertisement(scan_filter)
            isTestSuccessFull &= bleAdapter.disconnect()

        bleAdapter.startDiscovery(runTest.discoveryStartedCb)
        runTest.mainloop.run()
        bleAdapter.stopDiscovery()

        runTest._simple_connect()
        runTest.stopAdvertisement(scan_filter)
        # Discover all primary services
        isTestSuccessFull = runTest.discoverPrimaryServices()
        runTest.submitTestResult(isTestSuccessFull,
                                 runTest.discoverPrimaryServices)
        bleAdapter.gatt.updateLocalAttributeTable()
        isTestSuccessFull &= runTest.checkUUIDs(bleAdapter.gatt)
        # Check attribute table properties
        isTestSuccessFull &= runTest.checkProperties(bleAdapter.gatt)

        # Check read/write, simple connection
        for i in range(STRESS_TEST_MODULE_REPETITION):
            isTestSuccessFull &= runTest.readWriteSimpleConnection()
            runTest.submitTestResult(isTestSuccessFull,
                                     runTest.readWriteSimpleConnection)
            time.sleep(2)

        bleAdapter.subscribeForNotification(
            runTest.DUT_NOTIFY_CHAR_UUID)  # subscribe for next test
        bleAdapter.subscribeForNotification(
            runTest.DUT_INDICATE_CHAR_UUID)  # subscribe for next test

        # Check read/write, simple connection
        for i in range(STRESS_TEST_MODULE_REPETITION):
            # Enable and receive notification and indication then disable.
            bleAdapter.setNotificationCallBack(runTest.notificationCb)
            isTestSuccessFull = True
            runTest.mainloop.run()
            runTest.submitTestResult(isTestSuccessFull, runTest.notification)

            bleAdapter.setNotificationCallBack(runTest.indicationCb)
            isTestSuccessFull = True
            runTest.mainloop.run()
            runTest.submitTestResult(isTestSuccessFull, runTest.indication)
        time.sleep(5)

        isTestSuccessFull = bleAdapter.subscribeForNotification(
            runTest.DUT_NOTIFY_CHAR_UUID, subscribe=False)  # unsubscribe
        isTestSuccessFull = True
        runTest.submitTestResult(isTestSuccessFull, runTest.removeNotification)

        isTestSuccessFull = bleAdapter.subscribeForNotification(
            runTest.DUT_INDICATE_CHAR_UUID, subscribe=False)  # unsubscribe
        isTestSuccessFull = True
        runTest.submitTestResult(isTestSuccessFull, runTest.removeIndication)

        if STRESS_TEST_MODULE_REPETITION:
            runTest.submitTestResult(isTestSuccessFull, runTest.reConnection)

        time.sleep(2)

    runTest.printTestsSummary()