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 35 matching lines @@
                       new MockNotificationDelegate(id)) {}
 
   Notification GetResult() { return notification_; }
 
   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 251 matching lines @@
           2));
 
   CreateNotificationBridgeLinux();
   notification_bridge_linux_->Display(
       NotificationCommon::PERSISTENT, "", "", false,
       NotificationBuilder("1").SetNeverTimeout(false).GetResult());
   notification_bridge_linux_->Display(
       NotificationCommon::PERSISTENT, "", "", false,
       NotificationBuilder("2").SetNeverTimeout(true).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"))
+          .SetOriginUrl(GURL("https://google.com/"))
+          .GetResult());
+}