test_query_only_static_row: use flat reader

Signed-off-by: Piotr Jastrzebski <piotr@scylladb.com>
This commit is contained in:
Piotr Jastrzebski
2018-01-17 09:35:33 +01:00
parent 5a5a5149e3
commit c7ce24be06

View File

@@ -833,7 +833,7 @@ static void test_query_only_static_row(populate_fn populate) {
// fully populate cache
{
auto prange = dht::partition_range::make_ending_with(dht::ring_position(m1.decorated_key()));
assert_that(ms(s.schema(), prange, s.schema()->full_slice()))
assert_that(ms.make_flat_mutation_reader(s.schema(), prange, s.schema()->full_slice()))
.produces(m1)
.produces_end_of_stream();
}
@@ -844,7 +844,7 @@ static void test_query_only_static_row(populate_fn populate) {
.with_ranges({})
.build();
auto prange = dht::partition_range::make_ending_with(dht::ring_position(m1.decorated_key()));
assert_that(ms(s.schema(), prange, slice))
assert_that(ms.make_flat_mutation_reader(s.schema(), prange, slice))
.produces(m1, slice.row_ranges(*s.schema(), m1.key()))
.produces_end_of_stream();
}