From 145fa2e29e975104d002611491a335f0c2d5413b Mon Sep 17 00:00:00 2001
From: Alexandre Gobbo <alexandre.gobbo@psi.ch>
Date: Thu, 10 Oct 2019 14:47:19 +0200
Subject: [PATCH] Script execution

---
 script/test/TestZMQ2.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/script/test/TestZMQ2.py b/script/test/TestZMQ2.py
index 61d1b4e..368b70e 100644
--- a/script/test/TestZMQ2.py
+++ b/script/test/TestZMQ2.py
@@ -2,6 +2,7 @@ from ch.psi.pshell.bs import Stream
 from ch.psi.pshell.bs import Provider 
 from ch.psi.pshell.bs.ProviderConfig import SocketType;
 
+
 providers, streams = [], []
 for i in range(9000, 9008):
    p =  Provider("Provider"+str(i),"tcp://SFTEST-CVME-DBPM1:"+str(i), SocketType.PULL)
@@ -13,23 +14,40 @@ for i in range(9000, 9008):
    
 def check(s):
     print "Starting checking " + s.getName() 
+    divider = 3 if s.name == "Stream9000" else 1
     pid = s.take().pulseId
     while(True):
         s.waitCacheChange(10000)
         p=s.take().pulseId
-        if p!= pid + 1:
+        if p!= (pid + divider):
             print s.getName() + " error: received pid %d, expecting %d" % (p, (pid + 1))
         pid = p
+
+class StreamListener (DeviceListener):
+    def onCacheChanged(self, device, value, former, timestamp, valueChange)
+        if self.pid!= (value.pulseId + self.divider):
+             print s.getName() + " error: received pid %d, expecting %d" % (value.pulseId, (self.pid+ 1))
+        pid = value.pulseId
+
      
 try:
     for s in streams:
         s.start(True)
         s.waitCacheChange(5000)
         print s.getName() + " channels:" + str(s.getIdentifiers())
+    """
     args = []
     for s in streams:
         args.append((check,(s,)))
     parallelize(*args)
+    #check(streams[0])
+    """
+    "
+    for s in streams:
+        l = StreamListener()
+        l.divider = 3 if s.name == "Stream9000" else 1
+        l.pid = s.take().pulseId
+        s.addListener(listenerAI)
                 
 finally:
     try: