// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <atlbase.h>
#include <atlcom.h>
#include "base/bind.h"
#include "base/bind_helpers.h"
#include "chrome/common/automation_messages.h"
#include "chrome_frame/test/chrome_frame_test_utils.h"
#include "chrome_frame/test/test_server.h"
#include "chrome_frame/test/test_with_web_server.h"
#include "chrome_frame/urlmon_url_request.h"
#include "chrome_frame/urlmon_url_request_private.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gmock_mutant.h"
#include "testing/gtest/include/gtest/gtest.h"
using testing::CreateFunctor;
using chrome_frame_test::kChromeFrameLongNavigationTimeout;
static void AppendToStream(IStream* s, void* buffer, ULONG cb) {
ULONG bytes_written;
LARGE_INTEGER current_pos;
LARGE_INTEGER zero = {0};
// Remember current position.
ASSERT_HRESULT_SUCCEEDED(s->Seek(zero, STREAM_SEEK_CUR,
reinterpret_cast<ULARGE_INTEGER*>(¤t_pos)));
// Seek to the end.
ASSERT_HRESULT_SUCCEEDED(s->Seek(zero, STREAM_SEEK_END, NULL));
ASSERT_HRESULT_SUCCEEDED(s->Write(buffer, cb, &bytes_written));
ASSERT_EQ(cb, bytes_written);
// Seek to original position.
ASSERT_HRESULT_SUCCEEDED(s->Seek(current_pos, STREAM_SEEK_SET, NULL));
}
class MockUrlDelegate : public PluginUrlRequestDelegate {
public:
MOCK_METHOD9(OnResponseStarted, void(int request_id, const char* mime_type,
const char* headers, int size, base::Time last_modified,
const std::string& redirect_url, int redirect_status,
const net::HostPortPair& socket_address, uint64 upload_size));
MOCK_METHOD2(OnReadComplete, void(int request_id, const std::string& data));
MOCK_METHOD2(OnResponseEnd, void(int request_id,
const net::URLRequestStatus& status));
void PostponeReadRequest(chrome_frame_test::TimedMsgLoop* loop,
UrlmonUrlRequest* request, int bytes_to_read) {
loop->PostTask(FROM_HERE,
base::Bind(&MockUrlDelegate::RequestRead,
base::Unretained(this), request, bytes_to_read));
}
private:
void RequestRead(UrlmonUrlRequest* request, int bytes_to_read) {
request->Read(bytes_to_read);
}
};
// Simplest UrlmonUrlRequest. Retrieve a file from local web server.
TEST(UrlmonUrlRequestTest, Simple1) {
MockUrlDelegate mock;
chrome_frame_test::TimedMsgLoop loop;
testing::StrictMock<MockWebServer> mock_server(1337,
ASCIIToWide(chrome_frame_test::GetLocalIPv4Address()),
chrome_frame_test::GetTestDataFolder());
mock_server.ExpectAndServeAnyRequests(CFInvocation(CFInvocation::NONE));
CComObjectStackEx<UrlmonUrlRequest> request;
request.AddRef();
request.Initialize(&mock, 1, // request_id
WideToUTF8(mock_server.Resolve(L"chrome_frame_window_open.html")),
"get",
"", // referrer
"", // extra request
NULL, // upload data
ResourceType::MAIN_FRAME, // resource type
true,
0); // frame busting
testing::InSequence s;
EXPECT_CALL(mock, OnResponseStarted(1, testing::_, testing::_, testing::_,
testing::_, testing::_, testing::_,
testing::_, testing::_))
.Times(1)
.WillOnce(testing::IgnoreResult(testing::InvokeWithoutArgs(CreateFunctor(
&request, &UrlmonUrlRequest::Read, 512))));
EXPECT_CALL(mock, OnReadComplete(1, testing::Property(&std::string::size,
testing::Gt(0u))))
.Times(testing::AtLeast(1))
.WillRepeatedly(testing::InvokeWithoutArgs(CreateFunctor(&mock,
&MockUrlDelegate::PostponeReadRequest, &loop, &request, 64)));
EXPECT_CALL(mock, OnResponseEnd(1, testing::_))
.Times(1)
.WillOnce(QUIT_LOOP_SOON(loop, base::TimeDelta::FromSeconds(2)));
request.Start();
loop.RunFor(kChromeFrameLongNavigationTimeout);
request.Release();
}
// Same as Simple1 except we use the HEAD verb to fetch only the headers
// from the server.
TEST(UrlmonUrlRequestTest, Head) {
MockUrlDelegate mock;
chrome_frame_test::TimedMsgLoop loop;
// Use SimpleWebServer instead of the python server to support HEAD
// requests.
test_server::SimpleWebServer server(13337);
test_server::SimpleResponse head_response("/head", "");
server.AddResponse(&head_response);
CComObjectStackEx<UrlmonUrlRequest> request;
request.AddRef();
request.Initialize(&mock, 1, // request_id
base::StringPrintf("http://%s:13337/head", server.host().c_str()),
"head",
"", // referrer
"", // extra request
NULL, // upload data
ResourceType::MAIN_FRAME, // resource type
true,
0); // frame busting
testing::InSequence s;
EXPECT_CALL(mock, OnResponseStarted(1, testing::_, testing::_, testing::_,
testing::_, testing::_, testing::_,
testing::_, testing::_))
.Times(1)
.WillOnce(testing::IgnoreResult(testing::InvokeWithoutArgs(CreateFunctor(
&request, &UrlmonUrlRequest::Read, 512))));
// For HEAD requests we don't expect content reads.
EXPECT_CALL(mock, OnReadComplete(1, testing::_)).Times(0);
EXPECT_CALL(mock, OnResponseEnd(1, testing::_))
.Times(1)
.WillOnce(QUIT_LOOP_SOON(loop, base::TimeDelta::FromSeconds(2)));
request.Start();
loop.RunFor(kChromeFrameLongNavigationTimeout);
request.Release();
}
TEST(UrlmonUrlRequestTest, UnreachableUrl) {
MockUrlDelegate mock;
chrome_frame_test::TimedMsgLoop loop;
CComObjectStackEx<UrlmonUrlRequest> request;
testing::StrictMock<MockWebServer> mock_server(1337,
ASCIIToWide(chrome_frame_test::GetLocalIPv4Address()),
chrome_frame_test::GetTestDataFolder());
mock_server.ExpectAndServeAnyRequests(CFInvocation(CFInvocation::NONE));
GURL unreachable(WideToUTF8(mock_server.Resolve(
L"non_existing.html")));
request.AddRef();
request.Initialize(&mock, 1, // request_id
unreachable.spec(), "get",
"", // referrer
"", // extra request
NULL, // upload data
ResourceType::MAIN_FRAME, // resource type
true,
0); // frame busting
// Expect headers
EXPECT_CALL(mock, OnResponseStarted(1, testing::_,
testing::StartsWith("HTTP/1.1 404"),
testing::_, testing::_, testing::_,
testing::_, testing::_, testing::_))
.Times(1)
.WillOnce(QUIT_LOOP_SOON(loop, base::TimeDelta::FromSeconds(2)));
EXPECT_CALL(mock, OnResponseEnd(1, testing::Property(
&net::URLRequestStatus::error,
net::ERR_TUNNEL_CONNECTION_FAILED)))
.Times(testing::AtMost(1));
request.Start();
loop.RunFor(kChromeFrameLongNavigationTimeout);
request.Release();
}
TEST(UrlmonUrlRequestTest, ZeroLengthResponse) {
MockUrlDelegate mock;
chrome_frame_test::TimedMsgLoop loop;
testing::StrictMock<MockWebServer> mock_server(1337,
ASCIIToWide(chrome_frame_test::GetLocalIPv4Address()),
chrome_frame_test::GetTestDataFolder());
mock_server.ExpectAndServeAnyRequests(CFInvocation(CFInvocation::NONE));
CComObjectStackEx<UrlmonUrlRequest> request;
request.AddRef();
request.Initialize(&mock, 1, // request_id
WideToUTF8(mock_server.Resolve(L"empty.html")), "get",
"", // referrer
"", // extra request
NULL, // upload data
ResourceType::MAIN_FRAME, // resource type
true,
0); // frame busting
// Expect headers
EXPECT_CALL(mock, OnResponseStarted(1, testing::_, testing::_, testing::_,
testing::_, testing::_, testing::_,
testing::_, testing::_))
.Times(1)
.WillOnce(QUIT_LOOP(loop));
request.Start();
loop.RunFor(kChromeFrameLongNavigationTimeout);
EXPECT_FALSE(loop.WasTimedOut());
// Should stay quiet, since we do not ask for anything for awhile.
EXPECT_CALL(mock, OnResponseEnd(1, testing::_)).Times(0);
loop.RunFor(base::TimeDelta::FromSeconds(3));
// Invoke read. Only now the response end ("server closed the connection")
// is supposed to be delivered.
EXPECT_CALL(mock, OnResponseEnd(1, testing::Property(
&net::URLRequestStatus::is_success, true))).Times(1);
request.Read(512);
request.Release();
}
ACTION_P4(ManagerRead, loop, mgr, request_id, bytes_to_read) {
loop->PostTask(FROM_HERE,
base::Bind(&UrlmonUrlRequestManager::ReadUrlRequest,
base::Unretained(mgr), request_id, bytes_to_read));
}
ACTION_P3(ManagerEndRequest, loop, mgr, request_id) {
loop->PostTask(FROM_HERE, base::Bind(&UrlmonUrlRequestManager::EndUrlRequest,
base::Unretained(mgr), request_id,
net::URLRequestStatus()));
}
// Simplest test - retrieve file from local web server.
TEST(UrlmonUrlRequestManagerTest, Simple1) {
MockUrlDelegate mock;
chrome_frame_test::TimedMsgLoop loop;
testing::StrictMock<MockWebServer> mock_server(1337,
ASCIIToWide(chrome_frame_test::GetLocalIPv4Address()),
chrome_frame_test::GetTestDataFolder());
mock_server.ExpectAndServeAnyRequests(CFInvocation(CFInvocation::NONE));
scoped_ptr<UrlmonUrlRequestManager> mgr(new UrlmonUrlRequestManager());
mgr->set_delegate(&mock);
AutomationURLRequest r1;
r1.url = WideToUTF8(mock_server.Resolve(L"chrome_frame_window_open.html"));
r1.method = "get";
r1.resource_type = 0;
r1.load_flags = 0;
EXPECT_CALL(mock, OnResponseStarted(1, testing::_, testing::_, testing::_,
testing::_, testing::_, testing::_, testing::_,
testing::_))
.Times(1)
.WillOnce(ManagerRead(&loop, mgr.get(), 1, 512));
EXPECT_CALL(mock, OnReadComplete(1, testing::Property(&std::string::size,
testing::Gt(0u))))
.Times(testing::AtLeast(1))
.WillRepeatedly(ManagerRead(&loop, mgr.get(), 1, 2));
EXPECT_CALL(mock, OnResponseEnd(1, testing::_))
.Times(1)
.WillOnce(QUIT_LOOP_SOON(loop, base::TimeDelta::FromSeconds(2)));
mgr->StartUrlRequest(1, r1);
loop.RunFor(kChromeFrameLongNavigationTimeout);
mgr.reset();
}
TEST(UrlmonUrlRequestManagerTest, Abort1) {
MockUrlDelegate mock;
chrome_frame_test::TimedMsgLoop loop;
testing::StrictMock<MockWebServer> mock_server(1337,
ASCIIToWide(chrome_frame_test::GetLocalIPv4Address()),
chrome_frame_test::GetTestDataFolder());
mock_server.ExpectAndServeAnyRequests(CFInvocation(CFInvocation::NONE));
scoped_ptr<UrlmonUrlRequestManager> mgr(new UrlmonUrlRequestManager());
mgr->set_delegate(&mock);
AutomationURLRequest r1;
r1.url = WideToUTF8(mock_server.Resolve(L"chrome_frame_window_open.html"));
r1.method = "get";
r1.resource_type = 0;
r1.load_flags = 0;
EXPECT_CALL(mock, OnResponseStarted(1, testing::_, testing::_, testing::_,
testing::_, testing::_, testing::_, testing::_,
testing::_))
.Times(1)
.WillOnce(testing::DoAll(
ManagerEndRequest(&loop, mgr.get(), 1),
QUIT_LOOP_SOON(loop, base::TimeDelta::FromSeconds(3))));
EXPECT_CALL(mock, OnReadComplete(1, testing::_))
.Times(0);
EXPECT_CALL(mock, OnResponseEnd(1, testing::_))
.Times(0);
mgr->StartUrlRequest(1, r1);
loop.RunFor(kChromeFrameLongNavigationTimeout);
mgr.reset();
}