Allow third_party/bazel/ to use Mockito 2

PiperOrigin-RevId: 239185340
diff --git a/src/java_tools/junitrunner/javatests/com/google/testing/junit/runner/BUILD b/src/java_tools/junitrunner/javatests/com/google/testing/junit/runner/BUILD
index ee22122..3017c29 100644
--- a/src/java_tools/junitrunner/javatests/com/google/testing/junit/runner/BUILD
+++ b/src/java_tools/junitrunner/javatests/com/google/testing/junit/runner/BUILD
@@ -25,8 +25,8 @@
         "//third_party:jsr305",
         "//third_party:jsr330_inject",
         "//third_party:junit4",
-        "//third_party:mockito",
         "//third_party:truth",
+        "//third_party/java/mockito:mockito2_for_third_party",
     ],
 )
 
diff --git a/src/test/java/com/google/devtools/build/android/dexer/BUILD b/src/test/java/com/google/devtools/build/android/dexer/BUILD
index 5585dc2..6505701 100644
--- a/src/test/java/com/google/devtools/build/android/dexer/BUILD
+++ b/src/test/java/com/google/devtools/build/android/dexer/BUILD
@@ -32,7 +32,7 @@
         "//src/tools/android/java/com/google/devtools/build/android/dexer",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "@bazel_tools//tools/java/runfiles",
     ] + select({
diff --git a/src/test/java/com/google/devtools/build/lib/BUILD b/src/test/java/com/google/devtools/build/lib/BUILD
index 4214ed5..3587eef 100644
--- a/src/test/java/com/google/devtools/build/lib/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/BUILD
@@ -241,7 +241,7 @@
         "//src/main/java/com/google/devtools/build/lib/skyframe/serialization:constants",
         "//src/main/java/com/google/devtools/build/lib/skyframe/serialization/testutils",
         "//src/main/java/com/google/devtools/build/lib/vfs",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party/protobuf:protobuf_java",
     ],
 )
@@ -364,7 +364,7 @@
         "//src/main/java/com/google/devtools/build/lib/vfs",
         "//src/main/java/com/google/devtools/build/lib/vfs/inmemoryfs",
         "//src/main/java/com/google/devtools/common/options",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth8",
     ],
 )
@@ -386,7 +386,7 @@
         "//src/main/java/com/google/devtools/build/lib:out-err",
         "//src/main/java/com/google/devtools/build/lib:util",
         "//src/main/protobuf:bazel_flags_java_proto",
-        "//third_party:mockito",
+        "//third_party:mockito2",
     ],
 )
 
@@ -574,7 +574,7 @@
         "//src/main/java/com/google/devtools/common/options",
         "//src/main/protobuf:action_cache_java_proto",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
     ],
 )
 
@@ -614,7 +614,7 @@
         "//src/main/java/com/google/devtools/common/options",
         "//src/main/protobuf:action_cache_java_proto",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth8",
     ],
 )
@@ -699,7 +699,7 @@
         "//src/test/java/com/google/devtools/build/lib/skyframe:testutil",
         "//src/test/java/com/google/devtools/build/skyframe:testutil",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party/protobuf:protobuf_java",
         "@bazel_tools//tools/java/runfiles",
     ],
@@ -754,7 +754,7 @@
         "//src/main/java/com/google/devtools/build/lib/vfs",
         "//src/main/protobuf:extra_actions_base_java_proto",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
     ],
 )
 
@@ -793,7 +793,7 @@
         "//src/test/java/com/google/devtools/build/lib/skyframe/trimming:trimmable_test_fragments",
         "//third_party:auto_value",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
     ],
 )
 
@@ -892,7 +892,7 @@
         "//src/test/java/com/google/devtools/build/lib/skyframe:testutil",
         "//third_party:auto_value",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
     ],
 )
 
@@ -955,7 +955,7 @@
         "//src/main/protobuf:crosstool_config_java_proto",
         "//src/main/protobuf:extra_actions_base_java_proto",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party/protobuf:protobuf_java",
     ],
 )
@@ -995,7 +995,7 @@
         "//src/main/java/com/google/devtools/build/skyframe:skyframe-objects",
         "//src/main/protobuf:build_java_proto",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party/protobuf:protobuf_java",
     ],
 )
