def test_query_invokes_execute(self): with mock.patch('queries.tornado_session.TornadoSession._execute') as \ _execute: future = concurrent.Future() future.set_result(True) _execute.return_value = future obj = tornado_session.TornadoSession(io_loop=self.io_loop) result = yield obj.query('SELECT 1') _execute.assert_called_once_with('execute', 'SELECT 1', None)
def test_callproc_invokes_execute(self): with mock.patch('queries.tornado_session.TornadoSession._execute') as \ _execute: future = concurrent.Future() future.set_result(True) _execute.return_value = future obj = tornado_session.TornadoSession(io_loop=self.io_loop) result = yield obj.callproc('foo', ['bar']) _execute.assert_called_once_with('callproc', 'foo', ['bar'])
def setUp(self): super(SessionConnectTests, self).setUp() self.conn = mock.Mock() self.conn.fileno = mock.Mock(return_value=10) self.obj = tornado_session.TornadoSession(io_loop=self.io_loop) def create_connection(future): future.set_result(self.conn) self.obj._create_connection = create_connection
def test_validate_invokes_connect(self): with mock.patch('queries.tornado_session.TornadoSession._connect') as \ _connect: with mock.patch('queries.pool.PoolManager.free'): future = concurrent.Future() connection = mock.Mock() connection.fileno = mock.Mock(return_value=10) future.set_result(connection) _connect.return_value = future obj = tornado_session.TornadoSession(io_loop=self.io_loop) result = yield obj.validate() _connect.assert_called_once_with()
def test_pool_manager_add_failures_are_propagated(self): futures = [] def add_future(future, callback): futures.append((future, callback)) obj = tornado_session.TornadoSession() obj._ioloop = mock.Mock() obj._ioloop.add_future = add_future future = concurrent.Future() with mock.patch.object(obj._pool_manager, 'add') as add_method: add_method.side_effect = pool.PoolFullError(mock.Mock()) obj._create_connection(future) self.assertEqual(len(futures), 1) connected_future, callback = futures.pop() connected_future.set_result(True) callback(connected_future) self.assertIs(future.exception(), add_method.side_effect)
def setUp(self): self.obj = tornado_session.TornadoSession()
def test_query_error_index_error(self): obj = tornado_session.TornadoSession(io_loop=self.io_loop) with self.assertRaises(Exception): yield obj.query('SELECT * FROM foo WHERE bar=%s', [])