| Index: components/cdm/browser/media_drm_storage_impl_unittest.cc
|
| diff --git a/components/cdm/browser/media_drm_storage_impl_unittest.cc b/components/cdm/browser/media_drm_storage_impl_unittest.cc
|
| deleted file mode 100644
|
| index 6a1610ee5a364eeeaf0b9f679be63c82b5ed4dbb..0000000000000000000000000000000000000000
|
| --- a/components/cdm/browser/media_drm_storage_impl_unittest.cc
|
| +++ /dev/null
|
| @@ -1,196 +0,0 @@
|
| -// Copyright 2017 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 "components/cdm/browser/media_drm_storage_impl.h"
|
| -
|
| -#include <memory>
|
| -
|
| -#include "base/run_loop.h"
|
| -#include "base/test/test_message_loop.h"
|
| -#include "components/prefs/testing_pref_service.h"
|
| -#include "media/mojo/services/mojo_media_drm_storage.h"
|
| -#include "mojo/public/cpp/bindings/interface_request.h"
|
| -#include "mojo/public/cpp/bindings/strong_binding.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -#include "url/gurl.h"
|
| -#include "url/origin.h"
|
| -
|
| -namespace cdm {
|
| -
|
| -const char kMediaDrmStorage[] = "media.media_drm_storage";
|
| -const char kTestOrigin[] = "https://www.testorigin.com:80";
|
| -
|
| -class MediaDrmStorageImplTest : public ::testing::Test {
|
| - public:
|
| - MediaDrmStorageImplTest() {}
|
| -
|
| - void SetUp() override {
|
| - pref_service_.reset(new TestingPrefServiceSimple());
|
| - PrefRegistrySimple* registry = pref_service_->registry();
|
| - MediaDrmStorageImpl::RegisterProfilePrefs(registry);
|
| -
|
| - media::mojom::MediaDrmStoragePtr media_drm_storage_ptr;
|
| - auto request = mojo::MakeRequest(&media_drm_storage_ptr);
|
| -
|
| - media_drm_storage_.reset(
|
| - new media::MojoMediaDrmStorage(std::move(media_drm_storage_ptr)));
|
| -
|
| - // The created object will be destroyed on connection error.
|
| - new MediaDrmStorageImpl(nullptr, // Use null RenderFrameHost for testing.
|
| - pref_service_.get(), url::Origin(GURL(kTestOrigin)),
|
| - std::move(request));
|
| -
|
| - media_drm_storage_->Initialize(url::Origin(GURL(kTestOrigin)));
|
| - }
|
| -
|
| - void TearDown() override {
|
| - media_drm_storage_.reset();
|
| - base::RunLoop().RunUntilIdle();
|
| - }
|
| -
|
| - protected:
|
| - using SessionData = media::MediaDrmStorage::SessionData;
|
| -
|
| - void OnProvisioned() {
|
| - media_drm_storage_->OnProvisioned(ExpectResult(true));
|
| - }
|
| -
|
| - void SavePersistentSession(const std::string& session_id,
|
| - const std::vector<uint8_t>& key_set_id,
|
| - const std::string& mime_type,
|
| - bool success = true) {
|
| - media_drm_storage_->SavePersistentSession(
|
| - session_id, SessionData(key_set_id, mime_type), ExpectResult(success));
|
| - }
|
| -
|
| - void LoadPersistentSession(const std::string& session_id,
|
| - const std::vector<uint8_t>& expected_key_set_id,
|
| - const std::string& expected_mime_type) {
|
| - media_drm_storage_->LoadPersistentSession(
|
| - session_id, ExpectResult(base::MakeUnique<SessionData>(
|
| - expected_key_set_id, expected_mime_type)));
|
| - }
|
| -
|
| - void LoadPersistentSessionAndExpectFailure(const std::string& session_id) {
|
| - media_drm_storage_->LoadPersistentSession(
|
| - session_id, ExpectResult(std::unique_ptr<SessionData>()));
|
| - }
|
| -
|
| - void RemovePersistentSession(const std::string& session_id,
|
| - bool success = true) {
|
| - media_drm_storage_->RemovePersistentSession(session_id,
|
| - ExpectResult(success));
|
| - }
|
| -
|
| - void SaveAndLoadPersistentSession(const std::string& session_id,
|
| - const std::vector<uint8_t>& key_set_id,
|
| - const std::string& mime_type) {
|
| - SavePersistentSession(session_id, key_set_id, mime_type);
|
| - LoadPersistentSession(session_id, key_set_id, mime_type);
|
| - }
|
| -
|
| - media::MediaDrmStorage::ResultCB ExpectResult(bool expected_result) {
|
| - return base::BindOnce(&MediaDrmStorageImplTest::CheckResult,
|
| - base::Unretained(this), expected_result);
|
| - }
|
| -
|
| - media::MediaDrmStorage::LoadPersistentSessionCB ExpectResult(
|
| - std::unique_ptr<SessionData> expected_session_data) {
|
| - return base::BindOnce(&MediaDrmStorageImplTest::CheckLoadedSession,
|
| - base::Unretained(this),
|
| - base::Passed(&expected_session_data));
|
| - }
|
| -
|
| - void CheckResult(bool expected_result, bool result) {
|
| - EXPECT_EQ(expected_result, result);
|
| - }
|
| -
|
| - void CheckLoadedSession(std::unique_ptr<SessionData> expected_session_data,
|
| - std::unique_ptr<SessionData> session_data) {
|
| - if (!expected_session_data) {
|
| - EXPECT_FALSE(session_data);
|
| - return;
|
| - }
|
| -
|
| - EXPECT_EQ(expected_session_data->key_set_id, session_data->key_set_id);
|
| - EXPECT_EQ(expected_session_data->mime_type, session_data->mime_type);
|
| - }
|
| -
|
| - base::TestMessageLoop message_loop_;
|
| - std::unique_ptr<TestingPrefServiceSimple> pref_service_;
|
| - std::unique_ptr<media::MediaDrmStorage> media_drm_storage_;
|
| -};
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, OnProvisioned) {
|
| - OnProvisioned();
|
| - base::RunLoop().RunUntilIdle();
|
| -
|
| - // Verify the origin dictionary is created.
|
| - const base::DictionaryValue* storage_dict =
|
| - pref_service_->GetDictionary(kMediaDrmStorage);
|
| - EXPECT_TRUE(
|
| - storage_dict->GetDictionaryWithoutPathExpansion(kTestOrigin, nullptr));
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, OnProvisioned_Twice) {
|
| - OnProvisioned();
|
| - SaveAndLoadPersistentSession("session_id", {1, 0}, "mime/type1");
|
| - // Provisioning again will clear everything associated with the origin.
|
| - OnProvisioned();
|
| - LoadPersistentSessionAndExpectFailure("session_id");
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, SaveSession_Unprovisioned) {
|
| - SavePersistentSession("session_id", {1, 0}, "mime/type", false);
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, SaveSession_SaveTwice) {
|
| - OnProvisioned();
|
| - SaveAndLoadPersistentSession("session_id", {1, 0}, "mime/type1");
|
| - SaveAndLoadPersistentSession("session_id", {2, 3}, "mime/type2");
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, SaveAndLoadSession_LoadTwice) {
|
| - OnProvisioned();
|
| - SaveAndLoadPersistentSession("session_id", {1, 0}, "mime/type");
|
| - LoadPersistentSession("session_id", {1, 0}, "mime/type");
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, SaveAndLoadSession_SpecialCharacters) {
|
| - OnProvisioned();
|
| - SaveAndLoadPersistentSession("session.id", {1, 0}, "mime.type");
|
| - SaveAndLoadPersistentSession("session/id", {1, 0}, "mime/type");
|
| - SaveAndLoadPersistentSession("session-id", {1, 0}, "mime-type");
|
| - SaveAndLoadPersistentSession("session_id", {1, 0}, "mime_type");
|
| - SaveAndLoadPersistentSession("session,id", {1, 0}, "mime,type");
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, LoadSession_Unprovisioned) {
|
| - LoadPersistentSessionAndExpectFailure("session_id");
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, RemoveSession_Success) {
|
| - OnProvisioned();
|
| - SaveAndLoadPersistentSession("session_id", {1, 0}, "mime/type");
|
| - RemovePersistentSession("session_id", true);
|
| - LoadPersistentSessionAndExpectFailure("session_id");
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(MediaDrmStorageImplTest, RemoveSession_InvalidSession) {
|
| - OnProvisioned();
|
| - SaveAndLoadPersistentSession("session_id", {1, 0}, "mime/type");
|
| - RemovePersistentSession("invalid_session_id", true);
|
| - // Can still load "session_id" session.
|
| - LoadPersistentSession("session_id", {1, 0}, "mime/type");
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -} // namespace cdm
|
|
|