Convert ChromeActivityTestCaseBase direct children to JUnit4

chrome/android/javatests/src/org/chromium/chrome/browser/SmartClipProviderTest.java

 // Copyright 2014 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.chrome.browser;
 
 import android.annotation.TargetApi;
 import android.content.pm.ApplicationInfo;
 import android.content.pm.PackageManager;
 import android.os.Build;
 import android.os.Bundle;
 import android.os.Handler;
 import android.os.HandlerThread;
 import android.os.Message;
 import android.support.test.filters.MediumTest;
 import android.view.View;
 import android.view.ViewGroup;
 
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
 import org.chromium.base.ThreadUtils;
 import org.chromium.base.test.util.CallbackHelper;
+import org.chromium.base.test.util.CommandLineFlags;
 import org.chromium.base.test.util.Feature;
 import org.chromium.base.test.util.RetryOnFailure;
-import org.chromium.chrome.test.ChromeActivityTestCaseBase;
+import org.chromium.chrome.test.ChromeActivityTestRule;
+import org.chromium.chrome.test.ChromeJUnit4ClassRunner;
 
 import java.lang.reflect.Method;
 import java.util.concurrent.TimeoutException;
 
 /**
  * Tests for the SmartClipProvider.
  */
-public class SmartClipProviderTest
-        extends ChromeActivityTestCaseBase<ChromeActivity> implements Handler.Callback {
+@RunWith(ChromeJUnit4ClassRunner.class)
+@CommandLineFlags.Add({ChromeSwitches.DISABLE_FIRST_RUN_EXPERIENCE,
+        ChromeActivityTestRule.DISABLE_NETWORK_PREDICTION_FLAG})
+public class SmartClipProviderTest implements Handler.Callback {
     // This is a key for meta-data in the package manifest. It should NOT
     // change, as OEMs will use it when they look for the SmartClipProvider
     // interface.
+
+    @Rule
+    public ChromeActivityTestRule<ChromeActivity> mActivityTestRule =
+            new ChromeActivityTestRule<>(ChromeActivity.class);
+
     private static final String SMART_CLIP_PROVIDER_KEY =
             "org.chromium.content.browser.SMART_CLIP_PROVIDER";
 
     private static class MyCallbackHelper extends CallbackHelper {
         public String getTitle() {
             return mTitle;
         }
 
         public String getUrl() {
             return mUrl;
@@ skipping 22 matching lines @@
     }
 
     private ChromeActivity mActivity;
     private MyCallbackHelper mCallbackHelper;
     private HandlerThread mHandlerThread;
     private Handler mHandler;
     private Class<?> mSmartClipProviderClass;
     private Method mSetSmartClipResultHandlerMethod;
     private Method mExtractSmartClipDataMethod;
 
-    public SmartClipProviderTest() {
-        super(ChromeActivity.class);
-    }
-
-    @Override
-    public void startMainActivity() throws InterruptedException {
-        startMainActivityOnBlankPage();
-    }
-
-    @Override
+    @Before
     public void setUp() throws Exception {
-        super.setUp();
-        mActivity = getActivity();
+        mActivityTestRule.startMainActivityOnBlankPage();
+        mActivity = mActivityTestRule.getActivity();
         mCallbackHelper = new MyCallbackHelper();
         mHandlerThread = new HandlerThread("ContentViewTest thread");
         mHandlerThread.start();
         mHandler = new Handler(mHandlerThread.getLooper(), this);
 
         mSmartClipProviderClass = getSmartClipProviderClass();
-        assertNotNull(mSmartClipProviderClass);
+        Assert.assertNotNull(mSmartClipProviderClass);
         mSetSmartClipResultHandlerMethod = mSmartClipProviderClass.getDeclaredMethod(
                 "setSmartClipResultHandler", new Class[] { Handler.class });
         mExtractSmartClipDataMethod = mSmartClipProviderClass.getDeclaredMethod(
                 "extractSmartClipData",
                 new Class[] { Integer.TYPE, Integer.TYPE, Integer.TYPE, Integer.TYPE });
     }
 
+    @After
     @TargetApi(Build.VERSION_CODES.JELLY_BEAN_MR2)
-    @Override
     public void tearDown() throws Exception {
-        try {
-            if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.JELLY_BEAN_MR2) {
-                mHandlerThread.quitSafely();
-            } else {
-                mHandlerThread.quit();
-            }
-        } finally {
-            super.tearDown();
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.JELLY_BEAN_MR2) {
+            mHandlerThread.quitSafely();
+        } else {
+            mHandlerThread.quit();
         }
     }
 
     // Implements Handler.Callback
     @Override
     public boolean handleMessage(Message msg) {
         Bundle bundle = msg.getData();
-        assertNotNull(bundle);
+        Assert.assertNotNull(bundle);
         String url = bundle.getString("url");
         String title = bundle.getString("title");
         String text = bundle.getString("text");
         String html = bundle.getString("html");
         // We don't care about other values for now.
         mCallbackHelper.notifyCalled(title, url, text, html);
         return true;
     }
 
     // Create SmartClipProvider interface from package meta-data.
     private Class<?> getSmartClipProviderClass() throws Exception {
         ApplicationInfo ai = mActivity.getPackageManager().getApplicationInfo(
                 mActivity.getPackageName(), PackageManager.GET_META_DATA);
         Bundle bundle = ai.metaData;
         String className = bundle.getString(SMART_CLIP_PROVIDER_KEY);
-        assertNotNull(className);
+        Assert.assertNotNull(className);
         return Class.forName(className);
     }
 
     // Returns the first smart clip provider under the root view using DFS.
     private Object findSmartClipProvider(View v) {
         if (mSmartClipProviderClass.isInstance(v)) {
             return v;
         } else if (v instanceof ViewGroup) {
             ViewGroup viewGroup = (ViewGroup) v;
             int count = viewGroup.getChildCount();
             for (int i = 0; i < count; ++i) {
                 View c = viewGroup.getChildAt(i);
                 Object found = findSmartClipProvider(c);
                 if (found != null) return found;
             }
         }
         return null;
     }
 
+    @Test
     @MediumTest
     @Feature({"SmartClip"})
     @RetryOnFailure
     public void testSmartClipDataCallback() throws InterruptedException, TimeoutException {
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 // This emulates what OEM will be doing when they want to call
                 // functions on SmartClipProvider through view hierarchy.
 
-                Object scp =
-                        findSmartClipProvider(getActivity().findViewById(android.R.id.content));
-                assertNotNull(scp);
+                Object scp = findSmartClipProvider(
+                        mActivityTestRule.getActivity().findViewById(android.R.id.content));
+                Assert.assertNotNull(scp);
                 try {
                     mSetSmartClipResultHandlerMethod.invoke(scp, mHandler);
                     mExtractSmartClipDataMethod.invoke(scp, 10, 20, 100, 70);
                 } catch (Exception e) {
                     e.printStackTrace();
-                    fail();
+                    Assert.fail();
                 }
             }
         });
         mCallbackHelper.waitForCallback(0, 1);  // call count: 0 --> 1
