[Clank] Load origin id before create MediaDrmBridge

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 <memory>
 
 #include "base/android/build_info.h"
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
 #include "media/base/android/media_drm_bridge.h"
 #include "media/base/provision_fetcher.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #include "widevine_cdm_version.h"  // In SHARED_INTERMEDIATE_DIR.
 
 namespace media {
 
 #define EXPECT_TRUE_IF_WIDEVINE_AVAILABLE(a)                              \
   do {                                                                    \
@@ skipping 31 matching lines @@
   static std::unique_ptr<ProvisionFetcher> Create() {
     return base::WrapUnique(new MockProvisionFetcher);
   }
 
   // ProvisionFetcher implementation.
   void Retrieve(const std::string& default_url,
                 const std::string& request_data,
                 const ResponseCB& response_cb) override {}
 };
 
+void CheckMediaDrmBridge(bool expect_success,
+                         scoped_refptr<MediaDrmBridge> media_drm_bridge) {
+  if (expect_success) {
+    EXPECT_TRUE_IF_WIDEVINE_AVAILABLE(media_drm_bridge);
+  } else {
+    EXPECT_FALSE(media_drm_bridge);
+  }
+}
+
 }  // namespace (anonymous)
 
 TEST(MediaDrmBridgeTest, IsKeySystemSupported_Widevine) {
   // TODO(xhwang): Enable when b/13564917 is fixed.
   // EXPECT_TRUE_IF_AVAILABLE(
   //     IsKeySystemSupportedWithType(kWidevineKeySystem, kAudioMp4));
   EXPECT_TRUE_IF_WIDEVINE_AVAILABLE(
       IsKeySystemSupportedWithType(kWidevineKeySystem, kVideoMp4));
 
   if (base::android::BuildInfo::GetInstance()->sdk_int() <= 19) {
@@ skipping 18 matching lines @@
   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, CreateWithoutSessionSupport_Widevine) {
   base::MessageLoop message_loop_;
-  EXPECT_TRUE_IF_WIDEVINE_AVAILABLE(MediaDrmBridge::CreateWithoutSessionSupport(
-      kWidevineKeySystem, kDefault, base::Bind(&MockProvisionFetcher::Create)));
+  MediaDrmBridge::CreateWithoutSessionSupport(
+      kWidevineKeySystem, kDefault, base::Bind(&MockProvisionFetcher::Create),
+      base::BindOnce(&CheckMediaDrmBridge, true));
+
+  base::RunLoop().RunUntilIdle();
 }
 
 // Invalid key system is NOT supported regardless whether MediaDrm is available.
 TEST(MediaDrmBridgeTest, CreateWithoutSessionSupport_InvalidKeySystem) {
   base::MessageLoop message_loop_;
-  EXPECT_FALSE(MediaDrmBridge::CreateWithoutSessionSupport(
-      kInvalidKeySystem, kDefault, base::Bind(&MockProvisionFetcher::Create)));
+  MediaDrmBridge::CreateWithoutSessionSupport(
+      kInvalidKeySystem, kDefault, base::Bind(&MockProvisionFetcher::Create),
+      base::BindOnce(&CheckMediaDrmBridge, false));
+
+  base::RunLoop().RunUntilIdle();
 }
 
 TEST(MediaDrmBridgeTest, CreateWithSecurityLevel_Widevine) {
   base::MessageLoop message_loop_;
 
   // We test "L3" fully. But for "L1" we don't check the result as it depends on
   // whether the test device supports "L1".
-  EXPECT_TRUE_IF_WIDEVINE_AVAILABLE(MediaDrmBridge::CreateWithoutSessionSupport(
-      kWidevineKeySystem, kL3, base::Bind(&MockProvisionFetcher::Create)));
+  MediaDrmBridge::CreateWithoutSessionSupport(
+      kWidevineKeySystem, kL3, base::Bind(&MockProvisionFetcher::Create),
+      base::BindOnce(&CheckMediaDrmBridge, true));
 
   MediaDrmBridge::CreateWithoutSessionSupport(
-      kWidevineKeySystem, kL1, base::Bind(&MockProvisionFetcher::Create));
+      kWidevineKeySystem, kL1, base::Bind(&MockProvisionFetcher::Create),
+      base::BindOnce([](scoped_refptr<MediaDrmBridge>) {}));
+
+  base::RunLoop().RunUntilIdle();
 }
 
 }  // namespace media