Create CommandLineTestRule

base/test/android/javatests/src/org/chromium/base/test/util/CommandLineFlags.java

 // 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.base.test.util;
 
 import android.content.Context;
 
-import junit.framework.Assert;
+import org.junit.Assert;
 
 import org.chromium.base.BaseChromiumApplication;
 import org.chromium.base.CommandLine;
 import org.chromium.base.test.BaseTestResult.PreTestHook;
 import org.chromium.base.test.util.parameter.BaseParameter;
 
 import java.lang.annotation.ElementType;
 import java.lang.annotation.Inherited;
 import java.lang.annotation.Retention;
 import java.lang.annotation.RetentionPolicy;
@@ skipping 33 matching lines @@
      * Note that this can only remove flags added via {@link Add} above.
      */
     @Inherited
     @Retention(RetentionPolicy.RUNTIME)
     @Target({ElementType.METHOD, ElementType.TYPE})
     public @interface Remove {
         String[] value();
     }
 
     /**
+     * Sets up the CommandLine flags using a set of flag strings.
+     *
+     * This will add {@code flags} to the {@link org.chromium.base.CommandLine}.
+     */
+    public static void setUp(Context targetContext, Set<String> flags) {
+        Assert.assertNotNull("Unable to get a non-null target context.", targetContext);
+        CommandLine.reset();
+        BaseChromiumApplication.initCommandLine(targetContext);
+        for (String flag : flags) {
+            CommandLine.getInstance().appendSwitch(flag);
+        }
+    }
+
+    /**
      * Sets up the CommandLine with the appropriate flags.
      *
      * This will add the difference of the sets of flags specified by {@link CommandLineFlags.Add}
      * and {@link CommandLineFlags.Remove} to the {@link org.chromium.base.CommandLine}. Note that
      * trying to remove a flag set externally, i.e. by the command-line flags file, will not work.
      */
     public static void setUp(Context targetContext, AnnotatedElement element) {
-        Assert.assertNotNull("Unable to get a non-null target context.", targetContext);
-        CommandLine.reset();
-        BaseChromiumApplication.initCommandLine(targetContext);
-        Set<String> flags = getFlags(element);
-        for (String flag : flags) {
-            CommandLine.getInstance().appendSwitch(flag);
-        }
+        setUp(targetContext, getFlags(element));
     }
 
     private static Set<String> getFlags(AnnotatedElement element) {
         AnnotatedElement parent = (element instanceof Method)
                 ? ((Method) element).getDeclaringClass()
-                : ((Class) element).getSuperclass();
+                : ((Class<?>) element).getSuperclass();
         Set<String> flags = (parent == null) ? new HashSet<String>() : getFlags(parent);
 
         if (element.isAnnotationPresent(CommandLineFlags.Add.class)) {
             flags.addAll(
                     Arrays.asList(element.getAnnotation(CommandLineFlags.Add.class).value()));
         }
 
         if (element.isAnnotationPresent(CommandLineFlags.Remove.class)) {
             List<String> flagsToRemove =
                     Arrays.asList(element.getAnnotation(CommandLineFlags.Remove.class).value());
@@ skipping 51 matching lines @@
     public static class Parameter extends BaseParameter {
         public static final String PARAMETER_TAG = "cmdlinearg-parameter";
         public static final String ADD_ARG = "add";
         public static final String REMOVE_ARG = "remove";
 
         public Parameter(org.chromium.base.test.util.parameter.Parameter.Reader parameterReader) {
             super(PARAMETER_TAG, parameterReader);
         }
     }
 }