def test_close(self): config = self._setup_config() pika_string = 'socorro.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() assert len(conn_context_functor.pool) == 0
def test_force_reconnect(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() assert conn is conn_context_functor.pool[currentThread().getName()] conn_context_functor.force_reconnect() assert len(conn_context_functor.pool) == 0 conn2 = conn_context_functor.connection() assert not conn == conn2
def test_connection(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() assert conn is conn_context_functor.pool[currentThread().getName()] conn = conn_context_functor.connection('dwight') assert 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() assert conn is conn_context_functor.pool[currentThread().getName()]
def test_connection(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() assert conn is conn_context_functor.pool[currentThread().getName()] conn = conn_context_functor.connection('dwight') assert 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() assert conn is conn_context_functor.pool[currentThread().getName()]
def test_force_reconnect(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string) as mocked_pika_module: conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() self.assertTrue( conn is conn_context_functor.pool[currentThread().getName()] ) conn_context_functor.force_reconnect() self.assertEqual(len(conn_context_functor.pool), 0) conn2 = conn_context_functor.connection() self.assertFalse(conn == conn2)
def test_close(self): config = self._setup_config() pika_string = 'socorro.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_close(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string) as mocked_pika_module: conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() conn = conn_context_functor.connection('dwight') conn = conn_context_functor.connection('wilma') conn_context_functor.close() self.assertEqual(len(conn_context_functor.pool), 0)
def test_force_reconnect(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() assert conn is conn_context_functor.pool[currentThread().getName()] conn_context_functor.force_reconnect() assert len(conn_context_functor.pool) == 0 conn2 = conn_context_functor.connection() assert not conn == conn2
def test_force_reconnect(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string) as mocked_pika_module: conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection() self.assertTrue( conn is conn_context_functor.pool[currentThread().getName()]) conn_context_functor.force_reconnect() self.assertEqual(len(conn_context_functor.pool), 0) conn2 = conn_context_functor.connection() self.assertFalse(conn == conn2)
def test_close_connection(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection('dwight') assert conn is conn_context_functor.pool['dwight'] conn_context_functor.close_connection(conn) # should be no change assert conn is conn_context_functor.pool['dwight'] assert len(conn_context_functor.pool) == 1 conn_context_functor.close_connection(conn, True) with pytest.raises(KeyError): conn_context_functor.pool['dwight']() assert len(conn_context_functor.pool) == 0
def test_close_connection(self): config = self._setup_config() pika_string = 'socorro.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 = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string) as mocked_pika_module: conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection('dwight') self.assertTrue(conn is conn_context_functor.pool['dwight']) conn_context_functor.close_connection(conn) # should be no change self.assertTrue(conn is conn_context_functor.pool['dwight']) self.assertEqual(len(conn_context_functor.pool), 1) conn_context_functor.close_connection(conn, True) self.assertRaises(KeyError, lambda: conn_context_functor.pool['dwight']) self.assertEqual(len(conn_context_functor.pool), 0)
def test_close_connection(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string): conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection('dwight') assert conn is conn_context_functor.pool['dwight'] conn_context_functor.close_connection(conn) # should be no change assert conn is conn_context_functor.pool['dwight'] assert len(conn_context_functor.pool) == 1 conn_context_functor.close_connection(conn, True) with pytest.raises(KeyError): conn_context_functor.pool['dwight']() assert len(conn_context_functor.pool) == 0
def test_close_connection(self): config = self._setup_config() pika_string = 'socorro.external.rabbitmq.connection_context.pika' with patch(pika_string) as mocked_pika_module: conn_context_functor = ConnectionContextPooled(config) conn = conn_context_functor.connection('dwight') self.assertTrue( conn is conn_context_functor.pool['dwight'] ) conn_context_functor.close_connection(conn) # should be no change self.assertTrue( conn is conn_context_functor.pool['dwight'] ) self.assertEqual(len(conn_context_functor.pool), 1) conn_context_functor.close_connection(conn, True) self.assertRaises( KeyError, lambda : conn_context_functor.pool['dwight'] ) self.assertEqual(len(conn_context_functor.pool), 0)
def test_close_connection(self): config = self._setup_config() pika_string = 'socorro.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) assert conn_context_functor.config is conn_context_functor.local_config assert len(conn_context_functor.pool) == 0
def test_constructor(self): conn_context_functor = ConnectionContextPooled(self._setup_config) self.assertTrue( conn_context_functor.config is conn_context_functor.local_config ) self.assertEqual(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)