def test_pySubscriber_withInvalidTopic(self):
apiObj = ezmq.pyEZMQAPI()
apiObj.initialize()
subscriber_cb = callback()
subscriber = ezmq.pyEZMQSubscriber("localhost", int(5562), subscriber_cb)
self.assertNotEqual(subscriber, None)
subscriber.start()
self.assertRaises(ValueError, subscriber.subscribe, topic=4452)
self.assertRaises(ValueError, subscriber.subscribe, topic=0)
self.assertRaises(ValueError, subscriber.subscribe, topic=1.55)
topicString = "sample/string"
ret = subscriber.subscribe(topic=topicString)
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
self.assertRaises(ValueError, subscriber.unSubscribe, topic=4452)
self.assertRaises(ValueError, subscriber.unSubscribe, topic=22)
self.assertRaises(ValueError, subscriber.unSubscribe, topic=1.2589)
ret = subscriber.unSubscribe(topic=topicString)
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = subscriber.stop()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = apiObj.terminate()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
del subscriber, subscriber_cb, apiObj
def test_pyEZMQPublisher_EventData(self):
port = 5563
apiObj = ezmq.pyEZMQAPI()
apiObj.initialize()
obj = ezmq.pyEZMQPublisher(port)
ret = obj.start()
data = ezmq.pyEvent()
data.init()
data.set_device("device")
data.set_created(10)
data.set_modified(20)
data.set_id("id")
data.set_pushed(10)
data.set_origin(20)
read1 = data.add_reading()
read1.set_name("reading1")
read1.set_value("10")
read1.set_created(25)
read1.set_device("device")
read1.set_modified(20)
read1.set_id("id1")
read1.set_origin(25)
read1.set_pushed(1)
topic = "arya"
for i in range(0, 5):
ret = obj.publish(data)
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = obj.stop()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = apiObj.terminate()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
def test_pySubscriber_withTopicList(self):
apiObj = ezmq.pyEZMQAPI()
apiObj.initialize()
subscriber_cb = callback()
ip = "localhost"
port = 5562
subscriber = ezmq.pyEZMQSubscriber(ip, int(port), subscriber_cb)
self.assertNotEqual(subscriber, None)
self.assertEqual(subscriber.getIp(), ip)
self.assertEqual(subscriber.getPort(), port)
ret = subscriber.start()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
topicList = ["sample/topic1", "sample/topic2", "/topic3"]
ret = subscriber.subscribe(topic=topicList)
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = subscriber.unSubscribe(topic=topicList)
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = subscriber.stop()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = apiObj.terminate()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
del subscriber, subscriber_cb, apiObj
def test_errorString_N(self): self.assertNotEqual(ezmq.errorString(1), "EZMQ_OK") self.assertNotEqual(ezmq.errorString(2), "EZMQ_OK") self.assertNotEqual(ezmq.errorString(3), "EZMQ_OK") self.assertNotEqual(ezmq.errorString(0), "EZMQ_ERROR") self.assertNotEqual(ezmq.errorString(2), "EZMQ_ERROR") self.assertNotEqual(ezmq.errorString(0), "EZMQ_INVALID_TOPIC") self.assertNotEqual(ezmq.errorString(3), "EZMQ_INVALID_TOPIC") self.assertNotEqual(ezmq.errorString(2), "EZMQ_INVALID_CONTENT_TYPE") self.assertNotEqual(ezmq.errorString(1), "EZMQ_INVALID_CONTENT_TYPE")
def test_pyEZMQAPI_P(self): obj = ezmq.pyEZMQAPI() self.assertNotEqual(obj, None) status = obj.getStatus() self.assertEqual(ezmq.statusString(status), "EZMQ_Terminated") ret = obj.initialize() status = obj.getStatus() self.assertEqual(ezmq.errorString(ret), "EZMQ_OK") self.assertEqual(ezmq.statusString(status), "EZMQ_Initialized") ret = obj.terminate() status = obj.getStatus() self.assertEqual(ezmq.errorString(ret), "EZMQ_OK") self.assertEqual(ezmq.statusString(status), "EZMQ_Terminated")
def test_pyEZMQPublisher(self):
port = 5560
obj = ezmq.pyEZMQPublisher(port)
self.assertNotEqual(obj, None)
ret = obj.start()
self.assertEqual("EZMQ_ERROR", ezmq.errorString(ret))
apiObj = ezmq.pyEZMQAPI()
apiObj.initialize()
obj = ezmq.pyEZMQPublisher(port)
ret = obj.start()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = obj.stop()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = apiObj.terminate()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
def test_pyEZMQPublisher_ByteData(self):
port = 5562
apiObj = ezmq.pyEZMQAPI()
apiObj.initialize()
obj = ezmq.pyEZMQPublisher(port)
ret = obj.start()
event = bytearray([12, 24, 36, 48, 60])
length = len(event)
data = ezmq.pyEZMQByteData()
data.init(event, len(event))
for i in range(0, 5):
ret = obj.publish(data)
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = obj.stop()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
ret = apiObj.terminate()
self.assertEqual("EZMQ_OK", ezmq.errorString(ret))
def test_pyEZMQByteData_N(self):
data = bytearray([10, 20, 30, 40, 50])
obj = ezmq.pyEZMQByteData()
obj.init(data, len(data))
self.assertNotEqual(obj, None)
self.assertIsInstance(obj, ezmq.pyEZMQByteData)
self.assertNotEqual(obj.getByteData(), None)
for i in range (0, 10):
if i is not len(data):
self.assertNotEqual(obj.getLength(), i)
dataNew = bytearray([15, 99])
ret = obj.setByteData(dataNew, len(dataNew))
self.assertNotEqual(ezmq.errorString(ret), "EZMQ_ERROR")
self.assertNotEqual(obj.getByteData(), data)
dataNULL = bytearray([])
ret = obj.setByteData(dataNULL, len(dataNULL))
self.assertNotEqual(ezmq.errorString(ret), "EZMQ_OK")
def test_pyEZMQAPI_N(self): obj = ezmq.pyEZMQAPI() self.assertNotEqual(obj, None) status = obj.getStatus() self.assertEqual(ezmq.statusString(status), "EZMQ_Constructed") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Unknown") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Terminated") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Initialized") ret = obj.initialize() status = obj.getStatus() self.assertNotEqual(ezmq.errorString(ret), "EZMQ_ERROR") self.assertNotEqual(ezmq.errorString(ret), "EZMQ_INVALID_TOPIC") self.assertNotEqual(ezmq.errorString(ret), "EZMQ_INVALID_CONTENT_TYPE") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Unknown") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Terminated") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Constructed") ret = obj.terminate() status = obj.getStatus() self.assertNotEqual(ezmq.errorString(ret), "EZMQ_ERROR") self.assertNotEqual(ezmq.errorString(ret), "EZMQ_INVALID_TOPIC") self.assertNotEqual(ezmq.errorString(ret), "EZMQ_INVALID_CONTENT_TYPE") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Unknown") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Constructed") self.assertNotEqual(ezmq.statusString(status), "EZMQ_Initialized")
def publish(**kwargs):
print "PUBLISH CALLED"
port = 5562
if "port" in kwargs:
port = kwargs["port"]
topicName = ""
if "topic" in kwargs:
topicName = kwargs["topic"]
counter = 0
if "counter" in kwargs:
counter = kwargs["counter"]
secure = 1
if "secure" in kwargs:
secure = kwargs["secure"]
apiObj = ezmq.pyEZMQAPI()
apiObj.initialize()
publisher = ezmq.pyEZMQPublisher(int(port))
if secure == 1:
publisher.setServerPrivateKey("")
publisher.start()
while counter != 0 :
event = getByteArray()
length = len(event)
data = ezmq.pyEZMQByteData()
data.init(event, len(event))
d = bytearray(data.getByteData())
dl = data.getLength()
print " DATA : ", list(d[0:dl])
print " DATA LENGTH : ", dl
if topicName == "":
ret = publisher.publish(data)
else:
print " TOPIC ", topicName
ret = publisher.publish(data, topic=topicName)
print ' EVENT PUBLISH RESULT :: ', ezmq.errorString(ret)
counter -= 1
time.sleep(1)
publisher.stop()
apiObj.terminate()
def test_pyEZMQByteData_P(self):
data = bytearray([10, 20, 30, 40, 50])
obj = ezmq.pyEZMQByteData()
obj.init(data, len(data))
self.assertNotEqual(obj, None)
self.assertIsInstance(obj, ezmq.pyEZMQByteData)
d = bytearray(obj.getByteData())
dl = obj.getLength()
self.assertEqual(d[0:dl], data[0:len(data)])
self.assertEqual(dl, len(data))
dataNew = bytearray([15, 99])
ret = obj.setByteData(dataNew, len(dataNew))
d = bytearray(obj.getByteData())
dl = obj.getLength()
self.assertEqual(d[0:dl], dataNew[0:len(dataNew)])
self.assertEqual(obj.getLength(), len(dataNew))
self.assertEqual(ezmq.errorString(ret), "EZMQ_OK")
if ',' in topicStr:
topicSub = [str(elem) for elem in topicStr.split(",")]
else:
topicSub = topicStr
print "USING TOPIC :: ", topicStr
if secured == 1:
print "SECURITY ENABLED FOR SAMPLE."
print "-----------------------------------------------------------"
print "\nCreating ezmq API object"
apiObj = ezmq.pyEZMQAPI()
print "Python ezmqApi object created successfully"
print "EZMQAPI Status : ", ezmq.statusString(apiObj.getStatus())
print "Initializing ezmqApi object"
ret = apiObj.initialize()
print "Initializing API RESULT", ezmq.errorString(ret)
print "EZMQAPI Status : ", ezmq.statusString(apiObj.getStatus())
print "Creating ezmq Subscriber object"
cb = callback()
subscriber = ezmq.pyEZMQSubscriber(ip, int(port), cb)
if subscriber is None:
print "FAILED TO GET Subscription object for ", ip, port
print "RERUN Subscriber as :: \"$python subscriber.py ip=\'Publisher IP\' port=5562\""
exit()
print "Python ezmqSubscriber object created successfully"
print "\nSUBCRIPTION IP :", subscriber.getIp()
print "SUBCRIPTION PORT : ", subscriber.getPort()
if secured == 1:
print "Setting Server and Client Keys"
try:
if topicStr is "":
print "========>SUBSCRIBING WITH NO TOPIC"
print "========>TO USE TOPIC run subscriber topic=topicName"
else:
print "USING TOPIC :: ", topicStr
if secured == 1:
print "SECURITY ENABLED FOR SAMPLE."
print "-----------------------------------------------------------"
print "Creating ezmq API object"
apiObj = ezmq.pyEZMQAPI()
print "Python ezmqApi object created successfully"
print "EZMQAPI Status : ", ezmq.statusString(apiObj.getStatus())
print "\nInitializing ezmqApi object"
ret = apiObj.initialize()
print "Initializing API RESULT", ezmq.errorString(ret)
print "EZMQAPI Status : ", ezmq.statusString(apiObj.getStatus())
print "Creating ezmq Publisher object"
publisher = ezmq.pyEZMQPublisher(int(port))
print "Python ezmqPublisher object created successfully"
if secured == 1:
print "Setting Server key"
try:
publisher.setServerPrivateKey(gServerSecretKey)
except Exception as e:
print "Exception caught for Settings server And client keys\n", e
exit()
print "Successfully set server and client keys."
print "Starting ezmq publisher"
ret = publisher.start()
print "Starting Publisher RESULT", ezmq.errorString(ret)