-        assertEquals("about:blank", mCallbackHelper.getTitle());
-        assertEquals("about:blank", mCallbackHelper.getUrl());
-        assertNotNull(mCallbackHelper.getText());
-        assertNotNull(mCallbackHelper.getHtml());
+        Assert.assertEquals("about:blank", mCallbackHelper.getTitle());
+        Assert.assertEquals("about:blank", mCallbackHelper.getUrl());
+        Assert.assertNotNull(mCallbackHelper.getText());
+        Assert.assertNotNull(mCallbackHelper.getHtml());
     }
 
+    @Test
     @MediumTest
     @Feature({"SmartClip"})
     @RetryOnFailure
     public void testSmartClipNoHandlerDoesntCrash() throws InterruptedException, TimeoutException {
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
-                Object scp =
-                        findSmartClipProvider(getActivity().findViewById(android.R.id.content));
-                assertNotNull(scp);
+                Object scp = findSmartClipProvider(
+                        mActivityTestRule.getActivity().findViewById(android.R.id.content));
+                Assert.assertNotNull(scp);
                 try {
                     // Galaxy Note 4 has a bug where it doesn't always set the handler first; in
                     // that case, we shouldn't crash: http://crbug.com/710147
                     mExtractSmartClipDataMethod.invoke(scp, 10, 20, 100, 70);
 
                     // Add a wait for a valid callback to make sure we have time to
                     // hit the crash from the above call if any.
                     mSetSmartClipResultHandlerMethod.invoke(scp, mHandler);
                     mExtractSmartClipDataMethod.invoke(scp, 10, 20, 100, 70);
                 } catch (Exception e) {
                     e.printStackTrace();
-                    fail();
+                    Assert.fail();
                 }
             }
         });
         mCallbackHelper.waitForCallback(0, 1); // call count: 0 --> 1
     }
 }