Index: chromite/tests/build_image_test.py |
diff --git a/chromite/tests/build_image_test.py b/chromite/tests/build_image_test.py |
deleted file mode 100644 |
index 8c0c7fa98d9863b6142a0b20ff6be3795035526f..0000000000000000000000000000000000000000 |
--- a/chromite/tests/build_image_test.py |
+++ /dev/null |
@@ -1,130 +0,0 @@ |
-#!/usr/bin/python |
-# |
-# Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
-# Use of this source code is governed by a BSD-style license that can be |
-# found in the LICENSE file. |
- |
-"""Tests for build_image shell script. |
- |
-Note: |
- This script must be run from INSIDE chroot. |
- |
-Sample usage: |
- # (inside chroot) pushd ~/trunk/src/scripts/ |
- # run all test cases in this script |
- python chromite/tests/build_image_test.py |
- |
- # run all test cases in a test suite |
- python chromite/tests/build_image_test.py BuildImageTest |
- |
- # run a specific test |
- python chromite/tests/build_image_test.py BuildImageTest.testWithoutBoardExit |
-""" |
- |
-import os |
-import re |
-import sys |
-import unittest |
-sys.path.append(os.path.join(os.path.dirname(__file__), '../lib')) |
-from cros_build_lib import (RunCommand, IsInsideChroot, GetChromeosVersion, |
- GetOutputImageDir) |
- |
- |
-class BuildImageTest(unittest.TestCase): |
- """Test suite for build_image script.""" |
- |
- def setUp(self): |
- if not IsInsideChroot(): |
- raise RuntimeError('This script must be run from inside chroot.') |
- |
- def _CheckStringPresent(self, query_list, check_stdout=False): |
- """Check for presence of specific queries. |
- |
- Args: |
- query_list: a list of strings to look for. |
- check_stdout: a boolean. True == use stdout from child process. |
- Otherwise use its stderr. |
- """ |
- for query in query_list: |
- # Source error string defined in src/scripts/build_image |
- if check_stdout: |
- self.assertNotEqual(-1, self.output.find(query)) |
- else: |
- self.assertNotEqual(-1, self.error.find(query)) |
- |
- def _RunBuildImageCmd(self, cmd, assert_success=True): |
- """Run build_image with flags. |
- |
- Args: |
- cmd: a string. |
- assert_success: a boolean. True == check child process return code is 0. |
- False otherwise. |
- """ |
- Info ('About to run command: %s' % cmd) |
- cmd_result = RunCommand( |
- cmd, error_ok=True, exit_code=True, redirect_stdout=True, |
- redirect_stderr=True, shell=True) |
- self.output = cmd_result.output |
- self.error = cmd_result.error |
- Info ('output =\n%r' % self.output) |
- Info ('error =\n%r' % self.error) |
- |
- message = 'cmd should have failed! error:\n%s' % self.error |
- if assert_success: |
- self.assertEqual(0, cmd_result.returncode) |
- else: |
- self.assertNotEqual(0, cmd_result.returncode, message) |
- |
- def _VerifyOutputImagesExist(self, image_dir, image_list): |
- """Verify output images exist in image_dir. |
- |
- Args: |
- image_dir: a string, absolute path to output directory with images. |
- image_list: a list of strings, names of output images. |
- """ |
- for i in image_list: |
- image_path = os.path.join(image_dir, i) |
- self.assertTrue(os.path.exists(image_path)) |
- |
- def testWithoutBoardExit(self): |
- """Fail when no --board is specified.""" |
- self._RunBuildImageCmd('./build_image --board=""', assert_success=False) |
- self._CheckStringPresent(['ERROR', '--board is required']) |
- |
- def testIncompatibleInstallFlags(self): |
- """Fail when both --factory_install and --dev_install are set.""" |
- cmd = './build_image --board=x86-generic --factory_install --dev_install' |
- self._RunBuildImageCmd(cmd, assert_success=False) |
- self._CheckStringPresent(['ERROR', 'Incompatible flags']) |
- |
- def testIncompatibleRootfsFlags(self): |
- """Fail when rootfs partition is not large enough.""" |
- cmd = ('./build_image --board=x86-generic --rootfs_size=100' |
- ' --rootfs_hash_pad=10 --rootfs_partition_size=20') |
- self._RunBuildImageCmd(cmd, assert_success=False) |
- self._CheckStringPresent(['ERROR', 'bigger than partition']) |
- |
- def _BuildImageForBoard(self, board, image_list): |
- """Build image for specific board type. |
- |
- Args: |
- board: a string. |
- image_list: a list of strings, names of output images. |
- """ |
- cmd = './build_image --board=%s' % board |
- Info ('If all goes well, it takes ~5 min. to build an image...') |
- self._RunBuildImageCmd(cmd) |
- self._CheckStringPresent(['Image created in', 'copy to USB keyfob'], |
- check_stdout=True) |
- chromeos_version_str = GetChromeosVersion(self.output) |
- image_dir = GetOutputImageDir(board, chromeos_version_str) |
- self._VerifyOutputImagesExist(image_dir, image_list) |
- |
- def testBuildX86Generic(self): |
- """Verify we can build an x86-generic image.""" |
- self._BuildImageForBoard( |
- 'x86-generic', ['chromiumos_image.bin', 'chromiumos_base_image.bin']) |
- |
- |
-if __name__ == '__main__': |
- unittest.main() |