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 |