@Override public Subscriber getSubscriber() { return context == QUERY_SUBMISSION ? getPubscriber() : new DRPCQuerySubscriber(config, maxUncommittedMessages); }
@Override public Publisher getPublisher() { return context == QUERY_SUBMISSION ? getPubscriber() : new DRPCResultPublisher(config); }
private Consumer<Response> createResponseConsumer(String id) { // Create a closure with id return response -> handleResponse(id, response); }
@BeforeMethod public void setup() { DRPCConfig config = new DRPCConfig("src/test/resources/test_drpc_config.yaml"); Map stormConfig = new Config("src/test/resources/test_storm_config.yaml").getAll(Optional.empty()); config.set(DRPCConfig.STORM_CONFIG, stormConfig); publisher = new DRPCResultPublisher(config); collector = publisher.getCollector(); // Override the ReturnResults with our own that uses our collector and fails every two tuples injectedMockBolt = new MockReturnResults(collector, 5); publisher.setBolt(injectedMockBolt); }
@Test public void testQuerySubmissionSingleInstanceTypes() throws Exception { config.set(DRPCConfig.PUBSUB_CONTEXT_NAME, PubSub.Context.QUERY_SUBMISSION.name()); DRPCPubSub pubSub = new DRPCPubSub(config); Publisher publisher = pubSub.getPublisher(); Subscriber subscriber = pubSub.getSubscriber(); Assert.assertTrue(publisher instanceof DRPCQueryResultPubscriber); Assert.assertTrue(subscriber instanceof DRPCQueryResultPubscriber); }
public void addMessageParts(String id, String... contents) { int size = messageIDs.size(); int index = size; for (String content : contents) { List<Object> tuple = makeTuple(makeMessage(id, content, index - size), makeReturnInfo("fake" + id, "testHost", index)); tuples.offer(tuple); Object messageID = makeMessageID(id, index); messageIDs.offer(messageID); index++; } }
@BeforeMethod public void setup() { config = new DRPCConfig("src/test/resources/test_drpc_config.yaml"); config.set(DRPCConfig.DRPC_SERVERS, Collections.singletonList("foo.bar.bullet.drpc.com")); pubscriber = new DRPCQueryResultPubscriber(config); }
@Test(expectedExceptions = PubSubException.class) public void testFailing() throws Exception { injectedMockBolt.setFailNumber(1); Assert.assertEquals(injectedMockBolt.getCount(), 0); PubSubMessage message = new PubSubMessage("foo", "{}", new Metadata(null, makeReturnInfo("a", "testHost", 80))); publisher.send(message); }
/** * Create a DRPCPubSub using a {@link BulletConfig}. * * @param config The BulletConfig containing settings to create a new DRPCPubSub. * @throws PubSubException if there were errors in creating an instance. */ public DRPCPubSub(BulletConfig config) throws PubSubException { super(config); this.config = new DRPCConfig(config); maxUncommittedMessages = getRequiredConfig(Number.class, DRPCConfig.DRPC_MAX_UNCOMMITED_MESSAGES).intValue(); }
@Test public void testCleaningUp() { Assert.assertFalse(injectedMockBolt.isCleanedUp()); publisher.close(); Assert.assertTrue(injectedMockBolt.isCleanedUp()); } }
@Test public void testDefaultInitialization() { DRPCConfig config = makeEmpty(); Assert.assertEquals(config.get(DRPCConfig.DRPC_MAX_UNCOMMITED_MESSAGES), DRPCConfig.DEFAULT_DRPC_MAX_UNCOMMITED_MESSAGES); Assert.assertEquals(config.get(DRPCConfig.TOPOLOGY_NAME), DRPCConfig.DEFAULT_TOPOLOGY_NAME); Assert.assertEquals(config.get(DRPCConfig.QUERY_MAX_DURATION), DRPCConfig.DEFAULT_QUERY_MAX_DURATION); Assert.assertEquals(config.get(DRPCConfig.AGGREGATION_MAX_SIZE), DRPCConfig.DEFAULT_AGGREGATION_MAX_SIZE); }
private static DRPCConfig makeEmpty() { BulletStormConfig config = new BulletStormConfig(); config.set(BulletStormConfig.WINDOW_DISABLE, true); return new DRPCConfig(config); }
private Response getOkResponse(String data) { return getResponse(DRPCQueryResultPubscriber.OK_200, "Ok", data); }
@Test public void testQueryProcessingSingleInstanceTypes() throws Exception { config.set(DRPCConfig.PUBSUB_CONTEXT_NAME, PubSub.Context.QUERY_PROCESSING.name()); DRPCPubSub pubSub = new DRPCPubSub(config); Publisher publisher = pubSub.getPublisher(); Subscriber subscriber = pubSub.getSubscriber(); Assert.assertTrue(publisher instanceof DRPCResultPublisher); Assert.assertTrue(subscriber instanceof DRPCQuerySubscriber); }
@Override public Subscriber getSubscriber() { return context == QUERY_SUBMISSION ? getPubscriber() : new DRPCQuerySubscriber(config, maxUncommittedMessages); }
@Override public Publisher getPublisher() { return context == QUERY_SUBMISSION ? getPubscriber() : new DRPCResultPublisher(config); }
/** * Create a DRPCPubSub using a {@link BulletConfig}. * * @param config The BulletConfig containing settings to create a new DRPCPubSub. * @throws PubSubException if there were errors in creating an instance. */ public DRPCPubSub(BulletConfig config) throws PubSubException { super(config); this.config = new DRPCConfig(config); maxUncommittedMessages = getRequiredConfig(Number.class, DRPCConfig.DRPC_MAX_UNCOMMITED_MESSAGES).intValue(); }
private Consumer<Response> createResponseConsumer(String id) { // Create a closure with id return response -> handleResponse(id, response); }
private Response getNotOkResponse(int status) { return getResponse(status, "Error", null); }