media: Implement MediaDrmStorageImpl with tests

components/cdm/browser/media_drm_storage_impl_unittest.cc

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
new file mode 100644
index 0000000000000000000000000000000000000000..6a1610ee5a364eeeaf0b9f679be63c82b5ed4dbb
--- /dev/null
+++ b/components/cdm/browser/media_drm_storage_impl_unittest.cc
@@ -0,0 +1,196 @@
+// 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