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
82
83
84
|
#
# This file is part of IO-Socket-Timeout
#
# This software is copyright (c) 2013 by Damien "dams" Krotkine.
#
# This is free software; you can redistribute it and/or modify it under
# the same terms as the Perl 5 programming language system itself.
#
package TestTimeout;
use strict;
use warnings;
use Test::More;
use IO::Socket::Timeout;
use Test::TCP;
use POSIX qw(ETIMEDOUT ECONNRESET strerror);
use Exporter 'import';
require bytes;
sub create_server_with_timeout {
my ($connection_delay, $read_delay, $write_delay) = @_;
# Warning:
# $read_delay and $write_delay are seen from the *client* point of view
Test::TCP->new(
code => sub {
my $port = shift;
my $socket = IO::Socket::INET->new(
Listen => 5,
Reuse => 1,
Blocking => 1,
LocalPort => $port
) or die "ops $!";
my $buffer;
while (1) {
# First, establish connection
my $client = $socket->accept();
$client or next;
# Then get data (with delay)
if ( defined (my $message = <$client>) ) {
my $response = "S" . $message;
print $client $response;
$message = <$client>;
$response = "S" . $message;
sleep($read_delay);
print $client $response;
}
$client->close();
}
},
);
}
sub test {
my $class = shift;
my %p = @_;
my $server = create_server_with_timeout( $p{connection_delay},
$p{read_delay},
$p{write_delay},
);
my $client = IO::Socket::INET->new(
PeerHost => '127.0.0.1',
PeerPort => $server->port,
$p{connection_timeout} ? (Timeout => $p{connection_timeout} ) : (),
);
if (! $p{no_timeouts} ) {
IO::Socket::Timeout->enable_timeouts_on($client);
$p{read_timeout} and $client->read_timeout($p{read_timeout});
$p{write_timeout} and $client->write_timeout($p{write_timeout});
}
my $etimeout = strerror(ETIMEDOUT);
my $ereset = strerror(ECONNRESET);
$p{callback}->($client, $etimeout, $ereset);
}
1;
|
