[eme] Break mojo connection during call

media/base/mock_filters.h

 // Copyright (c) 2012 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_BASE_MOCK_FILTERS_H_
 #define MEDIA_BASE_MOCK_FILTERS_H_
 
 #include <stdint.h>
 
 #include <memory>
 #include <string>
 
 #include "base/callback.h"
 #include "base/macros.h"
 #include "media/base/audio_decoder.h"
 #include "media/base/audio_decoder_config.h"
 #include "media/base/audio_renderer.h"
 #include "media/base/cdm_context.h"
+#include "media/base/cdm_factory.h"
 #include "media/base/cdm_key_information.h"
 #include "media/base/cdm_promise.h"
+#include "media/base/cdm_promise_adapter.h"
 #include "media/base/content_decryption_module.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/decryptor.h"
 #include "media/base/demuxer.h"
 #include "media/base/media_track.h"
 #include "media/base/pipeline.h"
 #include "media/base/pipeline_status.h"
 #include "media/base/renderer.h"
 #include "media/base/renderer_client.h"
 #include "media/base/stream_parser.h"
@@ skipping 408 matching lines @@
   ~MockCdmSessionPromise() override;
 
   MOCK_METHOD1(resolve, void(const std::string&));
   MOCK_METHOD3(reject,
                void(CdmPromise::Exception, uint32_t, const std::string&));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockCdmSessionPromise);
 };
 
+class MockCdm : public ContentDecryptionModule {
+ public:
+  MockCdm();
+
+  // ContentDecryptionModule implementation.
+  // As move-only parameters aren't supported by mock methods, convert promises
+  // into IDs and pass them to On... methods.
+  void SetServerCertificate(const std::vector<uint8_t>& certificate,
+                            std::unique_ptr<SimpleCdmPromise> promise) override;
+  MOCK_METHOD2(OnSetServerCertificate,
+               void(const std::vector<uint8_t>& certificate,
+                    uint32_t promise_id));
+
+  void CreateSessionAndGenerateRequest(
+      CdmSessionType session_type,
+      EmeInitDataType init_data_type,
+      const std::vector<uint8_t>& init_data,
+      std::unique_ptr<NewSessionCdmPromise> promise) override;
+  MOCK_METHOD4(OnCreateSessionAndGenerateRequest,
+               void(CdmSessionType session_type,
+                    EmeInitDataType init_data_type,
+                    const std::vector<uint8_t>& init_data,
+                    uint32_t promise_id));

[xhwang, 2017/01/05 06:59:53]

Instead of using the promise adapter to convert the promise to an ID, I think we
could also do something like 

MOCK_METHOD4(OnCreateSessionAndGenerateRequest,
                                ....
                                std::unique_ptr<NewSessionCdmPromise>& promise);

Note that the promise is passed as a reference. I didn't try this but if it can
work it'll be simpler, as we don't need to keep the id around and ask the
MockCdm to resolve/reject the promise, we can just resolve/reject it directly.

[jrummell, 2017/01/05 23:47:22]

Done.

+
+  void LoadSession(CdmSessionType session_type,
+                   const std::string& session_id,
+                   std::unique_ptr<NewSessionCdmPromise> promise) override;
+  MOCK_METHOD3(OnLoadSession,
+               void(CdmSessionType session_type,
+                    const std::string& session_id,
+                    uint32_t promise_id));
+
+  void UpdateSession(const std::string& session_id,
+                     const std::vector<uint8_t>& response,
+                     std::unique_ptr<SimpleCdmPromise> promise) override;
+  MOCK_METHOD3(OnUpdateSession,
+               void(const std::string& session_id,
+                    const std::vector<uint8_t>& response,
+                    uint32_t promise_id));
+
+  void CloseSession(const std::string& session_id,
+                    std::unique_ptr<SimpleCdmPromise> promise) override;
+  MOCK_METHOD2(OnCloseSession,
+               void(const std::string& session_id, uint32_t promise_id));
+
+  void RemoveSession(const std::string& session_id,
+                     std::unique_ptr<SimpleCdmPromise> promise) override;
+  MOCK_METHOD2(OnRemoveSession,
+               void(const std::string& session_id, uint32_t promise_id));
+
+  MOCK_METHOD0(GetCdmContext, CdmContext*());
+
+  // As the promises are saved to |cdm_promise_adapter_|, these methods
+  // provide a way for the mock methods to fulfill the promise as necessary
+  // for the test.
+  void ResolvePromise(uint32_t promise_id);
+  void RejectPromise(uint32_t promise_id);
+  void ResolvePromiseWithSession(uint32_t promise_id,
+                                 const std::string& session_id);
+
+  // As MockCdmFactory returns the MockCdm provided to it, this sets the
+  // actual callbacks used when MockCdmFactory::Create() is called.
+  void SetCallbacks(
+      const SessionMessageCB& session_message_cb,
+      const SessionClosedCB& session_closed_cb,
+      const SessionKeysChangeCB& session_keys_change_cb,
+      const SessionExpirationUpdateCB& session_expiration_update_cb);
+  void CallSessionMessageCB(const std::string& session_id,
+                            ContentDecryptionModule::MessageType message_type,
+                            const std::vector<uint8_t>& message);
+  void CallSessionClosedCB(const std::string& session_id);
+  void CallSessionKeysChangeCB(const std::string& session_id,
+                               bool has_additional_usable_key,
+                               CdmKeysInfo keys_info);
+  void CallSessionExpirationUpdateCB(const std::string& session_id,
+                                     base::Time new_expiry_time);
+
+ protected:
+  ~MockCdm() override;
+
+ private:
+  // Callbacks.
+  SessionMessageCB session_message_cb_;
+  SessionClosedCB session_closed_cb_;
+  SessionKeysChangeCB session_keys_change_cb_;
+  SessionExpirationUpdateCB session_expiration_update_cb_;
+
+  // Keep track of the promises so that ResolvePromise(), etc. can be called
+  // by the mock methods.
+  CdmPromiseAdapter cdm_promise_adapter_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockCdm);
+};
+
+class MockCdmFactory : public CdmFactory {
+ public:
+  MockCdmFactory();
+  ~MockCdmFactory() override;
+
+  // CdmFactory implementation.
+  MOCK_METHOD8(
+      Create,
+      void(const std::string& key_system,
+           const GURL& security_origin,
+           const CdmConfig& cdm_config,
+           const SessionMessageCB& session_message_cb,
+           const SessionClosedCB& session_closed_cb,
+           const SessionKeysChangeCB& session_keys_change_cb,
+           const SessionExpirationUpdateCB& session_expiration_update_cb,
+           const CdmCreatedCB& cdm_created_cb));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockCdmFactory);
+};
+
 class MockStreamParser : public StreamParser {
  public:
   MockStreamParser();
   ~MockStreamParser() override;
 
   // StreamParser interface
   MOCK_METHOD8(
       Init,
       void(const InitCB& init_cb,
            const NewConfigCB& config_cb,
            const NewBuffersCB& new_buffers_cb,
            bool ignore_text_track,
            const EncryptedMediaInitDataCB& encrypted_media_init_data_cb,
            const NewMediaSegmentCB& new_segment_cb,
            const EndMediaSegmentCB& end_of_segment_cb,
            const scoped_refptr<MediaLog>& media_log));
   MOCK_METHOD0(Flush, void());
   MOCK_METHOD2(Parse, bool(const uint8_t*, int));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockStreamParser);
 };
 
 }  // namespace media
 
 #endif  // MEDIA_BASE_MOCK_FILTERS_H_