add logging

Author: SiyaoIsHiding

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

@@ -72,6 +72,8 @@
 @Category(IsolatedTests.class)
 public class PreparedStatementCachingIT {
 
+  private static final Logger LOG = LoggerFactory.getLogger(PreparedStatementCachingIT.class);
+
   private CustomCcmRule ccmRule = CustomCcmRule.builder().build();
 
   private SessionRule<CqlSession> sessionRule =
@@ -266,15 +268,46 @@ private void invalidationTestInner(
       // alter test_type_caching_2 to trigger cache invalidation and above events
       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
+      Uninterruptibles.sleepUninterruptibly(100, TimeUnit.MILLISECONDS);
       session.checkSchemaAgreement();
 
       // wait for latches and fail if they don't reach zero before timeout
-      assertThat(
-              Uninterruptibles.awaitUninterruptibly(
-                  preparedStmtCacheRemoveLatch, 120, TimeUnit.SECONDS))
+      // Use longer timeout for cache removal as it depends on complex event chain
+      boolean cacheRemovalSuccess =
+          Uninterruptibles.awaitUninterruptibly(
+              preparedStmtCacheRemoveLatch, 180, TimeUnit.SECONDS);
+      boolean typeChangeSuccess =
+          Uninterruptibles.awaitUninterruptibly(typeChangeEventLatch, 60, TimeUnit.SECONDS);
+
+      // Provide detailed diagnostics if either latch fails
+      if (!cacheRemovalSuccess || !typeChangeSuccess) {
+        String diagnostics =
+            String.format(
+                "Test failure diagnostics:\n"
+                    + "  - 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"
+                    + "  - Type change errors: %s\n"
+                    + "  - Removal event errors: %s",
+                cacheRemovalSuccess,
+                preparedStmtCacheRemoveLatch.getCount(),
+                typeChangeSuccess,
+                typeChangeEventLatch.getCount(),
+                getPreparedCacheSize(session),
+                changedTypes.keySet(),
+                removedQueryIds.get(),
+                typeChangeEventError.get(),
+                removedQueryEventError.get());
+        LOG.error("Prepared statement cache invalidation test failed: {}", diagnostics);
+      }
+
+      assertThat(cacheRemovalSuccess)
           .withFailMessage("preparedStmtCacheRemoveLatch did not trigger before timeout")
           .isTrue();
-      assertThat(Uninterruptibles.awaitUninterruptibly(typeChangeEventLatch, 20, TimeUnit.SECONDS))
+      assertThat(typeChangeSuccess)
           .withFailMessage("typeChangeEventLatch did not trigger before timeout")
           .isTrue();