"""Tests for blueberry.tests.bluetooth_pairing."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import logging

from mobly import test_runner
from mobly import signals
from blueberry.utils import asserts
from blueberry.utils import blueberry_base_test


class BluetoothPairingTest(blueberry_base_test.BlueberryBaseTest):
  """Test Class for Bluetooth Pairing Test.

  Test will reset the bluetooth settings on the phone and attempt to pair
  with the derived_bt_device specified in the configuration file.
  """

  def setup_class(self):
    """Standard Mobly setup class."""
    super(BluetoothPairingTest, self).setup_class()
    # Adds a use case that derived_bt_device initiates a pairing request to
    # primary_device. Enable this case if allow_pairing_reverse is 1.
    self.allow_pairing_reverse = int(self.user_params.get(
        'allow_pairing_reverse', 0))

    if self.android_devices:
      self.primary_device = self.android_devices[0]
      self.primary_device.init_setup()
      self.primary_device.sl4a_setup()

      if len(self.android_devices) > 1 and not self.derived_bt_devices:
        # In the case of pairing phone to phone, we need to treat the
        # secondary phone as a derived_bt_device in order for the generic script
        # to work with this android phone properly.
        self.derived_bt_device = self.android_devices[1]
        self.derived_bt_devices.append(self.derived_bt_device)
        self.derived_bt_device.init_setup()
        self.derived_bt_device.sl4a_setup()
      else:
        self.derived_bt_device = self.derived_bt_devices[0]
        self.derived_bt_device.factory_reset_bluetooth()
    else:
      # In the case of pairing mock to mock, at least 2 derived_bt_device is
      # required. The first derived_bt_device is treated as primary_device.
      self.primary_device = self.derived_bt_devices[0]
      self.primary_device.init_setup()
      self.derived_bt_device = self.derived_bt_devices[1]
      self.derived_bt_device.factory_reset_bluetooth()

  def setup_test(self):
    """Setup for pairing test."""
    logging.info('Setup Test for test_pair_and_bond')
    super(BluetoothPairingTest, self).setup_test()

  def test_pair_and_bond(self):
    """Test for pairing and bonding a phone with a bluetooth device.

    Initiates pairing from the phone and checks for 20 seconds that
    the device is connected.
    """
    device_list = [(self.primary_device, self.derived_bt_device)]
    if self.allow_pairing_reverse:
      device_list.append((self.derived_bt_device, self.primary_device))
    for initiator, receiver in device_list:
      # get mac address of device to pair with
      mac_address = receiver.get_bluetooth_mac_address()
      logging.info('Receiver BT MAC Address: %s', mac_address)
      # put device into pairing mode
      receiver.activate_pairing_mode()
      # initiate pairing from initiator
      initiator.set_target(receiver)
      with asserts.assert_not_raises(signals.ControllerError):
        initiator.pair_and_connect_bluetooth(mac_address)
      if self.allow_pairing_reverse and initiator != self.derived_bt_device:
        logging.info('===== Reversing Pairing =====')
        # Resets Bluetooth status for two sides.
        initiator.factory_reset_bluetooth()
        receiver.factory_reset_bluetooth()


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