From cee17f26c104928cb014c0019b6b9d2f2218d1f7 Mon Sep 17 00:00:00 2001
From: George Ma <georgecma@google.com>
Date: Mon, 17 Jun 2024 19:43:12 +0000
Subject: [PATCH] Tweak test case 'WHERE' clause

---
 .../org/apache/beam/sdk/io/cassandra/CassandraIOTest.java     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sdks/java/io/cassandra/src/test/java/org/apache/beam/sdk/io/cassandra/CassandraIOTest.java b/sdks/java/io/cassandra/src/test/java/org/apache/beam/sdk/io/cassandra/CassandraIOTest.java
index 1ca874d01e31..73dc7ae302c9 100644
--- a/sdks/java/io/cassandra/src/test/java/org/apache/beam/sdk/io/cassandra/CassandraIOTest.java
+++ b/sdks/java/io/cassandra/src/test/java/org/apache/beam/sdk/io/cassandra/CassandraIOTest.java
@@ -546,7 +546,7 @@ public void testReadWithQueryProvider() throws Exception {
   public void testReadWithQueryProviderWithWhereQuery() throws Exception {
     String query =
         String.format(
-            "select person_id, writetime(person_name) from %s.%s WHERE TRUE",
+            "select person_id, writetime(person_name) from %s.%s where person_id=10 AND person_department='logic'",
             CASSANDRA_KEYSPACE, CASSANDRA_TABLE);
 
     PCollection<Scientist> output =
@@ -561,7 +561,7 @@ public void testReadWithQueryProviderWithWhereQuery() throws Exception {
                 .withCoder(SerializableCoder.of(Scientist.class))
                 .withEntity(Scientist.class));
 
-    PAssert.thatSingleton(output.apply("Count", Count.globally())).isEqualTo(NUM_ROWS);
+    PAssert.thatSingleton(output.apply("Count", Count.globally())).isEqualTo(1L);
     PAssert.that(output)
         .satisfies(
             input -> {