ff18c837b7
This file uses std::cout and so should include <iostream>. Found with a patch to seastar that removes some redundant <iostream> includes. Signed-off-by: Rafael Ávila de Espíndola <espindola@scylladb.com> Message-Id: <20181218183816.34504-1-espindola@scylladb.com>
89 lines
2.0 KiB
C++
89 lines
2.0 KiB
C++
/*
|
|
* Copyright (C) 2018 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 <algorithm>
|
|
#include <random>
|
|
|
|
#include <boost/range/algorithm/generate.hpp>
|
|
#include <iostream>
|
|
|
|
namespace tests::random {
|
|
|
|
namespace internal {
|
|
|
|
inline std::random_device::result_type get_seed()
|
|
{
|
|
std::random_device rd;
|
|
auto seed = rd();
|
|
std::cout << "tests::random seed = " << seed << "\n";
|
|
return seed;
|
|
}
|
|
|
|
}
|
|
|
|
inline std::default_random_engine gen(internal::get_seed());
|
|
|
|
template<typename T>
|
|
T get_int() {
|
|
std::uniform_int_distribution<T> dist;
|
|
return dist(gen);
|
|
}
|
|
|
|
template<typename T>
|
|
T get_int(T max) {
|
|
std::uniform_int_distribution<T> dist(0, max);
|
|
return dist(gen);
|
|
}
|
|
|
|
template<typename T>
|
|
T get_int(T min, T max) {
|
|
std::uniform_int_distribution<T> dist(min, max);
|
|
return dist(gen);
|
|
}
|
|
|
|
inline bool get_bool() {
|
|
static std::bernoulli_distribution dist;
|
|
return dist(gen);
|
|
}
|
|
|
|
inline bytes get_bytes(size_t n) {
|
|
bytes b(bytes::initialized_later(), n);
|
|
boost::generate(b, [] { return get_int<bytes::value_type>(); });
|
|
return b;
|
|
}
|
|
|
|
inline bytes get_bytes() {
|
|
return get_bytes(get_int<unsigned>(128 * 1024));
|
|
}
|
|
|
|
inline sstring get_sstring(size_t n) {
|
|
sstring str(sstring::initialized_later(), n);
|
|
boost::generate(str, [] { return get_int<sstring::value_type>('a', 'z'); });
|
|
return str;
|
|
}
|
|
|
|
inline sstring get_sstring() {
|
|
return get_sstring(get_int<unsigned>(1024));
|
|
}
|
|
|
|
}
|