mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 14:48:21 +00:00
btest/generic: Add publish_hrw(), publish_rr() and logging tests
They currently use zeromq, but technically they should be valid for any other backend, too, even broker.
This commit is contained in:
Arne Welzel
parent
df69ec1279
commit
d816bfb249
11 changed files with 724 additions and 0 deletions
|
|
@ -0,0 +1,4 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
error in <...>/cluster-publish-errors.zeek, line 58: Publish of unknown record type 'Cluster::MyEvent' (Cluster::publish(topic, <internal>::#0))
|
||||
error in <...>/cluster-publish-errors.zeek, line 65: Publish of unknown record type 'Cluster::MyEvent' (Cluster::publish_hrw(Cluster::proxy_pool, key, <internal>::#0))
|
||||
error in <...>/cluster-publish-errors.zeek, line 72: Publish of unknown record type 'Cluster::MyEvent' (Cluster::publish_rr(Cluster::proxy_pool, key, <internal>::#0))
|
||||
201
testing/btest/Baseline/cluster.generic.logging-rr/rr2.log.sorted
Normal file
201
testing/btest/Baseline/cluster.generic.logging-rr/rr2.log.sorted
Normal file
|
|
@ -0,0 +1,201 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
0 worker-1
|
||||
0 worker-2
|
||||
1 worker-1
|
||||
1 worker-2
|
||||
2 worker-1
|
||||
2 worker-2
|
||||
3 worker-1
|
||||
3 worker-2
|
||||
4 worker-1
|
||||
4 worker-2
|
||||
5 worker-1
|
||||
5 worker-2
|
||||
6 worker-1
|
||||
6 worker-2
|
||||
7 worker-1
|
||||
7 worker-2
|
||||
8 worker-1
|
||||
8 worker-2
|
||||
9 worker-1
|
||||
9 worker-2
|
||||
10 worker-1
|
||||
10 worker-2
|
||||
11 worker-1
|
||||
11 worker-2
|
||||
12 worker-1
|
||||
12 worker-2
|
||||
13 worker-1
|
||||
13 worker-2
|
||||
14 worker-1
|
||||
14 worker-2
|
||||
15 worker-1
|
||||
15 worker-2
|
||||
16 worker-1
|
||||
16 worker-2
|
||||
17 worker-1
|
||||
17 worker-2
|
||||
18 worker-1
|
||||
18 worker-2
|
||||
19 worker-1
|
||||
19 worker-2
|
||||
20 worker-1
|
||||
20 worker-2
|
||||
21 worker-1
|
||||
21 worker-2
|
||||
22 worker-1
|
||||
22 worker-2
|
||||
23 worker-1
|
||||
23 worker-2
|
||||
24 worker-1
|
||||
24 worker-2
|
||||
25 worker-1
|
||||
25 worker-2
|
||||
26 worker-1
|
||||
26 worker-2
|
||||
27 worker-1
|
||||
27 worker-2
|
||||
28 worker-1
|
||||
28 worker-2
|
||||
29 worker-1
|
||||
29 worker-2
|
||||
30 worker-1
|
||||
30 worker-2
|
||||
31 worker-1
|
||||
31 worker-2
|
||||
32 worker-1
|
||||
32 worker-2
|
||||
33 worker-1
|
||||
33 worker-2
|
||||
34 worker-1
|
||||
34 worker-2
|
||||
35 worker-1
|
||||
35 worker-2
|
||||
36 worker-1
|
||||
36 worker-2
|
||||
37 worker-1
|
||||
37 worker-2
|
||||
38 worker-1
|
||||
38 worker-2
|
||||
39 worker-1
|
||||
39 worker-2
|
||||
40 worker-1
|
||||
40 worker-2
|
||||
41 worker-1
|
||||
41 worker-2
|
||||
42 worker-1
|
||||
42 worker-2
|
||||
43 worker-1
|
||||
43 worker-2
|
||||
44 worker-1
|
||||
44 worker-2
|
||||
45 worker-1
|
||||
45 worker-2
|
||||
46 worker-1
|
||||
46 worker-2
|
||||
47 worker-1
|
||||
47 worker-2
|
||||
48 worker-1
|
||||
48 worker-2
|
||||
49 worker-1
|
||||
49 worker-2
|
||||
50 worker-1
|
||||
50 worker-2
|
||||
51 worker-1
|
||||
51 worker-2
|
||||
52 worker-1
|
||||
52 worker-2
|
||||
53 worker-1
|
||||
53 worker-2
|
||||
54 worker-1
|
||||
54 worker-2
|
||||
55 worker-1
|
||||
55 worker-2
|
||||
56 worker-1
|
||||
56 worker-2
|
||||
57 worker-1
|
||||
57 worker-2
|
||||
58 worker-1
|
||||
58 worker-2
|
||||
59 worker-1
|
||||
59 worker-2
|
||||
60 worker-1
|
||||
60 worker-2
|
||||
61 worker-1
|
||||
61 worker-2
|
||||
62 worker-1
|
||||
62 worker-2
|
||||
63 worker-1
|
||||
63 worker-2
|
||||
64 worker-1
|
||||
64 worker-2
|
||||
65 worker-1
|
||||
65 worker-2
|
||||
66 worker-1
|
||||
66 worker-2
|
||||
67 worker-1
|
||||
67 worker-2
|
||||
68 worker-1
|
||||
68 worker-2
|
||||
69 worker-1
|
||||
69 worker-2
|
||||
70 worker-1
|
||||
70 worker-2
|
||||
71 worker-1
|
||||
71 worker-2
|
||||
72 worker-1
|
||||
72 worker-2
|
||||
73 worker-1
|
||||
73 worker-2
|
||||
74 worker-1
|
||||
74 worker-2
|
||||
75 worker-1
|
||||
75 worker-2
|
||||
76 worker-1
|
||||
76 worker-2
|
||||
77 worker-1
|
||||
77 worker-2
|
||||
78 worker-1
|
||||
78 worker-2
|
||||
79 worker-1
|
||||
79 worker-2
|
||||
80 worker-1
|
||||
80 worker-2
|
||||
81 worker-1
|
||||
81 worker-2
|
||||
82 worker-1
|
||||
82 worker-2
|
||||
83 worker-1
|
||||
83 worker-2
|
||||
84 worker-1
|
||||
84 worker-2
|
||||
85 worker-1
|
||||
85 worker-2
|
||||
86 worker-1
|
||||
86 worker-2
|
||||
87 worker-1
|
||||
87 worker-2
|
||||
88 worker-1
|
||||
88 worker-2
|
||||
89 worker-1
|
||||
89 worker-2
|
||||
90 worker-1
|
||||
90 worker-2
|
||||
91 worker-1
|
||||
91 worker-2
|
||||
92 worker-1
|
||||
92 worker-2
|
||||
93 worker-1
|
||||
93 worker-2
|
||||
94 worker-1
|
||||
94 worker-2
|
||||
95 worker-1
|
||||
95 worker-2
|
||||
96 worker-1
|
||||
96 worker-2
|
||||
97 worker-1
|
||||
97 worker-2
|
||||
98 worker-1
|
||||
98 worker-2
|
||||
99 worker-1
|
||||
99 worker-2
|
||||
|
|
@ -0,0 +1,42 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
got pong, 0, args, worker-2, args
|
||||
got pong, 0, args, worker-2, make_event
|
||||
got pong, 0, make_event, worker-2, args
|
||||
got pong, 0, make_event, worker-2, make_event
|
||||
got pong, 1, args, worker-1, args
|
||||
got pong, 1, args, worker-1, make_event
|
||||
got pong, 1, make_event, worker-1, args
|
||||
got pong, 1, make_event, worker-1, make_event
|
||||
got pong, 2, args, worker-2, args
|
||||
got pong, 2, args, worker-2, make_event
|
||||
got pong, 2, make_event, worker-2, args
|
||||
got pong, 2, make_event, worker-2, make_event
|
||||
got pong, 3, args, worker-1, args
|
||||
got pong, 3, args, worker-1, make_event
|
||||
got pong, 3, make_event, worker-1, args
|
||||
got pong, 3, make_event, worker-1, make_event
|
||||
got pong, 4, args, worker-1, args
|
||||
got pong, 4, args, worker-1, make_event
|
||||
got pong, 4, make_event, worker-1, args
|
||||
got pong, 4, make_event, worker-1, make_event
|
||||
got pong, 5, args, worker-2, args
|
||||
got pong, 5, args, worker-2, make_event
|
||||
got pong, 5, make_event, worker-2, args
|
||||
got pong, 5, make_event, worker-2, make_event
|
||||
got pong, 6, args, worker-1, args
|
||||
got pong, 6, args, worker-1, make_event
|
||||
got pong, 6, make_event, worker-1, args
|
||||
got pong, 6, make_event, worker-1, make_event
|
||||
got pong, 7, args, worker-2, args
|
||||
got pong, 7, args, worker-2, make_event
|
||||
got pong, 7, make_event, worker-2, args
|
||||
got pong, 7, make_event, worker-2, make_event
|
||||
got pong, 8, args, worker-1, args
|
||||
got pong, 8, args, worker-1, make_event
|
||||
got pong, 8, make_event, worker-1, args
|
||||
got pong, 8, make_event, worker-1, make_event
|
||||
got pong, 9, args, worker-1, args
|
||||
got pong, 9, args, worker-1, make_event
|
||||
got pong, 9, make_event, worker-1, args
|
||||
got pong, 9, make_event, worker-1, make_event
|
||||
have 40, finish!
|
||||
|
|
@ -0,0 +1,16 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
a node_up, manager
|
||||
a node_up, worker-2
|
||||
got ping, 1, args
|
||||
got ping, 1, make_event
|
||||
got ping, 3, args
|
||||
got ping, 3, make_event
|
||||
got ping, 4, args
|
||||
got ping, 4, make_event
|
||||
got ping, 6, args
|
||||
got ping, 6, make_event
|
||||
got ping, 8, args
|
||||
got ping, 8, make_event
|
||||
got ping, 9, args
|
||||
got ping, 9, make_event
|
||||
z got finish!
|
||||
|
|
@ -0,0 +1,12 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
a node_up, manager
|
||||
a node_up, worker-1
|
||||
got ping, 0, args
|
||||
got ping, 0, make_event
|
||||
got ping, 2, args
|
||||
got ping, 2, make_event
|
||||
got ping, 5, args
|
||||
got ping, 5, make_event
|
||||
got ping, 7, args
|
||||
got ping, 7, make_event
|
||||
z got finish!
|
||||
|
|
@ -0,0 +1,42 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
got pong, 0, args, worker-1, args
|
||||
got pong, 0, args, worker-1, make_event
|
||||
got pong, 0, make_event, worker-1, args
|
||||
got pong, 0, make_event, worker-1, make_event
|
||||
got pong, 1, args, worker-2, args
|
||||
got pong, 1, args, worker-2, make_event
|
||||
got pong, 1, make_event, worker-2, args
|
||||
got pong, 1, make_event, worker-2, make_event
|
||||
got pong, 2, args, worker-1, args
|
||||
got pong, 2, args, worker-1, make_event
|
||||
got pong, 2, make_event, worker-1, args
|
||||
got pong, 2, make_event, worker-1, make_event
|
||||
got pong, 3, args, worker-2, args
|
||||
got pong, 3, args, worker-2, make_event
|
||||
got pong, 3, make_event, worker-2, args
|
||||
got pong, 3, make_event, worker-2, make_event
|
||||
got pong, 4, args, worker-1, args
|
||||
got pong, 4, args, worker-1, make_event
|
||||
got pong, 4, make_event, worker-1, args
|
||||
got pong, 4, make_event, worker-1, make_event
|
||||
got pong, 5, args, worker-2, args
|
||||
got pong, 5, args, worker-2, make_event
|
||||
got pong, 5, make_event, worker-2, args
|
||||
got pong, 5, make_event, worker-2, make_event
|
||||
got pong, 6, args, worker-1, args
|
||||
got pong, 6, args, worker-1, make_event
|
||||
got pong, 6, make_event, worker-1, args
|
||||
got pong, 6, make_event, worker-1, make_event
|
||||
got pong, 7, args, worker-2, args
|
||||
got pong, 7, args, worker-2, make_event
|
||||
got pong, 7, make_event, worker-2, args
|
||||
got pong, 7, make_event, worker-2, make_event
|
||||
got pong, 8, args, worker-1, args
|
||||
got pong, 8, args, worker-1, make_event
|
||||
got pong, 8, make_event, worker-1, args
|
||||
got pong, 8, make_event, worker-1, make_event
|
||||
got pong, 9, args, worker-2, args
|
||||
got pong, 9, args, worker-2, make_event
|
||||
got pong, 9, make_event, worker-2, args
|
||||
got pong, 9, make_event, worker-2, make_event
|
||||
have 40, finish!
|
||||
|
|
@ -0,0 +1,14 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
a node_up, manager
|
||||
a node_up, worker-2
|
||||
got ping, 0, args
|
||||
got ping, 0, make_event
|
||||
got ping, 2, args
|
||||
got ping, 2, make_event
|
||||
got ping, 4, args
|
||||
got ping, 4, make_event
|
||||
got ping, 6, args
|
||||
got ping, 6, make_event
|
||||
got ping, 8, args
|
||||
got ping, 8, make_event
|
||||
z got finish!
|
||||
|
|
@ -0,0 +1,14 @@
|
|||
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
|
||||
a node_up, manager
|
||||
a node_up, worker-1
|
||||
got ping, 1, args
|
||||
got ping, 1, make_event
|
||||
got ping, 3, args
|
||||
got ping, 3, make_event
|
||||
got ping, 5, args
|
||||
got ping, 5, make_event
|
||||
got ping, 7, args
|
||||
got ping, 7, make_event
|
||||
got ping, 9, args
|
||||
got ping, 9, make_event
|
||||
z got finish!
|
||||
177
testing/btest/cluster/generic/logging-rr.zeek
Normal file
177
testing/btest/cluster/generic/logging-rr.zeek
Normal file
|
|
@ -0,0 +1,177 @@
|
|||
# @TEST-DOC: Testing round-robin of Log::write() across two loggers.
|
||||
#
|
||||
# @TEST-REQUIRES: have-zeromq
|
||||
#
|
||||
# @TEST-PORT: XPUB_PORT
|
||||
# @TEST-PORT: XSUB_PORT
|
||||
# @TEST-PORT: LOG_PULL_PORT_1
|
||||
# @TEST-PORT: LOG_PULL_PORT_2
|
||||
#
|
||||
# @TEST-EXEC: chmod +x ./check-log.sh
|
||||
#
|
||||
# @TEST-EXEC: cp $FILES/zeromq/cluster-layout-two-loggers.zeek cluster-layout.zeek
|
||||
# @TEST-EXEC: cp $FILES/zeromq/test-bootstrap.zeek zeromq-test-bootstrap.zeek
|
||||
#
|
||||
# @TEST-EXEC: zeek -b --parse-only common.zeek manager.zeek worker.zeek
|
||||
#
|
||||
# @TEST-EXEC: btest-bg-run manager "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=manager zeek -b ../manager.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run logger-1 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=logger-1 zeek -b ../common.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run logger-2 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=logger-2 zeek -b ../common.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run worker-1 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=worker-1 zeek -b ../worker.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run worker-2 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=worker-2 zeek -b ../worker.zeek >out"
|
||||
#
|
||||
# @TEST-EXEC: btest-bg-wait 10
|
||||
#
|
||||
# @TEST-EXEC: test $(grep -c worker-1 logger-1/rr2.log) -gt 10
|
||||
# @TEST-EXEC: test $(grep -c worker-2 logger-1/rr2.log) -gt 10
|
||||
# @TEST-EXEC: test $(grep -c worker-1 logger-2/rr2.log) -gt 10
|
||||
# @TEST-EXEC: test $(grep -c worker-2 logger-2/rr2.log) -gt 10
|
||||
|
||||
# @TEST-EXEC: zeek-cut < logger-1/rr2.log > rr2.log
|
||||
# @TEST-EXEC: zeek-cut < logger-2/rr2.log >> rr2.log
|
||||
# @TEST-EXEC: sort -n rr2.log > rr2.log.sorted
|
||||
# @TEST-EXEC: btest-diff rr2.log.sorted
|
||||
|
||||
# @TEST-START-FILE common.zeek
|
||||
@load ./zeromq-test-bootstrap.zeek
|
||||
|
||||
redef Log::default_rotation_interval = 0sec;
|
||||
redef Log::flush_interval = 0.03sec;
|
||||
redef Log::write_buffer_size = 7;
|
||||
|
||||
module LogRR;
|
||||
|
||||
export {
|
||||
redef enum Log::ID += { LOG1, LOG2 };
|
||||
type Info: record {
|
||||
c: count &log;
|
||||
from: string &log &default=Cluster::node;
|
||||
};
|
||||
|
||||
global go: event();
|
||||
global finish: event();
|
||||
}
|
||||
|
||||
event zeek_init()
|
||||
{
|
||||
Log::create_stream(LOG1, [$columns=Info, $path="rr1"]);
|
||||
Log::create_stream(LOG2, [$columns=Info, $path="rr2"]);
|
||||
}
|
||||
|
||||
event finish()
|
||||
{
|
||||
terminate();
|
||||
}
|
||||
# @TEST-END-FILE
|
||||
|
||||
# @TEST-START-FILE manager.zeek
|
||||
@load ./common.zeek
|
||||
|
||||
event check_ready()
|
||||
{
|
||||
if ( ! piped_exec("../check-log.sh", "") )
|
||||
{
|
||||
Reporter::error("check-log.sh failed");
|
||||
terminate();
|
||||
}
|
||||
|
||||
if ( file_size("DONE") >= 0 )
|
||||
{
|
||||
Cluster::publish(Cluster::worker_topic, LogRR::go);
|
||||
return;
|
||||
}
|
||||
|
||||
schedule 0.1sec { check_ready() };
|
||||
}
|
||||
|
||||
event zeek_init()
|
||||
{
|
||||
event check_ready();
|
||||
}
|
||||
|
||||
|
||||
global nodes_down: set[string];
|
||||
|
||||
event Cluster::node_down(name: string, id: string)
|
||||
{
|
||||
print current_time(), "node_down", name;
|
||||
add nodes_down[name];
|
||||
|
||||
if ( |nodes_down| == 2 ) # workers down
|
||||
Cluster::publish(Cluster::logger_topic, LogRR::finish);
|
||||
|
||||
if ( |nodes_down| == 4 ) # both loggers down
|
||||
terminate();
|
||||
}
|
||||
# @TEST-END-FILE
|
||||
|
||||
|
||||
# @TEST-START-FILE worker.zeek
|
||||
@load ./common.zeek
|
||||
|
||||
global do_write2 = F;
|
||||
|
||||
event write_log1(c: count)
|
||||
{
|
||||
if ( do_write2 )
|
||||
{
|
||||
Log::write(LogRR::LOG1, [$c=10000000]);
|
||||
return;
|
||||
}
|
||||
|
||||
Log::write(LogRR::LOG1, [$c=c]);
|
||||
Log::flush(LogRR::LOG1);
|
||||
schedule 0.05sec { write_log1(++c) };
|
||||
}
|
||||
|
||||
event write_log2(c: count)
|
||||
{
|
||||
if ( c == 100 )
|
||||
{
|
||||
terminate();
|
||||
return;
|
||||
}
|
||||
|
||||
Log::write(LogRR::LOG2, [$c=c]);
|
||||
schedule 0.012sec { write_log2(++c) };
|
||||
}
|
||||
|
||||
event LogRR::go()
|
||||
{
|
||||
do_write2 = T;
|
||||
event write_log2(0);
|
||||
}
|
||||
|
||||
event zeek_init()
|
||||
{
|
||||
event write_log1(0);
|
||||
}
|
||||
|
||||
# @TEST-END-FILE
|
||||
|
||||
@TEST-START-FILE check-log.sh
|
||||
#!/usr/bin/env bash
|
||||
#
|
||||
# This script regularly checks for the loggers rr1.log file until
|
||||
# both workers appear. Once this happens, creates a READY file
|
||||
# which will result in workers getting the "go" and sending writes
|
||||
# to rr2.log
|
||||
set -eux
|
||||
|
||||
LOGGERS="logger-1 logger-2"
|
||||
WORKERS="worker-1 worker-2"
|
||||
|
||||
for logger in $LOGGERS; do
|
||||
for worker in $WORKERS; do
|
||||
date +%s
|
||||
echo check $logger $worker
|
||||
if ! grep -q "${worker}" ../${logger}/rr1.log; then
|
||||
exit 0
|
||||
fi
|
||||
done
|
||||
done
|
||||
|
||||
echo "DONE"
|
||||
echo "DONE" > DONE
|
||||
exit 0
|
||||
@TEST-END-FILE
|
||||
101
testing/btest/cluster/generic/publish-hrw.zeek
Normal file
101
testing/btest/cluster/generic/publish-hrw.zeek
Normal file
|
|
@ -0,0 +1,101 @@
|
|||
# @TEST-DOC: Send ping/pong using publish_hrw(), publish() and make_event()
|
||||
#
|
||||
# @TEST-REQUIRES: have-zeromq
|
||||
#
|
||||
# @TEST-PORT: XPUB_PORT
|
||||
# @TEST-PORT: XSUB_PORT
|
||||
# @TEST-PORT: LOG_PULL_PORT
|
||||
#
|
||||
# @TEST-EXEC: cp $FILES/zeromq/cluster-layout-no-logger.zeek cluster-layout.zeek
|
||||
# @TEST-EXEC: cp $FILES/zeromq/test-bootstrap.zeek zeromq-test-bootstrap.zeek
|
||||
#
|
||||
# @TEST-EXEC: zeek -b --parse-only common.zeek manager.zeek worker.zeek
|
||||
#
|
||||
# @TEST-EXEC: btest-bg-run manager "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=manager zeek -b ../manager.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run worker-1 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=worker-1 zeek -b ../worker.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run worker-2 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=worker-2 zeek -b ../worker.zeek >out"
|
||||
#
|
||||
# @TEST-EXEC: btest-bg-wait 10
|
||||
# @TEST-EXEC: sort < ./manager/out > ./manager.sorted
|
||||
# @TEST-EXEC: sort < ./worker-1/out > ./worker-1.sorted
|
||||
# @TEST-EXEC: sort < ./worker-2/out > ./worker-2.sorted
|
||||
# @TEST-EXEC: btest-diff manager.sorted
|
||||
# @TEST-EXEC: btest-diff worker-1.sorted
|
||||
# @TEST-EXEC: btest-diff worker-2.sorted
|
||||
|
||||
# @TEST-START-FILE common.zeek
|
||||
@load ./zeromq-test-bootstrap.zeek
|
||||
|
||||
global finish: event();
|
||||
global ping: event(c: count, how: string);
|
||||
global pong: event(c: count, how: string, from: string, from_how: string);
|
||||
# @TEST-END-FILE
|
||||
|
||||
# @TEST-START-FILE manager.zeek
|
||||
@load ./common.zeek
|
||||
|
||||
global nodes_up: set[string];
|
||||
global nodes_down: set[string];
|
||||
global pongs: set[count, string, string, string];
|
||||
|
||||
global i = 0;
|
||||
|
||||
event send_hrw()
|
||||
{
|
||||
if (i >= 10 )
|
||||
return;
|
||||
|
||||
Cluster::publish_hrw(Cluster::worker_pool, cat(i), ping, i, "args");
|
||||
local e = Cluster::make_event(ping, i, "make_event");
|
||||
Cluster::publish_hrw(Cluster::worker_pool, cat(i), e);
|
||||
++i;
|
||||
|
||||
schedule 0.01sec { send_hrw() };
|
||||
}
|
||||
|
||||
event pong(c: count, how: string, from: string, from_how: string)
|
||||
{
|
||||
print "got pong", c, how, from, from_how;
|
||||
add pongs[c, how, from, from_how];
|
||||
|
||||
if ( |pongs| == 40 )
|
||||
{
|
||||
print "have 40, finish!";
|
||||
Cluster::publish(Cluster::worker_topic, finish);
|
||||
}
|
||||
}
|
||||
|
||||
event Cluster::node_up(name: string, id: string) {
|
||||
add nodes_up[name];
|
||||
if ( |nodes_up| == 2 ) {
|
||||
event send_hrw();
|
||||
}
|
||||
}
|
||||
|
||||
event Cluster::node_down(name: string, id: string) {
|
||||
add nodes_down[name];
|
||||
if ( |nodes_down| == 2 )
|
||||
terminate();
|
||||
}
|
||||
# @TEST-END-FILE
|
||||
|
||||
|
||||
# @TEST-START-FILE worker.zeek
|
||||
@load ./common.zeek
|
||||
|
||||
event ping(c: count, how: string) {
|
||||
print "got ping", c, how;
|
||||
Cluster::publish(Cluster::manager_topic, pong, c, how, Cluster::node, "args");
|
||||
local e = Cluster::make_event(pong, c, how, Cluster::node, "make_event");
|
||||
Cluster::publish(Cluster::manager_topic, e);
|
||||
}
|
||||
|
||||
event Cluster::node_up(name: string, id: string) {
|
||||
print "a node_up", name;
|
||||
}
|
||||
|
||||
event finish() &is_used {
|
||||
print "z got finish!";
|
||||
terminate();
|
||||
}
|
||||
# @TEST-END-FILE
|
||||
101
testing/btest/cluster/generic/publish-rr.zeek
Normal file
101
testing/btest/cluster/generic/publish-rr.zeek
Normal file
|
|
@ -0,0 +1,101 @@
|
|||
# @TEST-DOC: Send ping/pong using publish_rr(), publish() and make_event()
|
||||
#
|
||||
# @TEST-REQUIRES: have-zeromq
|
||||
#
|
||||
# @TEST-PORT: XPUB_PORT
|
||||
# @TEST-PORT: XSUB_PORT
|
||||
# @TEST-PORT: LOG_PULL_PORT
|
||||
#
|
||||
# @TEST-EXEC: cp $FILES/zeromq/cluster-layout-no-logger.zeek cluster-layout.zeek
|
||||
# @TEST-EXEC: cp $FILES/zeromq/test-bootstrap.zeek zeromq-test-bootstrap.zeek
|
||||
#
|
||||
# @TEST-EXEC: zeek -b --parse-only common.zeek manager.zeek worker.zeek
|
||||
#
|
||||
# @TEST-EXEC: btest-bg-run manager "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=manager zeek -b ../manager.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run worker-1 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=worker-1 zeek -b ../worker.zeek >out"
|
||||
# @TEST-EXEC: btest-bg-run worker-2 "ZEEKPATH=$ZEEKPATH:.. && CLUSTER_NODE=worker-2 zeek -b ../worker.zeek >out"
|
||||
#
|
||||
# @TEST-EXEC: btest-bg-wait 10
|
||||
# @TEST-EXEC: sort < ./manager/out > ./manager.sorted
|
||||
# @TEST-EXEC: sort < ./worker-1/out > ./worker-1.sorted
|
||||
# @TEST-EXEC: sort < ./worker-2/out > ./worker-2.sorted
|
||||
# @TEST-EXEC: btest-diff manager.sorted
|
||||
# @TEST-EXEC: btest-diff worker-1.sorted
|
||||
# @TEST-EXEC: btest-diff worker-2.sorted
|
||||
|
||||
# @TEST-START-FILE common.zeek
|
||||
@load ./zeromq-test-bootstrap.zeek
|
||||
|
||||
global finish: event();
|
||||
global ping: event(c: count, how: string);
|
||||
global pong: event(c: count, how: string, from: string, from_how: string);
|
||||
# @TEST-END-FILE
|
||||
|
||||
# @TEST-START-FILE manager.zeek
|
||||
@load ./common.zeek
|
||||
|
||||
global nodes_up: set[string];
|
||||
global nodes_down: set[string];
|
||||
global pongs: set[count, string, string, string];
|
||||
|
||||
global i = 0;
|
||||
|
||||
event send_rr()
|
||||
{
|
||||
if (i >= 10 )
|
||||
return;
|
||||
|
||||
Cluster::publish_rr(Cluster::worker_pool, "ping-key-args", ping, i, "args");
|
||||
local e = Cluster::make_event(ping, i, "make_event");
|
||||
Cluster::publish_rr(Cluster::worker_pool, "ping-key-event", e);
|
||||
++i;
|
||||
|
||||
schedule 0.01sec { send_rr() };
|
||||
}
|
||||
|
||||
event pong(c: count, how: string, from: string, from_how: string)
|
||||
{
|
||||
print "got pong", c, how, from, from_how;
|
||||
add pongs[c, how, from, from_how];
|
||||
|
||||
if ( |pongs| == 40 )
|
||||
{
|
||||
print "have 40, finish!";
|
||||
Cluster::publish(Cluster::worker_topic, finish);
|
||||
}
|
||||
}
|
||||
|
||||
event Cluster::node_up(name: string, id: string) {
|
||||
add nodes_up[name];
|
||||
if ( |nodes_up| == 2 ) {
|
||||
event send_rr();
|
||||
}
|
||||
}
|
||||
|
||||
event Cluster::node_down(name: string, id: string) {
|
||||
add nodes_down[name];
|
||||
if ( |nodes_down| == 2 )
|
||||
terminate();
|
||||
}
|
||||
# @TEST-END-FILE
|
||||
|
||||
|
||||
# @TEST-START-FILE worker.zeek
|
||||
@load ./common.zeek
|
||||
|
||||
event ping(c: count, how: string) {
|
||||
print "got ping", c, how;
|
||||
Cluster::publish(Cluster::manager_topic, pong, c, how, Cluster::node, "args");
|
||||
local e = Cluster::make_event(pong, c, how, Cluster::node, "make_event");
|
||||
Cluster::publish(Cluster::manager_topic, e);
|
||||
}
|
||||
|
||||
event Cluster::node_up(name: string, id: string) {
|
||||
print "a node_up", name;
|
||||
}
|
||||
|
||||
event finish() &is_used {
|
||||
print "z got finish!";
|
||||
terminate();
|
||||
}
|
||||
# @TEST-END-FILE
|
||||
Loading…
Add table
Add a link
Reference in a new issue