1#  Copyright (C) 2023 The Android Open Source Project
2#
3#  Licensed under the Apache License, Version 2.0 (the "License");
4#  you may not use this file except in compliance with the License.
5#  You may obtain a copy of the License at
6#
7#       http://www.apache.org/licenses/LICENSE-2.0
8#
9#  Unless required by applicable law or agreed to in writing, software
10#  distributed under the License is distributed on an "AS IS" BASIS,
11#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12#  See the License for the specific language governing permissions and
13#  limitations under the License.
14
15
16from bluetooth_test import bluetooth_base_test
17from mobly import asserts
18
19from utilities import constants
20from utilities.main_utils import common_main
21
22
23class BluetoothPalette(bluetooth_base_test.BluetoothBaseTest):
24    """Enable and Disable Bluetooth from Bluetooth Palette."""
25
26    def setup_test(self):
27        """Setup steps before any test is executed."""
28        # Pair the devices
29        self.bt_utils.pair_primary_to_secondary();
30
31    def test_enable_disable_bluetooth_button(self):
32        """Tests enable and disable functionality of bluetooth."""
33        self.call_utils.open_bluetooth_palette()
34        self.call_utils.wait_with_log(5)
35        asserts.assert_true(self.call_utils.is_bluetooth_connected(), 'Bluetooth Connected')
36        self.call_utils.click_bluetooth_button()
37        self.call_utils.wait_with_log(5)
38        asserts.assert_false(self.call_utils.is_bluetooth_connected(), 'Bluetooth Disconnected')
39
40
41if __name__ == '__main__':
42    common_main()
43