| 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 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 51 private: | 51 private: |
| 52 scoped_ptr<base::DictionaryValue> saved_value_; | 52 scoped_ptr<base::DictionaryValue> saved_value_; |
| 53 }; | 53 }; |
| 54 | 54 |
| 55 class PrivetURLFetcherTest : public ::testing::Test { | 55 class PrivetURLFetcherTest : public ::testing::Test { |
| 56 public: | 56 public: |
| 57 PrivetURLFetcherTest() { | 57 PrivetURLFetcherTest() { |
| 58 request_context_= new net::TestURLRequestContextGetter( | 58 request_context_= new net::TestURLRequestContextGetter( |
| 59 base::MessageLoopProxy::current()); | 59 base::MessageLoopProxy::current()); |
| 60 privet_urlfetcher_.reset(new PrivetURLFetcher( | 60 privet_urlfetcher_.reset(new PrivetURLFetcher( |
| 61 kSamplePrivetToken, | |
| 62 GURL(kSamplePrivetURL), | 61 GURL(kSamplePrivetURL), |
| 63 net::URLFetcher::POST, | 62 net::URLFetcher::POST, |
| 64 request_context_.get(), | 63 request_context_.get(), |
| 65 &delegate_)); | 64 &delegate_)); |
| 65 |
| 66 PrivetURLFetcher::SetTokenForHost( |
| 67 GURL(kSamplePrivetURL).GetOrigin().spec(), |
| 68 kSamplePrivetToken); |
| 66 } | 69 } |
| 67 virtual ~PrivetURLFetcherTest() { | 70 virtual ~PrivetURLFetcherTest() { |
| 68 } | 71 } |
| 69 | 72 |
| 70 void RunFor(base::TimeDelta time_period) { | 73 void RunFor(base::TimeDelta time_period) { |
| 71 base::CancelableCallback<void()> callback(base::Bind( | 74 base::CancelableCallback<void()> callback(base::Bind( |
| 72 &PrivetURLFetcherTest::Stop, base::Unretained(this))); | 75 &PrivetURLFetcherTest::Stop, base::Unretained(this))); |
| 73 base::MessageLoop::current()->PostDelayedTask( | 76 base::MessageLoop::current()->PostDelayedTask( |
| 74 FROM_HERE, callback.callback(), time_period); | 77 FROM_HERE, callback.callback(), time_period); |
| 75 | 78 |
| (...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 166 ASSERT_TRUE(fetcher != NULL); | 169 ASSERT_TRUE(fetcher != NULL); |
| 167 net::HttpRequestHeaders headers; | 170 net::HttpRequestHeaders headers; |
| 168 fetcher->GetExtraRequestHeaders(&headers); | 171 fetcher->GetExtraRequestHeaders(&headers); |
| 169 | 172 |
| 170 std::string header_token; | 173 std::string header_token; |
| 171 ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); | 174 ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); |
| 172 EXPECT_EQ(kSamplePrivetToken, header_token); | 175 EXPECT_EQ(kSamplePrivetToken, header_token); |
| 173 } | 176 } |
| 174 | 177 |
| 175 TEST_F(PrivetURLFetcherTest, Header2) { | 178 TEST_F(PrivetURLFetcherTest, Header2) { |
| 176 privet_urlfetcher_.reset(new PrivetURLFetcher( | 179 PrivetURLFetcher::SetTokenForHost( |
| 177 "", | 180 GURL(kSamplePrivetURL).GetOrigin().spec(), |
| 178 GURL(kSamplePrivetURL), | 181 ""); |
| 179 net::URLFetcher::POST, | |
| 180 request_context_.get(), | |
| 181 &delegate_)); | |
| 182 | 182 |
| 183 privet_urlfetcher_->AllowEmptyPrivetToken(); | 183 privet_urlfetcher_->AllowEmptyPrivetToken(); |
| 184 privet_urlfetcher_->Start(); | 184 privet_urlfetcher_->Start(); |
| 185 | 185 |
| 186 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); | 186 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); |
| 187 ASSERT_TRUE(fetcher != NULL); | 187 ASSERT_TRUE(fetcher != NULL); |
| 188 net::HttpRequestHeaders headers; | 188 net::HttpRequestHeaders headers; |
| 189 fetcher->GetExtraRequestHeaders(&headers); | 189 fetcher->GetExtraRequestHeaders(&headers); |
| 190 | 190 |
| 191 std::string header_token; | 191 std::string header_token; |
| (...skipping 10 matching lines...) Expand all Loading... |
| 202 net::OK)); | 202 net::OK)); |
| 203 fetcher->set_response_code(200); | 203 fetcher->set_response_code(200); |
| 204 | 204 |
| 205 EXPECT_CALL(delegate_, OnParsedJsonInternal(true)); | 205 EXPECT_CALL(delegate_, OnParsedJsonInternal(true)); |
| 206 fetcher->delegate()->OnURLFetchComplete(fetcher); | 206 fetcher->delegate()->OnURLFetchComplete(fetcher); |
| 207 } | 207 } |
| 208 | 208 |
| 209 } // namespace | 209 } // namespace |
| 210 | 210 |
| 211 } // namespace local_discovery | 211 } // namespace local_discovery |
| OLD | NEW |