diff options
Diffstat (limited to 'src/test')
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/skyframe/PackageValueTest.java | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/skyframe/PackageValueTest.java b/src/test/java/com/google/devtools/build/lib/skyframe/PackageValueTest.java index c8d9d9e03b..38cbbf77e0 100644 --- a/src/test/java/com/google/devtools/build/lib/skyframe/PackageValueTest.java +++ b/src/test/java/com/google/devtools/build/lib/skyframe/PackageValueTest.java @@ -36,7 +36,6 @@ import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.junit.runners.JUnit4; -import org.mockito.ArgumentCaptor; /** Basic tests for {@link PackageValue}. */ @RunWith(JUnit4.class) @@ -60,8 +59,8 @@ public class PackageValueTest { // Mock because all we need is to verify that we're properly delegating to Package deserializer. Package mockPackage = mock(Package.class); - when(mockDeserializer.deserialize(ArgumentCaptor.forClass(CodedInputStream.class).capture())) - .thenReturn(mockPackage); + when(mockDeserializer.deserialize( + any(DeserializationContext.class), any(CodedInputStream.class))).thenReturn(mockPackage); CodedInputStream codedIn = CodedInputStream.newInstance(new byte[] {1, 2, 3, 4}); PackageValue result = @@ -75,7 +74,8 @@ public class PackageValueTest { @Test public void testInterruptedExceptionRaisesIllegalStateException() throws Exception { InterruptedException staged = new InterruptedException("Stop that!"); - doThrow(staged).when(mockDeserializer).deserialize(any(CodedInputStream.class)); + doThrow(staged).when(mockDeserializer) + .deserialize(any(DeserializationContext.class), any(CodedInputStream.class)); try { underTest.deserialize( |