@@ -1054,7 +1054,7 @@
         "//src/main/java/com/google/devtools/build/lib/vfs",
         "//src/main/protobuf:command_server_java_proto",
         "//third_party:jsr305",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party/grpc:grpc-jar",
         "//third_party/protobuf:protobuf_java",
     ],
@@ -1275,7 +1275,7 @@
         "//src/main/protobuf:invocation_policy_java_proto",
         "//src/main/protobuf:test_status_java_proto",
         "//src/test/java/com/google/devtools/common/options:testutils",
-        "//third_party:mockito",
+        "//third_party:mockito2",
     ],
 )
 
@@ -1406,7 +1406,7 @@
         "//src/main/java/com/google/devtools/common/options",
         "//src/main/protobuf:spawn_java_proto",
         "//src/main/protobuf:test_status_java_proto",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party/protobuf:protobuf_java",
     ],
 )
@@ -1434,7 +1434,7 @@
         "//src/main/java/com/google/devtools/common/options",
         "//src/main/tools:process-wrapper",
         "//src/test/shell/integration:spend_cpu_time",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth8",
     ],
 )
@@ -1455,7 +1455,7 @@
         "//src/main/java/com/google/devtools/build/lib/collect/nestedset",
         "//src/main/java/com/google/devtools/build/lib/vfs",
         "//src/main/java/com/google/devtools/build/lib/vfs/inmemoryfs",
-        "//third_party:mockito",
+        "//third_party:mockito2",
     ],
 )
 
diff --git a/src/test/java/com/google/devtools/build/lib/bazel/debug/BUILD b/src/test/java/com/google/devtools/build/lib/bazel/debug/BUILD
index 11bc586..7a35ebf 100644
--- a/src/test/java/com/google/devtools/build/lib/bazel/debug/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/bazel/debug/BUILD
@@ -21,7 +21,7 @@
         "//src/main/java/com/google/devtools/build/lib/vfs:pathfragment",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/build/lib/bazel/repository/BUILD b/src/test/java/com/google/devtools/build/lib/bazel/repository/BUILD
index 737356c..8e15afd 100644
--- a/src/test/java/com/google/devtools/build/lib/bazel/repository/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/bazel/repository/BUILD
@@ -56,7 +56,7 @@
         "//third_party:jsr305",
         "//third_party:junit4",
         "//third_party:maven",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "@bazel_tools//tools/java/runfiles",
     ],
diff --git a/src/test/java/com/google/devtools/build/lib/bazel/repository/cache/BUILD b/src/test/java/com/google/devtools/build/lib/bazel/repository/cache/BUILD
index be882b9..4e7fd14 100644
--- a/src/test/java/com/google/devtools/build/lib/bazel/repository/cache/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/bazel/repository/cache/BUILD
@@ -23,7 +23,7 @@
         "//src/test/java/com/google/devtools/build/lib:testutil",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/BUILD b/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/BUILD
index cdd0374..5e724cb 100644
--- a/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/BUILD
@@ -25,7 +25,7 @@
         "//src/test/java/com/google/devtools/build/lib:testutil",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/HttpStreamTest.java b/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/HttpStreamTest.java
index 5b4f7fa..8e64f38 100644
--- a/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/HttpStreamTest.java
+++ b/src/test/java/com/google/devtools/build/lib/bazel/repository/downloader/HttpStreamTest.java
@@ -74,13 +74,14 @@
   @Before
   public void before() throws Exception {
     when(connection.getInputStream()).thenReturn(new ByteArrayInputStream(data));
-    when(progress.create(any(InputStream.class), any(URL.class), any(URL.class))).thenAnswer(
-        new Answer<InputStream>() {
-          @Override
-          public InputStream answer(InvocationOnMock invocation) throws Throwable {
-            return (InputStream) invocation.getArguments()[0];
-          }
-        });
+    when(progress.create(any(InputStream.class), any(), any(URL.class)))
+        .thenAnswer(
+            new Answer<InputStream>() {
+              @Override
+              public InputStream answer(InvocationOnMock invocation) throws Throwable {
+                return (InputStream) invocation.getArguments()[0];
+              }
+            });
   }
 
   @Test
diff --git a/src/test/java/com/google/devtools/build/lib/buildeventservice/BUILD b/src/test/java/com/google/devtools/build/lib/buildeventservice/BUILD
index effc600..bb26fcd 100644
--- a/src/test/java/com/google/devtools/build/lib/buildeventservice/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/buildeventservice/BUILD
@@ -31,7 +31,7 @@
         "//src/test/java/com/google/devtools/build/lib:testutil",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/grpc:grpc-jar",
         "@com_google_protobuf//:protobuf_java",
