Mercurial > hg > nginx-tests
view lib/Test/Nginx/SMTP.pm @ 40:3325fa3f515c
Tests: avoid using \r\n.
author | Maxim Dounin <mdounin@mdounin.ru> |
---|---|
date | Sat, 01 Nov 2008 16:13:02 +0300 |
parents | 0880e0fafde4 |
children | daa295331acd |
line wrap: on
line source
package Test::Nginx::SMTP; # (C) Maxim Dounin # Module for nginx smtp tests. ############################################################################### use warnings; use strict; use Test::More qw//; use IO::Socket; use Socket qw/ CRLF /; use Test::Nginx; use base qw/ IO::Socket::INET /; sub new { my $class = shift; my $self = return $class->SUPER::new( Proto => "tcp", PeerAddr => "localhost", PeerPort => 8025, @_ ) or die "Can't connect to nginx: $!\n"; $self->autoflush(1); return $self; } sub send { my ($self, $cmd) = @_; log_out($cmd); $self->print($cmd . CRLF); } sub read { my ($self) = @_; eval { alarm(2); local $SIG{ALRM} = sub { die "alarm\n" }; while (<$self>) { log_in($_); next if m/^\d\d\d-/; last; } alarm(0); }; alarm(0); if ($@) { return undef; } return $_; } sub check { my ($self, $regex, $name) = @_; Test::More::like($self->read(), $regex, $name); } sub ok { my $self = shift; $self->check(qr/^2\d\d /, @_); } ############################################################################### sub smtp_test_daemon { my $server = IO::Socket::INET->new( Proto => 'tcp', LocalPort => 8026, Listen => 5, Reuse => 1 ) or die "Can't create listening socket: $!\n"; while (my $client = $server->accept()) { $client->autoflush(1); print $client "220 fake esmtp server ready" . CRLF; while (<$client>) { if (/^quit/i) { print $client '221 quit ok' . CRLF; } elsif (/^(ehlo|helo)/i) { print $client '250 hello ok' . CRLF; } elsif (/^rset/i) { print $client '250 rset ok' . CRLF; } elsif (/^mail from:[^@]+$/i) { print $client '500 mail from error' . CRLF; } elsif (/^mail from:/i) { print $client '250 mail from ok' . CRLF; } elsif (/^rcpt to:[^@]+$/i) { print $client '500 rcpt to error' . CRLF; } elsif (/^rcpt to:/i) { print $client '250 rcpt to ok' . CRLF; } else { print $client "500 unknown command" . CRLF; } } close $client; } } ############################################################################### 1; ###############################################################################