add logging

Author: SiyaoIsHiding

integration-tests/src/test/java/com/datastax/oss/driver/core/cql/PreparedStatementCachingIT.java

@@ -123,9 +123,11 @@ private static RemovalListener<Object, Object> buildCacheRemoveCallback(
         @NonNull Optional<DefaultDriverContext> context) {
       return (evt) -> {
         try {
+          LOG.info("Cache removal callback triggered, cause: {}", evt.getCause());
           CompletableFuture<PreparedStatement> future =
               (CompletableFuture<PreparedStatement>) evt.getValue();
           ByteBuffer queryId = Uninterruptibles.getUninterruptibly(future).getId();
+          LOG.info("Firing PreparedStatementRemovalEvent for queryId: {}", queryId);
           context.ifPresent(
               ctx -> ctx.getEventBus().fire(new PreparedStatementRemovalEvent(queryId)));
         } catch (Exception e) {
@@ -224,10 +226,15 @@ private void invalidationTestInner(
       assertThat(getPreparedCacheSize(session)).isEqualTo(0);
       setupTestSchema.accept(session);
 
-      session.prepare(preparedStmtQueryType1);
-      ByteBuffer queryId2 = session.prepare(preparedStmtQueryType2).getId();
+      PreparedStatement stmt1 = session.prepare(preparedStmtQueryType1);
+      PreparedStatement stmt2 = session.prepare(preparedStmtQueryType2);
+      ByteBuffer queryId2 = stmt2.getId();
       assertThat(getPreparedCacheSize(session)).isEqualTo(2);
 
+      LOG.info("Prepared statements in cache:");
+      LOG.info("  Statement 1: {} (queryId: {})", preparedStmtQueryType1, stmt1.getId());
+      LOG.info("  Statement 2: {} (queryId: {})", preparedStmtQueryType2, stmt2.getId());
+
       CountDownLatch preparedStmtCacheRemoveLatch = new CountDownLatch(1);
       CountDownLatch typeChangeEventLatch = new CountDownLatch(expectedChangedTypes.size());
 
@@ -244,6 +251,8 @@ private void invalidationTestInner(
               TypeChangeEvent.class,
               (e) -> {
                 // expect one event per type changed and for every parent type that nests it
+                LOG.info("Received TypeChangeEvent for type: {} (changeType: {})",
+                    e.oldType.getName(), e.changeType);
                 if (Boolean.TRUE.equals(
                     changedTypes.putIfAbsent(e.oldType.getName().toString(), true))) {
                   // store an error if we see duplicate change event
@@ -256,16 +265,20 @@ private void invalidationTestInner(
           .register(
               PreparedStatementRemovalEvent.class,
               (e) -> {
+                LOG.info("Received PreparedStatementRemovalEvent for queryId: {}", e.queryId);
                 if (!removedQueryIds.compareAndSet(Optional.empty(), Optional.of(e.queryId))) {
                   // store an error if we see multiple cache invalidation events
                   // any non-empty error will fail the test so it's OK to do this multiple times
                   removedQueryEventError.set(
                       Optional.of("Unable to set reference for PS removal event"));
+                  LOG.warn("Multiple PreparedStatementRemovalEvents received, ignoring subsequent ones");
                 }
                 preparedStmtCacheRemoveLatch.countDown();
               });
 
       // alter test_type_caching_2 to trigger cache invalidation and above events
+      LOG.info("Executing ALTER TYPE test_type_caching_2 add i blob");
+      LOG.info("Expected to invalidate statement 2 (queryId: {}) due to type change", queryId2);
       session.execute("ALTER TYPE test_type_caching_2 add i blob");
 
       // Give a small delay to allow the schema change to propagate before checking agreement
@@ -288,16 +301,20 @@ private void invalidationTestInner(
                     + "  - Cache removal latch success: %s (count: %d)\n"
                     + "  - Type change latch success: %s (count: %d)\n"
                     + "  - Current cache size: %d\n"
-                    + "  - Changed types detected: %s\n"
-                    + "  - Removed query IDs: %s\n"
+                    + "  - Expected changed types: %s\n"
+                    + "  - Actual changed types detected: %s\n"
+                    + "  - Expected removed query ID: %s\n"
+                    + "  - Actual removed query IDs: %s\n"
                     + "  - Type change errors: %s\n"
                     + "  - Removal event errors: %s",
                 cacheRemovalSuccess,
                 preparedStmtCacheRemoveLatch.getCount(),
                 typeChangeSuccess,
                 typeChangeEventLatch.getCount(),
                 getPreparedCacheSize(session),
+                expectedChangedTypes,
                 changedTypes.keySet(),
+                queryId2,
                 removedQueryIds.get(),
                 typeChangeEventError.get(),
                 removedQueryEventError.get());