1#!/usr/bin/env python3
2#-*- coding: utf-8 -*-
3
4# Copyright (c) 2024 Huawei Device Co., Ltd.
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9#     http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17import time
18from devicetest.core.test_case import TestCase, CheckPoint
19from hypium import UiDriver
20from tools.get_source_path import get_source_path
21from tools.push_remove_source import push_source, remove_source
22
23
24class case19_switch002(TestCase):
25
26    def __init__(self, configs):
27        self.TAG = self.__class__.__name__
28        TestCase.__init__(self, self.TAG, configs)
29        self.tests = [
30            "test_step"
31        ]
32        self.driver = UiDriver(self.device1)
33        self.sn = self.device1.device_sn
34        self.source_path = {}
35
36    def setup(self):
37        self.log.info("case19_switch002 start")
38        self.driver.Screen.wake_up()
39        self.driver.Bluetooth.disable()
40        need_source = {"cfg": True, "fwk": False, "listen_test": True, "audio_ability": False, "ondemand": True,
41                       "proxy": True, "para": False}
42        self.source_path = get_source_path(need_source=need_source, casename="level0/case19_switch002")
43        push_source(source_path=self.source_path, driver=self.driver, sn=self.sn)
44
45    def test_step(self):
46        driver = self.driver
47        result = driver.System.execute_command("hidumper -ls")
48        assert "1494" not in result
49
50        driver.Bluetooth.enable()
51        result = driver.System.execute_command("hidumper -ls")
52        max_wait_time = 10
53        wait_time = 0
54        while ("1494" not in result and wait_time <= max_wait_time):
55            wait_time += 1
56            time.sleep(1)
57            result = driver.System.execute_command("hidumper -ls")
58        CheckPoint("After turning on Bluetooth, 1494 was successfully loaded")
59        assert "1494" in result
60
61        driver.Bluetooth.disable()
62        time.sleep(20)
63        result = driver.System.execute_command("hidumper -ls")
64        max_wait_time = 5
65        wait_time = 0
66        while ("1494" in result and wait_time <= max_wait_time):
67            wait_time += 1
68            time.sleep(1)
69            result = driver.System.execute_command("hidumper -ls")
70        CheckPoint("After turning off Bluetooth, 1494 was successfully unloaded")
71        assert "1494" not in result
72
73    def teardown(self):
74        self.driver.Screen.close()
75        remove_source(source_path=self.source_path, driver=self.driver, sn=self.sn)
76        self.log.info("case19_switch002 down")
77