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