diff --git a/src/test/java/com/google/devtools/build/lib/buildeventstream/BUILD b/src/test/java/com/google/devtools/build/lib/buildeventstream/BUILD
index 21619e5..7b4f570 100644
--- a/src/test/java/com/google/devtools/build/lib/buildeventstream/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/buildeventstream/BUILD
@@ -25,7 +25,7 @@
         "//src/main/java/com/google/devtools/common/options",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/build/lib/buildeventstream/transports/BUILD b/src/test/java/com/google/devtools/build/lib/buildeventstream/transports/BUILD
index e39b9d3..64c3263 100644
--- a/src/test/java/com/google/devtools/build/lib/buildeventstream/transports/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/buildeventstream/transports/BUILD
@@ -25,7 +25,7 @@
         "//src/test/java/com/google/devtools/build/lib:packages_testutil",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/protobuf:protobuf_java",
         "//third_party/protobuf:protobuf_java_util",
diff --git a/src/test/java/com/google/devtools/build/lib/exec/StandaloneTestStrategyTest.java b/src/test/java/com/google/devtools/build/lib/exec/StandaloneTestStrategyTest.java
index 28e636d..7719586 100644
--- a/src/test/java/com/google/devtools/build/lib/exec/StandaloneTestStrategyTest.java
+++ b/src/test/java/com/google/devtools/build/lib/exec/StandaloneTestStrategyTest.java
@@ -441,10 +441,10 @@
     when(spawnActionContext.exec(any(), any()))
         .thenAnswer(
             (invocation) -> {
-              Spawn spawn = invocation.getArgumentAt(0, Spawn.class);
+              Spawn spawn = invocation.getArgument(0);
               if (spawn.getOutputFiles().size() != 1) {
-                FileOutErr outErr =
-                    invocation.getArgumentAt(1, ActionExecutionContext.class).getFileOutErr();
+                ActionExecutionContext context = invocation.getArgument(1);
+                FileOutErr outErr = context.getFileOutErr();
                 try (OutputStream stream = outErr.getOutputStream()) {
                   stream.write("This will not appear in the test output: bla\n".getBytes(UTF_8));
                   stream.write((TestLogHelper.HEADER_DELIMITER + "\n").getBytes(UTF_8));
@@ -537,9 +537,9 @@
     when(spawnActionContext.exec(any(), any()))
         .thenAnswer(
             (invocation) -> {
-              Spawn spawn = invocation.getArgumentAt(0, Spawn.class);
-              FileOutErr outErr =
-                  invocation.getArgumentAt(1, ActionExecutionContext.class).getFileOutErr();
+              Spawn spawn = invocation.getArgument(0);
+              ActionExecutionContext context = invocation.getArgument(1);
+              FileOutErr outErr = context.getFileOutErr();
               called.add(outErr);
               if (spawn.getOutputFiles().size() != 1) {
                 try (OutputStream stream = outErr.getOutputStream()) {
diff --git a/src/test/java/com/google/devtools/build/lib/profiler/BUILD b/src/test/java/com/google/devtools/build/lib/profiler/BUILD
index e4cb29a..0b3ac0f 100644
--- a/src/test/java/com/google/devtools/build/lib/profiler/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/profiler/BUILD
@@ -29,7 +29,7 @@
         "//src/test/java/com/google/devtools/build/lib:testutil",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/build/lib/profiler/memory/BUILD b/src/test/java/com/google/devtools/build/lib/profiler/memory/BUILD
index b9c7617..6d34519 100644
--- a/src/test/java/com/google/devtools/build/lib/profiler/memory/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/profiler/memory/BUILD
@@ -18,7 +18,7 @@
         "//src/main/java/com/google/devtools/build/lib/vfs",
         "//third_party:guava",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/pprof:profile_java_proto",
     ],
diff --git a/src/test/java/com/google/devtools/build/lib/query2/BUILD b/src/test/java/com/google/devtools/build/lib/query2/BUILD
index 5366e0e..aa4d943 100644
--- a/src/test/java/com/google/devtools/build/lib/query2/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/query2/BUILD
@@ -85,7 +85,7 @@
         "//third_party:guava",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/build/lib/remote/BUILD b/src/test/java/com/google/devtools/build/lib/remote/BUILD
index 6b7f7d2..580845a 100644
--- a/src/test/java/com/google/devtools/build/lib/remote/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/remote/BUILD
@@ -44,7 +44,7 @@
         "//src/test/java/com/google/devtools/build/lib/remote/util",
         "//third_party:api_client",
         "//third_party:guava",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/grpc:grpc-jar",
         "//third_party/protobuf:protobuf_java",
diff --git a/src/test/java/com/google/devtools/build/lib/remote/blobstore/http/BUILD b/src/test/java/com/google/devtools/build/lib/remote/blobstore/http/BUILD
index 8d868d0..3325550 100644
--- a/src/test/java/com/google/devtools/build/lib/remote/blobstore/http/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/remote/blobstore/http/BUILD
@@ -21,7 +21,7 @@
         "//third_party:api_client",
         "//third_party:auth",
         "//third_party:guava",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:netty",
         "//third_party:truth",
     ],
diff --git a/src/test/java/com/google/devtools/build/lib/remote/logging/BUILD b/src/test/java/com/google/devtools/build/lib/remote/logging/BUILD
index 60e92a8a..e2372f9 100644
--- a/src/test/java/com/google/devtools/build/lib/remote/logging/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/remote/logging/BUILD
@@ -23,7 +23,7 @@
         "//src/test/java/com/google/devtools/build/lib:test_runner",
         "//src/test/java/com/google/devtools/build/lib:testutil",
         "//third_party:guava",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/grpc:grpc-jar",
         "//third_party/protobuf:protobuf_java",
diff --git a/src/test/java/com/google/devtools/build/lib/rules/cpp/BUILD b/src/test/java/com/google/devtools/build/lib/rules/cpp/BUILD
index fda9ab3..a1030f3 100644
--- a/src/test/java/com/google/devtools/build/lib/rules/cpp/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/rules/cpp/BUILD
@@ -65,7 +65,7 @@
         "//third_party:guava-testlib",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/protobuf:protobuf_java",
     ],
diff --git a/src/test/java/com/google/devtools/build/lib/rules/repository/BUILD b/src/test/java/com/google/devtools/build/lib/rules/repository/BUILD
index a66ac9b..a0d48f6 100644
--- a/src/test/java/com/google/devtools/build/lib/rules/repository/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/rules/repository/BUILD
@@ -39,7 +39,7 @@
         "//third_party:guava",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/protobuf:protobuf_java",
     ],
diff --git a/src/test/java/com/google/devtools/build/lib/shell/BUILD b/src/test/java/com/google/devtools/build/lib/shell/BUILD
index 3adcc7c..42e7a36 100644
--- a/src/test/java/com/google/devtools/build/lib/shell/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/shell/BUILD
@@ -46,7 +46,7 @@
         "//third_party:guava-testlib",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party:truth8",
     ],
@@ -266,7 +266,7 @@
         "//third_party:guava-testlib",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party:truth8",
     ],
diff --git a/src/test/java/com/google/devtools/build/lib/skyframe/BUILD b/src/test/java/com/google/devtools/build/lib/skyframe/BUILD
index 57fb1ed..8375528 100644
--- a/src/test/java/com/google/devtools/build/lib/skyframe/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/skyframe/BUILD
@@ -107,7 +107,7 @@
         "//third_party:guava-testlib",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/protobuf:protobuf_java",
     ],
diff --git a/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceASTFunctionTest.java b/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceASTFunctionTest.java
index 10018e2..c1bc87e 100644
--- a/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceASTFunctionTest.java
+++ b/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceASTFunctionTest.java
@@ -36,8 +36,8 @@
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
-import org.mockito.Matchers;
 import org.mockito.Mockito;
+import org.mockito.hamcrest.MockitoHamcrest;
 
 /**
  * Test for WorkspaceASTFunction.
@@ -69,9 +69,10 @@
 
   private SkyFunction.Environment getEnv() throws InterruptedException {
     SkyFunction.Environment env = Mockito.mock(SkyFunction.Environment.class);
-    Mockito.when(env.getValue(Matchers.argThat(new SkyKeyMatchers(FileValue.FILE))))
+    Mockito.when(env.getValue(MockitoHamcrest.argThat(new SkyKeyMatchers(FileValue.FILE))))
         .thenReturn(fakeWorkspaceFileValue);
-    Mockito.when(env.getValue(Matchers.argThat(new SkyKeyMatchers(SkyFunctions.PRECOMPUTED))))
+    Mockito.when(
+            env.getValue(MockitoHamcrest.argThat(new SkyKeyMatchers(SkyFunctions.PRECOMPUTED))))
         .thenReturn(new PrecomputedValue(Optional.<RootedPath>absent()));
     return env;
   }
diff --git a/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceFileFunctionTest.java b/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceFileFunctionTest.java
index f09bfd7..f6a2350 100644
--- a/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceFileFunctionTest.java
+++ b/src/test/java/com/google/devtools/build/lib/skyframe/WorkspaceFileFunctionTest.java
@@ -53,8 +53,8 @@
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
-import org.mockito.Matchers;
 import org.mockito.Mockito;
+import org.mockito.hamcrest.MockitoHamcrest;
 import org.mockito.invocation.InvocationOnMock;
 import org.mockito.stubbing.Answer;
 
@@ -165,10 +165,11 @@
 
   private SkyFunction.Environment getEnv() throws InterruptedException {
     SkyFunction.Environment env = Mockito.mock(SkyFunction.Environment.class);
-    Mockito.when(env.getValue(Matchers.argThat(new SkyKeyMatchers(FileValue.FILE))))
+    Mockito.when(env.getValue(MockitoHamcrest.argThat(new SkyKeyMatchers(FileValue.FILE))))
         .thenReturn(fakeWorkspaceFileValue);
     Mockito.when(
-            env.getValue(Matchers.argThat(new SkyKeyMatchers(WorkspaceFileValue.WORKSPACE_FILE))))
+            env.getValue(
+                MockitoHamcrest.argThat(new SkyKeyMatchers(WorkspaceFileValue.WORKSPACE_FILE))))
         .then(
             new Answer<SkyValue>() {
               @Override
@@ -177,7 +178,8 @@
                 return workspaceSkyFunc.compute(key, getEnv());
               }
             });
-    Mockito.when(env.getValue(Matchers.argThat(new SkyKeyMatchers(SkyFunctions.WORKSPACE_AST))))
+    Mockito.when(
+            env.getValue(MockitoHamcrest.argThat(new SkyKeyMatchers(SkyFunctions.WORKSPACE_AST))))
         .then(
             new Answer<SkyValue>() {
               @Override
@@ -186,7 +188,8 @@
                 return astSkyFunc.compute(key, getEnv());
               }
             });
-    Mockito.when(env.getValue(Matchers.argThat(new SkyKeyMatchers(SkyFunctions.PRECOMPUTED))))
+    Mockito.when(
+            env.getValue(MockitoHamcrest.argThat(new SkyKeyMatchers(SkyFunctions.PRECOMPUTED))))
         .then(
             new Answer<SkyValue>() {
               @Override
diff --git a/src/test/java/com/google/devtools/build/lib/skyframe/serialization/BUILD b/src/test/java/com/google/devtools/build/lib/skyframe/serialization/BUILD
index b6deccd..daeb7bf 100644
--- a/src/test/java/com/google/devtools/build/lib/skyframe/serialization/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/skyframe/serialization/BUILD
@@ -27,7 +27,7 @@
         "//third_party:guava-testlib",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
         "//third_party/protobuf:protobuf_java",
     ],
diff --git a/src/test/java/com/google/devtools/build/skyframe/BUILD b/src/test/java/com/google/devtools/build/skyframe/BUILD
index 7d590a3..a8e574d 100644
--- a/src/test/java/com/google/devtools/build/skyframe/BUILD
+++ b/src/test/java/com/google/devtools/build/skyframe/BUILD
@@ -64,7 +64,7 @@
         "//third_party:guava-testlib",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/common/options/BUILD b/src/test/java/com/google/devtools/common/options/BUILD
index 76d843b..d4e65f1 100644
--- a/src/test/java/com/google/devtools/common/options/BUILD
+++ b/src/test/java/com/google/devtools/common/options/BUILD
@@ -51,7 +51,7 @@
         "//third_party:jimfs",
         "//third_party:jsr305",
         "//third_party:junit4",
-        "//third_party:mockito",
+        "//third_party:mockito2",
         "//third_party:truth",
     ],
 )