mirror of
https://github.com/zeek/zeek.git
synced 2025-10-06 16:48:19 +00:00
15d74ac081
Mostly trying to standardize the way tests sleep for arbitrary amounts of time to make it easier to tell at which particular point the unit test actually may need the timeout interval increased (or else debugged further).
118 lines
4.6 KiB
Text
118 lines
4.6 KiB
Text
# @TEST-SERIALIZE: comm
|
|
#
|
|
# @TEST-EXEC: btest-bg-run manager-1 BROPATH=$BROPATH:.. CLUSTER_NODE=manager-1 bro %INPUT
|
|
# @TEST-EXEC: btest-bg-run worker-1 BROPATH=$BROPATH:.. CLUSTER_NODE=worker-1 bro %INPUT
|
|
# @TEST-EXEC: btest-bg-run worker-2 BROPATH=$BROPATH:.. CLUSTER_NODE=worker-2 bro %INPUT
|
|
# @TEST-EXEC: btest-bg-wait 15
|
|
# @TEST-EXEC: btest-diff manager-1/.stdout
|
|
|
|
@TEST-START-FILE cluster-layout.bro
|
|
redef Cluster::nodes = {
|
|
["manager-1"] = [$node_type=Cluster::MANAGER, $ip=127.0.0.1, $p=37757/tcp],
|
|
["worker-1"] = [$node_type=Cluster::WORKER, $ip=127.0.0.1, $p=37760/tcp, $manager="manager-1", $interface="eth0"],
|
|
["worker-2"] = [$node_type=Cluster::WORKER, $ip=127.0.0.1, $p=37761/tcp, $manager="manager-1", $interface="eth1"],
|
|
};
|
|
@TEST-END-FILE
|
|
|
|
redef Cluster::retry_interval = 1sec;
|
|
redef Broker::default_listen_retry = 1sec;
|
|
redef Broker::default_connect_retry = 1sec;
|
|
redef Log::default_rotation_interval = 0secs;
|
|
|
|
event bro_init() &priority=5
|
|
{
|
|
local r1: SumStats::Reducer = [$stream="test", $apply=set(SumStats::SAMPLE), $num_samples=5];
|
|
SumStats::create([$name="test",
|
|
$epoch=5secs,
|
|
$reducers=set(r1),
|
|
$epoch_result(ts: time, key: SumStats::Key, result: SumStats::Result) =
|
|
{
|
|
local r = result["test"];
|
|
print fmt("Host: %s Sampled observations: %d", key$host, r$sample_elements);
|
|
local sample_nums: vector of count = vector();
|
|
for ( sample in r$samples )
|
|
sample_nums[|sample_nums|] =r$samples[sample]$num;
|
|
|
|
print fmt(" %s", sort(sample_nums));
|
|
},
|
|
$epoch_finished(ts: time) =
|
|
{
|
|
terminate();
|
|
}]);
|
|
}
|
|
|
|
event Broker::peer_lost(endpoint: Broker::EndpointInfo, msg: string)
|
|
{
|
|
terminate();
|
|
}
|
|
|
|
global ready_for_data: event();
|
|
|
|
event bro_init()
|
|
{
|
|
Broker::auto_publish(Cluster::worker_topic, ready_for_data);
|
|
|
|
}
|
|
|
|
event ready_for_data()
|
|
{
|
|
if ( Cluster::node == "worker-1" )
|
|
{
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=5]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=22]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=94]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=50]);
|
|
# I checked the random numbers. seems legit.
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=51]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=61]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=61]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=71]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=81]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=91]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=101]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=111]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=121]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=131]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=141]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=151]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=161]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=171]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=181]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=191]);
|
|
|
|
SumStats::observe("test", [$host=6.5.4.3], [$num=2]);
|
|
SumStats::observe("test", [$host=7.2.1.5], [$num=1]);
|
|
}
|
|
if ( Cluster::node == "worker-2" )
|
|
{
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=75]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=30]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=3]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=57]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=52]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=61]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=1.2.3.4], [$num=95]);
|
|
SumStats::observe("test", [$host=6.5.4.3], [$num=5]);
|
|
SumStats::observe("test", [$host=7.2.1.5], [$num=91]);
|
|
SumStats::observe("test", [$host=10.10.10.10], [$num=5]);
|
|
}
|
|
}
|
|
|
|
@if ( Cluster::local_node_type() == Cluster::MANAGER )
|
|
|
|
global peer_count = 0;
|
|
event Broker::peer_added(endpoint: Broker::EndpointInfo, msg: string)
|
|
{
|
|
++peer_count;
|
|
if ( peer_count == 2 )
|
|
event ready_for_data();
|
|
}
|
|
|
|
@endif
|