Linux native notifications: Add attribution

chrome/browser/notifications/notification_platform_bridge_linux_unittest.cc

 // 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_platform_bridge_linux.h"
 
 #include <memory>
 #include <vector>
 
 #include "base/callback.h"
@@ skipping 43 matching lines @@
     notification_.set_image(image);
     return *this;
   }
 
   NotificationBuilder& SetItems(
       const std::vector<message_center::NotificationItem>& items) {
     notification_.set_items(items);
     return *this;
   }
 
+  NotificationBuilder& SetMessage(const base::string16& message) {
+    notification_.set_message(message);
+    return *this;
+  }
+
   NotificationBuilder& SetNeverTimeout(bool never_timeout) {
     notification_.set_never_timeout(never_timeout);
     return *this;
   }
 
+  NotificationBuilder& SetOriginUrl(const GURL& origin_url) {
+    notification_.set_origin_url(origin_url);
+    return *this;
+  }
+
   NotificationBuilder& SetProgress(int progress) {
     notification_.set_progress(progress);
     return *this;
   }
 
   NotificationBuilder& SetTitle(const base::string16& title) {
     notification_.set_title(title);
     return *this;
   }
 
@@ skipping 303 matching lines @@
           1));
 
   CreateNotificationBridgeLinux();
   notification_bridge_linux_->Display(
       NotificationCommon::PERSISTENT, "", "", false,
       NotificationBuilder("")
           .SetType(message_center::NOTIFICATION_TYPE_IMAGE)
           .SetImage(original_image)
           .GetResult());
 }
+
+TEST_F(NotificationPlatformBridgeLinuxTest, NotificationAttribution) {
+  EXPECT_CALL(*mock_notification_proxy_.get(),
+              MockCallMethodAndBlock(Calls("Notify"), _))
+      .WillOnce(OnNotify(
+          [](const NotificationRequest& request) {
+            EXPECT_EQ("Body text\ngoogle.com", request.body);
+          },
+          1));
+
+  CreateNotificationBridgeLinux();
+  notification_bridge_linux_->Display(
+      NotificationCommon::PERSISTENT, "", "", false,
+      NotificationBuilder("")
+          .SetMessage(base::UTF8ToUTF16("Body text"))

[Lei Zhang, 2017/05/15 22:19:59]

ACIIToUTF16() when you know it's ASCII.

[Tom (Use chromium acct), 2017/05/16 04:43:34]

Done.

+          .SetOriginUrl(GURL("https://google.com/"))

[Lei Zhang, 2017/05/15 22:19:59]

I'm not sure what guarantees the notification subsystem provides, but if we can
test with something more exotic as well, that would be good to have. So we can
see if the newly added UseOriginAsContextMessage() handler code escapes the
exotic origin properly.

[Tom (Use chromium acct), 2017/05/16 04:43:34]

Done. (maybe)

+          .GetResult());
+}