"""
  Copyright (C) 2023 The Android Open Source Project

  Licensed under the Apache License, Version 2.0 (the "License");
  you may not use this file except in compliance with the License.
  You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.



  Test Steps:
  (0. Flash device)
  1. Navigate to bluetooth settings and press the Bluetooth toggle (on device listing)
  to disconnect.
  2.  Confirm that Bluetooth, Media, and Audio are disabled
  3. Press the bluetooth toggle on the device listing to reconnect
  4. Confirm that Bluetooth, Media, and Audio are enabled

"""

import logging
from mobly import asserts
from utilities import constants
from utilities.main_utils import common_main
from bluetooth_test import bluetooth_base_test


class BluetoothDisconnectFromSettingsTest(bluetooth_base_test.BluetoothBaseTest):

    def setup_test(self):
        super().setup_test()
        # Pair the devices
        self.bt_utils.pair_primary_to_secondary()
        super().enable_recording()

    def test_disconnect_from_settings(self):
        for _ in range(2):
            # Log BT Connection State after pairing
            bt_connection_state=self.call_utils.get_bt_connection_status_using_adb_command(self.discoverer)
            logging.info("BT State after pairing : <%s>", bt_connection_state)

            # Allow time for connection to fully sync.
            self.call_utils.wait_with_log(constants.WAIT_THIRTY_SECONDS)

            bt_connection_state=self.call_utils.get_bt_connection_status_using_adb_command(self.discoverer)
            logging.info("BT Connection State after wait time : <%s>", bt_connection_state)

            # Navigate to Bluetooth Settings
            self.call_utils.open_bluetooth_settings_form_status_bar()
            self.call_utils.wait_with_log(constants.WAIT_TWO_SECONDS)
            # Press the Bluetooth toggle button to disconnect
            # (Recall that the toggle is the leftmost of the three buttons listed with the name)
            self.call_utils.press_bluetooth_toggle_on_device(self.target.mbs.btGetName())
            self.call_utils.wait_with_log(constants.WAIT_TWO_SECONDS)

            # Confirm that Bluetooth, Media, and Audio are disabled
            asserts.assert_true(
                self.call_utils.validate_three_preference_buttons(False),
                "Expected bluetooth, media, and phone buttons to be disabled after disconnection.")

            # Press the Bluetooth button to reconnect
            self.call_utils.press_bluetooth_toggle_on_device(self.target.mbs.btGetName())
            self.call_utils.wait_with_log(constants.SYNC_WAIT_TIME)

            # Confirm that Bluetooth, Media, and Audio are enabled
            asserts.assert_true(
                self.call_utils.validate_three_preference_buttons(True),
                "Expected bluetooth, media, and phone buttons to be enabled after reconnection.")


if __name__ == '__main__':
    # Take test args
    common_main()