[eme] Reject CDM calls after connection error

media/base/mock_filters.h

Index: media/base/mock_filters.h
diff --git a/media/base/mock_filters.h b/media/base/mock_filters.h
index bf58eea91c83ccd4425ed1655ff001b5c86db486..c583104dc4aa210e7c4ffce7760b075a44ddd7f5 100644
--- a/media/base/mock_filters.h
+++ b/media/base/mock_filters.h
@@ -17,6 +17,7 @@
 #include "media/base/audio_renderer.h"
 #include "media/base/cdm_context.h"
 #include "media/base/cdm_key_information.h"
+#include "media/base/cdm_promise.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/decryptor.h"
 #include "media/base/demuxer.h"
@@ -437,6 +438,37 @@ class MockStreamParser : public StreamParser {
   DISALLOW_COPY_AND_ASSIGN(MockStreamParser);
 };
 
+class MockCdmPromise : public SimpleCdmPromise {

[xhwang, 2016/12/14 18:43:54]

nit: please move these closer to other CDM related mocks, e.g. around line 417.

[jrummell, 2016/12/15 21:53:56]

Done.

+ public:
+  // |expect_success| is true if resolve() should be called, false if reject()
+  // is expected.
+  MockCdmPromise(bool expect_success);

[xhwang, 2016/12/14 18:43:54]

explicit

[jrummell, 2016/12/15 21:53:56]

Done.

+  ~MockCdmPromise() override;
+
+  MOCK_METHOD0(resolve, void());
+  MOCK_METHOD3(reject,
+               void(CdmPromise::Exception, uint32_t, const std::string&));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockCdmPromise);
+};
+
+class MockCdmSessionPromise : public NewSessionCdmPromise {
+ public:
+  // |expect_success| is true if resolve() should be called, false if reject()
+  // is expected. |new_session_id| is updated with the new session's ID on
+  // resolve().
+  MockCdmSessionPromise(bool expect_success, std::string* new_session_id);
+  ~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);
+};
+
 }  // namespace media
 
 #endif  // MEDIA_BASE_MOCK_FILTERS_H_