File Coverage

blib/lib/PocketIO/Test.pm

Criterion	Covered	Total	%
statement	42	45	93.3
branch	3	4	75.0
condition			n/a
subroutine	10	11	90.9
pod	2	2	100.0
total	57	62	91.9

package PocketIO::Test;

use strict;
use warnings;

use AnyEvent;
use AnyEvent::Socket;
use Test::TCP;
use Plack::Loader;

use parent qw(Exporter);
our @EXPORT = qw(test_pocketio http_get_session_id);

sub test_pocketio {
    my ($app, $client) = @_;

    test_tcp(
        client => $client,
        server => sub {
            my $port   = shift;
            my $server = Plack::Loader->load(
                'Twiggy',
                port => $port,
                host => ('127.0.0.1')
            );
            $server->run($app);
        },
    );
}

sub http_get_session_id {
    my $server = shift;
    my $port   = shift;

    my $cv = AnyEvent->condvar;

    my $session_id;

    $cv->begin;
    tcp_connect $server, $port, sub {
        my ($fh) = @_ or return $cv->send;

        syswrite $fh, <<"EOF";
GET /socket.io/1/ HTTP/1.1
Host: $server:$port

EOF

        my $buffer = '';

        my $read_watcher;
        $read_watcher = AnyEvent->io(
            fh   => $fh,
            poll => "r",
            cb   => sub {
                my $len = sysread $fh, my $chunk, 1024, 0;

                $buffer .= $chunk;

                if ($len <= 0) {
                    ($session_id) = $buffer =~ m/\r?\n(\d+):/;

                    undef $read_watcher;
                    return $cv->end;
                }
            }
        );
    };

    $cv->wait;

    return $session_id;
}

1;
__END__