Fix static mocking in TransactUtilsTest 85/80785/1
authorRobert Varga <robert.varga@pantheon.tech>
Mon, 11 Mar 2019 18:30:46 +0000 (19:30 +0100)
committerRobert Varga <robert.varga@pantheon.tech>
Mon, 11 Mar 2019 18:31:55 +0000 (19:31 +0100)
Change static mocks to doReturn().when(...) mocking style, which
allows them to properly work with Mockito 2.

Change-Id: Ie5b2f0223fd67b942a5c77cd56f5ccaa4fa870e1
JIRA: OVSDB-476
Signed-off-by: Robert Varga <robert.varga@pantheon.tech>
southbound/southbound-impl/src/test/java/org/opendaylight/ovsdb/southbound/ovsdb/transact/TransactUtilsTest.java

index e485b63989f1c7f3b10fab26e8eeea0896998015..edcfb2840bec30febad55e9b0a05b1c85945218c 100644 (file)
@@ -48,7 +48,6 @@ import org.opendaylight.ovsdb.southbound.SouthboundUtil;
 import org.opendaylight.yangtools.yang.binding.DataObject;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 import org.powermock.api.mockito.PowerMockito;
-import org.powermock.api.support.membermodification.MemberMatcher;
 import org.powermock.core.classloader.annotations.PrepareForTest;
 import org.powermock.modules.junit4.PowerMockRunner;
 
@@ -92,19 +91,17 @@ public class TransactUtilsTest {
     public void testExtractCreatedOrUpdated() {
         Map<InstanceIdentifier<DataObject>, DataObject> result = new HashMap<>();
 
-        PowerMockito.suppress(
-                MemberMatcher.method(TransactUtils.class, "extractUpdated", DataChangeEvent.class, Class.class));
-        PowerMockito.when(TransactUtils.extractUpdated(any(DataChangeEvent.class), eq(DataObject.class)))
-                .thenReturn(result);
+        PowerMockito.doReturn(result).when(TransactUtils.class);
+        TransactUtils.extractUpdated(any(DataChangeEvent.class), eq(DataObject.class));
 
         Map<InstanceIdentifier<DataObject>, DataObject> map = new HashMap<>();
         InstanceIdentifier<DataObject> iid = mock(InstanceIdentifier.class);
+
         DataObject db = mock(DataObject.class);
         map.put(iid, db);
-        PowerMockito.suppress(
-                MemberMatcher.method(TransactUtils.class, "extractCreated", DataChangeEvent.class, Class.class));
-        PowerMockito.when(TransactUtils.extractCreated(any(DataChangeEvent.class), eq(DataObject.class)))
-                .thenReturn(map);
+
+        PowerMockito.doReturn(map).when(TransactUtils.class);
+        TransactUtils.extractCreated(any(DataChangeEvent.class), eq(DataObject.class));
 
         Map<InstanceIdentifier<DataObject>, DataObject> testResult = new HashMap<>();
         testResult.put(iid, db);
@@ -118,19 +115,16 @@ public class TransactUtilsTest {
     public void testExtractCreatedOrUpdatedOrRemoved() {
         Map<InstanceIdentifier<DataObject>, DataObject> result = new HashMap<>();
 
-        PowerMockito.suppress(MemberMatcher.method(TransactUtils.class, "extractCreatedOrUpdated",
-                DataChangeEvent.class, Class.class));
-        PowerMockito.when(TransactUtils.extractCreatedOrUpdated(any(DataChangeEvent.class), eq(DataObject.class)))
-                .thenReturn(result);
+        PowerMockito.doReturn(result).when(TransactUtils.class);
+        TransactUtils.extractCreatedOrUpdated(any(DataChangeEvent.class), eq(DataObject.class));
 
         Map<InstanceIdentifier<DataObject>, DataObject> map = new HashMap<>();
         InstanceIdentifier<DataObject> iid = mock(InstanceIdentifier.class);
         DataObject db = mock(DataObject.class);
         map.put(iid, db);
-        PowerMockito.suppress(MemberMatcher.method(TransactUtils.class, "extractRemovedObjects",
-                DataChangeEvent.class, Class.class));
-        PowerMockito.when(TransactUtils.extractRemovedObjects(any(DataChangeEvent.class), eq(DataObject.class)))
-                .thenReturn(map);
+
+        PowerMockito.doReturn(map).when(TransactUtils.class);
+        TransactUtils.extractRemovedObjects(any(DataChangeEvent.class), eq(DataObject.class));
 
         Map<InstanceIdentifier<DataObject>, DataObject> testResult = new HashMap<>();
         testResult.put(iid, db);
@@ -160,20 +154,17 @@ public class TransactUtilsTest {
         assertEquals(HashSet.class, TransactUtils.extractRemoved(changes, klazz).getClass());
     }
 
-    @SuppressWarnings("unchecked")
     @Test
     public void testExtractRemovedObjects() {
         Set<InstanceIdentifier<DataObject>> iids = new HashSet<>();
-        PowerMockito.suppress(
-                MemberMatcher.method(TransactUtils.class, "extractRemoved", DataChangeEvent.class, Class.class));
-        PowerMockito.when(TransactUtils.extractRemoved(any(DataChangeEvent.class), eq(DataObject.class)))
-                .thenReturn(iids);
+
+        PowerMockito.doReturn(iids).when(TransactUtils.class);
+        TransactUtils.extractRemoved(any(DataChangeEvent.class), eq(DataObject.class));
 
         Map<InstanceIdentifier<DataObject>, DataObject> result = new HashMap<>();
-        PowerMockito.suppress(
-                MemberMatcher.method(TransactUtils.class, "extractOriginal", DataChangeEvent.class, Class.class));
-        PowerMockito.when(TransactUtils.extractOriginal(any(DataChangeEvent.class), eq(DataObject.class)))
-                .thenReturn(result);
+
+        PowerMockito.doReturn(result).when(TransactUtils.class);
+        TransactUtils.extractOriginal(any(DataChangeEvent.class), eq(DataObject.class));
 
         Class<DataObject> klazz = DataObject.class;
         DataChangeEvent changes = mock(DataChangeEvent.class);
@@ -226,15 +217,10 @@ public class TransactUtilsTest {
         ColumnSchema<GenericTableSchema, Map<String,String>> columnSchema = mock(ColumnSchema.class);
         InstanceIdentifierCodec instanceIdentifierCodec = mock(InstanceIdentifierCodec.class);
 
-        PowerMockito.suppress(MemberMatcher.method(TransactUtils.class, "stampInstanceIdentifierMutation",
-                TransactionBuilder.class,
-                InstanceIdentifier.class,
-                TableSchema.class,
-                ColumnSchema.class,
-                InstanceIdentifierCodec.class));
-        when(TransactUtils.stampInstanceIdentifierMutation(transaction, iid, tableSchema, columnSchema,
-                instanceIdentifierCodec))
-                .thenReturn(mock(Mutate.class));
+        PowerMockito.doReturn(mock(Mutate.class)).when(TransactUtils.class);
+        TransactUtils.stampInstanceIdentifierMutation(transaction, iid, tableSchema, columnSchema,
+            instanceIdentifierCodec);
+
         when(transaction.add(any(Operation.class))).thenReturn(transaction);
         TransactUtils.stampInstanceIdentifier(transaction, iid, tableSchema, columnSchema, instanceIdentifierCodec);
         verify(transaction).add(any(Operation.class));