Fixed plasma stream test to use FileDrops

ICRAR · May 28, 2021 · 70cd84e · 70cd84e
1 parent d59a93a
commit 70cd84e
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/daliuge-engine/test/apps/test_plasma.py b/daliuge-engine/test/apps/test_plasma.py
@@ -91,12 +91,15 @@ def test_plasma_stream(self):
         with tarfile.open('/daliuge/test/apps/data/test_ms.tar.gz', 'r') as ref:
             ref.extractall('/tmp/')
 
-        prod = MSStreamingPlasmaProducer('1', '1', input_file=in_file)
-        cons = MSStreamingPlasmaConsumer('2', '2', output_file=out_file)
+        prod = MSStreamingPlasmaProducer('1', '1')
+        cons = MSStreamingPlasmaConsumer('2', '2')
         drop = InMemoryDROP('3', '3')
-
-        drop.addStreamingConsumer(cons)
+        ms_in = FileDROP('4', '4', filepath=in_file)
+        ms_out = FileDROP('5', '5', filepath=out_file)
+        prod.addInput(ms_in)
         prod.addOutput(drop)
+        drop.addStreamingConsumer(cons)
+        cons.addOutput(ms_out)
 
         with droputils.DROPWaiterCtx(self, cons, 1000):
             prod.async_execute()