Build image part of http://codereview.chromium.org/913004/show

src/scripts/build_image

 #!/bin/bash
 
 # Copyright (c) 2009 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.
 
 # Script to build a bootable keyfob-based chromeos system image from within
 # a chromiumos setup. This assumes that all needed packages have been built into
 # the given target's root with binary packages turned on. This script will
 # build the Chrome OS image using only pre-built binary packages.
@@ skipping 19 matching lines @@
 DEFINE_boolean replace $FLAGS_FALSE \
   "Overwrite existing output, if any."
 DEFINE_boolean withdev $FLAGS_TRUE \
   "Include useful developer friendly utilities in the image."
 DEFINE_boolean installmask $FLAGS_TRUE \
   "Use INSTALL_MASK to shrink the resulting image."
 DEFINE_integer jobs -1 \
   "How many packages to build in parallel at maximum."
 DEFINE_boolean statefuldev $FLAGS_FALSE \
   "Install development packages on stateful partition -- still experimental"
+DEFINE_boolean withtest $FLAGS_TRUE \
+  "Include packages required for testing and prepare image for testing"
 
 # Parse command line.
 FLAGS "$@" || exit 1
 eval set -- "${FLAGS_ARGV}"
 
 # Only now can we die on error.  shflags functions leak non-zero error codes,
 # so will die prematurely if 'set -e' is specified before now.
 set -e
 
 if [ -z "$FLAGS_board" ] ; then
@@ skipping 214 matching lines @@
   sudo mount -n --bind "$DEV_IMAGE_ROOT" "$ROOT_FS_DIR/usr/local"
 fi
 
 # We "emerge --root=$ROOT_FS_DIR --root-deps=rdeps --usepkgonly" all of the
 # runtime packages for chrome os. This builds up a chrome os image from binary
 # packages with runtime dependencies only.  We use INSTALL_MASK to trim the
 # image size as much as possible.
 sudo INSTALL_MASK="$INSTALL_MASK" emerge-${BOARD} \
   --root="$ROOT_FS_DIR" --root-deps=rdeps \
   --usepkgonly chromeos $EMERGE_JOBS
+
+# Determine the root dir for development packages.
+ROOT_DEV_DIR="$ROOT_FS_DIR"
+[ $FLAGS_statefuldev -eq $FLAGS_TRUE ] && ROOT_DEV_DIR="$ROOT_FS_DIR/usr/local"
+
+# Install development packages.
 if [[ $FLAGS_withdev -eq $FLAGS_TRUE ]] ; then
-  # Determine the root dir for development packages
-  ROOT_DIR="$ROOT_FS_DIR"
-  [ $FLAGS_statefuldev -eq $FLAGS_TRUE ] && ROOT_DIR="$ROOT_FS_DIR/usr/local" 
-  
   sudo INSTALL_MASK="$INSTALL_MASK" emerge-${BOARD} \
-      --root="$ROOT_DIR" --root-deps=rdeps \
+      --root="$ROOT_DEV_DIR" --root-deps=rdeps \
       --usepkgonly chromeos-dev $EMERGE_JOBS
- 
-  if [ $FLAGS_statefuldev -eq $FLAGS_TRUE ] ; then
-    # Fix symlinks so they work on live system.
-    for path in bin include lib libexec sbin share; do
-      sudo unlink $DEV_IMAGE_ROOT/usr/$path
-      sudo ln -s /usr/local/$path $DEV_IMAGE_ROOT/usr/$path
-    done
-
-    # Fix exceptions.
-    sudo unlink "$DEV_IMAGE_ROOT/usr/lib64"
-    sudo unlink "$DEV_IMAGE_ROOT/usr/local"
-    
-    sudo ln -s "/usr/local/lib" "$DEV_IMAGE_ROOT/usr/lib64"
-    sudo ln -s "/usr/local" "$DEV_IMAGE_ROOT/usr/local"
-  fi 
 
   # The ldd tool is a useful shell script but lives in glibc; just copy it.
-  sudo cp -a "$(which ldd)" "${ROOT_FS_DIR}/usr/bin"
+  sudo cp -a "$(which ldd)" "${ROOT_DEV_DIR}/usr/bin"
+fi
+
+# Install packages required for testing.
+if [[ $FLAGS_withtest -eq $FLAGS_TRUE ]] ; then
+  sudo INSTALL_MASK="$INSTALL_MASK" emerge-${BOARD} \
+      --root="$ROOT_DEV_DIR" --root-deps=rdeps \
+      --usepkgonly chromeos-test $EMERGE_JOBS
+fi
+
+# Clean up links setup for stateful install of extra packages. 
+if [ $FLAGS_statefuldev -eq $FLAGS_TRUE ] ; then
+  # Fix symlinks so they work on live system.
+  for path in bin include lib libexec sbin share; do
+    sudo unlink $DEV_IMAGE_ROOT/usr/$path
+    sudo ln -s /usr/local/$path $DEV_IMAGE_ROOT/usr/$path
+  done
+
+  # Fix exceptions.
+  sudo unlink "$DEV_IMAGE_ROOT/usr/lib64"
+  sudo unlink "$DEV_IMAGE_ROOT/usr/local"
+
+  sudo ln -s "/usr/local/lib" "$DEV_IMAGE_ROOT/usr/lib64"
+  sudo ln -s "/usr/local" "$DEV_IMAGE_ROOT/usr/local" 
 fi
 
 # Perform any customizations on the root file system that are needed.
 WITH_DEV=""
 if [[ $FLAGS_withdev -eq $FLAGS_TRUE ]]; then
   WITH_DEV="--withdev"
 fi
 
 #TODO(sosa@chromium.org) - Does it make sense to leave /usr/local bound here?
 "${SCRIPTS_DIR}/customize_rootfs" \
@@ skipping 63 matching lines @@
 
 OUTSIDE_OUTPUT_DIR="../build/images/${FLAGS_board}/${IMAGE_SUBDIR}"
 echo "Done.  Image created in ${OUTPUT_DIR}"
 echo "To copy to USB keyfob, OUTSIDE the chroot, do something like:"
 echo "  ./image_to_usb.sh --from=${OUTSIDE_OUTPUT_DIR} --to=/dev/sdb"
 echo "To convert to VMWare image, OUTSIDE the chroot, do something like:"
 echo "  ./image_to_vmware.sh --from=${OUTSIDE_OUTPUT_DIR}"
 echo "from the scripts directory where you entered the chroot."
 
 trap - EXIT