Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(221)

Unified Diff: chromite/tests/build_image_test.py

Issue 6371018: Remove chromite from crosutils.git. It's been moved to chromite.git. (Closed) Base URL: http://git.chromium.org/git/crosutils.git@master
Patch Set: Created 9 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « chromite/specs/x86-generic.spec ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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()
« no previous file with comments | « chromite/specs/x86-generic.spec ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698