Ensure that Status checker are not shared by shard

When using legacy sharding, ensure that system checker
(that are stateful) are not shared between shards.

Test: local unit tests
Bug: 36776549
Change-Id: I59a3091734efd581b99ead5c42145383ce684f66

1 files changed, 21 insertions, 0 deletions

diff --git a/src/com/android/tradefed/invoker/ShardHelper.java b/src/com/android/tradefed/invoker/ShardHelper.java
index dd70dbce7..760ceb697 100644
--- a/src/com/android/tradefed/invoker/ShardHelper.java
+++ b/src/com/android/tradefed/invoker/ShardHelper.java
@@ -18,12 +18,15 @@ package com.android.tradefed.invoker;
 import com.android.tradefed.build.ExistingBuildProvider;
 import com.android.tradefed.build.IBuildInfo;
 import com.android.tradefed.config.ConfigurationException;
+import com.android.tradefed.config.ConfigurationFactory;
 import com.android.tradefed.config.IConfiguration;
 import com.android.tradefed.log.LogUtil.CLog;
 import com.android.tradefed.result.IShardableListener;
 import com.android.tradefed.result.ITestInvocationListener;
+import com.android.tradefed.suite.checker.ISystemStatusChecker;
 import com.android.tradefed.testtype.IRemoteTest;
 import com.android.tradefed.testtype.IShardableTest;
+import com.android.tradefed.util.QuotationAwareTokenizer;
 
 import java.util.ArrayList;
 import java.util.Collection;
@@ -71,6 +74,7 @@ public class ShardHelper {
             CLog.i("Rescheduling sharded config...");
             IConfiguration shardConfig = config.clone();
             shardConfig.setTest(testShard);
+            cloneStatusChecker(config, shardConfig);
 
             cloneBuildInfos(config, shardConfig, context);
 
@@ -92,6 +96,23 @@ public class ShardHelper {
     }
 
     /**
+     * Helper to clone {@link ISystemStatusChecker}s from the original config to the clonedConfig.
+     */
+    private static void cloneStatusChecker(IConfiguration oriConfig, IConfiguration clonedConfig) {
+        try {
+            IConfiguration deepCopy =
+                    ConfigurationFactory.getInstance()
+                            .createConfigurationFromArgs(
+                                    QuotationAwareTokenizer.tokenizeLine(
+                                            oriConfig.getCommandLine()));
+            clonedConfig.setSystemStatusCheckers(deepCopy.getSystemStatusCheckers());
+        } catch (ConfigurationException e) {
+            // should not happen
+            throw new RuntimeException("failed to deep copy a configuration", e);
+        }
+    }
+
+    /**
      * Attempt to shard given {@link IRemoteTest}.
      *
      * @param shardableTests the list of {@link IRemoteTest}s to add to