Tweak test case 'WHERE' clause

apache · Jun 17, 2024 · cee17f2 · cee17f2
1 parent 0db6193
commit cee17f2
Showing 1 changed file with 2 additions and 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
@@ -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 -> {