From caee8f891c7a994dad7ee78bb80873951a758c6e Mon Sep 17 00:00:00 2001
From: Tom Wai-Hong Tam <waihong@chromium.org>
Date: Sat, 20 Mar 2010 09:45:02 +0800
Subject: [PATCH] Modify image for mfg test, e.g. copying qualified component
 file to the image.

Review URL: http://codereview.chromium.org/1081002
---
 mod_image_for_test.sh | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/mod_image_for_test.sh b/mod_image_for_test.sh
index fb35572443..261cc9eae8 100755
--- a/mod_image_for_test.sh
+++ b/mod_image_for_test.sh
@@ -13,7 +13,10 @@
 get_default_board
 
 DEFINE_string board "$DEFAULT_BOARD" "Board for which the image was built"
+DEFINE_string qualdb "/tmp/run_remote_tests.*" \
+    "Location of qualified component file"
 DEFINE_string image "" "Location of the rootfs raw image file"
+DEFINE_boolean manuf $FLAGS_FALSE "Modify the image for manufacturing testing"
 DEFINE_boolean yes $FLAGS_FALSE "Answer yes to all prompts" "y"
 
 # Parse command line
@@ -100,6 +103,33 @@ MOD_SCRIPTS_ROOT="${GCLIENT_ROOT}/src/scripts/mod_for_test_scripts"
 sudo GCLIENT_ROOT=${GCLIENT_ROOT} ROOT_FS_DIR=${ROOT_FS_DIR} \
     "${MOD_SCRIPTS_ROOT}/test_setup.sh"
 
+# Run manufacturing test setup
+if [ ${FLAGS_manuf} -eq ${FLAGS_TRUE} ]; then
+  echo "Modifying image ${FLAGS_image} for manufacturing test..."
+
+  # Try to use the sytem component file in the most recent autotest result
+  FLAGS_qualdb=$(ls -dt ${FLAGS_qualdb} 2>&-| head -1)
+
+  # Try to append the full path to the file if FLAGS_qualdb is a directory
+  if [ ! -z ${FLAGS_qualdb} ] && [ -d ${FLAGS_qualdb} ]; then
+    # TODO(waihong): Handle multiple results to deliver to multiple images
+    FLAGS_qualdb="${FLAGS_qualdb}/platform_Components,*"
+    FLAGS_qualdb=$(ls -dt ${FLAGS_qualdb} 2>&-| head -1)
+    FLAGS_qualdb="${FLAGS_qualdb}/platform_Components/results/system_components"
+  fi
+
+  if [ ! -z ${FLAGS_qualdb} ] && [ -f ${FLAGS_qualdb} ]; then
+    # Copy the qualified component file to the image
+    echo "Copying ${FLAGS_qualdb} to the image."
+    sudo mkdir -p ${ROOT_FS_DIR}/usr/local/manufacturing
+    sudo cp -f ${FLAGS_qualdb} \
+      ${ROOT_FS_DIR}/usr/local/manufacturing/qualified_components
+  else
+    echo "No qualified component file found at: ${FLAGS_qualdb}"
+    exit 1
+  fi
+fi
+
 cleanup
 trap - EXIT