src/chrome/browser/geolocation/access_token_store_browsertest.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/bind.h"
   6 #include "base/bind_helpers.h"
   7 #include "base/strings/string_util.h"
   8 #include "base/strings/utf_string_conversions.h"
   9 #include "chrome/browser/geolocation/chrome_access_token_store.h"
  10 #include "chrome/test/base/in_process_browser_test.h"
  11 #include "chrome/test/base/ui_test_utils.h"
  12 #include "content/public/browser/access_token_store.h"
  13 #include "content/public/test/test_browser_thread.h"
  14
  15 using content::AccessTokenStore;
  16 using content::BrowserThread;
  17
  18 namespace {
  19
  20 // The token store factory implementation expects to be used from any well-known
  21 // chrome thread other than UI. We could use any arbitrary thread; IO is
  22 // a good choice as this is the expected usage.
  23 const BrowserThread::ID kExpectedClientThreadId = BrowserThread::IO;
  24 const char* kRefServerUrl1 = "https://test.domain.example/foo?id=bar.bar";
  25 const char* kRefServerUrl2 = "http://another.domain.example/foo?id=bar.bar#2";
  26 const char* kOldDefaultNetworkProviderUrl = "https://www.google.com/loc/json";
  27
  28 class GeolocationAccessTokenStoreTest
  29     : public InProcessBrowserTest {
  30  public:
  31   GeolocationAccessTokenStoreTest()
  32       : token_to_expect_(NULL), token_to_set_(NULL) {}
  33
  34   void DoTestStepAndWaitForResults(
  35       const char* ref_url, const string16* token_to_expect,
  36       const string16* token_to_set);
  37
  38   void OnAccessTokenStoresLoaded(
  39       AccessTokenStore::AccessTokenSet access_token_set,
  40       net::URLRequestContextGetter* context_getter);
  41
  42   scoped_refptr<AccessTokenStore> token_store_;
  43   GURL ref_url_;
  44   const string16* token_to_expect_;
  45   const string16* token_to_set_;
  46 };
  47
  48 void StartTestStepFromClientThread(
  49     scoped_refptr<AccessTokenStore>* store,
  50     const AccessTokenStore::LoadAccessTokensCallbackType& callback) {
  51   ASSERT_TRUE(BrowserThread::CurrentlyOn(kExpectedClientThreadId));
  52   if (store->get() == NULL)
  53     (*store) = new ChromeAccessTokenStore();
  54   (*store)->LoadAccessTokens(callback);
  55 }
  56
  57 struct TokenLoadClientForTest {
  58   void NotReachedCallback(AccessTokenStore::AccessTokenSet /*tokens*/,
  59                           net::URLRequestContextGetter* /*context_getter*/) {
  60     NOTREACHED() << "This request should have been canceled before callback";
  61   }
  62 };
  63
  64 void GeolocationAccessTokenStoreTest::DoTestStepAndWaitForResults(
  65     const char* ref_url, const string16* token_to_expect,
  66     const string16* token_to_set) {
  67   ref_url_ = GURL(ref_url);
  68   token_to_expect_ = token_to_expect;
  69   token_to_set_ = token_to_set;
  70
  71   BrowserThread::PostTask(
  72       kExpectedClientThreadId, FROM_HERE,
  73       base::Bind(
  74           &StartTestStepFromClientThread,
  75           &token_store_,
  76           base::Bind(
  77               &GeolocationAccessTokenStoreTest::OnAccessTokenStoresLoaded,
  78               base::Unretained(this))));
  79   content::RunMessageLoop();
  80 }
  81
  82 void GeolocationAccessTokenStoreTest::OnAccessTokenStoresLoaded(
  83     AccessTokenStore::AccessTokenSet access_token_set,
  84     net::URLRequestContextGetter* context_getter) {
  85   ASSERT_TRUE(BrowserThread::CurrentlyOn(kExpectedClientThreadId))
  86       << "Callback from token factory should be from the same thread as the "
  87          "LoadAccessTokenStores request was made on";
  88   DCHECK(context_getter);
  89   AccessTokenStore::AccessTokenSet::const_iterator item =
  90       access_token_set.find(ref_url_);
  91   if (!token_to_expect_) {
  92     EXPECT_TRUE(item == access_token_set.end());
  93   } else {
  94     EXPECT_FALSE(item == access_token_set.end());
  95     EXPECT_EQ(*token_to_expect_, item->second);
  96   }
  97
  98   if (token_to_set_) {
  99     scoped_refptr<AccessTokenStore> store(new ChromeAccessTokenStore());
 100     store->SaveAccessToken(ref_url_, *token_to_set_);
 101   }
 102   BrowserThread::PostTask(
 103       BrowserThread::UI, FROM_HERE, base::MessageLoop::QuitClosure());
 104 }
 105
 106 IN_PROC_BROWSER_TEST_F(GeolocationAccessTokenStoreTest, SetAcrossInstances) {
 107   const string16 ref_token1 = ASCIIToUTF16("jksdfo90,'s#\"#1*(");
 108   const string16 ref_token2 = ASCIIToUTF16("\1\2\3\4\5\6\7\10\11\12=023");
 109   ASSERT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::UI));
 110
 111   DoTestStepAndWaitForResults(kRefServerUrl1, NULL, &ref_token1);
 112   // Check it was set, and change to new value.
 113   DoTestStepAndWaitForResults(kRefServerUrl1, &ref_token1, &ref_token2);
 114   // And change back.
 115   DoTestStepAndWaitForResults(kRefServerUrl1, &ref_token2, &ref_token1);
 116   DoTestStepAndWaitForResults(kRefServerUrl1, &ref_token1, NULL);
 117
 118   // Set a second server URL
 119   DoTestStepAndWaitForResults(kRefServerUrl2, NULL, &ref_token2);
 120   DoTestStepAndWaitForResults(kRefServerUrl2, &ref_token2, NULL);
 121   DoTestStepAndWaitForResults(kRefServerUrl1, &ref_token1, NULL);
 122 }
 123
 124 IN_PROC_BROWSER_TEST_F(GeolocationAccessTokenStoreTest, OldUrlRemoval) {
 125   const string16 ref_token1 = ASCIIToUTF16("jksdfo90,'s#\"#1*(");
 126   ASSERT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::UI));
 127
 128   // Set a token for the old default network provider url.
 129   DoTestStepAndWaitForResults(kOldDefaultNetworkProviderUrl,
 130                               NULL, &ref_token1);
 131   // Check that the token related to the old default network provider url
 132   // was deleted.
 133   DoTestStepAndWaitForResults(kOldDefaultNetworkProviderUrl,
 134                               NULL, NULL);
 135 }
 136
 137 }  // namespace