Add badge to web notifications.

third_party/WebKit/Source/modules/notifications/NotificationDataTest.cpp

 // Copyright 2015 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 "modules/notifications/NotificationData.h"
 
 #include "bindings/core/v8/ExceptionState.h"
 #include "bindings/modules/v8/UnionTypesModules.h"
 #include "core/testing/NullExecutionContext.h"
 #include "modules/notifications/Notification.h"
 #include "modules/notifications/NotificationOptions.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/CurrentTime.h"
 #include "wtf/HashMap.h"
 #include "wtf/Vector.h"
 
 namespace blink {
 namespace {
 
 const char kNotificationTitle[] = "My Notification";
 
 const char kNotificationDir[] = "rtl";
 const char kNotificationLang[] = "nl";
 const char kNotificationBody[] = "Hello, world";
 const char kNotificationTag[] = "my_tag";
 const char kNotificationEmptyTag[] = "";
 const char kNotificationIcon[] = "https://example.com/icon.png";
 const char kNotificationIconInvalid[] = "https://invalid:icon:url";
+const char kNotificationBadge[] = "https://example.com/badge.png";
 const unsigned kNotificationVibration[] = { 42, 10, 20, 30, 40 };
 const unsigned long long kNotificationTimestamp = 621046800ull;
 const bool kNotificationRenotify = true;
 const bool kNotificationSilent = false;
 const bool kNotificationRequireInteraction = true;
 
 const char kNotificationActionAction[] = "my_action";
 const char kNotificationActionTitle[] = "My Action";
 const char kNotificationActionIcon[] = "https://example.com/action_icon.png";
 
@@ skipping 31 matching lines @@
 
         actions.append(action);
     }
 
     NotificationOptions options;
     options.setDir(kNotificationDir);
     options.setLang(kNotificationLang);
     options.setBody(kNotificationBody);
     options.setTag(kNotificationTag);
     options.setIcon(kNotificationIcon);
+    options.setBadge(kNotificationBadge);
     options.setVibrate(vibrationSequence);
     options.setTimestamp(kNotificationTimestamp);
     options.setRenotify(kNotificationRenotify);
     options.setSilent(kNotificationSilent);
     options.setRequireInteraction(kNotificationRequireInteraction);
     options.setActions(actions);
 
     // TODO(peter): Test |options.data| and |notificationData.data|.
 
     TrackExceptionState exceptionState;
     WebNotificationData notificationData = createWebNotificationData(getExecutionContext(), kNotificationTitle, options, exceptionState);
     ASSERT_FALSE(exceptionState.hadException());
 
     EXPECT_EQ(kNotificationTitle, notificationData.title);
 
     EXPECT_EQ(WebNotificationData::DirectionRightToLeft, notificationData.direction);
     EXPECT_EQ(kNotificationLang, notificationData.lang);
     EXPECT_EQ(kNotificationBody, notificationData.body);
     EXPECT_EQ(kNotificationTag, notificationData.tag);
 
-    // TODO(peter): Test WebNotificationData.icon and WebNotificationAction.icon when ExecutionContext::completeURL() works in this test.
+    // TODO(peter): Test the various icon properties when ExecutionContext::completeURL() works in this test.
 
     ASSERT_EQ(vibrationPattern.size(), notificationData.vibrate.size());
     for (size_t i = 0; i < vibrationPattern.size(); ++i)
         EXPECT_EQ(vibrationPattern[i], static_cast<unsigned>(notificationData.vibrate[i]));
 
     EXPECT_EQ(kNotificationTimestamp, notificationData.timestamp);
     EXPECT_EQ(kNotificationRenotify, notificationData.renotify);
     EXPECT_EQ(kNotificationSilent, notificationData.silent);
     EXPECT_EQ(kNotificationRequireInteraction, notificationData.requireInteraction);
     EXPECT_EQ(actions.size(), notificationData.actions.size());
@@ skipping 42 matching lines @@
     for (size_t i = 0; i < Notification::maxActions(); ++i) {
         NotificationAction action;
         action.setAction(kNotificationActionAction);
         action.setTitle(kNotificationActionTitle);
         action.setIcon(kNotificationIconInvalid);
         actions.append(action);
     }
 
     NotificationOptions options;
     options.setIcon(kNotificationIconInvalid);
+    options.setBadge(kNotificationIconInvalid);
     options.setActions(actions);
 
     TrackExceptionState exceptionState;
     WebNotificationData notificationData = createWebNotificationData(getExecutionContext(), kNotificationTitle, options, exceptionState);
     ASSERT_FALSE(exceptionState.hadException());
 
     EXPECT_TRUE(notificationData.icon.isEmpty());
+    EXPECT_TRUE(notificationData.badge.isEmpty());
     for (const auto& action : notificationData.actions)
         EXPECT_TRUE(action.icon.isEmpty());
 }
 
 TEST_F(NotificationDataTest, VibrationNormalization)
 {
     Vector<unsigned> unnormalizedPattern;
     for (size_t i = 0; i < WTF_ARRAY_LENGTH(kNotificationVibrationUnnormalized); ++i)
         unnormalizedPattern.append(kNotificationVibrationUnnormalized[i]);
 
@@ skipping 73 matching lines @@
     ASSERT_EQ(Notification::maxActions(), notificationData.actions.size());
 
     for (size_t i = 0; i < Notification::maxActions(); ++i) {
         WebString expectedAction = String::number(i);
         EXPECT_EQ(expectedAction, notificationData.actions[i].action);
     }
 }
 
 } // namespace
 } // namespace blink