Python QldbSession._no_throw_abort Examples

Example #1

File: test_qldb_session.py Project: tirumalesh123/amazon-qldb-driver-python

    def test_no_throw_abort_transaction_is_none(self, mock_session,
                                                mock_executor):
        qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
                                   mock_executor)
        qldb_session._no_throw_abort()

        mock_session._abort_transaction.assert_called_once_with()

Example #2

File: test_qldb_session.py Project: tirumalesh123/amazon-qldb-driver-python

    def test_no_throw_abort_client_error(self, mock_session,
                                         mock_logger_warning, mock_executor):
        mock_session._abort_transaction.side_effect = ClientError(
            MOCK_CLIENT_ERROR_MESSAGE, 'message')
        qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
                                   mock_executor)
        qldb_session._no_throw_abort()

        self.assertFalse(qldb_session._is_alive)
        mock_logger_warning.assert_called_once()

Example #3

File: test_qldb_session.py Project: saumehta9/amazon-qldb-driver-python

 def test_no_throw_abort_client_error(self, mock_session, mock_executor,
                                      mock_transaction,
                                      mock_logger_warning):
     qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
                                MOCK_RETRY_LIMIT, mock_executor)
     ce = ClientError(MOCK_CLIENT_ERROR_MESSAGE, MOCK_MESSAGE)
     mock_transaction.abort.side_effect = ce
     qldb_session._no_throw_abort(mock_transaction)
     mock_transaction.abort.assert_called_once_with()
     mock_logger_warning.assert_called_once()

Example #4

File: test_qldb_session.py Project: saumehta9/amazon-qldb-driver-python

 def test_no_throw_abort_with_none_transaction(self, mock_session,
                                               mock_executor):
     qldb_session = QldbSession(mock_session, MOCK_READ_AHEAD,
                                MOCK_RETRY_LIMIT, mock_executor)
     qldb_session._no_throw_abort(None)
     mock_session.abort_transaction.assert_called_once_with()