Files
scylladb/tests/cql_assertions.hh
Duarte Nunes 8ac4d7b2e8 tests: Add view_schema_test
This patch adds a set of tests for materialized view schema
handling, complementing the dtests for the same feature.

Signed-off-by: Duarte Nunes <duarte@scylladb.com>
2016-12-20 13:06:11 +00:00

72 lines
1.9 KiB
C++

/*
* Copyright (C) 2015 ScyllaDB
*/
/*
* This file is part of Scylla.
*
* Scylla is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Scylla is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include "transport/messages/result_message_base.hh"
#include "bytes.hh"
#include "core/shared_ptr.hh"
#include "core/future.hh"
class rows_assertions {
shared_ptr<transport::messages::result_message::rows> _rows;
public:
rows_assertions(shared_ptr<transport::messages::result_message::rows> rows);
rows_assertions with_size(size_t size);
rows_assertions is_empty();
rows_assertions with_row(std::initializer_list<bytes_opt> values);
// Verifies that the result has the following rows and only that rows, in that order.
rows_assertions with_rows(std::initializer_list<std::initializer_list<bytes_opt>> rows);
};
class result_msg_assertions {
shared_ptr<transport::messages::result_message> _msg;
public:
result_msg_assertions(shared_ptr<transport::messages::result_message> msg);
rows_assertions is_rows();
};
result_msg_assertions assert_that(shared_ptr<transport::messages::result_message> msg);
template<typename... T>
void assert_that_failed(future<T...>& f)
{
try {
f.get();
assert(f.failed());
}
catch (...) {
}
}
template<typename... T>
void assert_that_failed(future<T...>&& f)
{
try {
f.get();
assert(f.failed());
}
catch (...) {
}
}