Nose autodiscover & unittest2

This commit is contained in:
Stephen Diehl
2012-02-02 08:23:18 -05:00
parent 37e47abb14
commit b17f7759c8
2 changed files with 0 additions and 72 deletions
-6
View File
@@ -1,6 +0,0 @@
import unittest
from test.testmessaging import *
if __name__ == '__main__':
suite = unittest.TestLoader().loadTestsFromTestCase(MessagingTestCase)
unittest.TextTestRunner(verbosity=2).run(suite)
-66
View File
@@ -1,66 +0,0 @@
import unittest
import zmq
import logging
import tornado
from data.sources.equity import *
from data.feed import *
from data.transforms import MergedTransformsFeed, MovingAverage
from qbt_client import TestClient
class MessagingTestCase(unittest.TestCase):
def setUp(self):
tornado.options.parse_command_line()
self.total_data_count = 800
self.feed_config = {'emt1':{'sid':133, 'class':'RandomEquityTrades', 'count':400},
'emt2':{'sid':134, 'class':'RandomEquityTrades', 'count':400}}
self.feed = DataFeed(self.feed_config)
self.feed_proc = multiprocessing.Process(target=self.feed.run)
self.config = {}
self.config['name'] = '**merged feed**'
self.config['transforms'] = [{'name':'mavg1', 'class':'MovingAverage', 'hours':1},{'name':'mavg2', 'class':'MovingAverage', 'hours':2}]
def dtest_client(self):
#subscribe a client to the transformed feed
client = TestClient(self.feed, self.feed.feed_address)
feed_proc = multiprocessing.Process(target=self.feed.run)
feed_proc.start()
client.run()
self.assertEqual(self.feed.data_buffer.pending_messages(), 0, "The feed should be drained of all messages, found {n} remaining.".format(n=self.feed.data_buffer.pending_messages()))
self.assertEqual(self.total_data_count, client.received_count, "The client should have received ({n}) the same number of messages as the feed sent ({m}).".format(n=client.received_count, m=self.total_data_count))
def dtest_moving_average_to_client(self):
mavg = MovingAverage(self.feed, self.config['transforms'][0])
mavg_proc = multiprocessing.Process(target=mavg.run)
mavg_proc.start()
client = TestClient(self.feed, mavg.result_address, bind=True)
feed_proc = multiprocessing.Process(target=self.feed.run)
feed_proc.start()
client.run()
self.assertEqual(self.feed.data_buffer.pending_messages(), 0, "The feed should be drained of all messages.")
self.assertEqual(self.total_data_count, client.received_count, "The client should have received the same number of messages as the feed sent.")
def test_merged_to_client(self):
merger = MergedTransformsFeed(self.feed, self.config)
merger_proc = multiprocessing.Process(target=merger.run)
merger_proc.start()
client = TestClient(self.feed, merger.result_address)
feed_proc = multiprocessing.Process(target=self.feed.run)
feed_proc.start()
client.run()
self.assertEqual(self.feed.data_buffer.pending_messages(), 0, "The feed should be drained of all messages.")
self.assertEqual(self.total_data_count, client.received_count, "The client should have received the same number of messages as the feed sent.")