Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 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/media/router/discovery/dial/device_description_service. h" | 5 #include "chrome/browser/media/router/discovery/dial/device_description_service. h" |
| 6 | 6 |
| 7 #include "base/strings/stringprintf.h" | 7 #include "base/strings/stringprintf.h" |
| 8 #include "base/test/mock_callback.h" | 8 #include "base/test/mock_callback.h" |
| 9 #include "chrome/browser/media/router/discovery/dial/device_description_fetcher. h" | 9 #include "chrome/browser/media/router/discovery/dial/device_description_fetcher. h" |
| 10 #include "chrome/browser/media/router/discovery/dial/dial_device_data.h" | 10 #include "chrome/browser/media/router/discovery/dial/dial_device_data.h" |
| (...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 60 | 60 |
| 61 } // namespace | 61 } // namespace |
| 62 | 62 |
| 63 namespace media_router { | 63 namespace media_router { |
| 64 | 64 |
| 65 class TestSafeDialDeviceDescriptionParser | 65 class TestSafeDialDeviceDescriptionParser |
| 66 : public SafeDialDeviceDescriptionParser { | 66 : public SafeDialDeviceDescriptionParser { |
| 67 public: | 67 public: |
| 68 ~TestSafeDialDeviceDescriptionParser() override {} | 68 ~TestSafeDialDeviceDescriptionParser() override {} |
| 69 | 69 |
| 70 MOCK_METHOD2(Start, | 70 MOCK_METHOD2(StartMock, |
|
mark a. foltz
2017/05/09 22:28:44
s/StartMock/StartInternal/ for consistency with ot
tzik
2017/05/11 09:00:30
Done.
| |
| 71 void(const std::string& xml_text, | 71 void(const std::string& xml_text, |
| 72 const DeviceDescriptionCallback& callback)); | 72 DeviceDescriptionCallback& callback)); |
| 73 void Start(const std::string& xml_text, DeviceDescriptionCallback callback) { | |
| 74 StartMock(xml_text, callback); | |
| 75 } | |
| 73 }; | 76 }; |
| 74 | 77 |
| 75 class TestDeviceDescriptionService : public DeviceDescriptionService { | 78 class TestDeviceDescriptionService : public DeviceDescriptionService { |
| 76 public: | 79 public: |
| 77 TestDeviceDescriptionService( | 80 TestDeviceDescriptionService( |
| 78 const DeviceDescriptionParseSuccessCallback& success_cb, | 81 const DeviceDescriptionParseSuccessCallback& success_cb, |
| 79 const DeviceDescriptionParseErrorCallback& error_cb) | 82 const DeviceDescriptionParseErrorCallback& error_cb) |
| 80 : DeviceDescriptionService(success_cb, error_cb) {} | 83 : DeviceDescriptionService(success_cb, error_cb) {} |
| 81 | 84 |
| 82 MOCK_METHOD0(CreateSafeParser, SafeDialDeviceDescriptionParser*()); | 85 MOCK_METHOD0(CreateSafeParser, SafeDialDeviceDescriptionParser*()); |
| (...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 177 std::vector<DialDeviceData> devices = {device_data}; | 180 std::vector<DialDeviceData> devices = {device_data}; |
| 178 | 181 |
| 179 // Create Fetcher | 182 // Create Fetcher |
| 180 EXPECT_TRUE(fetcher_map_.empty()); | 183 EXPECT_TRUE(fetcher_map_.empty()); |
| 181 device_description_service()->GetDeviceDescriptions( | 184 device_description_service()->GetDeviceDescriptions( |
| 182 devices, profile_.GetRequestContext()); | 185 devices, profile_.GetRequestContext()); |
| 183 EXPECT_EQ(size_t(1), fetcher_map_.size()); | 186 EXPECT_EQ(size_t(1), fetcher_map_.size()); |
| 184 | 187 |
| 185 // Remove fetcher and create safe parser | 188 // Remove fetcher and create safe parser |
| 186 auto test_parser = base::MakeUnique<TestSafeDialDeviceDescriptionParser>(); | 189 auto test_parser = base::MakeUnique<TestSafeDialDeviceDescriptionParser>(); |
| 187 EXPECT_CALL(*test_parser, Start(_, _)); | 190 EXPECT_CALL(*test_parser, StartMock(_, _)); |
| 188 | 191 |
| 189 SetTestParser(std::move(test_parser)); | 192 SetTestParser(std::move(test_parser)); |
| 190 OnDeviceDescriptionFetchComplete(1); | 193 OnDeviceDescriptionFetchComplete(1); |
| 191 } | 194 } |
| 192 | 195 |
| 193 TEST_F(DeviceDescriptionServiceTest, TestGetDeviceDescriptionFetchURLError) { | 196 TEST_F(DeviceDescriptionServiceTest, TestGetDeviceDescriptionFetchURLError) { |
| 194 DialDeviceData device_data = CreateDialDeviceData(1); | 197 DialDeviceData device_data = CreateDialDeviceData(1); |
| 195 std::vector<DialDeviceData> devices; | 198 std::vector<DialDeviceData> devices; |
| 196 devices.push_back(device_data); | 199 devices.push_back(device_data); |
| 197 | 200 |
| (...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 304 DialDeviceData device_data_2 = CreateDialDeviceData(2); | 307 DialDeviceData device_data_2 = CreateDialDeviceData(2); |
| 305 DialDeviceData device_data_3 = CreateDialDeviceData(3); | 308 DialDeviceData device_data_3 = CreateDialDeviceData(3); |
| 306 | 309 |
| 307 std::vector<DialDeviceData> devices = {device_data_1, device_data_2, | 310 std::vector<DialDeviceData> devices = {device_data_1, device_data_2, |
| 308 device_data_3}; | 311 device_data_3}; |
| 309 | 312 |
| 310 // insert fetchers | 313 // insert fetchers |
| 311 device_description_service()->GetDeviceDescriptions( | 314 device_description_service()->GetDeviceDescriptions( |
| 312 devices, profile_.GetRequestContext()); | 315 devices, profile_.GetRequestContext()); |
| 313 auto test_parser = base::MakeUnique<TestSafeDialDeviceDescriptionParser>(); | 316 auto test_parser = base::MakeUnique<TestSafeDialDeviceDescriptionParser>(); |
| 314 EXPECT_CALL(*test_parser, Start(_, _)).Times(3); | 317 EXPECT_CALL(*test_parser, StartMock(_, _)).Times(3); |
| 315 | 318 |
| 316 EXPECT_FALSE(device_description_service()->parser_); | 319 EXPECT_FALSE(device_description_service()->parser_); |
| 317 SetTestParser(std::move(test_parser)); | 320 SetTestParser(std::move(test_parser)); |
| 318 OnDeviceDescriptionFetchComplete(1); | 321 OnDeviceDescriptionFetchComplete(1); |
| 319 | 322 |
| 320 EXPECT_TRUE(device_description_service()->parser_); | 323 EXPECT_TRUE(device_description_service()->parser_); |
| 321 OnDeviceDescriptionFetchComplete(2); | 324 OnDeviceDescriptionFetchComplete(2); |
| 322 OnDeviceDescriptionFetchComplete(3); | 325 OnDeviceDescriptionFetchComplete(3); |
| 323 | 326 |
| 324 EXPECT_FALSE(device_description_service()->parser_); | 327 EXPECT_FALSE(device_description_service()->parser_); |
| 325 } | 328 } |
| 326 | 329 |
| 327 } // namespace media_router | 330 } // namespace media_router |
| OLD | NEW |