Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1165)

Unified Diff: media/base/mock_filters.h

Issue 2592913002: [eme] Break mojo connection during call (Closed)
Patch Set: rebase Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « no previous file | media/base/mock_filters.cc » ('j') | media/mojo/clients/mojo_cdm_unittest.cc » ('J')
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: media/base/mock_filters.h
diff --git a/media/base/mock_filters.h b/media/base/mock_filters.h
index 3e6f8be44b3aeb027cb264a7b1739afa1835751a..332662a9c6650c89d708d4f178c86df5909a303a 100644
--- a/media/base/mock_filters.h
+++ b/media/base/mock_filters.h
@@ -16,8 +16,10 @@
#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"
@@ -446,6 +448,121 @@ class MockCdmSessionPromise : public NewSessionCdmPromise {
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 th
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();
« no previous file with comments | « no previous file | media/base/mock_filters.cc » ('j') | media/mojo/clients/mojo_cdm_unittest.cc » ('J')

Powered by Google App Engine
This is Rietveld 408576698