| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <atlbase.h> | 5 #include <atlbase.h> |
| 6 #include <atlcom.h> | 6 #include <atlcom.h> |
| 7 | 7 |
| 8 #include "base/strings/string16.h" | 8 #include "base/strings/string16.h" |
| 9 #include "base/strings/string_util.h" | 9 #include "base/strings/string_util.h" |
| 10 #include "base/strings/utf_string_conversions.h" | 10 #include "base/strings/utf_string_conversions.h" |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 61 TEST_F(HttpNegotiateTest, BeginningTransaction) { | 61 TEST_F(HttpNegotiateTest, BeginningTransaction) { |
| 62 static const int kBeginningTransactionIndex = 3; | 62 static const int kBeginningTransactionIndex = 3; |
| 63 CComObjectStackEx<TestHttpNegotiate> test_http; | 63 CComObjectStackEx<TestHttpNegotiate> test_http; |
| 64 IHttpNegotiate_BeginningTransaction_Fn original = | 64 IHttpNegotiate_BeginningTransaction_Fn original = |
| 65 reinterpret_cast<IHttpNegotiate_BeginningTransaction_Fn>( | 65 reinterpret_cast<IHttpNegotiate_BeginningTransaction_Fn>( |
| 66 (*reinterpret_cast<void***>( | 66 (*reinterpret_cast<void***>( |
| 67 static_cast<IHttpNegotiate*>( | 67 static_cast<IHttpNegotiate*>( |
| 68 &test_http)))[kBeginningTransactionIndex]); | 68 &test_http)))[kBeginningTransactionIndex]); |
| 69 | 69 |
| 70 base::string16 cf_ua( | 70 base::string16 cf_ua( |
| 71 ASCIIToWide(http_utils::GetDefaultUserAgentHeaderWithCFTag())); | 71 base::ASCIIToWide(http_utils::GetDefaultUserAgentHeaderWithCFTag())); |
| 72 base::string16 cf_tag(ASCIIToWide(http_utils::GetChromeFrameUserAgent())); | 72 base::string16 cf_tag( |
| 73 base::ASCIIToWide(http_utils::GetChromeFrameUserAgent())); |
| 73 | 74 |
| 74 EXPECT_NE(base::string16::npos, cf_ua.find(L"chromeframe/")); | 75 EXPECT_NE(base::string16::npos, cf_ua.find(L"chromeframe/")); |
| 75 | 76 |
| 76 struct TestCase { | 77 struct TestCase { |
| 77 const base::string16 original_headers_; | 78 const base::string16 original_headers_; |
| 78 const base::string16 delegate_additional_; | 79 const base::string16 delegate_additional_; |
| 79 const base::string16 expected_additional_; | 80 const base::string16 expected_additional_; |
| 80 HRESULT delegate_return_value_; | 81 HRESULT delegate_return_value_; |
| 81 } test_cases[] = { | 82 } test_cases[] = { |
| 82 { L"Accept: */*\r\n", | 83 { L"Accept: */*\r\n", |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 126 TEST_F(HttpNegotiateTest, BeginningTransactionUARemoval) { | 127 TEST_F(HttpNegotiateTest, BeginningTransactionUARemoval) { |
| 127 static const int kBeginningTransactionIndex = 3; | 128 static const int kBeginningTransactionIndex = 3; |
| 128 CComObjectStackEx<TestHttpNegotiate> test_http; | 129 CComObjectStackEx<TestHttpNegotiate> test_http; |
| 129 IHttpNegotiate_BeginningTransaction_Fn original = | 130 IHttpNegotiate_BeginningTransaction_Fn original = |
| 130 reinterpret_cast<IHttpNegotiate_BeginningTransaction_Fn>( | 131 reinterpret_cast<IHttpNegotiate_BeginningTransaction_Fn>( |
| 131 (*reinterpret_cast<void***>( | 132 (*reinterpret_cast<void***>( |
| 132 static_cast<IHttpNegotiate*>( | 133 static_cast<IHttpNegotiate*>( |
| 133 &test_http)))[kBeginningTransactionIndex]); | 134 &test_http)))[kBeginningTransactionIndex]); |
| 134 | 135 |
| 135 base::string16 nocf_ua( | 136 base::string16 nocf_ua( |
| 136 ASCIIToWide(http_utils::RemoveChromeFrameFromUserAgentValue( | 137 base::ASCIIToWide(http_utils::RemoveChromeFrameFromUserAgentValue( |
| 137 http_utils::GetDefaultUserAgentHeaderWithCFTag()))); | 138 http_utils::GetDefaultUserAgentHeaderWithCFTag()))); |
| 138 base::string16 cf_ua(ASCIIToWide( | 139 base::string16 cf_ua(base::ASCIIToWide( |
| 139 http_utils::AddChromeFrameToUserAgentValue(WideToASCII(nocf_ua)))); | 140 http_utils::AddChromeFrameToUserAgentValue(WideToASCII(nocf_ua)))); |
| 140 | 141 |
| 141 EXPECT_EQ(base::string16::npos, nocf_ua.find(L"chromeframe/")); | 142 EXPECT_EQ(base::string16::npos, nocf_ua.find(L"chromeframe/")); |
| 142 EXPECT_NE(base::string16::npos, cf_ua.find(L"chromeframe/")); | 143 EXPECT_NE(base::string16::npos, cf_ua.find(L"chromeframe/")); |
| 143 | 144 |
| 144 base::string16 ua_url(L"www.withua.com"); | 145 base::string16 ua_url(L"www.withua.com"); |
| 145 base::string16 no_ua_url(L"www.noua.com"); | 146 base::string16 no_ua_url(L"www.noua.com"); |
| 146 | 147 |
| 147 RegistryListPreferencesHolder& ua_holder = | 148 RegistryListPreferencesHolder& ua_holder = |
| 148 GetUserAgentPreferencesHolderForTesting(); | 149 GetUserAgentPreferencesHolderForTesting(); |
| (...skipping 130 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 279 | 280 |
| 280 // Check additional headers are preserved. | 281 // Check additional headers are preserved. |
| 281 EXPECT_THAT(AppendCFUserAgentString(NULL, | 282 EXPECT_THAT(AppendCFUserAgentString(NULL, |
| 282 L"Authorization: A Zoo That I Ruin\r\n" | 283 L"Authorization: A Zoo That I Ruin\r\n" |
| 283 L"User-Agent: Get a Nurse;\r\n" | 284 L"User-Agent: Get a Nurse;\r\n" |
| 284 L"Accept-Language: Cleanup a Cat Egg\r\n"), | 285 L"Accept-Language: Cleanup a Cat Egg\r\n"), |
| 285 AllOf(ContainsRegex("User-Agent: Get a Nurse; chromeframe.+\r\n"), | 286 AllOf(ContainsRegex("User-Agent: Get a Nurse; chromeframe.+\r\n"), |
| 286 HasSubstr("Authorization: A Zoo That I Ruin\r\n"), | 287 HasSubstr("Authorization: A Zoo That I Ruin\r\n"), |
| 287 HasSubstr("Accept-Language: Cleanup a Cat Egg\r\n"))); | 288 HasSubstr("Accept-Language: Cleanup a Cat Egg\r\n"))); |
| 288 } | 289 } |
| OLD | NEW |