Add testing for subscription from service workers

third_party/WebKit/Source/modules/push_messaging/PushManagerTest.cpp

 // Copyright 2016 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/push_messaging/PushManager.h"
 
 #include "bindings/modules/v8/UnionTypesModules.h"
 #include "core/dom/DOMArrayBuffer.h"
 #include "modules/push_messaging/PushSubscriptionOptions.h"
 #include "public/platform/WebString.h"
 #include "public/platform/modules/push_messaging/WebPushSubscriptionOptions.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 namespace {
 
 const char kValidKeyMarker = 0x04;
 const unsigned kValidKeyLength = 65;
 
+// NIST P-256 public key made available to tests. Must be an uncompressed
+// point in accordance with SEC1 2.3.3.
+const uint8_t kApplicationServerKey[65] = {
+    0x04, 0x55, 0x52, 0x6A, 0xA5, 0x6E, 0x8E, 0xAA, 0x47, 0x97, 0x36, 0x10, 0xC1,
+    0x66, 0x3C, 0x1E, 0x65, 0xBF, 0xA1, 0x7B, 0xEE, 0x48, 0xC9, 0xC6, 0xBB, 0xBF,
+    0x02, 0x18, 0x53, 0x72, 0x1D, 0x0C, 0x7B, 0xA9, 0xE3, 0x11, 0xB7, 0x03, 0x52,
+    0x21, 0xD3, 0x71, 0x90, 0x13, 0xA8, 0xC1, 0xCF, 0xED, 0x20, 0xF7, 0x1F, 0xD1,
+    0x7F, 0xF2, 0x76, 0xB6, 0x01, 0x20, 0xD8, 0x35, 0xA5, 0xD9, 0x3C, 0x43, 0xDF
+};
+
 TEST(PushManagerTest, ValidSenderKey)
 {
-    uint8_t senderKey[kValidKeyLength];
-    memset(senderKey, 0, sizeof(senderKey));
-    senderKey[0] = kValidKeyMarker;
     PushSubscriptionOptions options;
     options.setApplicationServerKey(
         ArrayBufferOrArrayBufferView::fromArrayBuffer(
-            DOMArrayBuffer::create(senderKey, kValidKeyLength)));
+            DOMArrayBuffer::create(kApplicationServerKey, kValidKeyLength)));
 
     TrackExceptionState exceptionState;
     WebPushSubscriptionOptions output = PushManager::toWebPushSubscriptionOptions(options, exceptionState);
     EXPECT_FALSE(exceptionState.hadException());
     EXPECT_EQ(output.applicationServerKey.length(), kValidKeyLength);
-    EXPECT_EQ(output.applicationServerKey, WebString::fromUTF8(reinterpret_cast<const char*>(senderKey), kValidKeyLength));
+
+    // Copy the key into a size+1 buffer so that it can be treated as a null
+    // terminated string for the purposes of EXPECT_EQ.
+    uint8_t senderKey[kValidKeyLength+1];
+    for (unsigned i = 0; i < kValidKeyLength; i++)
+        senderKey[i] = kApplicationServerKey[i];
+    senderKey[kValidKeyLength] = 0x0;
+    EXPECT_EQ(reinterpret_cast<const char*>(senderKey), output.applicationServerKey.latin1());
+    EXPECT_FALSE(output.applicationServerKey.isEmpty());
 }
 
 TEST(PushManagerTest, InvalidSenderKeyMarker)
 {
     uint8_t senderKey[kValidKeyLength];
     memset(senderKey, 0, sizeof(senderKey));
     senderKey[0] = 0x05;
     PushSubscriptionOptions options;
     options.setApplicationServerKey(
         ArrayBufferOrArrayBufferView::fromArrayBuffer(
@@ skipping 14 matching lines @@
         ArrayBufferOrArrayBufferView::fromArrayBuffer(
             DOMArrayBuffer::create(senderKey, kValidKeyLength - 1)));
 
     TrackExceptionState exceptionState;
     WebPushSubscriptionOptions output = PushManager::toWebPushSubscriptionOptions(options, exceptionState);
     EXPECT_TRUE(exceptionState.hadException());
 }
 
 } // namespace
 } // namespace blink