def thriftworker(self): return ThriftWorker(port_range=self.config.SERVICE_PORT_RANGE, protocol_factory=self.protocol_factory, pool_size=self.config.CONCURRENCY)
def test_custom_pool_size(self): app = ThriftWorker(pool_size=5) self.assertEqual(5, app.pool_size)
def setUp(self): super(CustomAppMixin, self).setUp() loop = self.loop = Loop() self.app = ThriftWorker(loop=loop)
def test_custom_proto_factory(self): factory = TBinaryProtocol.TBinaryProtocolFactory() app = ThriftWorker(protocol_factory=factory) self.assertIs(factory, app.protocol_factory)
def test_custom_port_range(self): app = ThriftWorker(port_range=(1, 10)) self.assertEqual((1, 10), app.port_range) app = ThriftWorker(port_range=('1', '10')) self.assertEqual((1, 10), app.port_range)
def test_custom_loop(self): custom_loop = Loop() app = ThriftWorker(loop=custom_loop) self.assertIs(custom_loop, app.loop)
def test_wrong_port_range(self): with self.assertRaises(ValueError): ThriftWorker(port_range=()) ThriftWorker(port_range=(1, 'test'))
def test_negative_pool_size(self): with self.assertRaises(ValueError): ThriftWorker(pool_size=-1)
def test_default_app(self): some_app = ThriftWorker() global_app = ThriftWorker.instance() self.assertIs(some_app, global_app)
def test_current_app(self): some_app = ThriftWorker() self.assertIs(some_app, get_current_app())