"""Tests for Connection switching feature of Triangle."""

import logging
import time

from mobly import test_runner
from blueberry.utils import triangle_base_test as base_test
from blueberry.utils import triangle_constants


class ConnectionSwitchingTest(base_test.TriangleBaseTest):
  """Connection Switching Test."""

  def setup_class(self):
    """Executes Connection switching setups.

    Pairs Phone to headset and Watch, then pairs and connect Watch to Headset,
    let Watch be last connected device of Headset.
    """
    super().setup_class()
    self.headset.power_on()
    self.pair_and_connect_phone_to_headset()
    self.pair_and_connect_phone_to_watch()
    self.pair_and_connect_watch_to_headset()

  def setup_test(self):
    """Makes sure that Headset is connected to Watch instead of Phone."""
    super().setup_test()
    self.phone.disconnect_bluetooth(self.headset.mac_address)
    self.watch.connect_bluetooth(self.headset.mac_address)
    self.assert_headset_a2dp_connection(connected=False, device=self.phone)
    self.assert_headset_hsp_connection(connected=False, device=self.phone)

  def test_trigger_connection_switching_when_headset_powered_on(self):
    """Test for triggering connection switching when Headset is powered on.

    Steps:
      1. Power off Headset.
      2. Wait 1 minute.
      3. Power on Headset, and then it will be reconnect.

    Verifications:
      The Headset connection is switched from Watch to Phone.
    """
    logging.info('Power off Headset and wait 1 minute.')
    self.headset.power_off()
    time.sleep(triangle_constants.WAITING_TIME_SEC)
    logging.info('Power on Headset.')
    self.headset.power_on()
    self.assert_headset_a2dp_connection(connected=True, device=self.phone)
    self.assert_headset_hsp_connection(connected=True, device=self.phone)

  def test_trigger_connection_switching_when_phone_tethered_watch(self):
    """Test for triggering connection switching when Phone is tethered to Watch.

    Steps:
      1. Disable Bluetooth on Phone.
      2. Wait 1 minute.
      3. Enable Bluetooth on Phone, and then Phone will be tethered to Watch.

    Verifications:
      The Headset connection is switched from Watch to Phone.
    """
    self.phone.log.info('Disable Bluetooth and wait 1 minute.')
    self.phone.mbs.btDisable()
    time.sleep(triangle_constants.WAITING_TIME_SEC)
    self.phone.log.info('Enable Bluetooth.')
    self.phone.mbs.btEnable()
    self.assert_headset_a2dp_connection(connected=True, device=self.phone)
    self.assert_headset_hsp_connection(connected=True, device=self.phone)


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