Repository: samza Updated Branches: refs/heads/1.0.0 20b30e2be -> 89fe8cd43
SAMZA-1965: SAMZA 1.0 DOCUMENTATION FOR TEST Framework Author: Sanil15 <sanil.jai...@gmail.com> Reviewers: Jagadish<jagad...@apache.org> Closes #751 from Sanil15/SAMZA-1965 Project: http://git-wip-us.apache.org/repos/asf/samza/repo Commit: http://git-wip-us.apache.org/repos/asf/samza/commit/89fe8cd4 Tree: http://git-wip-us.apache.org/repos/asf/samza/tree/89fe8cd4 Diff: http://git-wip-us.apache.org/repos/asf/samza/diff/89fe8cd4 Branch: refs/heads/1.0.0 Commit: 89fe8cd434bd0a25043067bcd7240988c943f45f Parents: 20b30e2 Author: Sanil15 <sanil.jai...@gmail.com> Authored: Thu Jan 17 11:17:26 2019 -0800 Committer: Jagadish <jvenkatra...@linkedin.com> Committed: Thu Jan 17 11:17:57 2019 -0800 ---------------------------------------------------------------------- .../versioned/api/test-framework.md | 332 +++++++++++++++++++ docs/learn/documentation/versioned/index.html | 1 + 2 files changed, 333 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/samza/blob/89fe8cd4/docs/learn/documentation/versioned/api/test-framework.md ---------------------------------------------------------------------- diff --git a/docs/learn/documentation/versioned/api/test-framework.md b/docs/learn/documentation/versioned/api/test-framework.md new file mode 100644 index 0000000..0f98fa9 --- /dev/null +++ b/docs/learn/documentation/versioned/api/test-framework.md @@ -0,0 +1,332 @@ +--- +layout: page +title: 'Testing Samza jobs: Integration Framework' +--- +<!-- + Licensed to the Apache Software Foundation (ASF) under one or more + contributor license agreements. See the NOTICE file distributed with + this work for additional information regarding copyright ownership. + The ASF licenses this file to You under the Apache License, Version 2.0 + (the "License"); you may not use this file except in compliance with + the License. You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. +--> + +# What is Samza's Integration Test Framework ? + +- Samza provides an Integration framework which allows you to test applications by quickly running them against a few messages and asserting on expected results. This alleviates the need to set up dependencies like Kafka, Yarn, Zookeeper to test your Samza applications +- Integration Framework can test the new StreamDSL (StreamApplication) and Task APIs (TaskApplication) as well as supports testing for legacy low level (StreamTask and AsyncStreamTask) samza jobs + +# Some Prerequisite Information +1. Your Samza job will be executed in single container mode and framework will set all the required configs for you to run your job (more on configs later) +2. Your Samza job will read from a special kind of bounded streams introduced in the next section, containing finite number of messages to make testing feasible. + + +# Key Concepts + +## Introduction to In Memory System and Streams + +1. With Samza 1.0 we now get the feature of using streams that are maintained in memory using an in memory system. +2. These in memory streams are described by InMemoryInputDescriptor, InMemoryOutputDescriptor and the corresponding system is described by InMemorySystemDescriptors +3. These streams are like Kafka streams but there lifecycle is maintained in memory which means they get initialized with your job, are available throughout its run and are destroyed after the test ends . + + + +## Introduction to TestRunner api +1. Samza 1.0 introduces a new TestRunner api to set up a test for Samza job, add configs, configure input/output streams, run the job in testing mode +2. TestRunner also provides utilities to consume contents of a stream once the test has ran successfully +3. TestRunner does basic config setup for you by default, you have flexibility to change these default configs if required +4. TestRunner supports stateless and stateful job testing. TestRunner works with InMemoryTables and RocksDB Tables + +## How To Write Test + +For example, here is a StreamApplication that validates and decorates page views with viewerâs profile information. + +{% highlight java %} + + class BadPageViewFilterApplication implements StreamApplication { + @Override + public void describe(StreamApplicationDescriptor appDesc) { ⦠} + } + + public class BadPageViewFilter implements StreamApplication { + @Override + public void describe(StreamApplicationDescriptor appDesc) { + KafkaSystemDescriptor kafka = new KafkaSystemDescriptor("test"); + InputDescriptor<PageView> pageViewInput = kafka.getInputDescriptor(âpage-viewsâ, new JsonSerdeV2<>(PageView.class)); + OutputDescriptor<DecoratedPageView> outputPageViews = kafka.getOutputDescriptor( âdecorated-page-viewsâ, new JsonSerdeV2<>(DecoratedPageView.class)); + MessageStream<PageView> pageViews = appDesc.getInputStream(pageViewInput); + pageViews.filter(this::isValidPageView) + .map(this::addProfileInformation) + .sendTo(appDesc.getOutputStream(outputPageViews)); + } + } + +{% endhighlight %} + + +There are 4 simple steps to write a test for your stream processing logic and assert on the output + +## Step 1: Construct an InMemorySystem +In the example we are writing we use a Kafka system called "test", so we will configure an equivalent in memory system (name should be the same as used in job) as shown below: + +{% highlight java %} + + InMemorySystemDescriptor inMemory = new InMemorySystemDescriptor("test"); + +{% endhighlight %} + +## Step 2: Initialize your input and output streams +1. TestRunner API uses a special kind of input and output streams called in memory streams which are easy to define and write assertions on. +2. Data in these streams are maintained in memory hence they always use a NoOpSerde<> +3. You need to configure all the stream that your job reads/writes to. +4. You can obtain handle of these streams from the system we initialized in previous step +5. We have two choices when we configure a stream type + +Input Stream described by InMemoryInputDescriptor, these streams need to be initialized with messages (data), since your job reads this. + +{% highlight java %} + + InMemoryInputDescriptor<PageView> pageViewInput = inMemory.getInputDescriptor(âpage-viewsâ, new NoOpSerde<>()); + +{% endhighlight %} +{% highlight jproperties %} + + INFO: Use the org.apache.samza.operators.KV as the message type ex: InMemoryInputDescriptor<KV<String,PageView>> as the message type + to use key of the KV (String here) as key and value as message (PageView here) for the IncomingMessageEnvelope in samza job, using all the other data types will result in key of the the IncomingMessageEnvelope set to null + +{% endhighlight %} + +Output Stream described by InMemoryOutputDescriptor, these streams need to be initialized with with a partition count and are empty since your job writes to these streams + +{% highlight java %} + + InMemoryOutputDescriptor<DecoratedPageView> outputPageViews = inMemory.getOutputDescriptor("decorated-page-views", new NoOpSerde<>()) + +{% endhighlight %} + +{% highlight jproperties %} + + Note: Input streams are immutable - ie., once they have been created you can't modify their contents eg: by adding new messages"All input streams are supposed to be bounded + +{% endhighlight %} + + +## Step 3: Create a TestRunner + +1. Initialize a TestRunner of your Samza job +2. Configure TestRunner with input streams and mock data to it +3. Configure TestRunner with output streams with a partition count +4. Add any configs if necessary +5. Run the test runner + +{% highlight java %} + + List<PageView> pageViews = generateData(...); + TestRunner + .of(new BadPageViewFilterApplication()) + .addInputStream(pageViewInput, pageViews) + .addOutputStream(outputPageViews, 10) + .run(Duration.ofMillis(1500)); + +{% endhighlight %} + +{% highlight jproperties %} + + Info: Use addConfig(Map<String, String> configs) or addConfig(String key, String value) to add/modify any config in the TestRunner + +{% endhighlight %} + + +## Step-4: Assert on the output stream + +You have the following choices for asserting the results of your tests + +1. You can use StreamAssert utils on your In Memory Streams to do consumption of all partitions + +{% highlight java %} + + // Consume multi-paritioned stream, key of the map represents partitionId + Map<Integer, PageView> expOutput; + StreamAssert.containsInOrder(outputPageViews, expectedOutput, Duration.ofMillis(1000)); + // Consume single paritioned stream + StreamAssert.containsInOrder(outputPageViews, Arrays.asList(...), Duration.ofMillis(1000)); + +{% endhighlight %} + + +2. You have the flexibility to define your custom assertions using API TestRunner.consumeStream() to assert on any partitions of the stream + +{% highlight java %} + + Assert.assertEquals( + TestRunner.consumeStream(outputPageViews,Duration.ofMillis(1000)).get(0).size(),1 + ); + +{% endhighlight %} + +Complete Glance at the code + +{% highlight java %} + + @Test + public void testStreamDSLApi() throws Exception { + // Generate Mock Data + List<PageView> pageViews = genrateMockInput(...); + List<DecoratedPageView> expectedOutput = genrateMockOutput(...); + + // Configure System and Stream Descriptors + InMemorySystemDescriptor inMemory = new InMemorySystemDescriptor("test"); + InMemoryInputDescriptor<PageView> pageViewInput = inMemory + .getInputDescriptor(âpage-viewsâ, new NoOpSerde<>()); + InMemoryOutputDescriptor<DecoratedPageView> outputPageView = inMemory + .getOutputDescriptor(âdecorated-page-viewsâ, new NoOpSerde<>()) + + // Configure the TestRunner + TestRunner + .of(new BadPageViewFilterApplication()) + .addInputStream(pageViewInput, pageViews) + .addOutputStream(outputPageView, 10) + .run(Duration.ofMillis(1500)); + + // Assert the results + StreamAssert.containsInOrder(expectedOutput, outputPageView, Duration.ofMillis(1000)); + } + +{% endhighlight %} + +### Example for Low Level Api: + +For a Low Level Task API + +{% highlight java %} + + public class BadPageViewFilter implements TaskApplication { + @Override + public void describe(TaskApplicationDescriptor appDesc) { + // Add input, output streams and tables + KafkaSystemDescriptor<String, PageViewEvent> kafkaSystem = + new KafkaSystemDescriptor(âkafkaâ) + .withConsumerZkConnect(myZkServers) + .withProducerBootstrapServers(myBrokers); + KVSerde<String, PageViewEvent> serde = + KVSerde.of(new StringSerde(), new JsonSerdeV2<PageViewEvent>()); + // Add input, output streams and tables + appDesc.withInputStream(kafkaSystem.getInputDescriptor(âpageViewEventâ, serde)) + .withOutputStream(kafkaSystem.getOutputDescriptor(âgoodPageViewEventâ, serde)) + .withTable(new RocksDBTableDescriptor( + âbadPageUrlTableâ, KVSerde.of(new StringSerde(), new IntegerSerde()) + .withTaskFactory(new BadPageViewTaskFactory()); + } + } + + public class BadPageViewTaskFactory implements StreamTaskFactory { + @Override + public StreamTask createInstance() { + // Add input, output streams and tables + return new BadPageViewFilterTask(); + } + } + + public class BadPageViewFilterTask implements StreamTask { + @Override + public void process(IncomingMessageEnvelope envelope, + MessageCollector collector, + TaskCoordinator coordinator) { + // process message synchronously + } + } + + + @Test + public void testBadPageViewFilterTaskApplication() { + List<PageView> badPageViews = Arrays.asList(generatePageViews(..)); + List<Profile> expectedGoodPageViews = Arrays.asList(generatePageViews(..)); + + InMemorySystemDescriptor inMemory = new InMemorySystemDescriptor("kafka"); + + InMemoryInputDescriptor<PageView> pageViewInput = inMemory + .getInputDescriptor("pageViewEvent", new NoOpSerde<>()); + + InMemoryOutputDescriptor<PageView> pageViewOutput = inMemory + .getOutputDescriptor("goodPageViewEvent", new NoOpSerde<>()); + + TestRunner + .of(new BadPageViewFilter()) + .addInputStream(pageViewInput, badPageViews) + .addOutputStream(pageViewOutput, 1) + .run(Duration.ofSeconds(2)); + + StreamAssert.containsInOrder(expectedGoodPageViews, pageViewOutput, Duration.ofMillis(1000)); + } + +{% endhighlight %} + + +Follow a similar approach for Legacy Low Level API, just provide the classname +(class implementing StreamTask or AsyncStreamTask) to TestRunner + +{% highlight java %} + + public class MultiplyByTenStreamTask implements StreamTask { + @Override + public void process(IncomingMessageEnvelope envelope, MessageCollector collector, TaskCoordinator coordinator) + throws Exception { + Integer obj = (Integer) envelope.getMessage(); + collector.send(new OutgoingMessageEnvelope(new SystemStream("test", "output"), + envelope.getKey(), envelope.getKey(), obj * 10)); + } + } + + @Test + public void testLowLevelApi() throws Exception { + List<Integer> inputList = Arrays.asList(1, 2, 3, 4, 5); + List<Integer> outputList = Arrays.asList(10, 20, 30, 40, 50); + + InMemorySystemDescriptor inMemory = new InMemorySystemDescriptor("test"); + + InMemoryInputDescriptor<Integer> numInput = inMemory + .getInputDescriptor("input", new NoOpSerde<Integer>()); + + InMemoryOutputDescriptor<Integer> numOutput = inMemory + .getOutputDescriptor("output", new NoOpSerde<Integer>()); + + TestRunner + .of(MyStreamTestTask.class) + .addInputStream(numInput, inputList) + .addOutputStream(numOutput, 1) + .run(Duration.ofSeconds(1)); + + Assert.assertThat(TestRunner.consumeStream(imod, Duration.ofMillis(1000)).get(0), + IsIterableContainingInOrder.contains(outputList.toArray()));; + } + + +{% endhighlight %} + +## Stateful Testing + +1. There is no additional config/changes required for TestRunner apis for testing samza jobs using StreamApplication or TaskApplication APIs +2. Legacy task api only supports RocksDbTable and needs following configs to be added to TestRunner. + For example if your job is using a RocksDbTable named "my-store" with key and msg serde of String type +{% highlight java %} + + Map<String, String> config = new HashMap<>(); + config.put("stores.my-store.factory", "org.apache.samza.storage.kv.RocksDbKeyValueStorageEngineFactory"); + config.out("serializers.registry.string.class", "org.apache.samza.serializers.StringSerdeFactory"); + config.put("stores.my-store.key.serde", "string"); + config.put("stores.my-store.msg.serde", "string"); + + TestRunner + .of(...) + .addConfig(config) + ... + +{% endhighlight %} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/samza/blob/89fe8cd4/docs/learn/documentation/versioned/index.html ---------------------------------------------------------------------- diff --git a/docs/learn/documentation/versioned/index.html b/docs/learn/documentation/versioned/index.html index 2d5446b..b565b42 100644 --- a/docs/learn/documentation/versioned/index.html +++ b/docs/learn/documentation/versioned/index.html @@ -30,6 +30,7 @@ title: Documentation <li><a href="api/high-level-api.html">High Level Streams API</a></li> <li><a href="api/low-level-api.html">Low Level Task API</a></li> <li><a href="api/table-api.html">Table API</a></li> + <li><a href="api/test-framework.html">Testing Samza</a></li> <li><a href="api/samza-sql.html">Samza SQL</a></li> <li><a href="https://beam.apache.org/documentation/runners/samza/">Apache BEAM</a></li> </ul>