@Test public void testPrintSinkStdErr() throws Exception { PrintSinkFunction<String> printSink = new PrintSinkFunction<>(true); printSink.setRuntimeContext(new MockStreamingRuntimeContext(false, 1, 0)); printSink.open(new Configuration()); printSink.invoke("hello world!", SinkContextUtil.forTimestamp(0)); assertEquals("Print to System.err", printSink.toString()); assertEquals("hello world!" + line, arrayErrorStream.toString()); printSink.close(); }
@Test public void testPrintSinkWithPrefix() throws Exception { PrintSinkFunction<String> printSink = new PrintSinkFunction<>(); printSink.setRuntimeContext(new MockStreamingRuntimeContext(false, 2, 1)); printSink.open(new Configuration()); printSink.invoke("hello world!", SinkContextUtil.forTimestamp(0)); assertEquals("Print to System.out", printSink.toString()); assertEquals("2> hello world!" + line, arrayOutputStream.toString()); printSink.close(); }
@Test public void testPrintSinkStdOut() throws Exception { PrintSinkFunction<String> printSink = new PrintSinkFunction<>(); printSink.setRuntimeContext(new MockStreamingRuntimeContext(false, 1, 0)); printSink.open(new Configuration()); printSink.invoke("hello world!", SinkContextUtil.forTimestamp(0)); assertEquals("Print to System.out", printSink.toString()); assertEquals("hello world!" + line, arrayOutputStream.toString()); printSink.close(); }
@Test public void testPrintSinkWithIdentifierAndPrefix() throws Exception { PrintSinkFunction<String> printSink = new PrintSinkFunction<>("mySink", false); printSink.setRuntimeContext(new MockStreamingRuntimeContext(false, 2, 1)); printSink.open(new Configuration()); printSink.invoke("hello world!", SinkContextUtil.forTimestamp(0)); assertEquals("Print to System.out", printSink.toString()); assertEquals("mySink:2> hello world!" + line, arrayOutputStream.toString()); printSink.close(); }
@Test public void testPrintSinkWithIdentifierButNoPrefix() throws Exception { PrintSinkFunction<String> printSink = new PrintSinkFunction<>("mySink", false); printSink.setRuntimeContext(new MockStreamingRuntimeContext(false, 1, 0)); printSink.open(new Configuration()); printSink.invoke("hello world!", SinkContextUtil.forTimestamp(0)); assertEquals("Print to System.out", printSink.toString()); assertEquals("mySink> hello world!" + line, arrayOutputStream.toString()); printSink.close(); }