Add a NotificationTemplateBuilder for Action Center toasts

chrome/browser/notifications/notification_template_builder_unittest.cc

Index: chrome/browser/notifications/notification_template_builder_unittest.cc
diff --git a/chrome/browser/notifications/notification_template_builder_unittest.cc b/chrome/browser/notifications/notification_template_builder_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..2ef75525a1cb0a4fafa88e8195257420ca6f49db
--- /dev/null
+++ b/chrome/browser/notifications/notification_template_builder_unittest.cc
@@ -0,0 +1,96 @@
+// 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/notifications/notification_template_builder.h"
+
+#include <memory>
+#include <string>
+
+#include "base/macros.h"
+#include "base/memory/ptr_util.h"
+#include "base/strings/utf_string_conversions.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/message_center/notification.h"
+
+using message_center::Notification;
+using message_center::NotifierId;
+using message_center::RichNotificationData;
+
+namespace {
+
+const char kNotificationId[] = "notification_id";
+const char kNotificationTitle[] = "My Title";
+const char kNotificationMessage[] = "My Message";
+const char kNotificationOrigin[] = "https://example.com";
+
+// Intermediary format for the options available when creating a notification,
+// with default values specific to this test suite to avoid endless repetition.
+struct NotificationData {
+  NotificationData()
+      : id(kNotificationId),
+        title(kNotificationTitle),
+        message(kNotificationMessage),
+        origin(kNotificationOrigin) {}
+
+  std::string id;
+  std::string title;
+  std::string message;
+  GURL origin;
+};
+
+}  // namespace
+
+class NotificationTemplateBuilderTest : public ::testing::Test {
+ public:
+  NotificationTemplateBuilderTest() = default;
+  ~NotificationTemplateBuilderTest() override = default;
+
+ protected:
+  // Builds the message_center::Notification object and then the template for
+  // the given |notification_data|, and writes that to |*xml_template|. Calls
+  // must be wrapped in ASSERT_NO_FATAL_FAILURE().
+  void BuildTemplate(const NotificationData& notification_data,
+                     std::string* xml_template) {
+    GURL origin_url(notification_data.origin);
+
+    Notification notification(
+        message_center::NOTIFICATION_TYPE_SIMPLE, notification_data.id,
+        base::UTF8ToUTF16(notification_data.title),
+        base::UTF8ToUTF16(notification_data.message), gfx::Image() /* icon */,
+        base::string16() /* display_source */, origin_url,
+        NotifierId(origin_url), RichNotificationData(), nullptr /* delegate */);
+
+    template_ =
+        NotificationTemplateBuilder::Build(notification_data.id, notification);
+    ASSERT_TRUE(template_);
+
+    *xml_template = template_->GetNotificationTemplate();
+  }
+
+ protected:
+  std::unique_ptr<NotificationTemplateBuilder> template_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(NotificationTemplateBuilderTest);
+};
+
+TEST_F(NotificationTemplateBuilderTest, SimpleToast) {
+  NotificationData notification_data;
+  std::string xml_template;
+
+  ASSERT_NO_FATAL_FAILURE(BuildTemplate(notification_data, &xml_template));
+
+  const char kExpectedXml[] = R"(<toast launch="notification_id">
+ <visual>
+  <binding template="ToastText04">
+   <text id="1">My Title</text>
+   <text id="2">My Message</text>
+   <text id="3">example.com</text>
+  </binding>
+ </visual>
+</toast>
+)";
+
+  EXPECT_EQ(xml_template, kExpectedXml);
+}