def test_start(self):
"""
test datastream worker pool
"""
s = SocketSession(ip='x')
s.keywords = ['google', 'bieber']
s.save()
w = WorkerPool()
w.start()
while not RawStreamQueue.objects.count():
time.sleep(0.1)
w.terminate()
self.assertGreater(RawStreamQueue.objects.count(), 0)
def test_worker(self):
"""
test single twitter worker
"""
s = SocketSession(ip='x')
s.keywords = ['google', 'bieber']
s.save()
kill = threading.Event()
w = TwitterWorker(kill)
w.setDaemon(True)
w.start()
while not RawStreamQueue.objects.count():
time.sleep(0.1)
kill.set()
self.assertGreater(RawStreamQueue.objects.count(), 0)
