android_webview/tools/WebViewShell/src/org/chromium/webview_shell/WebViewLayoutTestRunner.java

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 package org.chromium.webview_shell;
   6
   7 import android.os.Bundle;
   8 import android.test.InstrumentationTestRunner;
   9
  10 /**
  11  * Customized test runner for running instrumentation tests in WebViewBrowserTests.
  12  */
  13 public class WebViewLayoutTestRunner extends InstrumentationTestRunner {
  14     private String mModeArgument;
  15     private static final String MODE_REBASELINE = "rebaseline";
  16
  17     @Override
  18     public void onCreate(Bundle arguments) {
  19         super.onCreate(arguments);
  20         if (arguments != null) {
  21             mModeArgument = arguments.getString("mode");
  22         }
  23     }
  24
  25     public boolean isRebaseline() {
  26         return mModeArgument != null ? mModeArgument.equals(MODE_REBASELINE) : false;
  27     }
  28 }