コード例 #1
0
ファイル: fastboot_test.py プロジェクト: WhySoGeeky/DroidPot
class fastboot_testCase(unittest.TestCase):
    def setUp(self):
        self.fastboot = Fastboot()
        self.adb = Adb()
        self.adb.reboot_bootloader()

    def tearDown(self):
        self.fastboot.reboot()

        while len(self.adb.devices()) == 0:
            time.sleep(5)

    @unittest.skip("skip reboot test")
    def test_reboot(self):
        self.fastboot.reboot()
        result = True

        print "[!] waiting for phone to reboot..."
        for i in range(0, 5):
            if len(self.adb.devices()) == 0:
                time.sleep(10)
                result = False
            else:
                result = True
                break

            if i == 4:
                print "[!] Reboot time out"

        self.assertTrue(result, msg="fastboot reboot failed")

    @unittest.skip("skip reboot bootloader test")
    def test_reboot_bootloader(self):
        result = False
        self.fastboot.reboot_bootloader()

        devices = self.fastboot.devices()

        print "[!] waiting for phone to enter bootloader mode..."
        for i in range(0, 5):
            devices = self.fastboot.devices()

            if len(devices) != 0:
                result = True
                break
            else:
                time.sleep(10)

            if i == 4:
                print "[!] Reboot time out"

        self.assertTrue(result, msg="fastboot reboot-bootloader failed")

    def test_devices(self):
        result = False
        self.fastboot.reboot_bootloader()

        time.sleep(5)
        devices = self.fastboot.devices()

        if len(devices) > 0:
            result = True

        self.assertTrue(result, msg="fastboot devices did not detect any devices")