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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
// 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/command_line.h"
#include "base/compiler_specific.h"
#include "base/memory/weak_ptr.h"
#include "base/message_loop.h"
#include "chrome/browser/extensions/app_notify_channel_setup.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/testing_pref_service.h"
#include "chrome/test/base/testing_profile.h"
#include "content/test/test_browser_thread.h"
#include "content/test/test_url_fetcher_factory.h"
#include "googleurl/src/gurl.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
const int kRouteId = 4;
const int kCallbackId = 5;
class TestDelegate : public AppNotifyChannelSetup::Delegate,
public base::SupportsWeakPtr<TestDelegate> {
public:
TestDelegate() : was_called_(false) {}
virtual ~TestDelegate() {}
virtual void AppNotifyChannelSetupComplete(
const std::string& channel_id,
const std::string& error,
int route_id,
int callback_id) OVERRIDE {
EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::UI));
EXPECT_FALSE(was_called_);
was_called_ = true;
error_ = error;
route_id_ = route_id;
callback_id_ = callback_id;
MessageLoop::current()->Quit();
}
// Called to check that we were called with the expected arguments.
void ExpectWasCalled(const std::string& expected_channel_id,
const std::string& expected_error) {
EXPECT_TRUE(was_called_);
EXPECT_EQ(expected_error, error_);
EXPECT_EQ(kRouteId, route_id_);
EXPECT_EQ(kCallbackId, callback_id_);
}
private:
// Has our callback been called yet?
bool was_called_;
// When our AppNotifyChannelSetupComplete method is called, we copy the
// arguments into these member variables.
std::string channel_id_;
std::string error_;
int route_id_;
int callback_id_;
DISALLOW_COPY_AND_ASSIGN(TestDelegate);
};
} // namespace
class AppNotifyChannelSetupTest : public testing::Test {
public:
AppNotifyChannelSetupTest() : ui_thread_(BrowserThread::UI, &message_loop_) {}
virtual ~AppNotifyChannelSetupTest() {}
virtual void SetChannelServerUrl(GURL channel_server_url) {
CommandLine* command_line = CommandLine::ForCurrentProcess();
command_line->AppendSwitchASCII(switches::kAppNotifyChannelServerURL,
channel_server_url.spec());
}
virtual void SetLoggedInUser(const std::string username) {
TestingPrefService* prefs = profile_.GetTestingPrefService();
prefs->SetUserPref(prefs::kGoogleServicesUsername,
new StringValue(username));
}
virtual void RunServerTest(bool url_fetch_should_succeed,
const std::string& expected_code,
const std::string& expected_error) {
GURL page_url("http://www.google.com");
GURL server_url("http://dummy.google.com/app_notify_setup");
FakeURLFetcherFactory factory;
factory.SetFakeResponse(server_url.spec(),
"whatever",
url_fetch_should_succeed);
SetChannelServerUrl(server_url);
SetLoggedInUser("user@gmail.com");
scoped_refptr<AppNotifyChannelSetup > setup =
new AppNotifyChannelSetup(&profile_,
"1234",
page_url,
kRouteId,
kCallbackId,
delegate_.AsWeakPtr());
setup->Start();
message_loop_.Run();
delegate_.ExpectWasCalled(expected_code, expected_error);
}
protected:
MessageLoop message_loop_;
content::TestBrowserThread ui_thread_;
TestingProfile profile_;
TestDelegate delegate_;
};
TEST_F(AppNotifyChannelSetupTest, NotAvailable) {
GURL url("http://www.google.com");
scoped_refptr<AppNotifyChannelSetup > setup =
new AppNotifyChannelSetup(&profile_,
"1234",
url,
kRouteId,
kCallbackId,
delegate_.AsWeakPtr());
setup->Start();
message_loop_.Run();
delegate_.ExpectWasCalled(
std::string(), std::string("not_available"));
}
TEST_F(AppNotifyChannelSetupTest, ServerSuccess) {
RunServerTest(true, "dummy_do_not_use", "");
}
TEST_F(AppNotifyChannelSetupTest, ServerFailure) {
RunServerTest(false, "", "channel_service_error");
}
|
