Index: chrome_frame/test/http_negotiate_unittest.cc |
diff --git a/chrome_frame/test/http_negotiate_unittest.cc b/chrome_frame/test/http_negotiate_unittest.cc |
deleted file mode 100644 |
index c726811d3acffaea16ba7fdb472eaccd943686ce..0000000000000000000000000000000000000000 |
--- a/chrome_frame/test/http_negotiate_unittest.cc |
+++ /dev/null |
@@ -1,289 +0,0 @@ |
-// 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/strings/string16.h" |
-#include "base/strings/string_util.h" |
-#include "base/strings/utf_string_conversions.h" |
-#include "base/win/scoped_bstr.h" |
-#include "base/win/scoped_comptr.h" |
-#include "chrome_frame/html_utils.h" |
-#include "chrome_frame/http_negotiate.h" |
-#include "chrome_frame/registry_list_preferences_holder.h" |
-#include "chrome_frame/test/chrome_frame_test_utils.h" |
-#include "chrome_frame/utils.h" |
-#include "gmock/gmock.h" |
-#include "gtest/gtest.h" |
- |
-class HttpNegotiateTest : public testing::Test { |
- protected: |
- HttpNegotiateTest() { |
- } |
-}; |
- |
-class TestHttpNegotiate |
- : public CComObjectRootEx<CComMultiThreadModel>, |
- public IHttpNegotiate { |
- public: |
- TestHttpNegotiate() |
- : beginning_transaction_ret_(S_OK), additional_headers_(NULL) { |
- } |
- |
-BEGIN_COM_MAP(TestHttpNegotiate) |
- COM_INTERFACE_ENTRY(IHttpNegotiate) |
-END_COM_MAP() |
- STDMETHOD(BeginningTransaction)(LPCWSTR url, LPCWSTR headers, // NOLINT |
- DWORD reserved, // NOLINT |
- LPWSTR* additional_headers) { // NOLINT |
- if (additional_headers_) { |
- int len = lstrlenW(additional_headers_); |
- len++; |
- *additional_headers = reinterpret_cast<wchar_t*>( |
- ::CoTaskMemAlloc(len * sizeof(wchar_t))); |
- lstrcpyW(*additional_headers, additional_headers_); |
- } |
- return beginning_transaction_ret_; |
- } |
- |
- STDMETHOD(OnResponse)(DWORD response_code, LPCWSTR response_header, |
- LPCWSTR request_header, |
- LPWSTR* additional_request_headers) { |
- return S_OK; |
- } |
- |
- HRESULT beginning_transaction_ret_; |
- const wchar_t* additional_headers_; |
-}; |
- |
-TEST_F(HttpNegotiateTest, BeginningTransaction) { |
- static const int kBeginningTransactionIndex = 3; |
- CComObjectStackEx<TestHttpNegotiate> test_http; |
- IHttpNegotiate_BeginningTransaction_Fn original = |
- reinterpret_cast<IHttpNegotiate_BeginningTransaction_Fn>( |
- (*reinterpret_cast<void***>( |
- static_cast<IHttpNegotiate*>( |
- &test_http)))[kBeginningTransactionIndex]); |
- |
- base::string16 cf_ua( |
- base::ASCIIToWide(http_utils::GetDefaultUserAgentHeaderWithCFTag())); |
- base::string16 cf_tag( |
- base::ASCIIToWide(http_utils::GetChromeFrameUserAgent())); |
- |
- EXPECT_NE(base::string16::npos, cf_ua.find(L"chromeframe/")); |
- |
- struct TestCase { |
- const base::string16 original_headers_; |
- const base::string16 delegate_additional_; |
- const base::string16 expected_additional_; |
- HRESULT delegate_return_value_; |
- } test_cases[] = { |
- { L"Accept: */*\r\n", |
- L"", |
- cf_ua + L"\r\n", |
- S_OK }, |
- { L"Accept: */*\r\n", |
- L"", |
- L"", |
- E_OUTOFMEMORY }, |
- { L"", |
- L"Accept: */*\r\n", |
- L"Accept: */*\r\n" + cf_ua + L"\r\n", |
- S_OK }, |
- { L"User-Agent: Bingo/1.0\r\n", |
- L"", |
- L"User-Agent: Bingo/1.0 " + cf_tag + L"\r\n", |
- S_OK }, |
- { L"User-Agent: NotMe/1.0\r\n", |
- L"User-Agent: MeMeMe/1.0\r\n", |
- L"User-Agent: MeMeMe/1.0 " + cf_tag + L"\r\n", |
- S_OK }, |
- { L"", |
- L"User-Agent: MeMeMe/1.0\r\n", |
- L"User-Agent: MeMeMe/1.0 " + cf_tag + L"\r\n", |
- S_OK }, |
- }; |
- |
- for (int i = 0; i < arraysize(test_cases); ++i) { |
- TestCase& test = test_cases[i]; |
- wchar_t* additional = NULL; |
- test_http.beginning_transaction_ret_ = test.delegate_return_value_; |
- test_http.additional_headers_ = test.delegate_additional_.c_str(); |
- HttpNegotiatePatch::BeginningTransaction(original, &test_http, |
- L"http://www.google.com", test.original_headers_.c_str(), 0, |
- &additional); |
- EXPECT_TRUE(additional != NULL); |
- |
- if (additional) { |
- // Check against the expected additional headers. |
- EXPECT_EQ(test.expected_additional_, base::string16(additional)); |
- ::CoTaskMemFree(additional); |
- } |
- } |
-} |
- |
-TEST_F(HttpNegotiateTest, BeginningTransactionUARemoval) { |
- static const int kBeginningTransactionIndex = 3; |
- CComObjectStackEx<TestHttpNegotiate> test_http; |
- IHttpNegotiate_BeginningTransaction_Fn original = |
- reinterpret_cast<IHttpNegotiate_BeginningTransaction_Fn>( |
- (*reinterpret_cast<void***>( |
- static_cast<IHttpNegotiate*>( |
- &test_http)))[kBeginningTransactionIndex]); |
- |
- base::string16 nocf_ua( |
- base::ASCIIToWide(http_utils::RemoveChromeFrameFromUserAgentValue( |
- http_utils::GetDefaultUserAgentHeaderWithCFTag()))); |
- base::string16 cf_ua(base::ASCIIToWide( |
- http_utils::AddChromeFrameToUserAgentValue(WideToASCII(nocf_ua)))); |
- |
- EXPECT_EQ(base::string16::npos, nocf_ua.find(L"chromeframe/")); |
- EXPECT_NE(base::string16::npos, cf_ua.find(L"chromeframe/")); |
- |
- base::string16 ua_url(L"www.withua.com"); |
- base::string16 no_ua_url(L"www.noua.com"); |
- |
- RegistryListPreferencesHolder& ua_holder = |
- GetUserAgentPreferencesHolderForTesting(); |
- ua_holder.AddStringForTesting(no_ua_url); |
- |
- struct TestCase { |
- const base::string16 url_; |
- const base::string16 original_headers_; |
- const base::string16 delegate_additional_; |
- const base::string16 expected_additional_; |
- } test_cases[] = { |
- { ua_url, |
- L"", |
- L"Accept: */*\r\n" + cf_ua + L"\r\n", |
- L"Accept: */*\r\n" + cf_ua + L"\r\n" }, |
- { ua_url, |
- L"", |
- L"Accept: */*\r\n" + nocf_ua + L"\r\n", |
- L"Accept: */*\r\n" + cf_ua + L"\r\n" }, |
- { no_ua_url, |
- L"", |
- L"Accept: */*\r\n" + cf_ua + L"\r\n", |
- L"Accept: */*\r\n" + nocf_ua + L"\r\n" }, |
- { no_ua_url, |
- L"", |
- L"Accept: */*\r\n" + nocf_ua + L"\r\n", |
- L"Accept: */*\r\n" + nocf_ua + L"\r\n" }, |
- }; |
- |
- for (int i = 0; i < arraysize(test_cases); ++i) { |
- TestCase& test = test_cases[i]; |
- wchar_t* additional = NULL; |
- test_http.beginning_transaction_ret_ = S_OK; |
- test_http.additional_headers_ = test.delegate_additional_.c_str(); |
- HttpNegotiatePatch::BeginningTransaction(original, &test_http, |
- test.url_.c_str(), test.original_headers_.c_str(), 0, |
- &additional); |
- EXPECT_TRUE(additional != NULL); |
- |
- if (additional) { |
- // Check against the expected additional headers. |
- EXPECT_EQ(test.expected_additional_, base::string16(additional)) |
- << "Iteration: " << i; |
- ::CoTaskMemFree(additional); |
- } |
- } |
-} |
- |
- |
-class TestInternetProtocolSink |
- : public CComObjectRootEx<CComMultiThreadModel>, |
- public IInternetProtocolSink { |
- public: |
- TestInternetProtocolSink() : status_(0) { |
- // Create an instance of IE to fullfill the requirements of being able |
- // to detect whether a sub-frame or top-frame is being loaded (see |
- // IsSubFrameRequest) and to be able to mark an IBrowserService |
- // implementation as a target for CF navigation. |
- HRESULT hr = browser_.CreateInstance(CLSID_InternetExplorer); |
- CHECK(SUCCEEDED(hr)); |
- if (SUCCEEDED(hr)) { |
- browser_->Navigate(base::win::ScopedBstr(L"about:blank"), |
- NULL, NULL, NULL, NULL); |
- } |
- } |
- |
- ~TestInternetProtocolSink() { |
- if (browser_) |
- browser_->Quit(); |
- } |
- |
-BEGIN_COM_MAP(TestInternetProtocolSink) |
- COM_INTERFACE_ENTRY(IInternetProtocolSink) |
- COM_INTERFACE_ENTRY_AGGREGATE(IID_IServiceProvider, browser_) |
-END_COM_MAP() |
- |
- // IInternetProtocolSink. |
- STDMETHOD(Switch)(PROTOCOLDATA* data) { |
- NOTREACHED(); |
- return S_OK; |
- } |
- |
- STDMETHOD(ReportProgress)(ULONG status, LPCWSTR text) { |
- status_ = status; |
- status_text_ = text ? text : L""; |
- return S_OK; |
- } |
- |
- STDMETHOD(ReportData)(DWORD bscf, ULONG progress, ULONG progress_max) { |
- NOTREACHED(); |
- return S_OK; |
- } |
- |
- STDMETHOD(ReportResult)(HRESULT hr, DWORD err, LPCWSTR result) { |
- NOTREACHED(); |
- return S_OK; |
- } |
- |
- ULONG last_status() const { |
- return status_; |
- } |
- |
- const base::string16& last_status_text() const { |
- return status_text_; |
- } |
- |
- protected: |
- ULONG status_; |
- base::string16 status_text_; |
- base::win::ScopedComPtr<IWebBrowser2> browser_; |
-}; |
- |
-using testing::AllOf; |
-using testing::ContainsRegex; |
-using testing::HasSubstr; |
- |
-TEST(AppendUserAgent, Append) { |
- EXPECT_THAT(AppendCFUserAgentString(NULL, NULL), |
- testing::ContainsRegex("User-Agent:.+chromeframe.+\r\n")); |
- |
- // Check Http headers are reasonably parsed. |
- EXPECT_THAT(AppendCFUserAgentString(L"Bad User-Agent: Age Tuners;\r\n", NULL), |
- AllOf(ContainsRegex("User-Agent:.+chromeframe.+\r\n"), |
- testing::Not(testing::HasSubstr("Age Tuners")))); |
- |
- // Honor headers User-Agent, if additional headers does not specify one. |
- EXPECT_THAT(AppendCFUserAgentString(L"User-Agent: A Tense Rug;\r\n", NULL), |
- ContainsRegex("User-Agent: A Tense Rug; chromeframe.+\r\n")); |
- |
- // Honor additional headers User-Agent. |
- EXPECT_THAT(AppendCFUserAgentString(L"User-Agent: Near Guest;\r\n", |
- L"User-Agent: Rat see Gun;\r\n"), |
- ContainsRegex("User-Agent: Rat see Gun; chromeframe.+\r\n")); |
- |
- // Check additional headers are preserved. |
- EXPECT_THAT(AppendCFUserAgentString(NULL, |
- L"Authorization: A Zoo That I Ruin\r\n" |
- L"User-Agent: Get a Nurse;\r\n" |
- L"Accept-Language: Cleanup a Cat Egg\r\n"), |
- AllOf(ContainsRegex("User-Agent: Get a Nurse; chromeframe.+\r\n"), |
- HasSubstr("Authorization: A Zoo That I Ruin\r\n"), |
- HasSubstr("Accept-Language: Cleanup a Cat Egg\r\n"))); |
-} |