/*
 * Copyright 2015 Cloudius Systems
 */

#define BOOST_TEST_DYN_LINK

#include <seastar/core/thread.hh>
#include <seastar/tests/test-utils.hh>

#include "tests/cql_test_env.hh"
#include "tests/mutation_source_test.hh"
#include "tests/result_set_assertions.hh"
#include "service/storage_proxy.hh"
#include "partition_slice_builder.hh"

static query::result to_data_query_result(mutation_reader& reader, const query::partition_slice& slice) {
    query::result::builder builder(slice);
    auto now = gc_clock::now();
    while (true) {
        mutation_opt mo = reader().get0();
        if (!mo) {
            break;
        }
        auto pb = builder.add_partition(mo->key());
        mo->partition().query(pb, *mo->schema(), now);
    }
    return builder.build();
}

static query::result_set to_result_set(schema_ptr s, mutation_reader& reader) {
    auto slice = partition_slice_builder(*s).build();
    return query::result_set::from_raw_result(s, slice, to_data_query_result(reader, slice));
}

SEASTAR_TEST_CASE(test_make_local_reader) {
    return do_with_cql_env([](cql_test_env& e) {
        return seastar::async([&] {
            e.execute_cql("create keyspace ks2 with replication = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 };").get();
            e.execute_cql("create table ks2.cf (k blob, v int, primary key (k));").get();
            e.execute_cql(
                "begin unlogged batch \n"
                    "  insert into ks2.cf (k, v) values (0x01, 0); \n"
                    "  insert into ks2.cf (k, v) values (0x02, 0); \n"
                    "  insert into ks2.cf (k, v) values (0x03, 0); \n"
                    "  insert into ks2.cf (k, v) values (0x04, 0); \n"
                    "  insert into ks2.cf (k, v) values (0x05, 0); \n"
                    "apply batch;").get();

            auto s = e.local_db().find_schema("ks2", "cf");
            auto reader = service::get_storage_proxy().local().make_local_reader(s->id(), query::full_partition_range);

            assert_that(to_result_set(s, reader))
                .has_size(5)
                .has(a_row().with_column(bytes("k"), bytes("\01")))
                .has(a_row().with_column(bytes("k"), bytes("\02")))
                .has(a_row().with_column(bytes("k"), bytes("\03")))
                .has(a_row().with_column(bytes("k"), bytes("\04")))
                .has(a_row().with_column(bytes("k"), bytes("\05")));
        });
    });
}