def test_force_reconnect(self): config = self._setup_config() pika_string = 'collector.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() ok_( conn is conn_context_functor.pool[currentThread().getName()] ) conn_context_functor.force_reconnect() eq_(len(conn_context_functor.pool), 0) conn2 = conn_context_functor.connection() ok_(not conn == conn2)
def test_close(self): config = self._setup_config() pika_string = 'collector.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn_context_functor.connection() conn_context_functor.connection('dwight') conn_context_functor.connection('wilma') conn_context_functor.close() eq_(len(conn_context_functor.pool), 0)
def test_connection(self): config = self._setup_config() pika_string = 'collector.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() ok_( conn is conn_context_functor.pool[currentThread().getName()] ) conn = conn_context_functor.connection('dwight') ok_( conn is conn_context_functor.pool['dwight'] ) # get the same connection again to make sure it really is the same conn = conn_context_functor.connection() ok_( conn is conn_context_functor.pool[currentThread().getName()] )
def test_close_connection(self): config = self._setup_config() pika_string = 'collector.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection('dwight') ok_( conn is conn_context_functor.pool['dwight'] ) conn_context_functor.close_connection(conn) # should be no change ok_( conn is conn_context_functor.pool['dwight'] ) eq_(len(conn_context_functor.pool), 1) conn_context_functor.close_connection(conn, True) assert_raises( KeyError, lambda : conn_context_functor.pool['dwight'] ) eq_(len(conn_context_functor.pool), 0)
def test_constructor(self): conn_context_functor = ConnectionContextPooled(self._setup_config) ok_( conn_context_functor.config is conn_context_functor.local_config ) eq_(len(conn_context_functor.pool), 0)