Alternative: Convert Chrome NativeLibraryTestBase tests example

content/public/test/android/javatests/src/org/chromium/content/browser/test/NativeLibraryTestRule.java

 // Copyright 2017 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.content.browser.test;
 
 import android.support.test.InstrumentationRegistry;
 
+import org.junit.Assert;
 import org.junit.rules.TestRule;
 import org.junit.runner.Description;
 import org.junit.runners.model.Statement;
 
+import org.chromium.base.test.SetUpTestRule;
+
 /**
  * TestRule that adds support for loading and dealing with native libraries.
  *
  * NativeLibraryTestRule does not interact with any Activity.
  */
-public class NativeLibraryTestRule implements TestRule {
+public class NativeLibraryTestRule implements TestRule, SetUpTestRule<NativeLibraryTestRule> {
     private final NativeLibraryTestCommon mTestCommon = new NativeLibraryTestCommon();
+    private NativeLibraryInitializationOption mInitOption;
+    private boolean mShouldSetUp = false;
+
+    public enum NativeLibraryInitializationOption { INIT_BROWSER_PROCESS, NO_BROWSER_PROCESS }
 
     /**
      * Loads the native library on the activity UI thread (must not be called from the UI thread).
      */
     public void loadNativeLibraryNoBrowserProcess() {
         mTestCommon.handleNativeInitialization(false, InstrumentationRegistry.getInstrumentation());
     }
 
     /**
      * Loads the native library on the activity UI thread (must not be called from the UI thread).
      * After loading the library, this will initialize the browser process.
      */
     public void loadNativeLibraryAndInitBrowserProcess() {
         mTestCommon.handleNativeInitialization(true, InstrumentationRegistry.getInstrumentation());
     }
 
+    public NativeLibraryTestRule setInitOption(NativeLibraryInitializationOption option) {
+        mInitOption = option;
+        return this;
+    }
+
+    @Override
+    public NativeLibraryTestRule shouldSetUp(boolean runSetUp) {
+        mShouldSetUp = runSetUp;
+        return this;
+    }
+
     @Override
     public Statement apply(Statement base, Description description) {
-        return base;
+        return new SetUpStatement(base, this, mShouldSetUp);
+    }
+
+    @Override
+    public void setUp() {
+        Assert.assertNotNull("Initialization option not set, please call "
+                        + "NativeLibraryTestRule#setInitOption to set up",
+                mInitOption);
+        if (mInitOption.equals(NativeLibraryInitializationOption.INIT_BROWSER_PROCESS)) {
+            loadNativeLibraryAndInitBrowserProcess();
+        } else if (mInitOption.equals(NativeLibraryInitializationOption.NO_BROWSER_PROCESS)) {
+            loadNativeLibraryNoBrowserProcess();
+        } else {
+            throw new AssertionError("Initialization option not recognized");
+        }
     }
 }