88 lines
2.2 KiB
C++
88 lines
2.2 KiB
C++
|
#include "test.hpp"
|
||
|
|
#include "libtorrent/socket.hpp"
|
||
|
|
#include "libtorrent/connection_queue.hpp"
|
||
|
|
#include "libtorrent/http_connection.hpp"
|
||
|
|
|
||
|
|
using namespace libtorrent;
|
||
|
|
|
||
|
|
io_service ios;
|
||
|
|
connection_queue cq(ios);
|
||
|
|
|
||
|
|
int connect_handler_called = 0;
|
||
|
|
int handler_called = 0;
|
||
|
|
int data_size = 0;
|
||
|
|
int http_status = 0;
|
||
|
|
asio::error_code error_code;
|
||
|
|
|
||
|
|
void print_http_header(http_parser const& p)
|
||
|
|
{
|
||
|
|
std::cerr << p.status_code() << " " << p.message() << std::endl;
|
||
|
|
|
||
|
|
for (std::map<std::string, std::string>::const_iterator i
|
||
|
|
= p.headers().begin(), end(p.headers().end()); i != end; ++i)
|
||
|
|
{
|
||
|
|
std::cerr << i->first << ": " << i->second << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_connect_handler(http_connection& c)
|
||
|
|
{
|
||
|
|
++connect_handler_called;
|
||
|
|
TEST_CHECK(c.socket().is_open());
|
||
|
|
TEST_CHECK(c.socket().remote_endpoint().address() == address::from_string("127.0.0.1"));
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_handler(asio::error_code const& ec, http_parser const& parser, char const* data, int size)
|
||
|
|
{
|
||
|
|
++handler_called;
|
||
|
|
data_size = size;
|
||
|
|
error_code = ec;
|
||
|
|
|
||
|
|
if (parser.header_finished()) http_status = parser.status_code();
|
||
|
|
print_http_header(parser);
|
||
|
|
|
||
|
|
cq.close();
|
||
|
|
}
|
||
|
|
|
||
|
|
void reset_globals()
|
||
|
|
{
|
||
|
|
connect_handler_called = 0;
|
||
|
|
handler_called = 0;
|
||
|
|
data_size = 0;
|
||
|
|
http_status = 0;
|
||
|
|
error_code = asio::error_code();
|
||
|
|
}
|
||
|
|
|
||
|
|
void run_test(char const* url, int size, int status, asio::error_code const& ec)
|
||
|
|
{
|
||
|
|
reset_globals();
|
||
|
|
|
||
|
|
std::cerr << " ===== TESTING: " << url << " =====" << std::endl;
|
||
|
|
|
||
|
|
boost::shared_ptr<http_connection> h(new http_connection(ios, cq
|
||
|
|
, &::http_handler, true, &::http_connect_handler));
|
||
|
|
h->get(url);
|
||
|
|
ios.reset();
|
||
|
|
ios.run();
|
||
|
|
|
||
|
|
std::cerr << "connect_handler_called: " << connect_handler_called << std::endl;
|
||
|
|
std::cerr << "handler_called: " << handler_called << std::endl;
|
||
|
|
std::cerr << "status: " << http_status << std::endl;
|
||
|
|
std::cerr << "size: " << data_size << std::endl;
|
||
|
|
std::cerr << "error_code: " << error_code.message() << std::endl;
|
||
|
|
TEST_CHECK(connect_handler_called == 1);
|
||
|
|
TEST_CHECK(handler_called == 1);
|
||
|
|
TEST_CHECK(data_size == size || size == -1);
|
||
|
|
TEST_CHECK(error_code == ec);
|
||
|
|
TEST_CHECK(http_status == status);
|
||
|
|
}
|
||
|
|
|
||
|
|
int test_main()
|
||
|
|
{
|
||
|
|
run_test("http://127.0.0.1/disk_io.png", 17809, 200, asio::error_code());
|
||
|
|
run_test("http://127.0.0.1/non-existing-file", -1, 404, asio::error::eof);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|