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_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_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_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)
