use of com.ibm.streamsx.topology.tester.Tester in project streamsx.topology by IBMStreams.
the class FileStreamsTest method runDirectoryWatcher.
private void runDirectoryWatcher(final Topology t, int numberOfFiles, int repeat) throws Exception {
final Path dir = Files.createTempDirectory("testdw");
final String[] files = new String[numberOfFiles];
for (int i = 0; i < files.length; i++) {
files[i] = dir.resolve("A" + (numberOfFiles - i)).toAbsolutePath().toString();
}
TStream<String> fileNames = FileStreams.directoryWatcher(t, dir.toAbsolutePath().toString());
// Create the files from within the topology to ensure it works
// in distributed and standalone.
//
// Due to vagaries / delays that can occur in operator startup
// in different processes (distributed mode), delay the initial
// file creation process to give the watcher a chance to startup.
//
// e.g., with numberOfFiles=20 & repeat=1, each group of files
// only lasts 20*(10ms*2) => 200ms. That can easily happen before
// the watcher is started and has done its first dir.listFiles(),
// with the result being not seeing/processing the expected number
// of files.
addStartupDelay(BeaconStreams.single(t)).sink(createFiles(files, repeat));
Tester tester = t.getTester();
Condition<Long> expectedCount = tester.tupleCount(fileNames, numberOfFiles * repeat);
String[] expectedFileNames = files;
if (repeat != 1) {
expectedFileNames = new String[files.length * repeat];
for (int r = 0; r < repeat; r++) System.arraycopy(files, 0, expectedFileNames, r * files.length, files.length);
}
Condition<List<String>> expectedNames = tester.stringContentsUnordered(fileNames, expectedFileNames);
complete(tester, expectedCount, 60, TimeUnit.SECONDS);
assertTrue(expectedCount.toString(), expectedCount.valid());
assertTrue(expectedNames.toString(), expectedNames.valid());
deleteFiles(dir, files);
}
use of com.ibm.streamsx.topology.tester.Tester in project streamsx.topology by IBMStreams.
the class SimpleEmbeddedTest method testTwoStreams.
@Test
public void testTwoStreams() throws Exception {
Topology topology = new Topology("testTwoStreams");
TStream<String> hw = topology.strings("Hello", "World!", "Test!!");
SPLStream hws = SPLStreams.stringToSPLStream(hw);
TStream<String> hw2 = StringStreams.contains(hw, "e");
SPLStream hw2s = SPLStreams.stringToSPLStream(hw2);
Tester tester = topology.getTester();
StreamCounter<Tuple> counter = tester.splHandler(hws, new StreamCounter<Tuple>());
StreamCollector<LinkedList<Tuple>, Tuple> collector = tester.splHandler(hws, StreamCollector.newLinkedListCollector());
StreamCounter<Tuple> counter2 = tester.splHandler(hw2s, new StreamCounter<Tuple>());
StreamCollector<LinkedList<Tuple>, Tuple> collector2 = tester.splHandler(hw2s, StreamCollector.newLinkedListCollector());
StreamsContextFactory.getStreamsContext(StreamsContext.Type.EMBEDDED_TESTER).submit(topology).get();
assertEquals(3, counter.getTupleCount());
assertEquals("Hello", collector.getTuples().get(0).getString(0));
assertEquals("World!", collector.getTuples().get(1).getString(0));
assertEquals("Test!!", collector.getTuples().get(2).getString(0));
assertEquals(2, counter2.getTupleCount());
assertEquals("Hello", collector2.getTuples().get(0).getString(0));
assertEquals("Test!!", collector2.getTuples().get(1).getString(0));
}
use of com.ibm.streamsx.topology.tester.Tester in project streamsx.topology by IBMStreams.
the class KafkaStreamsTest method completeAndValidateUnordered.
// would be nice if Tester provided this too
private void completeAndValidateUnordered(String msg, Topology topology, TStream<String> stream, int secTimeout, String... expected) throws Exception {
Tester tester = topology.getTester();
Condition<Long> sCount = tester.tupleCount(stream, expected.length);
Condition<List<String>> sContents = tester.stringContentsUnordered(stream, expected);
complete(tester, sCount, secTimeout, TimeUnit.SECONDS);
assertTrue(msg + " contents:" + sContents, sContents.valid());
assertTrue("valid:" + sCount, sCount.valid());
}
use of com.ibm.streamsx.topology.tester.Tester in project streamsx.topology by IBMStreams.
the class PythonFunctionalOperatorsTest method testValues.
/**
* Test that specific values in Python
* make their way into SPL correctly
* when returning as a tuple.
* @throws Exception
*/
@Test
public void testValues() throws Exception {
Topology topology = new Topology("testValues");
addTestToolkit(topology);
SPLStream pysrc = SPL.invokeSource(topology, "com.ibm.streamsx.topology.pytest.pysource::SpecificValues", null, ALL_PYTHON_TYPES_SCHEMA);
StreamSchema sparseSchema = Type.Factory.getStreamSchema("tuple<int32 a, int32 b, int32 c, int32 d, int32 e>");
Tester tester = topology.getTester();
Condition<Long> expectedCount = tester.tupleCount(pysrc, 1);
Condition<List<Tuple>> outTuples = tester.tupleContents(pysrc);
SPLStream pysparse = SPL.invokeSource(topology, "com.ibm.streamsx.topology.pytest.pysource::SparseTuple", null, sparseSchema);
SPLStream pysparsemap = SPL.invokeOperator("com.ibm.streamsx.topology.pytest.pymap::SparseTupleMap", pysparse, sparseSchema.extend("int32", "f"), null);
Condition<Long> expectedCountSparse = tester.tupleCount(pysparse, 1);
Condition<List<Tuple>> sparseTupleOut = tester.tupleContents(pysparse);
Condition<Long> expectedCountSparseMap = tester.tupleCount(pysparsemap, 1);
Condition<List<Tuple>> sparseTupleMapOut = tester.tupleContents(pysparsemap);
// getConfig().put(ContextProperties.TRACING_LEVEL, TraceLevel.DEBUG);
complete(tester, expectedCount.and(expectedCountSparse, expectedCountSparseMap), 20, TimeUnit.SECONDS);
assertTrue(expectedCount.valid());
assertTrue(expectedCountSparse.valid());
assertTrue(expectedCountSparseMap.valid());
Tuple r1 = outTuples.getResult().get(0);
assertTrue(r1.getBoolean("b"));
// signed integers
// 23, -2525, 3252352, -2624565653,
assertEquals(r1.getByte("i8"), 23);
assertEquals(r1.getShort("i16"), -2525);
assertEquals(r1.getInt("i32"), 3252352);
assertEquals(r1.getLong("i64"), -2624565653L);
// unsigned int
// 72, 6873, 43665588, 357568872
assertEquals(r1.getString("u8"), "72");
assertEquals(r1.getString("u16"), "6873");
assertEquals(r1.getString("u32"), "43665588");
assertEquals(r1.getString("u64"), "357568872");
// floats
// 4367.34, -87657525334.22
assertEquals(r1.getFloat("f32"), 4367.34f, 0.1);
assertEquals(r1.getDouble("f64"), -87657525334.22d, 0.1);
// rstring, Unicode data
assertEquals("⡍⠔⠙⠖ ⡊ ⠙⠕⠝⠰⠞ ⠍⠑⠁⠝ ⠞⠕ ⠎⠁⠹ ⠹⠁⠞ ⡊ ⠅⠝⠪⠂ ⠕⠋ ⠍⠹", r1.getString("r"));
// complex(-23.0, 325.38), complex(-35346.234, 952524.93)
assertEquals(((Complex) r1.getObject("c32")).getReal(), -23.0, 0.1);
assertEquals(((Complex) r1.getObject("c32")).getImaginary(), 325.38, 0.1);
assertEquals(((Complex) r1.getObject("c64")).getReal(), -35346.234, 0.1);
assertEquals(((Complex) r1.getObject("c64")).getImaginary(), 952524.93, 0.1);
// Timestamp Timestamp(781959759, 9320, 76)
assertEquals(781959759L, r1.getTimestamp("ts").getSeconds());
assertEquals(9320, r1.getTimestamp("ts").getNanoseconds());
assertEquals(76, r1.getTimestamp("ts").getMachineId());
// ["a", "Streams!", "2H₂ + O₂ ⇌ 2H₂O, R = 4.7 kΩ, ⌀ 200 mm"]
{
@SuppressWarnings("unchecked") List<RString> lr = (List<RString>) r1.getObject("lr");
assertEquals(3, lr.size());
assertEquals("a", lr.get(0).getString());
assertEquals("Streams!", lr.get(1).getString());
assertEquals("2H₂ + O₂ ⇌ 2H₂O, R = 4.7 kΩ, ⌀ 200 mm", lr.get(2).getString());
}
// [345,-4578],
{
int[] li32 = (int[]) r1.getObject("li32");
assertEquals(2, li32.length);
assertEquals(345, li32[0]);
assertEquals(-4578, li32[1]);
}
// [9983, -4647787587, 0]
{
long[] li64 = (long[]) r1.getObject("li64");
assertEquals(3, li64.length);
assertEquals(9983L, li64[0]);
assertEquals(-4647787587L, li64[1]);
assertEquals(0L, li64[2]);
}
{
@SuppressWarnings("unchecked") List<Integer> lui32 = (List<Integer>) r1.getObject("lui32");
assertEquals(1, lui32.size());
assertEquals("87346", Integer.toUnsignedString(lui32.get(0)));
}
{
@SuppressWarnings("unchecked") List<Long> lui64 = (List<Long>) r1.getObject("lui64");
assertEquals(2, lui64.size());
assertEquals("45433674", Long.toUnsignedString(lui64.get(0)));
assertEquals("41876984848", Long.toUnsignedString(lui64.get(1)));
}
// 4.269986E+05, -8.072285E+02 -6.917091E-08 7.735085E8
{
float[] li32 = (float[]) r1.getObject("lf32");
assertEquals(4, li32.length);
assertEquals(4.269986E+05f, li32[0], 0.1);
assertEquals(-8.072285E+02f, li32[1], 0.1);
assertEquals(-6.917091E-08f, li32[2], 0.1);
assertEquals(7.735085E8f, li32[3], 0.1);
}
{
double[] lf64 = (double[]) r1.getObject("lf64");
assertEquals(1, lf64.length);
assertEquals(765.46477e19, lf64[0], 0.1);
}
{
boolean[] lb = (boolean[]) r1.getObject("lb");
assertEquals(3, lb.length);
assertTrue(lb[0]);
assertFalse(lb[1]);
assertTrue(lb[2]);
}
assertTrue(r1.getMap("mi32r").isEmpty());
assertTrue(r1.getMap("mru32").isEmpty());
{
Map<?, ?> mri32 = r1.getMap("mri32");
assertEquals(2, mri32.size());
System.out.println("mri32:" + mri32);
assertTrue(mri32.containsKey(new RString("abc")));
assertTrue(mri32.containsKey(new RString("многоязычных")));
assertEquals(35320, mri32.get(new RString("abc")));
assertEquals(-236325, mri32.get(new RString("многоязычных")));
}
assertTrue(r1.getMap("mu32r").isEmpty());
assertTrue(r1.getMap("mi32i32").isEmpty());
assertTrue(r1.getMap("mu32u32").isEmpty());
assertTrue(r1.getMap("mrr").isEmpty());
assertTrue(r1.getMap("mf64f64").isEmpty());
assertTrue(r1.getMap("mf64i32").isEmpty());
assertTrue(r1.getMap("mf64u32").isEmpty());
assertTrue(r1.getMap("mf64r").isEmpty());
assertTrue(r1.getMap("mrf64").isEmpty());
// Sparse tuple handling - source
assertEquals(1, sparseTupleOut.getResult().size());
Tuple st = sparseTupleOut.getResult().get(0);
// set by op
assertEquals(37, st.getInt("a"));
// default as None in tuple
assertEquals(0, st.getInt("b"));
// default as None in tuple
assertEquals(0, st.getInt("c"));
// set by op
assertEquals(-46, st.getInt("d"));
// default as no value (short tuple)
assertEquals(0, st.getInt("e"));
// Sparse tuple handling - map
assertEquals(1, sparseTupleMapOut.getResult().size());
Tuple stm = sparseTupleMapOut.getResult().get(0);
// set by op
assertEquals(37 + 81, stm.getInt("a"));
// set by op
assertEquals(23, stm.getInt("b"));
// default as None in tuple
assertEquals(0, stm.getInt("c"));
// default to matching input
assertEquals(-46, stm.getInt("d"));
// set by op
assertEquals(34, stm.getInt("e"));
// default as no value (short tuple)
assertEquals(0, stm.getInt("f"));
}
use of com.ibm.streamsx.topology.tester.Tester in project streamsx.topology by IBMStreams.
the class SimpleStandaloneTest method testSimple.
@Test
public void testSimple() throws Exception {
Topology topology = new Topology("testSimple");
TStream<String> hw = topology.strings("Hello", "World!", "Test!!");
SPLStream hws = SPLStreams.stringToSPLStream(hw);
Tester tester = topology.getTester();
StreamCounter<Tuple> counter = tester.splHandler(hws, new StreamCounter<Tuple>());
StreamCollector<LinkedList<Tuple>, Tuple> collector = tester.splHandler(hws, StreamCollector.newLinkedListCollector());
StreamsContextFactory.getStreamsContext(StreamsContext.Type.STANDALONE_TESTER).submit(topology).get();
assertEquals(3, counter.getTupleCount());
assertEquals("Hello", collector.getTuples().get(0).getString(0));
assertEquals("World!", collector.getTuples().get(1).getString(0));
assertEquals("Test!!", collector.getTuples().get(2).getString(0));
}
Aggregations