Java Examples for org.mockito.Mockito.anyMap

@Test
@SuppressWarnings("unchecked")
public void pagedExecutionShouldNotGenerateCountQueryIfQueryReportedNoResults() {
    when(sessionMock.query(eq(User.class), anyString(), anyMap())).thenReturn(Collections.<User>emptyList());
    GraphQueryMethod queryMethod = new GraphQueryMethod(method, new DefaultRepositoryMetadata(UserRepository.class), factory);
    GraphParameterAccessor accessor = new GraphParametersParameterAccessor(queryMethod, new Object[] { "", new PageRequest(0, 1) });
    GraphQueryExecution.PagedExecution execution = new GraphQueryExecution.PagedExecution(sessionMock, accessor);
    Query query = new Query("", "noop", new HashMap<>());
    execution.execute(query, User.class);
    verify(sessionMock).query(eq(User.class), anyString(), anyMap());
    verify(sessionMock, never()).queryForObject(eq(Integer.class), any(String.class), anyMap());
}

@Test
public void testGetFormDisplayTaskWithoutProcess() {
    long taskId = 1;
    final TaskImpl task = new TaskImpl();
    task.setId(taskId);
    task.setName("TaskName");
    task.setTaskData(new TaskDataImpl());
    when(taskService.getTaskById(taskId)).thenReturn(task);
    final String form = formProviderService.getFormDisplayTask(1);
    assertEquals("", form);
    verify(dataService, never()).getProcessesByDeploymentIdProcessId(anyString(), anyString());
    verify(formProvider).render(eq(task.getName()), eq(task), isNull(ProcessDefinition.class), anyMap());
}

// Test zero source tasks and onVertexStarted is called
// before onVertexStateUpdated.
@Test(timeout = 5000)
public void testZeroSourceTasksWithVertexStartedFirst() {
    Configuration conf = new Configuration();
    ShuffleVertexManagerBase manager;
    final String mockSrcVertexId1 = "Vertex1";
    final String mockSrcVertexId2 = "Vertex2";
    final String mockSrcVertexId3 = "Vertex3";
    final String mockManagedVertexId = "Vertex4";
    final List<Integer> scheduledTasks = Lists.newLinkedList();
    // source vertices have 0 tasks.
    final VertexManagerPluginContext mockContext = createVertexManagerContext(mockSrcVertexId1, 0, mockSrcVertexId2, 0, mockSrcVertexId3, 1, mockManagedVertexId, 4, scheduledTasks, null);
    // check initialization
    // Tez notified of reconfig
    manager = createManager(conf, mockContext, 0.1f, 0.1f);
    manager.onVertexStarted(emptyCompletions);
    verify(mockContext, times(1)).vertexReconfigurationPlanned();
    // The edge between destination and source vertex mockSrcVertexId3 is
    // broadcast type. Thus mockSrcVertexId3 isn't counted as bipartiteSource.
    Assert.assertTrue(manager.bipartiteSources == 2);
    // check waiting for notification before scheduling
    Assert.assertFalse(manager.pendingTasks.isEmpty());
    // source vertices have 0 tasks. triggers scheduling
    manager.onVertexStateUpdated(new VertexStateUpdate(mockSrcVertexId1, VertexState.CONFIGURED));
    manager.onVertexStateUpdated(new VertexStateUpdate(mockSrcVertexId2, VertexState.CONFIGURED));
    manager.onVertexStateUpdated(new VertexStateUpdate(mockSrcVertexId3, VertexState.CONFIGURED));
    Assert.assertTrue(manager.pendingTasks.isEmpty());
    verify(mockContext, times(1)).reconfigureVertex(eq(1), any(VertexLocationHint.class), anyMap());
    // reconfig done
    verify(mockContext, times(1)).doneReconfiguringVertex();
    // all tasks scheduled and parallelism changed
    Assert.assertTrue(scheduledTasks.size() == 1);
    scheduledTasks.clear();
// TODO TEZ-1714 locking verify(mockContext, times(1)).vertexManagerDone(); // notified after scheduling all tasks
}

@Test
@SuppressWarnings("unchecked")
public void pagedExecutionShouldNotGenerateCountQueryIfQueryReportedNoResults() {
    when(sessionMock.query(eq(User.class), anyString(), anyMap())).thenReturn(Collections.<User>emptyList());
    GraphQueryMethod queryMethod = new GraphQueryMethod(method, new DefaultRepositoryMetadata(UserRepository.class), factory);
    GraphParameterAccessor accessor = new GraphParametersParameterAccessor(queryMethod, new Object[] { "", new PageRequest(0, 1) });
    GraphQueryExecution.PagedExecution execution = new GraphQueryExecution.PagedExecution(sessionMock, accessor);
    Query query = new Query("", "noop", new HashMap<>());
    execution.execute(query, User.class);
    verify(sessionMock).query(eq(User.class), anyString(), anyMap());
    verify(sessionMock, never()).queryForObject(eq(Integer.class), any(String.class), anyMap());
}

@Test
public void testWriteRepoAttributes() throws Exception {
    serializer.writeRepoAttributes(meta, json);
    verify(json, atLeastOnce()).writeObjectField(JobEntryBaseJsonSerializer.JSON_PROPERTY_FIELDS, anyList());
    verify(json, atLeastOnce()).writeObjectField(JobEntryBaseJsonSerializer.JSON_PROPERTY_ATTRIBUTES, anyMap());
}