Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/printing/cloud_print/privet_notifications.h" | 5 #include "chrome/browser/printing/cloud_print/privet_notifications.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
| 10 #include "base/threading/thread_task_runner_handle.h" | 10 #include "base/threading/thread_task_runner_handle.h" |
| (...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 62 } | 62 } |
| 63 | 63 |
| 64 const std::string& GetName() override { return name_; } | 64 const std::string& GetName() override { return name_; } |
| 65 | 65 |
| 66 private: | 66 private: |
| 67 std::string name_; | 67 std::string name_; |
| 68 scoped_refptr<net::URLRequestContextGetter> request_context_; | 68 scoped_refptr<net::URLRequestContextGetter> request_context_; |
| 69 }; | 69 }; |
| 70 | 70 |
| 71 explicit MockPrivetHttpFactory(net::URLRequestContextGetter* request_context) | 71 explicit MockPrivetHttpFactory(net::URLRequestContextGetter* request_context) |
| 72 : request_context_(request_context) { | 72 : request_context_(request_context) {} |
|
Lei Zhang
2016/06/01 02:15:17
I let "git cl format" reformat the file.
| |
| 73 } | |
| 74 | 73 |
| 75 std::unique_ptr<PrivetHTTPResolution> CreatePrivetHTTP( | 74 std::unique_ptr<PrivetHTTPResolution> CreatePrivetHTTP( |
| 76 const std::string& name) override { | 75 const std::string& name) override { |
| 77 return base::WrapUnique(new MockResolution(name, request_context_.get())); | 76 return base::WrapUnique(new MockResolution(name, request_context_.get())); |
| 78 } | 77 } |
| 79 | 78 |
| 80 private: | 79 private: |
| 81 scoped_refptr<net::URLRequestContextGetter> request_context_; | 80 scoped_refptr<net::URLRequestContextGetter> request_context_; |
| 82 }; | 81 }; |
| 83 | 82 |
| 84 class PrivetNotificationsListenerTest : public ::testing::Test { | 83 class PrivetNotificationsListenerTest : public ::testing::Test { |
| 85 public: | 84 public: |
| 86 PrivetNotificationsListenerTest() | 85 PrivetNotificationsListenerTest() |
| 87 : request_context_(new net::TestURLRequestContextGetter( | 86 : request_context_(new net::TestURLRequestContextGetter( |
| 88 base::ThreadTaskRunnerHandle::Get())) { | 87 base::ThreadTaskRunnerHandle::Get())) { |
| 89 notification_listener_.reset(new PrivetNotificationsListener( | 88 notification_listener_.reset(new PrivetNotificationsListener( |
| 90 std::unique_ptr<PrivetHTTPAsynchronousFactory>( | 89 std::unique_ptr<PrivetHTTPAsynchronousFactory>( |
| 91 new MockPrivetHttpFactory(request_context_.get())), | 90 new MockPrivetHttpFactory(request_context_.get())), |
| 92 &mock_delegate_)); | 91 &mock_delegate_)); |
| 93 | 92 |
| 94 description_.name = kExampleDeviceHumanName; | 93 description_.name = kExampleDeviceHumanName; |
| 95 description_.description = kExampleDeviceDescription; | 94 description_.description = kExampleDeviceDescription; |
| 96 } | 95 } |
| 97 | 96 |
| 98 virtual ~PrivetNotificationsListenerTest() { | 97 virtual ~PrivetNotificationsListenerTest() {} |
| 99 } | |
| 100 | 98 |
| 101 bool SuccessfulResponseToInfo(const std::string& response) { | 99 bool SuccessfulResponseToInfo(const std::string& response) { |
| 102 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); | 100 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); |
| 103 EXPECT_TRUE(fetcher); | 101 EXPECT_TRUE(fetcher); |
| 104 EXPECT_EQ(GURL(kDeviceInfoURL), fetcher->GetOriginalURL()); | 102 EXPECT_EQ(GURL(kDeviceInfoURL), fetcher->GetOriginalURL()); |
| 105 | 103 |
| 106 if (!fetcher || GURL(kDeviceInfoURL) != fetcher->GetOriginalURL()) | 104 if (!fetcher || GURL(kDeviceInfoURL) != fetcher->GetOriginalURL()) |
| 107 return false; | 105 return false; |
| 108 | 106 |
| 109 fetcher->SetResponseString(response); | 107 fetcher->SetResponseString(response); |
| 110 fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::SUCCESS, | 108 fetcher->set_status( |
| 111 net::OK)); | 109 net::URLRequestStatus(net::URLRequestStatus::SUCCESS, net::OK)); |
| 112 fetcher->set_response_code(200); | 110 fetcher->set_response_code(200); |
| 113 fetcher->delegate()->OnURLFetchComplete(fetcher); | 111 fetcher->delegate()->OnURLFetchComplete(fetcher); |
| 114 return true; | 112 return true; |
| 115 } | 113 } |
| 116 | 114 |
| 117 protected: | 115 protected: |
| 118 StrictMock<MockPrivetNotificationsListenerDeleagate> mock_delegate_; | 116 StrictMock<MockPrivetNotificationsListenerDeleagate> mock_delegate_; |
| 119 std::unique_ptr<PrivetNotificationsListener> notification_listener_; | 117 std::unique_ptr<PrivetNotificationsListener> notification_listener_; |
| 120 base::MessageLoop message_loop_; | 118 base::MessageLoop message_loop_; |
| 121 scoped_refptr<net::TestURLRequestContextGetter> request_context_; | 119 scoped_refptr<net::TestURLRequestContextGetter> request_context_; |
| 122 net::TestURLFetcherFactory fetcher_factory_; | 120 net::TestURLFetcherFactory fetcher_factory_; |
| 123 DeviceDescription description_; | 121 DeviceDescription description_; |
| 124 }; | 122 }; |
| 125 | 123 |
| 126 TEST_F(PrivetNotificationsListenerTest, DisappearReappearTest) { | 124 TEST_F(PrivetNotificationsListenerTest, DisappearReappearTest) { |
| 127 EXPECT_CALL(mock_delegate_, PrivetNotify( | 125 EXPECT_CALL(mock_delegate_, PrivetNotify(1, true)); |
| 128 1, | |
| 129 true)); | |
| 130 | 126 |
| 131 notification_listener_->DeviceChanged( | 127 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 132 kExampleDeviceName, | |
| 133 description_); | |
| 134 | 128 |
| 135 SuccessfulResponseToInfo(kInfoResponseUptime20); | 129 SuccessfulResponseToInfo(kInfoResponseUptime20); |
| 136 | 130 |
| 137 EXPECT_CALL(mock_delegate_, PrivetRemoveNotification()); | 131 EXPECT_CALL(mock_delegate_, PrivetRemoveNotification()); |
| 138 | 132 |
| 139 notification_listener_->DeviceRemoved( | 133 notification_listener_->DeviceRemoved(kExampleDeviceName); |
| 140 kExampleDeviceName); | |
| 141 | 134 |
| 142 notification_listener_->DeviceChanged( | 135 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 143 kExampleDeviceName, | |
| 144 description_); | |
| 145 | 136 |
| 146 description_.id = kExampleDeviceID; | 137 description_.id = kExampleDeviceID; |
| 147 | 138 |
| 148 notification_listener_->DeviceChanged( | 139 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 149 kExampleDeviceName, | |
| 150 description_); | |
| 151 } | 140 } |
| 152 | 141 |
| 153 TEST_F(PrivetNotificationsListenerTest, RegisterTest) { | 142 TEST_F(PrivetNotificationsListenerTest, RegisterTest) { |
| 154 EXPECT_CALL(mock_delegate_, PrivetNotify( | 143 EXPECT_CALL(mock_delegate_, PrivetNotify(1, true)); |
| 155 1, | |
| 156 true)); | |
| 157 | 144 |
| 158 notification_listener_->DeviceChanged( | 145 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 159 kExampleDeviceName, | |
| 160 description_); | |
| 161 | 146 |
| 162 SuccessfulResponseToInfo(kInfoResponseUptime20); | 147 SuccessfulResponseToInfo(kInfoResponseUptime20); |
| 163 | 148 |
| 164 EXPECT_CALL(mock_delegate_, PrivetRemoveNotification()); | 149 EXPECT_CALL(mock_delegate_, PrivetRemoveNotification()); |
| 165 | 150 |
| 166 description_.id = kExampleDeviceID; | 151 description_.id = kExampleDeviceID; |
| 167 | 152 |
| 168 notification_listener_->DeviceChanged( | 153 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 169 kExampleDeviceName, | |
| 170 description_); | |
| 171 } | 154 } |
| 172 | 155 |
| 173 TEST_F(PrivetNotificationsListenerTest, HighUptimeTest) { | 156 TEST_F(PrivetNotificationsListenerTest, HighUptimeTest) { |
| 174 notification_listener_->DeviceChanged( | 157 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 175 kExampleDeviceName, | |
| 176 description_); | |
| 177 | 158 |
| 178 SuccessfulResponseToInfo(kInfoResponseUptime3600); | 159 SuccessfulResponseToInfo(kInfoResponseUptime3600); |
| 179 | 160 |
| 180 description_.id = kExampleDeviceID; | 161 description_.id = kExampleDeviceID; |
| 181 | 162 |
| 182 notification_listener_->DeviceChanged( | 163 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 183 kExampleDeviceName, | |
| 184 description_); | |
| 185 } | 164 } |
| 186 | 165 |
| 187 TEST_F(PrivetNotificationsListenerTest, HTTPErrorTest) { | 166 TEST_F(PrivetNotificationsListenerTest, HTTPErrorTest) { |
| 188 notification_listener_->DeviceChanged( | 167 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 189 kExampleDeviceName, | |
| 190 description_); | |
| 191 | 168 |
| 192 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); | 169 net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); |
| 193 | 170 |
| 194 fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::SUCCESS, | 171 fetcher->set_status( |
| 195 net::OK)); | 172 net::URLRequestStatus(net::URLRequestStatus::SUCCESS, net::OK)); |
| 196 fetcher->set_response_code(200); | 173 fetcher->set_response_code(200); |
| 197 fetcher->delegate()->OnURLFetchComplete(fetcher); | 174 fetcher->delegate()->OnURLFetchComplete(fetcher); |
| 198 } | 175 } |
| 199 | 176 |
| 200 TEST_F(PrivetNotificationsListenerTest, DictionaryErrorTest) { | 177 TEST_F(PrivetNotificationsListenerTest, DictionaryErrorTest) { |
| 201 notification_listener_->DeviceChanged( | 178 notification_listener_->DeviceChanged(kExampleDeviceName, description_); |
| 202 kExampleDeviceName, | |
| 203 description_); | |
| 204 | 179 |
| 205 SuccessfulResponseToInfo(kInfoResponseNoUptime); | 180 SuccessfulResponseToInfo(kInfoResponseNoUptime); |
| 206 } | 181 } |
| 207 | 182 |
| 208 } // namespace | 183 } // namespace |
| 209 | 184 |
| 210 } // namespace cloud_print | 185 } // namespace cloud_print |
| OLD | NEW |