@Test public void testConfigureAccumuloInputFormatWithIterators() throws Exception { AccumuloConnectionParameters accumuloParams = new AccumuloConnectionParameters(conf); ColumnMapper columnMapper = new ColumnMapper( conf.get(AccumuloSerDeParameters.COLUMN_MAPPINGS), conf.get(AccumuloSerDeParameters.DEFAULT_STORAGE_TYPE), columnNames, columnTypes); Set<Pair<Text, Text>> cfCqPairs = inputformat.getPairCollection(columnMapper.getColumnMappings()); List<IteratorSetting> iterators = new ArrayList<IteratorSetting>(); Set<Range> ranges = Collections.singleton(new Range()); String instanceName = "realInstance"; String zookeepers = "host1:2181,host2:2181,host3:2181"; IteratorSetting cfg = new IteratorSetting(50, PrimitiveComparisonFilter.class); cfg.addOption(PrimitiveComparisonFilter.P_COMPARE_CLASS, StringCompare.class.getName()); cfg.addOption(PrimitiveComparisonFilter.COMPARE_OPT_CLASS, Equal.class.getName()); cfg.addOption(PrimitiveComparisonFilter.CONST_VAL, "dave"); cfg.addOption(PrimitiveComparisonFilter.COLUMN, "person:name"); iterators.add(cfg); cfg = new IteratorSetting(50, PrimitiveComparisonFilter.class); cfg.addOption(PrimitiveComparisonFilter.P_COMPARE_CLASS, IntCompare.class.getName()); cfg.addOption(PrimitiveComparisonFilter.COMPARE_OPT_CLASS, Equal.class.getName()); cfg.addOption(PrimitiveComparisonFilter.CONST_VAL, "50"); cfg.addOption(PrimitiveComparisonFilter.COLUMN, "person:age"); iterators.add(cfg); ZooKeeperInstance zkInstance = Mockito.mock(ZooKeeperInstance.class); HiveAccumuloTableInputFormat mockInputFormat = Mockito.mock(HiveAccumuloTableInputFormat.class); // Stub out the ZKI mock Mockito.when(zkInstance.getInstanceName()).thenReturn(instanceName); Mockito.when(zkInstance.getZooKeepers()).thenReturn(zookeepers); // Call out to the real configure method Mockito.doCallRealMethod() .when(mockInputFormat) .configure(conf, zkInstance, con, accumuloParams, columnMapper, iterators, ranges); // Also compute the correct cf:cq pairs so we can assert the right argument was passed Mockito.doCallRealMethod() .when(mockInputFormat) .getPairCollection(columnMapper.getColumnMappings()); mockInputFormat.configure( conf, zkInstance, con, accumuloParams, columnMapper, iterators, ranges); // Verify that the correct methods are invoked on AccumuloInputFormat Mockito.verify(mockInputFormat).setZooKeeperInstance(conf, instanceName, zookeepers, false); Mockito.verify(mockInputFormat).setConnectorInfo(conf, USER, new PasswordToken(PASS)); Mockito.verify(mockInputFormat).setInputTableName(conf, TEST_TABLE); Mockito.verify(mockInputFormat) .setScanAuthorizations(conf, con.securityOperations().getUserAuthorizations(USER)); Mockito.verify(mockInputFormat).addIterators(conf, iterators); Mockito.verify(mockInputFormat).setRanges(conf, ranges); Mockito.verify(mockInputFormat).fetchColumns(conf, cfCqPairs); }
@Before public void initComponents() { view = new MvpView() {}; presenter = Mockito.mock(MvpPresenter.class); callback = Mockito.mock(PartialMvpDelegateCallbackImpl.class); Mockito.doCallRealMethod().when(callback).setPresenter(presenter); Mockito.doCallRealMethod().when(callback).getPresenter(); Mockito.when(callback.getMvpView()).thenReturn(view); delegate = new ViewGroupMvpDelegateImpl<>(callback); }
public static PrintJob createTestPrintJob(PrintFileProcessor processor) throws InappropriateDeviceException, Exception { PrintJob printJob = Mockito.mock(PrintJob.class); Printer printer = Mockito.mock(Printer.class); PrinterConfiguration printerConfiguration = Mockito.mock(PrinterConfiguration.class); SlicingProfile slicingProfile = Mockito.mock(SlicingProfile.class); InkConfig inkConfiguration = Mockito.mock(InkConfig.class); eGENERICGCodeControl gCode = Mockito.mock(eGENERICGCodeControl.class); SerialCommunicationsPort serialPort = Mockito.mock(SerialCommunicationsPort.class); MonitorDriverConfig monitorConfig = Mockito.mock(MonitorDriverConfig.class); MachineConfig machine = Mockito.mock(MachineConfig.class); Mockito.when(printJob.getJobFile()).thenReturn(new File("jobname.txt")); Mockito.when(printJob.getPrinter()).thenReturn(printer); Mockito.when(printer.getPrinterFirmwareSerialPort()).thenReturn(serialPort); Mockito.when(printJob.getPrintFileProcessor()).thenReturn(processor); Mockito.when(printer.getConfiguration()).thenReturn(printerConfiguration); Mockito.when(printer.waitForPauseIfRequired()).thenReturn(true); Mockito.when(printer.isPrintActive()).thenReturn(true); Mockito.when(printerConfiguration.getSlicingProfile()).thenReturn(slicingProfile); Mockito.when(slicingProfile.getSelectedInkConfig()).thenReturn(inkConfiguration); Mockito.when(slicingProfile.getDirection()).thenReturn(BuildDirection.Bottom_Up); Mockito.when(printer.getGCodeControl()).thenReturn(gCode); Mockito.when(slicingProfile.getgCodeLift()).thenReturn("Lift z"); Mockito.doCallRealMethod() .when(gCode) .executeGCodeWithTemplating(Mockito.any(PrintJob.class), Mockito.anyString()); Mockito.when(printer.getConfiguration().getMachineConfig()).thenReturn(machine); Mockito.when(printer.getConfiguration().getMachineConfig().getMonitorDriverConfig()) .thenReturn(monitorConfig); return printJob; }
@Test public void testConfigureAccumuloInputFormatWithAuthorizations() throws Exception { AccumuloConnectionParameters accumuloParams = new AccumuloConnectionParameters(conf); conf.set(AccumuloSerDeParameters.AUTHORIZATIONS_KEY, "foo,bar"); ColumnMapper columnMapper = new ColumnMapper( conf.get(AccumuloSerDeParameters.COLUMN_MAPPINGS), conf.get(AccumuloSerDeParameters.DEFAULT_STORAGE_TYPE), columnNames, columnTypes); Set<Pair<Text, Text>> cfCqPairs = inputformat.getPairCollection(columnMapper.getColumnMappings()); List<IteratorSetting> iterators = Collections.emptyList(); Set<Range> ranges = Collections.singleton(new Range()); String instanceName = "realInstance"; String zookeepers = "host1:2181,host2:2181,host3:2181"; ZooKeeperInstance zkInstance = Mockito.mock(ZooKeeperInstance.class); HiveAccumuloTableInputFormat mockInputFormat = Mockito.mock(HiveAccumuloTableInputFormat.class); // Stub out the ZKI mock Mockito.when(zkInstance.getInstanceName()).thenReturn(instanceName); Mockito.when(zkInstance.getZooKeepers()).thenReturn(zookeepers); // Call out to the real configure method Mockito.doCallRealMethod() .when(mockInputFormat) .configure(conf, zkInstance, con, accumuloParams, columnMapper, iterators, ranges); // Also compute the correct cf:cq pairs so we can assert the right argument was passed Mockito.doCallRealMethod() .when(mockInputFormat) .getPairCollection(columnMapper.getColumnMappings()); mockInputFormat.configure( conf, zkInstance, con, accumuloParams, columnMapper, iterators, ranges); // Verify that the correct methods are invoked on AccumuloInputFormat Mockito.verify(mockInputFormat).setZooKeeperInstance(conf, instanceName, zookeepers, false); Mockito.verify(mockInputFormat).setConnectorInfo(conf, USER, new PasswordToken(PASS)); Mockito.verify(mockInputFormat).setInputTableName(conf, TEST_TABLE); Mockito.verify(mockInputFormat).setScanAuthorizations(conf, new Authorizations("foo,bar")); Mockito.verify(mockInputFormat).addIterators(conf, iterators); Mockito.verify(mockInputFormat).setRanges(conf, ranges); Mockito.verify(mockInputFormat).fetchColumns(conf, cfCqPairs); }
@Test public void testCreateTicket() throws Exception { Ticket ticket = new Ticket(); ticket.setTicketId(1); Mockito.when(ticketDao.createTicket(ticket)).thenReturn(ticket); Mockito.doCallRealMethod().when(emailSenderService).newTicketCreated(ticket.getEmail()); MessageResponse messageResponse = ticketService.createTicket(ticket); assertTrue(messageResponse.getCode() == 1); }
@Test public void testConfigureMockAccumuloInputFormat() throws Exception { AccumuloConnectionParameters accumuloParams = new AccumuloConnectionParameters(conf); ColumnMapper columnMapper = new ColumnMapper( conf.get(AccumuloSerDeParameters.COLUMN_MAPPINGS), conf.get(AccumuloSerDeParameters.DEFAULT_STORAGE_TYPE), columnNames, columnTypes); Set<Pair<Text, Text>> cfCqPairs = inputformat.getPairCollection(columnMapper.getColumnMappings()); List<IteratorSetting> iterators = Collections.emptyList(); Set<Range> ranges = Collections.singleton(new Range()); HiveAccumuloTableInputFormat mockInputFormat = Mockito.mock(HiveAccumuloTableInputFormat.class); // Call out to the real configure method Mockito.doCallRealMethod() .when(mockInputFormat) .configure(conf, mockInstance, con, accumuloParams, columnMapper, iterators, ranges); // Also compute the correct cf:cq pairs so we can assert the right argument was passed Mockito.doCallRealMethod() .when(mockInputFormat) .getPairCollection(columnMapper.getColumnMappings()); mockInputFormat.configure( conf, mockInstance, con, accumuloParams, columnMapper, iterators, ranges); // Verify that the correct methods are invoked on AccumuloInputFormat Mockito.verify(mockInputFormat).setMockInstance(conf, mockInstance.getInstanceName()); Mockito.verify(mockInputFormat).setConnectorInfo(conf, USER, new PasswordToken(PASS)); Mockito.verify(mockInputFormat).setInputTableName(conf, TEST_TABLE); Mockito.verify(mockInputFormat) .setScanAuthorizations(conf, con.securityOperations().getUserAuthorizations(USER)); Mockito.verify(mockInputFormat).addIterators(conf, iterators); Mockito.verify(mockInputFormat).setRanges(conf, ranges); Mockito.verify(mockInputFormat).fetchColumns(conf, cfCqPairs); }
private void mockSession() { final EventLoop eventLoop = Mockito.mock(EventLoop.class); final Channel channel = Mockito.mock(Channel.class); final ChannelPipeline pipeline = Mockito.mock(ChannelPipeline.class); Mockito.doReturn(null) .when(eventLoop) .schedule(any(Runnable.class), any(long.class), any(TimeUnit.class)); Mockito.doReturn(eventLoop).when(channel).eventLoop(); Mockito.doReturn(Boolean.TRUE).when(channel).isWritable(); Mockito.doReturn(null).when(channel).close(); Mockito.doReturn(pipeline).when(channel).pipeline(); Mockito.doCallRealMethod().when(channel).toString(); Mockito.doReturn(pipeline).when(pipeline).addLast(Mockito.any(ChannelHandler.class)); Mockito.doReturn(new DefaultChannelPromise(channel)) .when(channel) .writeAndFlush(any(Notification.class)); Mockito.doReturn(new InetSocketAddress("localhost", 12345)).when(channel).remoteAddress(); Mockito.doReturn(new InetSocketAddress("localhost", 12345)).when(channel).localAddress(); final List<BgpParameters> params = Lists.newArrayList( new BgpParametersBuilder() .setOptionalCapabilities( Lists.newArrayList( new OptionalCapabilitiesBuilder() .setCParameters( new CParametersBuilder() .addAugmentation( CParameters1.class, new CParameters1Builder() .setMultiprotocolCapability( new MultiprotocolCapabilityBuilder() .setAfi(Ipv4AddressFamily.class) .setSafi(UnicastSubsequentAddressFamily.class) .build()) .build()) .build()) .build())) .build()); this.session = new BGPSessionImpl( this.classic, channel, new OpenBuilder() .setBgpIdentifier(new Ipv4Address("1.1.1.1")) .setHoldTimer(50) .setMyAsNumber(72) .setBgpParameters(params) .build(), 30, null); }
@Test public void testUpdateRetryWithError() { Repository<Entity> mockRepo = Mockito.spy(repository); Map<String, Object> studentBody = buildTestStudentEntity(); Map<String, Object> studentMetaData = new HashMap<String, Object>(); Entity entity = new MongoEntity("student", null, studentBody, studentMetaData); int noOfRetries = 3; Mockito.doThrow(new InvalidDataAccessApiUsageException("Test Exception")) .when(mockRepo) .update("student", entity, false); Mockito.doCallRealMethod().when(mockRepo).updateWithRetries("student", entity, noOfRetries); try { mockRepo.updateWithRetries("student", entity, noOfRetries); } catch (InvalidDataAccessApiUsageException ex) { assertEquals(ex.getMessage(), "Test Exception"); } Mockito.verify(mockRepo, Mockito.times(noOfRetries)).update("student", entity, false); }
@Test @Ignore("Need to work out how to check set subscription values") public void testPassingInvalidSubscriptionTypeSetsSubscriptionToNone() throws Exception { IQ request = toIq( readStanzaAsString("/iq/pubsub/subscribe/authorizationPendingGrantReply.stanza") .replaceFirst( "subscription='subscribed'", "subscription='i-can-haz-all-the-items'")); NodeAffiliation subscriptionMockActor = Mockito.mock(NodeAffiliation.class); Mockito.when(subscriptionMockActor.getAffiliation()).thenReturn(Affiliations.owner); NodeSubscription subscriptionMockSubscriber = Mockito.mock(NodeSubscription.class); Mockito.when(subscriptionMockSubscriber.getSubscription()).thenReturn(Subscriptions.subscribed); Mockito.when(dataStore.nodeExists(node)).thenReturn(true); Mockito.when(dataStore.getUserAffiliation(node, jid)).thenReturn(subscriptionMockActor); Mockito.doCallRealMethod() .when(dataStore) .addUserSubscription(Mockito.any(NodeSubscription.class)); Mockito.when(dataStore.getUserSubscription(node, new JID(subscriber))) .thenReturn(subscriptionMockSubscriber); event.setChannelManager(dataStore); event.process(element, jid, request, null); NodeSubscription subscriptionMock = new NodeSubscriptionImpl( node, new JID("*****@*****.**"), new JID("*****@*****.**"), Subscriptions.none); /* * subscriptionMock Mockito.anyString(), Mockito.any(JID.class), * Mockito.any(JID.class), Mockito.eq(Subscriptions.none)); */ Mockito.verify(dataStore).addUserSubscription(subscriptionMock); }
@Test public void testCreateRetryWithError() { Repository<Entity> mockRepo = Mockito.spy(repository); Map<String, Object> studentBody = buildTestStudentEntity(); Map<String, Object> studentMetaData = new HashMap<String, Object>(); int noOfRetries = 5; Mockito.doThrow(new MongoException("Test Exception")) .when(((MongoEntityRepository) mockRepo)) .internalCreate("student", null, studentBody, studentMetaData, "student"); Mockito.doCallRealMethod() .when(mockRepo) .createWithRetries("student", null, studentBody, studentMetaData, "student", noOfRetries); try { mockRepo.createWithRetries( "student", null, studentBody, studentMetaData, "student", noOfRetries); } catch (MongoException ex) { assertEquals(ex.getMessage(), "Test Exception"); } Mockito.verify((MongoEntityRepository) mockRepo, Mockito.times(noOfRetries)) .internalCreate("student", null, studentBody, studentMetaData, "student"); }