def test_abort_or_close_when_closed(self, mock_session, mock_executor):
qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
MOCK_RETRY_LIMIT, mock_executor)
qldb_session._is_closed = True
self.assertFalse(qldb_session._abort_or_close())
mock_session.abort_transaction.assert_not_called()
def test_close(self, mock_session, mock_executor):
qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
MOCK_RETRY_LIMIT, mock_executor)
qldb_session._is_closed = False
qldb_session.close()
self.assertTrue(qldb_session._is_closed)
mock_session.close.assert_called_once_with()
def test_close_twice(self, mock_session, mock_executor):
qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
mock_executor, MOCK_DRIVER_RELEASE)
qldb_session._is_closed = False
qldb_session._close()
qldb_session._close()
self.assertTrue(qldb_session._is_closed)
def test_throw_if_closed_when_closed(self, mock_executor, mock_session,
mock_release_session,
mock_logger_error):
qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
mock_executor, mock_release_session)
qldb_session._is_closed = True
self.assertRaises(SessionClosedError, qldb_session._throw_if_closed)
mock_logger_error.assert_called_once()
def test_throw_if_closed_session_closed_error(self, mock_session,
mock_logger_error,
mock_executor):
qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
MOCK_RETRY_LIMIT, mock_executor)
qldb_session._is_closed = True
self.assertRaises(SessionClosedError, qldb_session.throw_if_closed)
mock_logger_error.assert_called_once()
def test_throw_if_closed_not_closed(self, mock_executor, mock_session,
mock_release_session,
mock_logger_error):
qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
mock_executor, mock_release_session)
qldb_session._is_closed = False
result = qldb_session._throw_if_closed()
self.assertEqual(result, None)
mock_logger_error.assert_not_called()