#  Copyright (C) 2025 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.

import logging
from mobly import base_test
from mobly import test_runner
from mobly.controllers import android_device
from utilities.main_utils import common_main

class AndroidRebootTest(base_test.BaseTestClass):

    def setup_class(self):
        # Setup required for the entire test class. This executes
        # before any test cases are run.
        self.android_devices = self.register_controller(android_device)
        self.num_reboots = self.user_params['num_reboots']

    def test_reboot_and_check_fingerprint_stability(self):
        # Number of times the reboot operation will be performed.
        num_reboots = self.num_reboots
        num_successful_reboots = 0

        # Ensure there is at least one device available.
        if not self.android_devices:
            raise AssertionError('No Android devices are registered. Please make '
                                 'sure at least one device is connected and '
                                 'configured properly.')

        ad = self.android_devices[0]  # Get the first device object.
        # Record the fingerprint before the first reboot.
        initial_fingerprint = ad.adb.shell('getprop ro.build.fingerprint')
        logging.info('Initial fingerprint: %s', initial_fingerprint)

        for i in range(num_reboots):
            logging.info('Rebooting: Attempt #%d', i + 1)

            # Reboot the device and wait until it's back online.
            ad.reboot()
            ad.wait_for_boot_completion()

            # Check the fingerprint after the reboot.
            current_fingerprint = ad.adb.shell('getprop ro.build.fingerprint')
            logging.info('Fingerprint after reboot #%d: %s', i + 1, current_fingerprint)

            # Verify that the fingerprint remains the same after each reboot.
            if initial_fingerprint != current_fingerprint:
                raise AssertionError('The ro.build.fingerprint has changed after '
                                     'reboot #%d. Initial: %s, after: %s' %
                                     (i + 1, initial_fingerprint, current_fingerprint))
            else:
                num_successful_reboots += 1

        logging.info('Reboot Stability Test: %d/%d successful reboots.',
                     num_successful_reboots, num_reboots)

if __name__ == '__main__':
    common_main()