38a54df863
People keep tripping over the old copyrights and copy-pasting them to new files. Search and replace "Cloudius Systems" with "ScyllaDB". Message-Id: <1460013664-25966-1-git-send-email-penberg@scylladb.com>
107 lines
3.3 KiB
C++
107 lines
3.3 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/>.
|
|
*/
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
#include "cql_assertions.hh"
|
|
#include "transport/messages/result_message.hh"
|
|
#include "to_string.hh"
|
|
#include "bytes.hh"
|
|
|
|
static inline void fail(sstring msg) {
|
|
throw std::runtime_error(msg);
|
|
}
|
|
|
|
rows_assertions::rows_assertions(shared_ptr<transport::messages::result_message::rows> rows)
|
|
: _rows(rows)
|
|
{ }
|
|
|
|
rows_assertions
|
|
rows_assertions::with_size(size_t size) {
|
|
auto row_count = _rows->rs().size();
|
|
if (row_count != size) {
|
|
fail(sprint("Expected %d row(s) but got %d", size, row_count));
|
|
}
|
|
return {*this};
|
|
}
|
|
|
|
rows_assertions
|
|
rows_assertions::is_empty() {
|
|
auto row_count = _rows->rs().size();
|
|
if (row_count != 0) {
|
|
auto&& first_row = *_rows->rs().rows().begin();
|
|
fail(sprint("Expected no rows, but got %d. First row: %s", row_count, to_string(first_row)));
|
|
}
|
|
return {*this};
|
|
}
|
|
|
|
rows_assertions
|
|
rows_assertions::with_row(std::initializer_list<bytes_opt> values) {
|
|
std::vector<bytes_opt> expected_row(values);
|
|
for (auto&& row : _rows->rs().rows()) {
|
|
if (row == expected_row) {
|
|
return {*this};
|
|
}
|
|
}
|
|
fail("Expected row not found");
|
|
return {*this};
|
|
}
|
|
|
|
// Verifies that the result has the following rows and only that rows, in that order.
|
|
rows_assertions
|
|
rows_assertions::with_rows(std::initializer_list<std::initializer_list<bytes_opt>> rows) {
|
|
auto actual_i = _rows->rs().rows().begin();
|
|
auto actual_end = _rows->rs().rows().end();
|
|
int row_nr = 0;
|
|
for (auto&& row : rows) {
|
|
if (actual_i == actual_end) {
|
|
fail(sprint("Expected more rows (%d), got %d", rows.size(), _rows->rs().size()));
|
|
}
|
|
auto& actual = *actual_i;
|
|
if (!std::equal(
|
|
std::begin(row), std::end(row),
|
|
std::begin(actual), std::end(actual))) {
|
|
fail(sprint("row %d differs, expected %s got %s", row_nr, to_string(row), to_string(actual)));
|
|
}
|
|
++actual_i;
|
|
++row_nr;
|
|
}
|
|
if (actual_i != actual_end) {
|
|
fail(sprint("Expected less rows (%d), got %d. Next row is: %s", rows.size(), _rows->rs().size(),
|
|
to_string(*actual_i)));
|
|
}
|
|
return {*this};
|
|
}
|
|
|
|
result_msg_assertions::result_msg_assertions(shared_ptr<transport::messages::result_message> msg)
|
|
: _msg(msg)
|
|
{ }
|
|
|
|
rows_assertions result_msg_assertions::is_rows() {
|
|
auto rows = dynamic_pointer_cast<transport::messages::result_message::rows>(_msg);
|
|
BOOST_REQUIRE(rows);
|
|
return rows_assertions(rows);
|
|
}
|
|
|
|
result_msg_assertions assert_that(shared_ptr<transport::messages::result_message> msg) {
|
|
return result_msg_assertions(msg);
|
|
}
|