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
Beispiel #2
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)