54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
#
|
||
|
|
# 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.
|
||
|
|
#
|
||
|
|
"""Tests for the fastboot command line utility.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import re
|
||
|
|
import subprocess
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
class DevicesTest(unittest.TestCase):
|
||
|
|
"""Tests for `fastboot devices`."""
|
||
|
|
|
||
|
|
|
||
|
|
def test_devices(self):
|
||
|
|
"""Ensure that the format of `fastboot devices` does not change.
|
||
|
|
|
||
|
|
`fastboot devices` should alternate between a line containing the
|
||
|
|
device's serial number and fastboot state and an empty line
|
||
|
|
"""
|
||
|
|
output = subprocess.check_output(["fastboot", "devices"])
|
||
|
|
|
||
|
|
previous_line_was_empty = True
|
||
|
|
for line in output.decode().splitlines():
|
||
|
|
if previous_line_was_empty:
|
||
|
|
if not re.match(r"[a-zA-Z\d]+\s+(bootloader|fastbootd)", line):
|
||
|
|
self.fail("%s does not match the expected format <serial no>\\s+(bootloader|fastbootd)" % line)
|
||
|
|
previous_line_was_empty = False
|
||
|
|
else:
|
||
|
|
if line:
|
||
|
|
self.fail("Expected an empty line. Received '%s'" % line)
|
||
|
|
previous_line_was_empty = True
|
||
|
|
|
||
|
|
if len(output) == 0:
|
||
|
|
self.fail("Output is empty. Are any devices connected?")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|