HDDS-1559. Fix TestReplicationManager. Contributed by Bharat Viswanadham. (#856)

hadoop-hdds/server-scm/src/test/java/org/apache/hadoop/hdds/scm/container/TestReplicationManager.java

@@ -96,10 +96,10 @@ public class TestReplicationManager {
     containerPlacementPolicy = Mockito.mock(ContainerPlacementPolicy.class);
 
     Mockito.when(containerPlacementPolicy.chooseDatanodes(
-        Mockito.anyListOf(DatanodeDetails.class), null,
-        Mockito.anyInt(), Mockito.anyLong()))
-        .thenAnswer(invocation -> {
-          int count = (int) invocation.getArguments()[1];
+        Mockito.anyListOf(DatanodeDetails.class),
+        Mockito.anyListOf(DatanodeDetails.class), Mockito.anyInt(),
+        Mockito.anyLong())).thenAnswer(invocation -> {
+          int count = (int) invocation.getArguments()[2];
           return IntStream.range(0, count)
               .mapToObj(i -> randomDatanodeDetails())
               .collect(Collectors.toList());