Added grub.py that will help to automate grub test cases. #884
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,122 @@ | ||
| #!/usr/bin/env python3 | ||
| # OpenPOWER Automated Test Project | ||
| # | ||
| # Contributors Listed Below - COPYRIGHT 2017 | ||
| # [+] International Business Machines Corp. | ||
| # | ||
| # | ||
| # 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 unittest | ||
| import time | ||
| import pexpect | ||
|
|
||
| import OpTestConfiguration | ||
| import OpTestLogger | ||
| from common.OpTestSystem import OpSystemState | ||
| from common.OpTestError import OpTestError | ||
|
|
||
| log = OpTestLogger.optest_logger_glob.get_logger(__name__) | ||
|
|
||
|
|
||
| class Grub(unittest.TestCase): | ||
| ''' | ||
| Test case to measure boot timing | ||
|
There was a problem hiding this comment. can we reword like
|
||
| ''' | ||
| def setUp(self): | ||
| self.conf = OpTestConfiguration.conf | ||
| self.cv_SYSTEM = self.conf.system() | ||
| self.cv_HOST = self.conf.host() | ||
| self.console = self.cv_SYSTEM.console | ||
|
|
||
| def measure_boot_time(self): | ||
|
There was a problem hiding this comment. Can we pick another name to convey the exact use case? boot_time is very generic. |
||
| ''' | ||
| Measure boot timing from reboot to grub menu and from grub menu to kernel loading | ||
|
|
||
| ''' | ||
| # Start timing | ||
| start_time = time.time() | ||
| log.info("Starting boot time measurement at: %s" % time.strftime("%Y-%m-%d %H:%M:%S")) | ||
| try: | ||
| # Ensure we're in the OS | ||
| if self.cv_SYSTEM.get_state() != OpSystemState.OS: | ||
| log.info("Moving to OS state...") | ||
| self.cv_SYSTEM.goto_state(OpSystemState.OS) | ||
| # Get console connection | ||
| self.console = self.cv_SYSTEM.console | ||
| # Issue reboot command | ||
| log.info("Issuing reboot command...") | ||
| self.console.pty.sendline("reboot") | ||
| # Wait for system to start rebooting | ||
| log.info("Waiting for system to start rebooting...") | ||
| time.sleep(5) # Give system time to start rebooting | ||
| # Wait for GRUB menu with more patterns | ||
| log.info("Waiting for GRUB menu...") | ||
| self.console.pty.expect([ | ||
| 'GNU GRUB', | ||
| 'Press any key to enter the menu', | ||
| 'Welcome to GRUB', | ||
| 'Booting a command list', | ||
| 'Booting \'GNU/Linux\'', | ||
| 'Loading Linux', | ||
| 'Loading initial ramdisk', | ||
| 'Starting kernel', | ||
| 'Linux version' | ||
| ], timeout=300) | ||
| grub_time = time.time() - start_time | ||
| log.info("GRUB menu detected at: %.2f seconds" % grub_time) | ||
| # Wait for kernel loading with more patterns | ||
| log.info("Waiting for kernel loading...") | ||
| self.console.pty.expect([ | ||
| 'Loading Linux', | ||
| 'Loading initial ramdisk', | ||
| 'Booting the kernel', | ||
| 'Starting kernel', | ||
| 'Linux version', | ||
| 'Command line: BOOT_IMAGE' | ||
| ], timeout=300) | ||
| kernel_time = time.time() - start_time | ||
| log.info("Kernel loading detected at: %.2f seconds" % kernel_time) | ||
| # Calculate time from GRUB to kernel | ||
| grub_to_kernel = kernel_time - grub_time | ||
| log.info("Time from GRUB menu to kernel loading: %.2f seconds" % grub_to_kernel) | ||
| except pexpect.TIMEOUT: | ||
| log.error("Timeout waiting for boot sequence") | ||
| log.error("Last console output: %s" % self.console.pty.before) | ||
| raise OpTestError("Boot sequence timeout") | ||
|
There was a problem hiding this comment. Instead of Error out, can we able to recover the system using test ? or it kept manual |
||
| except Exception as e: | ||
| log.error("Unexpected error during boot sequence: %s" % str(e)) | ||
| raise | ||
|
There was a problem hiding this comment. raising any error? |
||
| return { | ||
| 'reboot_to_grub': grub_time, | ||
| 'grub_to_kernel': grub_to_kernel, | ||
| 'total_boot_time': kernel_time | ||
| } | ||
|
|
||
| def runTest(self): | ||
| ''' | ||
| Run the boot time measurement test | ||
| ''' | ||
| log.info("Starting boot time measurement test") | ||
| boot_times = self.measure_boot_time() | ||
| # Print results | ||
|
There was a problem hiding this comment. Can you please remove this comment, even wherever possible, if required, push as a logger, or method doc string |
||
| log.info("Boot Time Results:") | ||
| log.info("-----------------") | ||
| log.info("Reboot to GRUB menu: %.2f seconds" % boot_times['reboot_to_grub']) | ||
| log.info("GRUB menu to kernel loading: %.2f seconds" % boot_times['grub_to_kernel']) | ||
| log.info("Total boot time: %.2f seconds" % boot_times['total_boot_time']) | ||
| # Verify boot times are within reasonable limits | ||
| if boot_times['reboot_to_grub'] > 300: | ||
| self.fail("Time to GRUB menu exceeded 180 seconds") | ||
| if boot_times['grub_to_kernel'] > 300: | ||
| self.fail("Time from GRUB to kernel loading exceeded 180 seconds") | ||
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
2025