Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(262)

Side by Side Diff: chrome/browser/local_discovery/privet_url_fetcher_unittest.cc

Issue 166373006: Use persistent map of privet tokens for PrivetURLFetcher (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 6 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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
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
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
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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698