RELAND: Media Remoting end to end integration tests.

media/test/fake_encrypted_media.h

+// 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.
+
+#ifndef MEDIA_TEST_FAKE_ENCRYPTED_MEDIA_H_
+#define MEDIA_TEST_FAKE_ENCRYPTED_MEDIA_H_
+
+#include "media/base/cdm_context.h"
+#include "media/base/content_decryption_module.h"
+
+namespace media {
+
+class AesDecryptor;
+
+// Note: Tests using this class only exercise the DecryptingDemuxerStream path.
+// They do not exercise the Decrypting{Audio|Video}Decoder path.
+class FakeEncryptedMedia {
+ public:
+  // Defines the behavior of the "app" that responds to EME events.
+  class AppBase {
+   public:
+    virtual ~AppBase() {}
+
+    virtual void OnSessionMessage(
+        const std::string& session_id,
+        ContentDecryptionModule::MessageType message_type,
+        const std::vector<uint8_t>& message,
+        AesDecryptor* decryptor) = 0;
+
+    virtual void OnSessionClosed(const std::string& session_id) = 0;
+
+    virtual void OnSessionKeysChange(const std::string& session_id,
+                                     bool has_additional_usable_key,
+                                     CdmKeysInfo keys_info) = 0;
+
+    virtual void OnEncryptedMediaInitData(EmeInitDataType init_data_type,
+                                          const std::vector<uint8_t>& init_data,
+                                          AesDecryptor* decryptor) = 0;
+  };
+
+  FakeEncryptedMedia(AppBase* app);
+  ~FakeEncryptedMedia();
+  CdmContext* GetCdmContext();
+  // Callbacks for firing session events. Delegate to |app_|.
+  void OnSessionMessage(const std::string& session_id,
+                        ContentDecryptionModule::MessageType message_type,
+                        const std::vector<uint8_t>& message);
+  void OnSessionClosed(const std::string& session_id);
+  void OnSessionKeysChange(const std::string& session_id,
+                           bool has_additional_usable_key,
+                           CdmKeysInfo keys_info);
+  void OnEncryptedMediaInitData(EmeInitDataType init_data_type,
+                                const std::vector<uint8_t>& init_data);
+
+ private:
+  class TestCdmContext : public CdmContext {
+   public:
+    TestCdmContext(Decryptor* decryptor);
+    Decryptor* GetDecryptor() final;
+    int GetCdmId() const final;
+
+   private:
+    Decryptor* decryptor_;
+  };
+
+  scoped_refptr<AesDecryptor> decryptor_;
+  TestCdmContext cdm_context_;
+  std::unique_ptr<AppBase> app_;
+
+  DISALLOW_COPY_AND_ASSIGN(FakeEncryptedMedia);
+};
+
+}  // namespace media
+
+#endif  // MEDIA_TEST_FAKE_ENCRYPTED_MEDIA_H_