65 lines
1.5 KiB
C++
65 lines
1.5 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>
|
|
|
|
namespace tests::random {
|
|
|
|
static std::random_device rd;
|
|
static std::default_random_engine gen(rd());
|
|
|
|
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);
|
|
}
|
|
|
|
bool get_bool() {
|
|
static std::bernoulli_distribution dist;
|
|
return dist(gen);
|
|
}
|
|
|
|
bytes get_bytes(size_t n) {
|
|
bytes b(bytes::initialized_later(), n);
|
|
boost::generate(b, [] { return get_int<bytes::value_type>(); });
|
|
return b;
|
|
}
|
|
|
|
bytes get_bytes() {
|
|
return get_bytes(get_int<unsigned>(128 * 1024));
|
|
}
|
|
|
|
}
|