def test_start(self):
        """
        test datastream worker pool
        """
        s = SocketSession(ip='x')
        s.keywords = ['google', 'bieber']
        s.save()

        w = WorkerPool()
        w.start()
        time.sleep(1)
        w.terminate()

        self.assertGreater(RawStreamQueue.objects.count(), 0)
    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)
import time
import logging

from smm.datastream.workerPool import WorkerPool
from smm import models


logger = logging.getLogger('start-collector')


#connect to db
models.connect()

s = models.SocketSession()
s.ip='x'
s.keywords = ['google']
s.save()

# init pool
pool = WorkerPool()

try:

    pool.start()
    logger.info("started WorkerPool with size %s", len(pool.pool))
    while True:
        time.sleep(60)
        logging.info('RawStreamQueue size: %s', models.RawStreamQueue.objects.count())

except (KeyboardInterrupt, SystemExit):
    pool.terminate()
import time
import logging

from smm.datastream.workerPool import WorkerPool
from smm import models


logger = logging.getLogger('start-collector')


#connect to db
models.connect()

# init pool
pool = WorkerPool()

try:

    pool.start()
    logger.info("started WorkerPool with size %s", len(pool.pool))
    while True:
        time.sleep(60)
        logging.info('RawStreamQueue size: %s', models.RawStreamQueue.objects.count())

except (KeyboardInterrupt, SystemExit):
    pool.terminate()