mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
0827a7e52b
CR or LF is included in a line, because they are not allowed in RFC5321. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@55324 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
102 lines
2.4 KiB
Ruby
102 lines
2.4 KiB
Ruby
# frozen_string_literal: false
|
|
require 'net/smtp'
|
|
require 'stringio'
|
|
require 'test/unit'
|
|
|
|
module Net
|
|
class TestSMTP < Test::Unit::TestCase
|
|
class FakeSocket
|
|
attr_reader :write_io
|
|
|
|
def initialize out = "250 OK\n"
|
|
@write_io = StringIO.new
|
|
@read_io = StringIO.new out
|
|
end
|
|
|
|
def writeline line
|
|
@write_io.write "#{line}\r\n"
|
|
end
|
|
|
|
def readline
|
|
line = @read_io.gets
|
|
raise 'ran out of input' unless line
|
|
line.chop
|
|
end
|
|
end
|
|
|
|
def test_critical
|
|
smtp = Net::SMTP.new 'localhost', 25
|
|
|
|
assert_raise RuntimeError do
|
|
smtp.send :critical do
|
|
raise 'fail on purpose'
|
|
end
|
|
end
|
|
|
|
assert_kind_of Net::SMTP::Response, smtp.send(:critical),
|
|
'[Bug #9125]'
|
|
end
|
|
|
|
def test_esmtp
|
|
smtp = Net::SMTP.new 'localhost', 25
|
|
assert smtp.esmtp
|
|
assert smtp.esmtp?
|
|
|
|
smtp.esmtp = 'omg'
|
|
assert_equal 'omg', smtp.esmtp
|
|
assert_equal 'omg', smtp.esmtp?
|
|
end
|
|
|
|
def test_rset
|
|
smtp = Net::SMTP.new 'localhost', 25
|
|
smtp.instance_variable_set :@socket, FakeSocket.new
|
|
|
|
assert smtp.rset
|
|
end
|
|
|
|
def test_mailfrom
|
|
sock = FakeSocket.new
|
|
smtp = Net::SMTP.new 'localhost', 25
|
|
smtp.instance_variable_set :@socket, sock
|
|
assert smtp.mailfrom("foo@example.com").success?
|
|
assert_equal "MAIL FROM:<foo@example.com>\r\n", sock.write_io.string
|
|
end
|
|
|
|
def test_rcptto
|
|
sock = FakeSocket.new
|
|
smtp = Net::SMTP.new 'localhost', 25
|
|
smtp.instance_variable_set :@socket, sock
|
|
assert smtp.rcptto("foo@example.com").success?
|
|
assert_equal "RCPT TO:<foo@example.com>\r\n", sock.write_io.string
|
|
end
|
|
|
|
def test_auth_plain
|
|
sock = FakeSocket.new
|
|
smtp = Net::SMTP.new 'localhost', 25
|
|
smtp.instance_variable_set :@socket, sock
|
|
assert smtp.auth_plain("foo", "bar").success?
|
|
assert_equal "AUTH PLAIN AGZvbwBiYXI=\r\n", sock.write_io.string
|
|
end
|
|
|
|
def test_crlf_injection
|
|
smtp = Net::SMTP.new 'localhost', 25
|
|
smtp.instance_variable_set :@socket, FakeSocket.new
|
|
|
|
assert_raise(ArgumentError) do
|
|
smtp.mailfrom("foo\r\nbar")
|
|
end
|
|
|
|
assert_raise(ArgumentError) do
|
|
smtp.mailfrom("foo\rbar")
|
|
end
|
|
|
|
assert_raise(ArgumentError) do
|
|
smtp.mailfrom("foo\nbar")
|
|
end
|
|
|
|
assert_raise(ArgumentError) do
|
|
smtp.rcptto("foo\r\nbar")
|
|
end
|
|
end
|
|
end
|
|
end
|