1 files changed, 38 insertions, 0 deletions
diff --git a/android_webview/tools/system_webview_shell/layout_tests/src/org/chromium/webview_shell/test/WebViewLayoutTestRunner.java b/android_webview/tools/system_webview_shell/layout_tests/src/org/chromium/webview_shell/test/WebViewLayoutTestRunner.java
new file mode 100644
index 0000000..44b745f
--- /dev/null
+++ b/android_webview/tools/system_webview_shell/layout_tests/src/org/chromium/webview_shell/test/WebViewLayoutTestRunner.java
@@ -0,0 +1,38 @@
+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.webview_shell.test;
+
+import android.os.Bundle;
+import android.test.AndroidTestRunner;
+import android.test.InstrumentationTestRunner;
+
+import org.chromium.test.reporter.TestStatusListener;
+
+/**
+ * Customized test runner for running instrumentation tests in WebViewBrowserTests.
+ */
+public class WebViewLayoutTestRunner extends InstrumentationTestRunner {
+    private String mModeArgument;
+    private static final String MODE_REBASELINE = "rebaseline";
+
+    @Override
+    public void onCreate(Bundle arguments) {
+        super.onCreate(arguments);
+        if (arguments != null) {
+            mModeArgument = arguments.getString("mode");
+        }
+    }
+
+    public boolean isRebaseline() {
+        return mModeArgument != null ? mModeArgument.equals(MODE_REBASELINE) : false;
+    }
+
+    @Override
+    protected AndroidTestRunner getAndroidTestRunner() {
+        AndroidTestRunner runner = super.getAndroidTestRunner();
+        runner.addTestListener(new TestStatusListener(getContext()));
+        return runner;
+    }
+}
+\ No newline at end of file