OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "base/message_loop/message_loop.h" | |
6 #include "chrome/browser/local_discovery/cloud_print_account_manager.h" | |
7 #include "net/http/http_request_headers.h" | |
8 #include "net/url_request/test_url_fetcher_factory.h" | |
9 #include "net/url_request/url_request_test_util.h" | |
10 #include "testing/gmock/include/gmock/gmock.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 namespace local_discovery { | |
14 | |
15 namespace { | |
16 | |
17 const char kSampleResponse[] = "{" | |
18 " \"success\": true," | |
19 " \"xsrf_token\": \"sample\"," | |
20 " \"request\" : { " | |
21 " \"users\": [\"first@gmail.com\", \"second@gmail.com\"]" | |
22 " } " | |
23 "}"; | |
24 | |
25 const char kSampleResponseFailure[] = "{" | |
26 " \"success\": false," | |
27 "}"; | |
28 | |
29 class MockCallback { | |
30 public: | |
31 MOCK_METHOD2(CloudPrintAccountsResolved, void( | |
32 const std::vector<std::string>& account, | |
33 const std::string& xsrf_token)); | |
34 }; | |
35 | |
36 class CloudPrintAccountManagerTest : public testing::Test { | |
37 public: | |
38 CloudPrintAccountManagerTest() | |
39 : request_context_( | |
40 new net::TestURLRequestContextGetter( | |
41 base::MessageLoopProxy::current())), | |
42 account_manager_( | |
43 request_context_.get(), | |
44 "https://www.google.com/cloudprint", | |
45 1, | |
46 base::Bind( | |
47 &MockCallback::CloudPrintAccountsResolved, | |
48 base::Unretained(&mock_callback_))) { | |
49 } | |
50 | |
51 virtual ~CloudPrintAccountManagerTest() { | |
52 } | |
53 | |
54 protected: | |
55 base::MessageLoop message_loop_; | |
56 scoped_refptr<net::TestURLRequestContextGetter> request_context_; | |
57 net::TestURLFetcherFactory fetcher_factory_; | |
58 MockCallback mock_callback_; | |
59 CloudPrintAccountManager account_manager_; | |
60 }; | |
61 | |
62 TEST_F(CloudPrintAccountManagerTest, Success) { | |
63 account_manager_.Start(); | |
64 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); | |
65 | |
66 net::HttpRequestHeaders headers; | |
67 std::string proxy; | |
68 fetcher->GetExtraRequestHeaders(&headers); | |
69 EXPECT_TRUE(headers.GetHeader("X-Cloudprint-Proxy", &proxy)); | |
70 EXPECT_EQ("Chrome", proxy); | |
71 EXPECT_EQ(GURL("https://www.google.com/cloudprint/list?proxy=none&user=1"), | |
72 fetcher->GetOriginalURL()); | |
73 | |
74 fetcher->SetResponseString(kSampleResponse); | |
75 fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::SUCCESS, | |
76 net::OK)); | |
77 fetcher->set_response_code(200); | |
78 | |
79 std::vector<std::string> expected_users; | |
80 expected_users.push_back("first@gmail.com"); | |
81 expected_users.push_back("second@gmail.com"); | |
82 | |
83 EXPECT_CALL(mock_callback_, | |
84 CloudPrintAccountsResolved(expected_users, "sample")); | |
85 fetcher->delegate()->OnURLFetchComplete(fetcher); | |
86 } | |
87 | |
88 TEST_F(CloudPrintAccountManagerTest, FailureJSON) { | |
89 account_manager_.Start(); | |
90 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); | |
91 | |
92 fetcher->SetResponseString(kSampleResponseFailure); | |
93 fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::SUCCESS, | |
94 net::OK)); | |
95 fetcher->set_response_code(200); | |
96 | |
97 EXPECT_CALL(mock_callback_, | |
98 CloudPrintAccountsResolved(std::vector<std::string>(), "")); | |
99 fetcher->delegate()->OnURLFetchComplete(fetcher); | |
100 } | |
101 | |
102 } // namespace | |
103 | |
104 } // namespace local_discovery | |
OLD | NEW |