1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
use strict;
use warnings FATAL => 'all';
use Apache::Test;
use Apache::TestRequest;
use Apache::TestUtil;
use Apache::TestConfig ();
# not reliable, hangs for some people:
# my @test_cases = ( "ping0", "ping1" x 10, "ping2" x 100, "ping3" x 1024, "ping4" x 4096, "sendquit");
my @test_cases = ( "ping0", "ping1" x 10, "ping2" x 100, "ping3" x 1024, "sendquit");
my $total_tests = 2;
plan tests => $total_tests, need 'AnyEvent::WebSocket::Client',
need_module('proxy_http', 'lua'), need_min_apache_version('2.4.47');
require AnyEvent;
require AnyEvent::WebSocket::Client;
my $config = Apache::Test::config();
my $hostport = Apache::TestRequest::hostport();
my $client = AnyEvent::WebSocket::Client->new(timeout => 5);
my $quit_program = AnyEvent->condvar;
my $responses = 0;
my $surprised = 0;
$client->connect("ws://$hostport/proxy/wsoc")->cb(sub {
our $connection = eval { shift->recv };
t_debug("wsoc connected");
if($@) {
# handle error...
warn $@;
$quit_program->send();
return;
}
# AnyEvent::WebSocket::Connection does not pass the PONG message down to the callback
# my $actualpingmsg = AnyEvent::WebSocket::Message->new(opcode => 0x09, body => "xxx");
# $connection->send($actualpingmsg);
foreach (@test_cases){
$connection->send($_);
}
$connection->on(finish => sub {
t_debug("finish");
});
# recieve message from the websocket...
$connection->on(each_message => sub {
# $connection is the same connection object
# $message isa AnyEvent::WebSocket::Message
my($connection, $message) = @_;
$responses++;
t_debug("wsoc msg received: " . substr($message->body, 0, 5). " opcode " . $message->opcode);
if ("sendquit" eq $message->body) {
$connection->send('quit');
t_debug("closing");
$connection->close; # doesn't seem to close TCP.
$quit_program->send();
}
elsif ($message->body =~ /^ping(\d)/) {
my $offset = $1;
if ($message->body ne $test_cases[$offset]) {
$surprised++;
}
}
else {
$surprised++;
}
});
});
$quit_program->recv;
ok t_cmp($surprised, 0);
ok t_cmp($responses, scalar(@test_cases) );
|
