| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "chrome/browser/local_discovery/privet_url_fetcher.h" | 5 #include "chrome/browser/local_discovery/privet_url_fetcher.h" |
| 6 #include "net/url_request/test_url_fetcher_factory.h" | 6 #include "net/url_request/test_url_fetcher_factory.h" |
| 7 #include "net/url_request/url_request_test_util.h" | 7 #include "net/url_request/url_request_test_util.h" |
| 8 #include "testing/gmock/include/gmock/gmock.h" | 8 #include "testing/gmock/include/gmock/gmock.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
| 10 | 10 |
| (...skipping 191 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 202 | 202 |
| 203 std::string header_token; | 203 std::string header_token; |
| 204 ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); | 204 ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); |
| 205 EXPECT_EQ(kSamplePrivetToken, header_token); | 205 EXPECT_EQ(kSamplePrivetToken, header_token); |
| 206 } | 206 } |
| 207 | 207 |
| 208 TEST_F(PrivetURLFetcherTest, Header2) { | 208 TEST_F(PrivetURLFetcherTest, Header2) { |
| 209 PrivetURLFetcher::SetTokenForHost(GURL(kSamplePrivetURL).GetOrigin().spec(), | 209 PrivetURLFetcher::SetTokenForHost(GURL(kSamplePrivetURL).GetOrigin().spec(), |
| 210 ""); | 210 ""); |
| 211 | 211 |
| 212 privet_urlfetcher_->AllowEmptyPrivetToken(); | 212 privet_urlfetcher_->SendEmptyPrivetToken(); |
| 213 privet_urlfetcher_->Start(); | 213 privet_urlfetcher_->Start(); |
| 214 | 214 |
| 215 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); | 215 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); |
| 216 ASSERT_TRUE(fetcher != NULL); |
| 217 net::HttpRequestHeaders headers; |
| 218 fetcher->GetExtraRequestHeaders(&headers); |
| 219 |
| 220 std::string header_token; |
| 221 ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); |
| 222 EXPECT_EQ(kEmptyPrivetToken, header_token); |
| 223 } |
| 224 |
| 225 TEST_F(PrivetURLFetcherTest, AlwaysSendEmpty) { |
| 226 PrivetURLFetcher::SetTokenForHost(GURL(kSamplePrivetURL).GetOrigin().spec(), |
| 227 "SampleToken"); |
| 228 |
| 229 privet_urlfetcher_->SendEmptyPrivetToken(); |
| 230 privet_urlfetcher_->Start(); |
| 231 |
| 232 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); |
| 216 ASSERT_TRUE(fetcher != NULL); | 233 ASSERT_TRUE(fetcher != NULL); |
| 217 net::HttpRequestHeaders headers; | 234 net::HttpRequestHeaders headers; |
| 218 fetcher->GetExtraRequestHeaders(&headers); | 235 fetcher->GetExtraRequestHeaders(&headers); |
| 219 | 236 |
| 220 std::string header_token; | 237 std::string header_token; |
| 221 ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); | 238 ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); |
| 222 EXPECT_EQ(kEmptyPrivetToken, header_token); | 239 EXPECT_EQ(kEmptyPrivetToken, header_token); |
| 223 } | 240 } |
| 224 | 241 |
| 225 TEST_F(PrivetURLFetcherTest, FetchHasError) { | 242 TEST_F(PrivetURLFetcherTest, FetchHasError) { |
| (...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 275 net::OK)); | 292 net::OK)); |
| 276 fetcher->set_response_code(200); | 293 fetcher->set_response_code(200); |
| 277 | 294 |
| 278 EXPECT_CALL(delegate_, OnFileInternal()); | 295 EXPECT_CALL(delegate_, OnFileInternal()); |
| 279 fetcher->delegate()->OnURLFetchComplete(fetcher); | 296 fetcher->delegate()->OnURLFetchComplete(fetcher); |
| 280 } | 297 } |
| 281 | 298 |
| 282 } // namespace | 299 } // namespace |
| 283 | 300 |
| 284 } // namespace local_discovery | 301 } // namespace local_discovery |
| OLD | NEW |