52 lines
1.7 KiB
C++
52 lines
1.7 KiB
C++
|
// Copyright 2022 Memgraph Ltd.
|
||
|
|
//
|
||
|
|
// Use of this software is governed by the Business Source License
|
||
|
|
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
||
|
|
// License, and you may not use this file except in compliance with the Business Source License.
|
||
|
|
//
|
||
|
|
// As of the Change Date specified in that file, in accordance with
|
||
|
|
// the Business Source License, use of this software will be governed
|
||
|
|
// by the Apache License, Version 2.0, included in the file
|
||
|
|
// licenses/APL.txt.
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <rapidcheck.h>
|
||
|
|
|
||
|
|
#include "testing_constants.hpp"
|
||
|
|
|
||
|
|
namespace memgraph::tests::simulation {
|
||
|
|
|
||
|
|
struct ClusterConfig {
|
||
|
|
int servers;
|
||
|
|
int replication_factor;
|
||
|
|
int shards;
|
||
|
|
|
||
|
|
friend std::ostream &operator<<(std::ostream &in, const ClusterConfig &cluster) {
|
||
|
|
in << "ClusterConfig { servers: " << cluster.servers << ", replication_factor: " << cluster.replication_factor
|
||
|
|
<< ", shards: " << cluster.shards << " }";
|
||
|
|
return in;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace memgraph::tests::simulation
|
||
|
|
|
||
|
|
// Required namespace for rapidcheck generator
|
||
|
|
namespace rc {
|
||
|
|
|
||
|
|
using memgraph::tests::simulation::ClusterConfig;
|
||
|
|
|
||
|
|
template <>
|
||
|
|
struct Arbitrary<ClusterConfig> {
|
||
|
|
static Gen<ClusterConfig> arbitrary() {
|
||
|
|
return gen::build<ClusterConfig>(
|
||
|
|
// gen::inRange is [inclusive min, exclusive max)
|
||
|
|
gen::set(&ClusterConfig::servers, gen::inRange(kMinimumServers, kMaximumServers)),
|
||
|
|
gen::set(&ClusterConfig::replication_factor,
|
||
|
|
gen::inRange(kMinimumReplicationFactor, kMaximumReplicationFactor)),
|
||
|
|
gen::set(&ClusterConfig::shards, gen::inRange(kMinimumShards, kMaximumShards)));
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace rc
|