public static BulkEmitter createTestBulkEmitter(int batchSize, int interval, BatchOperations batchOperations) { return spy(new BulkEmitter(batchSize, interval, batchOperations)); }
@Override public BulkEmitter createInstance(int batchSize, int deliveryInterval, ClientObjectFactory clientObjectFactory, FailoverPolicy failoverPolicy) { BulkEmitter bulkEmitter = new BulkEmitter(batchSize, deliveryInterval, clientObjectFactory.createBatchOperations()); bulkEmitter.addListener(clientObjectFactory.createBatchListener(failoverPolicy)); return bulkEmitter; }
@Override public BatchEmitter createInstance(int batchSize, int deliveryInterval, ClientObjectFactory clientObjectFactory, FailoverPolicy failoverPolicy) { if (spiedEmitter == null) { BulkEmitter emitter = new BulkEmitter(batchSize, deliveryInterval, clientObjectFactory.createBatchOperations()); emitter.addListener(clientObjectFactory.createBatchListener(failoverPolicy)); spiedEmitter = Mockito.spy(emitter); } return spiedEmitter; } }