def test_close(self, mock_client, mock_qldb_session1, mock_qldb_session2):
mock_client.return_value = mock_client
mock_client.max_pool_connections = DEFAULT_MAX_CONCURRENT_TRANSACTIONS
qldb_driver = QldbDriver(MOCK_LEDGER_NAME)
qldb_driver._pool = Queue()
qldb_driver._pool.put(mock_qldb_session1)
qldb_driver._pool.put(mock_qldb_session2)
qldb_driver.close()
mock_qldb_session1._end_session.assert_called_once_with()
mock_qldb_session2._end_session.assert_called_once_with()
def test_get_session_no_session_in_pool(
self, mock_client, mock_qldb_session, mock_bounded_semaphore,
mock_atomic_integer, mock_logger_debug, mock_create_new_session):
mock_create_new_session.return_value = mock_qldb_session
mock_client.return_value = mock_client
mock_client.max_pool_connections = DEFAULT_MAX_CONCURRENT_TRANSACTIONS
qldb_driver = QldbDriver(MOCK_LEDGER_NAME)
qldb_driver._pool = Queue()
session = qldb_driver._get_session(False)
self.assertEqual(session, mock_qldb_session)
mock_bounded_semaphore().acquire.assert_called_once_with(
timeout=DEFAULT_TIMEOUT_SECONDS)
mock_atomic_integer().decrement.assert_called_once_with()
self.assertEqual(mock_logger_debug.call_count, 2)
mock_create_new_session.assert_called_once_with()
