public static CommandLineJobInfo createRandom() { CommandLineJobInfo result = new CommandLineJobInfo(); Random random = new Random(); String command = CommonUtils.randomString(random.nextInt(10)); JobConfInfo jobConfInfo = JobConfInfoTest.createRandom(); result.setCommand(command); result.setConf(jobConfInfo); return result; }
@Test public void requestBlockLocationTest() throws Exception { Map<String, String> params = new HashMap<>(); params.put("blockId", "1"); params.put("sessionId", "1"); params.put("initialBytes", "1"); String blockLocation = CommonUtils.randomString(10); Mockito.doReturn(blockLocation) .when(mBlockWorker) .createBlock(Mockito.anyLong(), Mockito.anyLong(), Mockito.anyString(), Mockito.anyLong()); TestCaseFactory.newWorkerTestCase( getEndpoint(BlockWorkerClientRestServiceHandler.REQUEST_BLOCK_LOCATION), params, "POST", blockLocation, mResource) .run(); Mockito.verify(mBlockWorker) .createBlock(Mockito.anyLong(), Mockito.anyLong(), Mockito.anyString(), Mockito.anyLong()); }