Use persistent map of privet tokens for PrivetURLFetcher

chrome/browser/local_discovery/privet_url_fetcher_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/local_discovery/privet_url_fetcher.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ skipping 71 matching lines @@
   scoped_ptr<base::DictionaryValue> saved_value_;
   bool raw_mode_;
 };
 
 class PrivetURLFetcherTest : public ::testing::Test {
  public:
   PrivetURLFetcherTest() {
     request_context_= new net::TestURLRequestContextGetter(
         base::MessageLoopProxy::current());
     privet_urlfetcher_.reset(new PrivetURLFetcher(
-        kSamplePrivetToken,
         GURL(kSamplePrivetURL),
         net::URLFetcher::POST,
         request_context_.get(),
         &delegate_));
+
+    PrivetURLFetcher::SetTokenForHost(GURL(kSamplePrivetURL).GetOrigin().spec(),
+                                      kSamplePrivetToken);
   }
   virtual ~PrivetURLFetcherTest() {
   }
 
   void RunFor(base::TimeDelta time_period) {
     base::CancelableCallback<void()> callback(base::Bind(
         &PrivetURLFetcherTest::Stop, base::Unretained(this)));
     base::MessageLoop::current()->PostDelayedTask(
         FROM_HERE, callback.callback(), time_period);
 
@@ skipping 90 matching lines @@
   ASSERT_TRUE(fetcher != NULL);
   net::HttpRequestHeaders headers;
   fetcher->GetExtraRequestHeaders(&headers);
 
   std::string header_token;
   ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token));
   EXPECT_EQ(kSamplePrivetToken, header_token);
 }
 
 TEST_F(PrivetURLFetcherTest, Header2) {
-  privet_urlfetcher_.reset(new PrivetURLFetcher(
-      "",
-      GURL(kSamplePrivetURL),
-      net::URLFetcher::POST,
-      request_context_.get(),
-      &delegate_));
+  PrivetURLFetcher::SetTokenForHost(GURL(kSamplePrivetURL).GetOrigin().spec(),
+                                    "");
 
   privet_urlfetcher_->AllowEmptyPrivetToken();
   privet_urlfetcher_->Start();
 
   net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0);
   ASSERT_TRUE(fetcher != NULL);
   net::HttpRequestHeaders headers;
   fetcher->GetExtraRequestHeaders(&headers);
 
   std::string header_token;
@@ skipping 54 matching lines @@
                                             net::OK));
   fetcher->set_response_code(200);
 
   EXPECT_CALL(delegate_, OnFileInternal());
   fetcher->delegate()->OnURLFetchComplete(fetcher);
 }
 
 }  // namespace
 
 }  // namespace local_discovery