129 lines
3.1 KiB
Ruby
129 lines
3.1 KiB
Ruby
|
require 'em_test_helper'
|
||
|
|
||
|
class TestHttpClient2 < Test::Unit::TestCase
|
||
|
class TestServer < EM::Connection
|
||
|
end
|
||
|
|
||
|
def setup
|
||
|
@port = next_port
|
||
|
end
|
||
|
|
||
|
# #connect returns an object which has made a connection to an HTTP server
|
||
|
# and exposes methods for making HTTP requests on that connection.
|
||
|
# #connect can take either a pair of parameters (a host and a port),
|
||
|
# or a single parameter which is a Hash.
|
||
|
#
|
||
|
def test_connect
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
EM.start_server '127.0.0.1', @port, TestServer
|
||
|
silent do
|
||
|
EM::P::HttpClient2.connect '127.0.0.1', @port
|
||
|
EM::P::HttpClient2.connect( :host=>'127.0.0.1', :port=>@port )
|
||
|
end
|
||
|
EM.stop
|
||
|
}
|
||
|
end
|
||
|
|
||
|
def test_bad_port
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
EM.start_server '127.0.0.1', @port, TestServer
|
||
|
assert_raises( ArgumentError ) {
|
||
|
silent { EM::P::HttpClient2.connect '127.0.0.1', "xxx" }
|
||
|
}
|
||
|
EM.stop
|
||
|
}
|
||
|
end
|
||
|
|
||
|
def test_bad_server
|
||
|
err = nil
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
http = silent { EM::P::HttpClient2.connect '127.0.0.1', 9999 }
|
||
|
d = http.get "/"
|
||
|
d.errback { err = true; d.internal_error; EM.stop }
|
||
|
}
|
||
|
assert(err)
|
||
|
end
|
||
|
|
||
|
def test_get
|
||
|
content = nil
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
http = silent { EM::P::HttpClient2.connect :host => "google.com", :port => 80, :version => '1.0' }
|
||
|
d = http.get "/"
|
||
|
d.callback {
|
||
|
content = d.content
|
||
|
EM.stop
|
||
|
}
|
||
|
}
|
||
|
assert(content)
|
||
|
end
|
||
|
|
||
|
# Not a pipelined request because we wait for one response before we request the next.
|
||
|
# XXX this test is broken because it sends the second request to the first connection
|
||
|
# XXX right before the connection closes
|
||
|
def _test_get_multiple
|
||
|
content = nil
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
http = silent { EM::P::HttpClient2.connect "google.com", :version => '1.0' }
|
||
|
d = http.get "/"
|
||
|
d.callback {
|
||
|
e = http.get "/"
|
||
|
e.callback {
|
||
|
content = e.content
|
||
|
EM.stop
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
assert(content)
|
||
|
end
|
||
|
|
||
|
def test_get_pipeline
|
||
|
headers, headers2 = nil, nil
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
http = silent { EM::P::HttpClient2.connect "google.com", 80 }
|
||
|
d = http.get("/")
|
||
|
d.callback {
|
||
|
headers = d.headers
|
||
|
}
|
||
|
e = http.get("/")
|
||
|
e.callback {
|
||
|
headers2 = e.headers
|
||
|
}
|
||
|
EM.tick_loop { EM.stop if headers && headers2 }
|
||
|
EM.add_timer(1) { EM.stop }
|
||
|
}
|
||
|
assert(headers)
|
||
|
assert(headers2)
|
||
|
end
|
||
|
|
||
|
def test_authheader
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
EM.start_server '127.0.0.1', @port, TestServer
|
||
|
http = silent { EM::P::HttpClient2.connect '127.0.0.1', 18842 }
|
||
|
d = http.get :url=>"/", :authorization=>"Basic xxx"
|
||
|
d.callback {EM.stop}
|
||
|
d.errback {EM.stop}
|
||
|
}
|
||
|
end
|
||
|
|
||
|
def test_https_get
|
||
|
omit_unless(EM.ssl?)
|
||
|
d = nil
|
||
|
EM.run {
|
||
|
setup_timeout(1)
|
||
|
http = silent { EM::P::HttpClient2.connect :host => 'www.google.com', :port => 443, :ssl => true, :version => '1.0' }
|
||
|
d = http.get "/"
|
||
|
d.callback {EM.stop}
|
||
|
d.errback {EM.stop}
|
||
|
}
|
||
|
assert_equal(200, d.status)
|
||
|
end
|
||
|
|
||
|
end
|