Examples with MockStateStoreSupplier org.apache.kafka.test.MockStateStoreSupplier used on opensource projects

Example 6 with MockStateStoreSupplier

use of org.apache.kafka.test.MockStateStoreSupplier in project kafka by apache.

the class TopologyBuilderTest method testAddStateStoreWithDuplicates.

@Test(expected = TopologyBuilderException.class)
public void testAddStateStoreWithDuplicates() {
    final TopologyBuilder builder = new TopologyBuilder();
    builder.addStateStore(new MockStateStoreSupplier("store", false));
    builder.addStateStore(new MockStateStoreSupplier("store", false));
}

Example 7 with MockStateStoreSupplier

use of org.apache.kafka.test.MockStateStoreSupplier in project kafka by apache.

the class TopologyBuilderTest method testAddStateStoreWithSource.

@Test(expected = TopologyBuilderException.class)
public void testAddStateStoreWithSource() {
    final TopologyBuilder builder = new TopologyBuilder();
    builder.addSource("source-1", "topic-1");
    builder.addStateStore(new MockStateStoreSupplier("store", false), "source-1");
}

Example 8 with MockStateStoreSupplier

use of org.apache.kafka.test.MockStateStoreSupplier in project kafka by apache.

the class TopologyBuilderTest method testAddStateStoreWithNonExistingProcessor.

@Test(expected = TopologyBuilderException.class)
public void testAddStateStoreWithNonExistingProcessor() {
    final TopologyBuilder builder = new TopologyBuilder();
    builder.addStateStore(new MockStateStoreSupplier("store", false), "no-such-processsor");
}

Example 9 with MockStateStoreSupplier

use of org.apache.kafka.test.MockStateStoreSupplier in project kafka by apache.

the class TopologyBuilderTest method testTopicGroupsByStateStore.

@Test
public void testTopicGroupsByStateStore() {
    final TopologyBuilder builder = new TopologyBuilder();
    builder.setApplicationId("X");
    builder.addSource("source-1", "topic-1", "topic-1x");
    builder.addSource("source-2", "topic-2");
    builder.addSource("source-3", "topic-3");
    builder.addSource("source-4", "topic-4");
    builder.addSource("source-5", "topic-5");
    builder.addProcessor("processor-1", new MockProcessorSupplier(), "source-1");
    builder.addProcessor("processor-2", new MockProcessorSupplier(), "source-2");
    builder.addStateStore(new MockStateStoreSupplier("store-1", false), "processor-1", "processor-2");
    builder.addProcessor("processor-3", new MockProcessorSupplier(), "source-3");
    builder.addProcessor("processor-4", new MockProcessorSupplier(), "source-4");
    builder.addStateStore(new MockStateStoreSupplier("store-2", false), "processor-3", "processor-4");
    builder.addProcessor("processor-5", new MockProcessorSupplier(), "source-5");
    StateStoreSupplier supplier = new MockStateStoreSupplier("store-3", false);
    builder.addStateStore(supplier);
    builder.connectProcessorAndStateStores("processor-5", "store-3");
    Map<Integer, TopicsInfo> topicGroups = builder.topicGroups();
    Map<Integer, TopicsInfo> expectedTopicGroups = new HashMap<>();
    final String store1 = ProcessorStateManager.storeChangelogTopic("X", "store-1");
    final String store2 = ProcessorStateManager.storeChangelogTopic("X", "store-2");
    final String store3 = ProcessorStateManager.storeChangelogTopic("X", "store-3");
    expectedTopicGroups.put(0, new TopicsInfo(Collections.<String>emptySet(), mkSet("topic-1", "topic-1x", "topic-2"), Collections.<String, InternalTopicConfig>emptyMap(), Collections.singletonMap(store1, new InternalTopicConfig(store1, Collections.singleton(InternalTopicConfig.CleanupPolicy.compact), Collections.<String, String>emptyMap()))));
    expectedTopicGroups.put(1, new TopicsInfo(Collections.<String>emptySet(), mkSet("topic-3", "topic-4"), Collections.<String, InternalTopicConfig>emptyMap(), Collections.singletonMap(store2, new InternalTopicConfig(store2, Collections.singleton(InternalTopicConfig.CleanupPolicy.compact), Collections.<String, String>emptyMap()))));
    expectedTopicGroups.put(2, new TopicsInfo(Collections.<String>emptySet(), mkSet("topic-5"), Collections.<String, InternalTopicConfig>emptyMap(), Collections.singletonMap(store3, new InternalTopicConfig(store3, Collections.singleton(InternalTopicConfig.CleanupPolicy.compact), Collections.<String, String>emptyMap()))));
    assertEquals(3, topicGroups.size());
    assertEquals(expectedTopicGroups, topicGroups);
}

Example 10 with MockStateStoreSupplier

use of org.apache.kafka.test.MockStateStoreSupplier in project kafka by apache.

the class TopologyBuilderTest method shouldAssociateStateStoreNameWhenStateStoreSupplierIsInternal.

@Test
public void shouldAssociateStateStoreNameWhenStateStoreSupplierIsInternal() throws Exception {
    final TopologyBuilder builder = new TopologyBuilder();
    builder.addSource("source", "topic");
    builder.addProcessor("processor", new MockProcessorSupplier(), "source");
    builder.addStateStore(new MockStateStoreSupplier("store", false), "processor");
    final Map<String, List<String>> stateStoreNameToSourceTopic = builder.stateStoreNameToSourceTopics();
    assertEquals(1, stateStoreNameToSourceTopic.size());
    assertEquals(Collections.singletonList("topic"), stateStoreNameToSourceTopic.get("store"));
}