Add MediaDrmBridge::AddKeySystemUuidMapping().

media/base/android/media_drm_bridge_unittest.cc

 // Copyright 2014 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 "base/basictypes.h"
 #include "base/logging.h"
 #include "media/base/android/media_drm_bridge.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #include "widevine_cdm_version.h"  // In SHARED_INTERMEDIATE_DIR.
 
 namespace media {
 
 #define EXPECT_TRUE_IF_AVAILABLE(a)                   \
   do {                                                \
     if (!MediaDrmBridge::IsAvailable()) {             \
       VLOG(0) << "MediaDrm not supported on device."; \
       EXPECT_FALSE(a);                                \
     } else {                                          \
       EXPECT_TRUE(a);                                 \
     }                                                 \
   } while (0)
 
 const char kAudioMp4[] = "audio/mp4";
 const char kVideoMp4[] = "video/mp4";
 const char kAudioWebM[] = "audio/webm";
 const char kVideoWebM[] = "video/webm";
 const char kInvalidKeySystem[] = "invalid.keysystem";
+const char kAcmeKeySystem[] = "com.acme.keysystem";

[xhwang, 2014/05/15 15:10:29]

s/acme/foo or example to make it more obvious?

[ycheo (away), 2014/05/16 04:53:38]

Done.

+const uint8 kWidevineUuid[16] = {
+    0xED, 0xEF, 0x8B, 0xA9, 0x79, 0xD6, 0x4A, 0xCE,
+    0xA3, 0xC8, 0x27, 0xDC, 0xD5, 0x1D, 0x21, 0xED };
 const MediaDrmBridge::SecurityLevel kLNone =
     MediaDrmBridge::SECURITY_LEVEL_NONE;
 const MediaDrmBridge::SecurityLevel kL1 = MediaDrmBridge::SECURITY_LEVEL_1;
 const MediaDrmBridge::SecurityLevel kL3 = MediaDrmBridge::SECURITY_LEVEL_3;
 
 // Helper functions to avoid typing "MediaDrmBridge::" in tests.
 
 static bool IsKeySystemSupported(const std::string& key_system) {
   return MediaDrmBridge::IsKeySystemSupported(key_system);
 }
@@ skipping 47 matching lines @@
   EXPECT_FALSE(IsKeySystemSupported(kInvalidKeySystem));
   EXPECT_FALSE(IsKeySystemSupportedWithType(kInvalidKeySystem, kAudioMp4));
   EXPECT_FALSE(IsKeySystemSupportedWithType(kInvalidKeySystem, kVideoMp4));
   EXPECT_FALSE(IsKeySystemSupportedWithType(kInvalidKeySystem, kAudioWebM));
   EXPECT_FALSE(IsKeySystemSupportedWithType(kInvalidKeySystem, kVideoWebM));
   EXPECT_FALSE(IsKeySystemSupportedWithType(kInvalidKeySystem, "unknown"));
   EXPECT_FALSE(IsKeySystemSupportedWithType(kInvalidKeySystem, "video/avi"));
   EXPECT_FALSE(IsKeySystemSupportedWithType(kInvalidKeySystem, "audio/mp3"));
 }
 
+TEST(MediaDrmBridgeTest, AddNewKeySystemName) {
+  EXPECT_FALSE(IsKeySystemSupported(kAcmeKeySystem));
+
+  MediaDrmBridge::AddKeySystem(kAcmeKeySystem,
+      // Use WV uuid, because it is the only MediaDrm we can guarentee

[xhwang, 2014/05/15 15:10:29]

s/uuid/UUID

s/MediaDrm/key system

[ycheo (away), 2014/05/16 04:53:38]

Done.

+      // that it is installed in the target device.

[xhwang, 2014/05/15 15:10:29]

nit: target/test

[ycheo (away), 2014/05/16 04:53:38]

Done.

+      std::vector<uint8>(kWidevineUuid,
+                         kWidevineUuid + arraysize(kWidevineUuid)));
+
+  EXPECT_TRUE_IF_AVAILABLE(IsKeySystemSupported(kAcmeKeySystem));
+  EXPECT_TRUE_IF_AVAILABLE(
+      IsKeySystemSupportedWithType(kAcmeKeySystem, kVideoMp4));
+}
+
+TEST(MediaDrmBridgeTest, ShouldNotOverwriteExistingKeySystem) {
+  EXPECT_TRUE_IF_AVAILABLE(IsKeySystemSupported(kWidevineKeySystem));
+
+  // Add WV keysystem again with the invalid uuid.
+  MediaDrmBridge::AddKeySystem(kWidevineKeySystem, std::vector<uint8>(16, 99));
+
+  EXPECT_TRUE_IF_AVAILABLE(IsKeySystemSupported(kWidevineKeySystem));
+}
+
 }  // namespace media