[Media Router] Add DialMediaSinkService and DeviceDescriptionService

chrome/browser/media/router/discovery/dial/device_description_service_unittest.cc

Index: chrome/browser/media/router/discovery/dial/device_description_service_unittest.cc
diff --git a/chrome/browser/media/router/discovery/dial/device_description_service_unittest.cc b/chrome/browser/media/router/discovery/dial/device_description_service_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..8b28da0420c03336f500a6831f97a1c7481ad8ef
--- /dev/null
+++ b/chrome/browser/media/router/discovery/dial/device_description_service_unittest.cc
@@ -0,0 +1,311 @@
+// Copyright 2017 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 "chrome/browser/media/router/discovery/dial/device_description_service.h"
+
+#include "base/strings/stringprintf.h"
+#include "base/test/mock_callback.h"
+#include "chrome/browser/media/router/discovery/dial/device_description_fetcher.h"
+#include "chrome/browser/media/router/discovery/dial/dial_device_data.h"
+#include "chrome/browser/media/router/discovery/dial/parsed_dial_device_description.h"
+#include "chrome/browser/media/router/discovery/dial/safe_dial_device_description_parser.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using ::testing::_;
+using ::testing::Return;
+using ::testing::SaveArg;
+
+namespace media_router {
+
+class TestSafeDialDeviceDescriptionParser
+    : public SafeDialDeviceDescriptionParser {
+ public:
+  void Start(const std::string& xml_text,

[imcheng, 2017/04/25 01:40:12]

Do you intend to mock out this method?

[zhaobin, 2017/04/26 18:50:04]

Done.

+             const DeviceDescriptionCallback& callback) override {}
+};
+
+class TestDeviceDescriptionService : public DeviceDescriptionService {
+ public:
+  TestDeviceDescriptionService(
+      const DeviceDescriptionParseSuccessCallback& success_cb,
+      const DeviceDescriptionParseErrorCallback& error_cb)
+      : DeviceDescriptionService(success_cb, error_cb) {}
+
+  MOCK_METHOD0(CreateSafeParser, SafeDialDeviceDescriptionParser*());
+};
+
+class DeviceDescriptionServiceTest : public ::testing::Test {
+ public:
+  DeviceDescriptionServiceTest()
+      : device_description_service_(mock_success_cb_.Get(),
+                                    mock_error_cb_.Get()),
+        fetcher_map_(
+            device_description_service_.device_description_fetcher_map_),
+        description_map_(device_description_service_.description_map_),
+        thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {}
+
+  TestDeviceDescriptionService* device_description_service() {
+    return &device_description_service_;
+  }
+
+  // Create Test Data

[mark a. foltz, 2017/04/25 21:06:26]

These static methods to create test data could go into an anonymous namespace at
the top level - just to keep this class focused on methods that manage the test
maps.  I don't feel strongly though.

[zhaobin, 2017/04/26 18:50:04]

Done.

+  static DialDeviceData CreateDialDeviceData(int num) {
+    DialDeviceData device_data(
+        base::StringPrintf("Device id %d", num),
+        GURL(base::StringPrintf("http://192.168.1.%d/dd.xml", num)),
+        base::Time::Now());
+    device_data.set_label(base::StringPrintf("Device label %d", num));
+    return device_data;
+  }
+
+  static DialDeviceDescriptionData CreateDialDeviceDescriptionData(int num) {
+    return DialDeviceDescriptionData(
+        "", GURL(base::StringPrintf("http://192.168.1.%d/apps", num)));
+  }
+
+  static chrome::mojom::DialDeviceDescriptionPtr CreateDialDeviceDescriptionPtr(
+      int num) {
+    chrome::mojom::DialDeviceDescriptionPtr description_ptr =
+        chrome::mojom::DialDeviceDescription::New();
+    description_ptr->friendly_name =
+        base::StringPrintf("Friendly name %d", num);
+    description_ptr->model_name = base::StringPrintf("Model name %d", num);
+    description_ptr->unique_id = base::StringPrintf("Unique ID %d", num);
+    return description_ptr;
+  }
+
+  static ParsedDialDeviceDescription CreateParsedDialDeviceDescription(
+      int num) {
+    ParsedDialDeviceDescription description_data;
+    description_data.app_url =
+        GURL(base::StringPrintf("http://192.168.1.%d/apps", num));
+    description_data.friendly_name =
+        base::StringPrintf("Friendly name %d", num);
+    description_data.model_name = base::StringPrintf("Model name %d", num);
+    description_data.unique_id = base::StringPrintf("Unique ID %d", num);
+    return description_data;
+  }
+
+  void AddToCache(const std::string& device_label,
+                  const ParsedDialDeviceDescription& description_data,
+                  bool expired) {
+    DeviceDescriptionService::CacheEntry cache_entry;
+    cache_entry.expire_time = base::Time::Now();
+    if (!expired)
+      cache_entry.expire_time += base::TimeDelta::FromHours(12);
+    cache_entry.description_data = description_data;
+    description_map_[device_label] = cache_entry;
+  }
+
+  void OnDeviceDescriptionFetchComplete(int num) {
+    if (!device_description_service_.parser_) {
+      device_description_service_.parser_ =
+          base::MakeUnique<TestSafeDialDeviceDescriptionParser>();
+    }
+
+    auto device_data = CreateDialDeviceData(num);
+    auto description_response_data = CreateDialDeviceDescriptionData(num);
+    auto parsed_description_data = CreateParsedDialDeviceDescription(num);
+
+    EXPECT_CALL(mock_success_cb_, Run(device_data, parsed_description_data));
+
+    device_description_service_.OnDeviceDescriptionFetchComplete(
+        device_data, description_response_data);
+    device_description_service_.OnParsedDeviceDescription(
+        device_data, description_response_data.app_url,
+        CreateDialDeviceDescriptionPtr(num));
+  }
+
+  void TestOnParsedDeviceDescription(
+      chrome::mojom::DialDeviceDescriptionPtr description_ptr,
+      const std::string& error_message) {
+    GURL app_url("http://192.168.1.1/apps");
+    auto device_data = CreateDialDeviceData(1);
+    auto description_data = CreateParsedDialDeviceDescription(1);
+
+    if (!error_message.empty()) {
+      EXPECT_CALL(mock_error_cb_, Run(device_data, error_message));
+    } else {
+      EXPECT_CALL(mock_success_cb_, Run(device_data, description_data));
+    }
+    device_description_service()->OnParsedDeviceDescription(
+        device_data, app_url, std::move(description_ptr));
+  }
+
+ protected:
+  base::MockCallback<
+      DeviceDescriptionService::DeviceDescriptionParseSuccessCallback>
+      mock_success_cb_;
+  base::MockCallback<
+      DeviceDescriptionService::DeviceDescriptionParseErrorCallback>
+      mock_error_cb_;
+
+  TestDeviceDescriptionService device_description_service_;
+  std::map<std::string, std::unique_ptr<DeviceDescriptionFetcher>>&
+      fetcher_map_;
+  std::map<std::string, DeviceDescriptionService::CacheEntry>& description_map_;
+
+  const content::TestBrowserThreadBundle thread_bundle_;
+  TestingProfile profile_;
+};
+
+TEST_F(DeviceDescriptionServiceTest, TestGetDeviceDescriptionFromCache) {
+  auto device_data = CreateDialDeviceData(1);
+  auto description_data = CreateParsedDialDeviceDescription(1);
+  EXPECT_CALL(mock_success_cb_, Run(device_data, description_data));
+
+  AddToCache(device_data.label(), description_data, false /* expired */);
+
+  std::vector<DialDeviceData> devices = {device_data};
+  device_description_service()->GetDeviceDescriptions(devices, nullptr);
+}
+
+TEST_F(DeviceDescriptionServiceTest, TestGetDeviceDescriptionFetchURL) {
+  DialDeviceData device_data = CreateDialDeviceData(1);
+  std::vector<DialDeviceData> devices = {device_data};
+
+  // Create Fetcher
+  EXPECT_TRUE(fetcher_map_.empty());
+  device_description_service()->GetDeviceDescriptions(
+      devices, profile_.GetRequestContext());
+  EXPECT_EQ(size_t(1), fetcher_map_.size());
+
+  // Remove fetcher and create safe parser
+  OnDeviceDescriptionFetchComplete(1);
+}
+
+TEST_F(DeviceDescriptionServiceTest, TestGetDeviceDescriptionFetchURLError) {
+  DialDeviceData device_data = CreateDialDeviceData(1);
+  std::vector<DialDeviceData> devices;
+  devices.push_back(device_data);
+
+  // Create Fetcher
+  EXPECT_TRUE(fetcher_map_.empty());
+  device_description_service()->GetDeviceDescriptions(
+      devices, profile_.GetRequestContext());
+  EXPECT_EQ(size_t(1), fetcher_map_.size());
+
+  EXPECT_CALL(mock_error_cb_, Run(device_data, ""));
+
+  device_description_service()->OnDeviceDescriptionFetchError(device_data, "");
+  EXPECT_TRUE(fetcher_map_.empty());
+}
+
+TEST_F(DeviceDescriptionServiceTest,
+       TestGetDeviceDescriptionRemoveOutDatedFetchers) {
+  DialDeviceData device_data_1 = CreateDialDeviceData(1);
+  DialDeviceData device_data_2 = CreateDialDeviceData(2);
+  DialDeviceData device_data_3 = CreateDialDeviceData(3);
+
+  std::vector<DialDeviceData> devices;
+  devices.push_back(device_data_1);
+  devices.push_back(device_data_2);
+
+  // insert fetchers
+  device_description_service()->GetDeviceDescriptions(
+      devices, profile_.GetRequestContext());
+
+  // Keep fetchers no exist in current device list and remove fetchers with
+  // different description url.
+  GURL new_url_2 = GURL("http://example.com");
+  device_data_2.set_device_description_url(new_url_2);
+
+  devices.clear();
+  devices.push_back(device_data_2);
+  devices.push_back(device_data_3);
+  device_description_service()->GetDeviceDescriptions(
+      devices, profile_.GetRequestContext());
+
+  EXPECT_EQ(size_t(3), fetcher_map_.size());
+
+  auto* description_fetcher = fetcher_map_[device_data_2.label()].get();
+  EXPECT_EQ(new_url_2, description_fetcher->device_description_url());
+
+  EXPECT_CALL(mock_error_cb_, Run(_, _)).Times(3);
+  device_description_service()->OnDeviceDescriptionFetchError(device_data_1,
+                                                              "");
+  device_description_service()->OnDeviceDescriptionFetchError(device_data_2,

[mark a. foltz, 2017/04/25 21:06:26]

Is this asserting that the original fetcher for |device_data_2| is erroring out
when its URL got updated?
Should the new fetcher created by GetDeviceDescriptions succeed?

[zhaobin, 2017/04/26 18:50:04]

No, |device_data_2| will fail quietly without any error...These 3
OnDeviceDescriptionFetchError() calls are simply clean up calls (or
~URLFetcher() will have DCHECK failures).

+                                                              "");
+  device_description_service()->OnDeviceDescriptionFetchError(device_data_3,
+                                                              "");
+}
+
+TEST_F(DeviceDescriptionServiceTest, TestCleanUpCacheEntries) {
+  DialDeviceData device_data_1 = CreateDialDeviceData(1);
+  DialDeviceData device_data_2 = CreateDialDeviceData(2);
+  DialDeviceData device_data_3 = CreateDialDeviceData(3);
+
+  AddToCache(device_data_1.label(), ParsedDialDeviceDescription(),
+             true /* expired */);
+  AddToCache(device_data_2.label(), ParsedDialDeviceDescription(),
+             true /* expired */);
+  AddToCache(device_data_3.label(), ParsedDialDeviceDescription(),
+             false /* expired */);
+
+  device_description_service_.CleanUpCacheEntries();
+  EXPECT_EQ(size_t(1), description_map_.size());

[mark a. foltz, 2017/04/25 21:06:26]

Nit: Check that device 3 specifically is not removed from the cache.

[zhaobin, 2017/04/26 18:50:04]

Done.

+
+  AddToCache(device_data_3.label(), ParsedDialDeviceDescription(),
+             true /* expired*/);
+  device_description_service_.CleanUpCacheEntries();
+  EXPECT_TRUE(description_map_.empty());
+}
+
+TEST_F(DeviceDescriptionServiceTest, TestOnParsedDeviceDescription) {
+  GURL app_url("http://192.168.1.1/apps");
+  DialDeviceData device_data = CreateDialDeviceData(1);
+
+  // null_ptr
+  std::string error_message = "Failed to parse device description xml";
+  TestOnParsedDeviceDescription(nullptr, error_message);
+
+  // Empty field
+  error_message = "Failed to process fetch result";
+  TestOnParsedDeviceDescription(chrome::mojom::DialDeviceDescription::New(),
+                                error_message);
+
+  // Valid device description ptr and put in cache
+  auto description_ptr = CreateDialDeviceDescriptionPtr(1);
+  TestOnParsedDeviceDescription(std::move(description_ptr), "");
+  EXPECT_EQ(size_t(1), description_map_.size());
+
+  // Valid device description ptr and skip cache.
+  size_t cache_num = 256;
+  for (size_t i = 0; i < cache_num; i++) {
+    AddToCache(std::to_string(i), ParsedDialDeviceDescription(),
+               false /* expired */);
+  }
+
+  EXPECT_EQ(size_t(cache_num + 1), description_map_.size());
+  description_ptr = CreateDialDeviceDescriptionPtr(1);
+  TestOnParsedDeviceDescription(std::move(description_ptr), "");
+  EXPECT_EQ(size_t(cache_num + 1), description_map_.size());
+}
+
+TEST_F(DeviceDescriptionServiceTest, TestSafeParserProperlyCreated) {
+  DialDeviceData device_data_1 = CreateDialDeviceData(1);
+  DialDeviceData device_data_2 = CreateDialDeviceData(2);
+  DialDeviceData device_data_3 = CreateDialDeviceData(3);
+
+  std::vector<DialDeviceData> devices = {device_data_1, device_data_2,
+                                         device_data_3};
+
+  // insert fetchers
+  device_description_service()->GetDeviceDescriptions(
+      devices, profile_.GetRequestContext());
+
+  EXPECT_FALSE(device_description_service()->parser_);
+  OnDeviceDescriptionFetchComplete(1);
+
+  EXPECT_TRUE(device_description_service()->parser_);
+  OnDeviceDescriptionFetchComplete(2);
+  OnDeviceDescriptionFetchComplete(3);
+
+  EXPECT_FALSE(device_description_service()->parser_);
+}
+
+}  // namespace media_router