"""
Bluetooth Base Test

This test class serves as a base class for others to inherit from.
It also serves as a device-cleaner to help reset devices between tests.

"""

import sys
import logging
import pprint


from mobly import asserts
from mobly import base_test
from mobly import test_runner
from mobly.controllers import android_device

from mbs_utils import spectatio_utils
from mbs_utils import bt_utils

class BluetoothBaseTest(base_test.BaseTestClass):


    def setup_class(self):
        # Registering android_device controller module, and declaring that the test
        # requires at least two Android devices.
        # This setup will need to be overwritten or extended if a test uses three devices.

        self.ads = self.register_controller(android_device, min_number=2)
        # The device used to discover Bluetooth devices.
        self.discoverer = android_device.get_device(
            self.ads, label='auto')
        # Sets the tag that represents this device in logs.
        self.discoverer.debug_tag = 'discoverer'
        # The device that is expected to be discovered
        self.target = android_device.get_device(self.ads, label='phone')
        self.target.debug_tag = 'target'
        self.target.load_snippet('mbs', android_device.MBS_PACKAGE)
        self.discoverer.load_snippet('mbs', android_device.MBS_PACKAGE)

        self.call_utils = (spectatio_utils.CallUtils(self.discoverer))

        self.bt_utils = (bt_utils.BTUtils(self.discoverer, self.target))

    def setup_test(self):
        # Make sure bluetooth is on.
        logging.info("Running basic test setup.")
        self.target.mbs.btEnable()
        self.discoverer.mbs.btEnable()

    def teardown_test(self):
        # Turn Bluetooth off on both devices.
        logging.info("Running basic test teardown.")
        # unpair target from discoverer
        discoverer_address = self.discoverer.mbs.btGetAddress()
        target_paired_devices = self.target.mbs.btGetPairedDevices()
        _, target_paired_addresses = self.bt_utils.get_info_from_devices(target_paired_devices)
        if discoverer_address in target_paired_addresses:
          logging.info(f"forget {discoverer_address}")
          self.target.mbs.btUnpairDevice(discoverer_address)
        # unpair discoverer from target
        target_address = self.target.mbs.btGetAddress()
        discoverer_paired_devices = self.discoverer.mbs.btGetPairedDevices()
        _, discoverer_paired_addresses = self.bt_utils.get_info_from_devices(discoverer_paired_devices)
        if target_address in discoverer_paired_addresses:
          logging.info(f"forget {target_address}")
          self.discoverer.mbs.btUnpairDevice(target_address)
        # disable BT
        self.target.mbs.btDisable()
        self.discoverer.mbs.btDisable()


if __name__ == '__main__':
    # Pass test arguments after '--' to the test runner.
    # Needed for Mobly Test Runner
    if '--' in sys.argv:
        index = sys.argv.index('--')
        sys.argv = sys.argv[:1] + sys.argv[index + 1:]
    test_runner.main()