media: Add MockCdmClient

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_key_information.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/decryptor.h"
 #include "media/base/demuxer.h"
+#include "media/base/media_keys.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"
 #include "media/base/text_track.h"
 #include "media/base/text_track_config.h"
 #include "media/base/time_source.h"
 #include "media/base/video_decoder.h"
@@ skipping 292 matching lines @@
   MOCK_METHOD5(addWebVTTCue, void(const base::TimeDelta& start,
                                   const base::TimeDelta& end,
                                   const std::string& id,
                                   const std::string& content,
                                   const std::string& settings));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockTextTrack);
 };
 
+// Mock CDM callbacks.
+// TODO(xhwang): This could be a subclass of CdmClient if we plan to add one.
+// See http://crbug.com/657940
+class MockCdmClient {
+ public:
+  MockCdmClient();
+  virtual ~MockCdmClient();
+
+  MOCK_METHOD3(OnSessionMessage,
+               void(const std::string& session_id,
+                    MediaKeys::MessageType message_type,
+                    const std::vector<uint8_t>& message));
+  MOCK_METHOD1(OnSessionClosed, void(const std::string& session_id));
+
+  // MOCK methods don't work with move-only types like CdmKeysInfo. Add an extra
+  // OnSessionKeysChangeCalled() function to work around this.
+  MOCK_METHOD2(OnSessionKeysChangeCalled,
+               void(const std::string& session_id,
+                    bool has_additional_usable_key));
+  void OnSessionKeysChange(const std::string& session_id,
+                           bool has_additional_usable_key,
+                           CdmKeysInfo keys_info) {
+    keys_info_.swap(keys_info);
+    OnSessionKeysChangeCalled(session_id, has_additional_usable_key);
+  }
+
+  MOCK_METHOD2(OnSessionExpirationUpdate,
+               void(const std::string& session_id, base::Time new_expiry_time));
+
+  const CdmKeysInfo& keys_info() const { return keys_info_; }
+
+ private:
+  CdmKeysInfo keys_info_;
+};
+
 class MockDecryptor : public Decryptor {
  public:
   MockDecryptor();
   virtual ~MockDecryptor();
 
   MOCK_METHOD2(RegisterNewKeyCB, void(StreamType stream_type,
                                       const NewKeyCB& new_key_cb));
   MOCK_METHOD3(Decrypt, void(StreamType stream_type,
                              const scoped_refptr<DecoderBuffer>& encrypted,
                              const DecryptCB& decrypt_cb));
@@ skipping 52 matching lines @@
   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_