"""Tests for Bluetooth connection with Android device and a Bluetooth device."""

import time

from mobly import test_runner
from blueberry.utils import blueberry_base_test

# Connection state change sleep time in seconds.
CONNECTION_STATE_CHANGE_SLEEP_SEC = 5


class BluetoothConnectionTest(blueberry_base_test.BlueberryBaseTest):
  """Test Class for Bluetooth connection testing.

  Attributes:
    primary_device: A primary device under test.
    derived_bt_device: A Bluetooth device which is used to connected to the
        primary device in the test.
  """

  def setup_class(self):
    super().setup_class()
    self.primary_device = self.android_devices[0]
    self.primary_device.init_setup()

    self.derived_bt_device = self.derived_bt_devices[0]
    self.derived_bt_device.factory_reset_bluetooth()
    self.mac_address = self.derived_bt_device.get_bluetooth_mac_address()
    self.derived_bt_device.activate_pairing_mode()
    self.primary_device.pair_and_connect_bluetooth(self.mac_address)

  def setup_test(self):
    super().setup_test()
    # Checks if A2DP and HSP profiles are connected.
    self.wait_for_a2dp_and_hsp_connection_state(connected=True)
    # Buffer between tests.
    time.sleep(CONNECTION_STATE_CHANGE_SLEEP_SEC)

  def wait_for_a2dp_and_hsp_connection_state(self, connected):
    """Asserts that A2DP and HSP connections are in the expected state.

    Args:
      connected: bool, True if the expected state is connected else False.
    """
    self.primary_device.wait_for_a2dp_connection_state(self.mac_address,
                                                       connected)
    self.primary_device.wait_for_hsp_connection_state(self.mac_address,
                                                      connected)

  def test_disconnect_and_connect(self):
    """Test for DUT disconnecting and then connecting to the remote device."""
    self.primary_device.log.info('Disconnecting the device "%s"...' %
                                 self.mac_address)
    self.primary_device.disconnect_bluetooth(self.mac_address)
    self.wait_for_a2dp_and_hsp_connection_state(connected=False)
    # Buffer time for connection state change.
    time.sleep(CONNECTION_STATE_CHANGE_SLEEP_SEC)
    self.primary_device.log.info('Connecting the device "%s"...' %
                                 self.mac_address)
    self.primary_device.connect_bluetooth(self.mac_address)
    self.wait_for_a2dp_and_hsp_connection_state(connected=True)

  def test_reconnect_when_enabling_bluetooth(self):
    """Test for DUT reconnecting to the remote device when Bluetooth enabled."""
    self.primary_device.log.info('Turning off Bluetooth...')
    self.primary_device.sl4a.bluetoothToggleState(False)
    self.primary_device.wait_for_bluetooth_toggle_state(enabled=False)
    self.primary_device.wait_for_disconnection_success(self.mac_address)
    time.sleep(CONNECTION_STATE_CHANGE_SLEEP_SEC)
    self.primary_device.log.info('Turning on Bluetooth...')
    self.primary_device.sl4a.bluetoothToggleState(True)
    self.primary_device.wait_for_bluetooth_toggle_state(enabled=True)
    self.primary_device.wait_for_connection_success(self.mac_address)
    self.wait_for_a2dp_and_hsp_connection_state(connected=True)

  def test_reconnect_when_connected_device_powered_on(self):
    """Test for the remote device reconnecting to DUT.

    Tests that DUT can be disconnected when the remoted device is powerd off,
    and then reconnected when the remote device is powered on.
    """
    self.primary_device.log.info(
        'The connected device "%s" is being powered off...' % self.mac_address)
    self.derived_bt_device.power_off()
    self.primary_device.wait_for_disconnection_success(self.mac_address)
    self.wait_for_a2dp_and_hsp_connection_state(connected=False)
    time.sleep(CONNECTION_STATE_CHANGE_SLEEP_SEC)
    self.derived_bt_device.power_on()
    self.primary_device.log.info(
        'The connected device "%s" is being powered on...' % self.mac_address)
    self.primary_device.wait_for_connection_success(self.mac_address)
    self.wait_for_a2dp_and_hsp_connection_state(connected=True)


if __name__ == '__main__':
  test_runner.main()