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