media_foundation_cdm_factory_unittest.cc 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "media/cdm/win/media_foundation_cdm_factory.h"
  5. #include <memory>
  6. #include "base/bind.h"
  7. #include "base/memory/raw_ptr.h"
  8. #include "base/test/gmock_callback_support.h"
  9. #include "base/test/mock_callback.h"
  10. #include "base/test/task_environment.h"
  11. #include "media/base/mock_filters.h"
  12. #include "media/base/test_helpers.h"
  13. #include "media/base/win/mf_helpers.h"
  14. #include "media/base/win/mf_mocks.h"
  15. #include "media/cdm/mock_helpers.h"
  16. #include "testing/gmock/include/gmock/gmock.h"
  17. #include "testing/gtest/include/gtest/gtest.h"
  18. #include "third_party/abseil-cpp/absl/types/optional.h"
  19. using base::test::RunOnceCallback;
  20. using ::testing::_;
  21. using ::testing::DoAll;
  22. using ::testing::IsEmpty;
  23. using ::testing::IsNull;
  24. using ::testing::NotNull;
  25. using ::testing::Return;
  26. using ::testing::StrictMock;
  27. namespace media {
  28. const char kClearKeyKeySystem[] = "org.w3.clearkey";
  29. const CdmConfig kClearKeyHardwareSecureCdmConfig = {kClearKeyKeySystem, true,
  30. true, true};
  31. using Microsoft::WRL::ComPtr;
  32. class MediaFoundationCdmFactoryTest : public testing::Test {
  33. public:
  34. MediaFoundationCdmFactoryTest()
  35. : mf_cdm_factory_(MakeComPtr<MockMFCdmFactory>()),
  36. mf_cdm_access_(MakeComPtr<MockMFCdmAccess>()),
  37. mf_cdm_(MakeComPtr<MockMFCdm>()) {
  38. auto cdm_helper =
  39. std::make_unique<StrictMock<MockCdmAuxiliaryHelper>>(nullptr);
  40. cdm_helper_ = cdm_helper.get();
  41. cdm_factory_ =
  42. std::make_unique<MediaFoundationCdmFactory>(std::move(cdm_helper));
  43. }
  44. ~MediaFoundationCdmFactoryTest() override = default;
  45. HRESULT GetMockCdmFactory(
  46. bool expect_success,
  47. ComPtr<IMFContentDecryptionModuleFactory>& mf_cdm_factory) {
  48. if (!expect_success)
  49. return E_FAIL;
  50. mf_cdm_factory = mf_cdm_factory_;
  51. return S_OK;
  52. }
  53. void SetCreateCdmFactoryCallbackForTesting(bool expect_success) {
  54. cdm_factory_->SetCreateCdmFactoryCallbackForTesting(
  55. kClearKeyKeySystem,
  56. base::BindRepeating(&MediaFoundationCdmFactoryTest::GetMockCdmFactory,
  57. base::Unretained(this), expect_success));
  58. }
  59. protected:
  60. void Create() {
  61. cdm_factory_->Create(
  62. kClearKeyHardwareSecureCdmConfig,
  63. base::BindRepeating(&MockCdmClient::OnSessionMessage,
  64. base::Unretained(&cdm_client_)),
  65. base::BindRepeating(&MockCdmClient::OnSessionClosed,
  66. base::Unretained(&cdm_client_)),
  67. base::BindRepeating(&MockCdmClient::OnSessionKeysChange,
  68. base::Unretained(&cdm_client_)),
  69. base::BindRepeating(&MockCdmClient::OnSessionExpirationUpdate,
  70. base::Unretained(&cdm_client_)),
  71. cdm_created_cb_.Get());
  72. task_environment_.RunUntilIdle();
  73. }
  74. base::test::TaskEnvironment task_environment_;
  75. StrictMock<MockCdmClient> cdm_client_;
  76. ComPtr<MockMFCdmFactory> mf_cdm_factory_;
  77. ComPtr<MockMFCdmAccess> mf_cdm_access_;
  78. ComPtr<MockMFCdm> mf_cdm_;
  79. raw_ptr<StrictMock<MockCdmAuxiliaryHelper>> cdm_helper_ = nullptr;
  80. std::unique_ptr<MediaFoundationCdmFactory> cdm_factory_;
  81. base::MockCallback<CdmCreatedCB> cdm_created_cb_;
  82. };
  83. TEST_F(MediaFoundationCdmFactoryTest, Create) {
  84. SetCreateCdmFactoryCallbackForTesting(/*expect_success=*/true);
  85. COM_EXPECT_CALL(mf_cdm_factory_, IsTypeSupported(NotNull(), IsNull()))
  86. .WillOnce(Return(TRUE));
  87. COM_EXPECT_CALL(mf_cdm_factory_, CreateContentDecryptionModuleAccess(
  88. NotNull(), NotNull(), _, _))
  89. .WillOnce(DoAll(SetComPointee<3>(mf_cdm_access_.Get()), Return(S_OK)));
  90. EXPECT_CALL(*cdm_helper_, GetMediaFoundationCdmData(_))
  91. .WillOnce(RunOnceCallback<0>(std::make_unique<MediaFoundationCdmData>(
  92. base::UnguessableToken::Create(), absl::nullopt, base::FilePath())));
  93. COM_EXPECT_CALL(mf_cdm_access_, CreateContentDecryptionModule(NotNull(), _))
  94. .WillOnce(DoAll(SetComPointee<1>(mf_cdm_.Get()), Return(S_OK)));
  95. EXPECT_CALL(cdm_created_cb_, Run(NotNull(), _));
  96. Create();
  97. }
  98. TEST_F(MediaFoundationCdmFactoryTest, CreateCdmFactoryFail) {
  99. SetCreateCdmFactoryCallbackForTesting(/*expect_success=*/false);
  100. EXPECT_CALL(*cdm_helper_, GetMediaFoundationCdmData(_))
  101. .WillOnce(RunOnceCallback<0>(std::make_unique<MediaFoundationCdmData>(
  102. base::UnguessableToken::Create(), absl::nullopt, base::FilePath())));
  103. EXPECT_CALL(cdm_created_cb_, Run(IsNull(), _));
  104. Create();
  105. }
  106. TEST_F(MediaFoundationCdmFactoryTest, IsTypeSupportedFail) {
  107. SetCreateCdmFactoryCallbackForTesting(/*expect_success=*/true);
  108. COM_EXPECT_CALL(mf_cdm_factory_, IsTypeSupported(NotNull(), IsNull()))
  109. .WillOnce(Return(FALSE));
  110. EXPECT_CALL(*cdm_helper_, GetMediaFoundationCdmData(_))
  111. .WillOnce(RunOnceCallback<0>(std::make_unique<MediaFoundationCdmData>(
  112. base::UnguessableToken::Create(), absl::nullopt, base::FilePath())));
  113. EXPECT_CALL(cdm_created_cb_, Run(IsNull(), _));
  114. Create();
  115. }
  116. TEST_F(MediaFoundationCdmFactoryTest, CreateCdmAccessFail) {
  117. SetCreateCdmFactoryCallbackForTesting(/*expect_success=*/true);
  118. COM_EXPECT_CALL(mf_cdm_factory_, IsTypeSupported(NotNull(), IsNull()))
  119. .WillOnce(Return(TRUE));
  120. COM_EXPECT_CALL(mf_cdm_factory_, CreateContentDecryptionModuleAccess(
  121. NotNull(), NotNull(), _, _))
  122. .WillOnce(Return(E_FAIL));
  123. EXPECT_CALL(*cdm_helper_, GetMediaFoundationCdmData(_))
  124. .WillOnce(RunOnceCallback<0>(std::make_unique<MediaFoundationCdmData>(
  125. base::UnguessableToken::Create(), absl::nullopt, base::FilePath())));
  126. EXPECT_CALL(cdm_created_cb_, Run(IsNull(), _));
  127. Create();
  128. }
  129. TEST_F(MediaFoundationCdmFactoryTest, NullCdmOriginIdFail) {
  130. SetCreateCdmFactoryCallbackForTesting(/*expect_success=*/true);
  131. EXPECT_CALL(*cdm_helper_, GetMediaFoundationCdmData(_))
  132. .WillOnce(RunOnceCallback<0>(std::make_unique<MediaFoundationCdmData>(
  133. base::UnguessableToken::Null(), absl::nullopt, base::FilePath())));
  134. EXPECT_CALL(cdm_created_cb_, Run(IsNull(), _));
  135. Create();
  136. }
  137. TEST_F(MediaFoundationCdmFactoryTest, CreateCdmFail) {
  138. SetCreateCdmFactoryCallbackForTesting(/*expect_success=*/true);
  139. COM_EXPECT_CALL(mf_cdm_factory_, IsTypeSupported(NotNull(), IsNull()))
  140. .WillOnce(Return(TRUE));
  141. COM_EXPECT_CALL(mf_cdm_factory_, CreateContentDecryptionModuleAccess(
  142. NotNull(), NotNull(), _, _))
  143. .WillOnce(DoAll(SetComPointee<3>(mf_cdm_access_.Get()), Return(S_OK)));
  144. EXPECT_CALL(*cdm_helper_, GetMediaFoundationCdmData(_))
  145. .WillOnce(RunOnceCallback<0>(std::make_unique<MediaFoundationCdmData>(
  146. base::UnguessableToken::Create(), absl::nullopt, base::FilePath())));
  147. COM_EXPECT_CALL(mf_cdm_access_, CreateContentDecryptionModule(NotNull(), _))
  148. .WillOnce(DoAll(SetComPointee<1>(mf_cdm_.Get()), Return(E_FAIL)));
  149. EXPECT_CALL(cdm_created_cb_, Run(IsNull(), _));
  150. Create();
  151. }
  152. } // namespace media