From a3682ec59fed3440c748170535ae09ec9c66c7c8 Mon Sep 17 00:00:00 2001 From: Eddie Hebert Date: Wed, 22 Aug 2012 13:32:38 -0400 Subject: [PATCH] Removes unused TestClient class. --- tests/client.py | 81 ------------------------------------------------- 1 file changed, 81 deletions(-) delete mode 100644 tests/client.py diff --git a/tests/client.py b/tests/client.py deleted file mode 100644 index 3ceede6f..00000000 --- a/tests/client.py +++ /dev/null @@ -1,81 +0,0 @@ -import logging - -import zipline.protocol as zp -from zipline.core.component import Component -from zipline.protocol import CONTROL_PROTOCOL, COMPONENT_TYPE - -LOGGER = logging.getLogger('ZiplineLogger') - -class TestClient(Component): - - def init(self): - self.received_count = 0 - self.prev_dt = None - - self.result_streams = [] - - # Maximum outgoing result streams, really shouldn't ever - # need more than 1. - self.max_outgoing = 5 - - @property - def get_id(self): - return "TEST_CLIENT" - - @property - def get_type(self): - return COMPONENT_TYPE.SINK - - def open(self): - self.data_feed = self.connect_result() - - def result_stream(self, zmq_socket, context=None): - """ - Asynchronously grab a socket to stream results out on. - """ - ctx = context or zmq.Context.instance() - sock = ctx.socket(zmq.PULL) - sock.bind(zmq_socket) - - # Add - self.result_streams.append( sock ) - - def do_work(self): - socks = dict(self.poll.poll(self.heartbeat_timeout)) - - if socks.get(self.control_in) == self.zmq.POLLIN: - msg = self.control_in.recv() - - if socks.get(self.data_feed) == self.zmq.POLLIN: - msg = self.data_feed.recv() - #logger.info('msg:' + str(msg)) - - if msg == str(CONTROL_PROTOCOL.DONE): - LOGGER.info("Client is DONE!") - self.signal_done() - return - - self.received_count += 1 - - try: - event = self.unframe(msg) - - # deserialization error - except zp.INVALID_MERGE_FRAME as exc: - return self.signal_exception(exc) - - if self.prev_dt != None: - if not event['dt'] >= self.prev_dt: - raise Exception( - "Message out of order: {date} after {prev}".format( - date = event['dt'], prev = self.prev_dt - ) - ) - else: - self.prev_dt = event.dt - - if self.received_count % 100 == 0: - LOGGER.info("received {n} messages".format(n=self.received_count)) - - def unframe(self, msg): - return zp.MERGE_UNFRAME(msg)