[cov,e2e] Enable code coverage collection in e2e tests

This change updates the orchestrator end-to-end tests to support
code coverage collection.

Change-Id: I40f041a245fa7a3e14177aa62514d9f529521c35
Signed-off-by: Yi-Hsuan Deng <[email protected]>

Author: sasdf

sw/host/provisioning/ft_lib/src/lib.rs

@@ -667,6 +667,10 @@ pub fn run_ft_personalize(
     response
         .stats
         .log_elapsed_time("second-bootstrap-done", second_t0);
+
+    #[cfg(feature = "ot_coverage_enabled")]
+    let _ = UartConsole::wait_for_coverage(spi_console, Duration::from_secs(10));
+
     Ok(())
 }
 

sw/host/provisioning/orchestrator/tests/BUILD

@@ -149,10 +149,16 @@ sh_test(
             ":orchestrator_{}_zip".format(fpga),
             "@python3",
             cfg["orchestrator_cfg"],
+            cfg["owner_fw"],
+            cfg["rom_ext"],
+            "//sw/host/provisioning/orchestrator/src:data_dependencies",
+            "//util/coverage/collect_cc_coverage",
         ],
         env = {
             "PYTHON": "$(location @python3//:python3)",
             "SKU_CONFIG_PATH": "$(location {})".format(cfg["orchestrator_cfg"]),
+            "COLLECT_CC_COVERAGE": "$(location //util/coverage/collect_cc_coverage)",
+            "CC_CODE_COVERAGE_SCRIPT": "",  # Disable built-in cc coverage
             "FPGA": "{}".format(fpga),
         },
         tags = [

sw/host/provisioning/orchestrator/tests/e2e.sh

@@ -14,19 +14,28 @@ cp sw/host/provisioning/orchestrator/src/orchestrator.zip $TEST_TMPDIR
 
 ORCHESTRATOR_PATH=$TEST_TMPDIR/orchestrator.zip
 
-# This script is run by a Bazel sh_test rule, which sets RUNFILES_DIR to point
-# at the test's runfiles. However, if RUNFILES_DIR is set, orchestrator.zip will
-# inherit its value instead of setting it to the proper directory. This breaks
-# runfile resolution, so we unset this variable here.
-unset RUNFILES_DIR
+(
+  # This script is run by a Bazel sh_test rule, which sets RUNFILES_DIR to point
+  # at the test's runfiles. However, if RUNFILES_DIR is set, orchestrator.zip will
+  # inherit its value instead of setting it to the proper directory. This breaks
+  # runfile resolution, so we unset this variable here.
+  unset RUNFILES_DIR
 
-# Run tool. The path to the --sku-config parameter is relative to the
-# runfiles-dir.
-$PYTHON ${ORCHESTRATOR_PATH} \
-  --sku-config=${SKU_CONFIG_PATH} \
-  --test-unlock-token="0x11111111_11111111_11111111_11111111" \
-  --test-exit-token="0x22222222_22222222_22222222_22222222" \
-  --fpga=${FPGA} \
-  --non-interactive \
-  "$@" \
-  --db-path=$TEST_TMPDIR/registry.sqlite
+  # Run tool. The path to the --sku-config parameter is relative to the
+  # runfiles-dir.
+  $PYTHON ${ORCHESTRATOR_PATH} \
+    --sku-config=${SKU_CONFIG_PATH} \
+    --test-unlock-token="0x11111111_11111111_11111111_11111111" \
+    --test-exit-token="0x22222222_22222222_22222222_22222222" \
+    --fpga=${FPGA} \
+    --non-interactive \
+    "$@" \
+    --db-path=$TEST_TMPDIR/registry.sqlite
+)
+
+# Post-process the coverage profile if coverage mode is enabled.
+if [[ "$COVERAGE" == "1" ]]; then
+  export COLLECT_CC_COVERAGE="$PWD/$COLLECT_CC_COVERAGE"
+  cd "$ROOT"
+  "$COLLECT_CC_COVERAGE"
+fi