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 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 -> {