@Test public void partitionThenBucketChangesNoCreateAsPartitionEstablished() throws Exception { when(mockRecordInspector.extractRecordIdentifier(RECORD)).thenReturn(ROW__ID_B0_R0, ROW__ID_INSERT); when(mockBucketIdResolver.computeBucketId(RECORD)).thenReturn(0, 0); when(mockPartitionHelper.getPathForPartition(PARTITION_B)).thenReturn(PATH_B); coordinator.delete(PARTITION_B, RECORD); /* PbB0 */ coordinator.insert(PARTITION_B, RECORD); /* PbB0 */ verify(mockPartitionHelper, never()).createPartitionIfNotExists(anyList()); }
any(byte[].class), any(PTableType.class), anyMap(), anyList(), any(byte[][].class), eq(false), eq(false), eq(false)); reset(connectionQueryServices);
@Test public void testUpdateWorkflow() { WorkflowDef workflowDef = new WorkflowDef(); List<WorkflowDef> listOfWorkflowDef = new ArrayList<>(); listOfWorkflowDef.add(workflowDef); metadataResource.update(listOfWorkflowDef); verify(mockMetadataService, times(1)).updateWorkflowDef(anyList()); }
when(store.remove(anyList(), anyInt())).thenAnswer(new Answer<boolean[]>() { @Override public boolean[] answer(InvocationOnMock invocationOnMock)
@Nonnull static PurchaseVerifier mockVerifier(@Nonnull PurchaseVerifier verifier, final boolean verified) { doAnswer(new Answer<Object>() { @Override public Object answer(InvocationOnMock invocation) throws Throwable { final List<Purchase> purchases = (List<Purchase>) invocation.getArguments()[0]; final RequestListener<List<Purchase>> l = (RequestListener) invocation.getArguments()[1]; l.onSuccess(verified ? new ArrayList<Purchase>(purchases) : Collections.<Purchase>emptyList()); return null; } }).when(verifier).verify(anyList(), any(RequestListener.class)); return verifier; }
@Test public void getDuplicatePatientsByAttributes_shouldCallDaoGivenAttributes() throws Exception { when(patientDaoMock.getDuplicatePatientsByAttributes(anyList())).thenReturn( Collections.singletonList(mock(Patient.class))); final List<Patient> duplicatePatients = patientService .getDuplicatePatientsByAttributes(Arrays.asList("some attribute", "another attribute")); verify(patientDaoMock, times(1)).getDuplicatePatientsByAttributes(anyList()); Assert.assertEquals(duplicatePatients.size(), 1); }
@Test public void universalSuccessCheckerShouldSucceedOnNoEvents() { Checker universalChecker = Mockito.mock(Checker.class); Mockito.when(universalChecker.check(Matchers.anyList())).thenReturn(validResult); assertNoError(() -> universalChecker, ImmutableList.<Event>of()); }
@Test public void universalSuccessCheckerShouldSucceed() { Checker universalChecker = Mockito.mock(Checker.class); Mockito.when(universalChecker.check(Matchers.anyList())).thenReturn(validResult); assertNoError(() -> universalChecker, eventList); }
@Test @SuppressWarnings("unchecked") public void testEnableCustomizers() { AvaticaServerConfiguration avaticaServerConfiguration = null; HttpServer.Builder builder = mock(HttpServer.Builder.class); Configuration conf = new Configuration(false); conf.set(QueryServices.QUERY_SERVER_CUSTOMIZERS_ENABLED, "true"); QueryServer queryServer = new QueryServer(); queryServer.enableServerCustomizersIfNecessary(builder, conf, avaticaServerConfiguration); verify(builder).withServerCustomizers(anyList(), any(Class.class)); } }
@Test public void testCreateProviderToPhysicalNetworkFailure() throws ResourceAllocationException { NetworkService networkService = Mockito.mock(NetworkService.class); addNetworkServiceProviderCmd._networkService = networkService; Mockito.when(networkService.addProviderToPhysicalNetwork(Matchers.anyLong(), Matchers.anyString(), Matchers.anyLong(), Matchers.anyList())).thenReturn(null); try { addNetworkServiceProviderCmd.create(); } catch (ServerApiException exception) { Assert.assertEquals("Failed to add service provider entity to physical network", exception.getDescription()); } }
@Test public void testCreateProviderToPhysicalNetworkSuccess() { NetworkService networkService = Mockito.mock(NetworkService.class); addNetworkServiceProviderCmd._networkService = networkService; PhysicalNetworkServiceProvider physicalNetworkServiceProvider = Mockito.mock(PhysicalNetworkServiceProvider.class); Mockito.when(networkService.addProviderToPhysicalNetwork(Matchers.anyLong(), Matchers.anyString(), Matchers.anyLong(), Matchers.anyList())).thenReturn( physicalNetworkServiceProvider); try { addNetworkServiceProviderCmd.create(); } catch (ResourceAllocationException e) { e.printStackTrace(); } }
@Test public void testSetAsync() throws Exception { OfflineStore lds = mock(OfflineStore.class); when(lds.unpinAllObjectsAsync(anyString())).thenReturn(Task.<Void>forResult(null)); when(lds.pinAllObjectsAsync(anyString(), anyList(), anyBoolean())) .thenReturn(Task.forResult(null)); Parse.setLocalDatastore(lds); OfflineObjectStore<ParseUser> store = new OfflineObjectStore<>(ParseUser.class, PIN_NAME, null); ParseUser user = mock(ParseUser.class); ParseTaskUtils.wait(store.setAsync(user)); verify(lds, times(1)).unpinAllObjectsAsync(PIN_NAME); verify(user, times(1)).pinInBackground(PIN_NAME, false); }
ParseCorePlugins.getInstance().registerQueryController(queryController); OfflineStore lds = mock(OfflineStore.class); when(lds.pinAllObjectsAsync(anyString(), anyList(), anyBoolean())) .thenReturn(Task.forResult(null)); when(lds.unpinAllObjectsAsync(anyString())).thenReturn(Task.<Void>forResult(null)); when(lds.pinAllObjectsAsync(anyString(), anyList(), anyBoolean())) .thenReturn(Task.<Void>forResult(null)); Parse.setLocalDatastore(lds);
/** * Initializes the mocks. */ @SuppressWarnings("unchecked") @BeforeClass public void init() { processor = mock(IIndexQueryRestrictionProcessor.class); when(processor.areAllRestrictionsFulfilled(anyObject(), anyList())).thenReturn(true); }
@Test public void testImportTableExecutesHiveImportWithCreateTableAndLoadDataCommands() throws Exception { doNothing().when(hs2ClientSpy).executeHiveImport(anyList()); hs2ClientSpy.importTable(); verify(hs2ClientSpy, times(1)).executeHiveImport(asList(CREATE_TABLE_STATEMENT, LOAD_DATA_STATEMENT)); }
@Test public void testCreateTableExecutesHiveImportWithCreateTableCommandOnly() throws Exception { doNothing().when(hs2ClientSpy).executeHiveImport(anyList()); hs2ClientSpy.createTable(); verify(hs2ClientSpy, times(1)).executeHiveImport(asList(CREATE_TABLE_STATEMENT)); }
@Before public void setup() { MockitoAnnotations.initMocks(this); when((StateInternals) mockStepContext.stateInternals()).thenReturn(stateInternals); when(mockEvaluationContext.createSideInputReader(anyList())) .thenReturn( SideInputContainer.create(mockEvaluationContext, Collections.emptyList()) .createReaderForViews(Collections.emptyList())); }
@Test public void noClassCacheWarningShouldBeLoggedOnce() throws Exception { when(connection.isConnected()).thenReturn(true); BusinessException businessException = new BusinessException(AgentManagementErrorCodeEnum.AGENT_DOES_NOT_EXIST); doThrow(businessException).when(connection).analyzeJmxAttributes(anyLong(), Matchers.anyList()); jmxSensor.mbeanServerAdded(mBeanServer); verify(agentStatisticsLogger, times(1)).noClassCacheAvailable(); } }
@Test public void testExecuteHiveImportThrowsRuntimeExceptionWhenExecuteCommandsThrows() throws Exception { SQLException sqlException = mock(SQLException.class); doThrow(sqlException).when(hs2ClientSpy).executeCommands(anyList()); expectedException.expect(RuntimeException.class); expectedException.expectMessage("Error executing Hive import."); hs2ClientSpy.executeHiveImport(TEST_COMMANDS); }
@Test public void appliesImplicitConversionOnConverterWhenSourceAndReturnTypeDoesNotMatch() throws MuleException { // Converter(C->D), payload B: uses implicit conversion B->C Transformer converter1 = new MockConverterBuilder().from(dataTypeC).to(dataTypeD).returning(new D()).build(); Transformer converter2 = new MockConverterBuilder().from(dataTypeB).to(dataTypeC).returning(new C()).build(); when(conversionResolver.resolve(any(DataType.class), anyList())).thenReturn(converter2); Message message = of(new B()); message = transformationService.applyTransformers(message, null, converter1); assertTrue(message.getPayload().getValue() instanceof D); verifyTransformerExecuted(converter1); verifyTransformerExecuted(converter2); }