[eme] Reject CDM calls after connection error

media/base/mock_filters.cc

 // 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.
 
 #include "media/base/mock_filters.h"
 
 #include "base/logging.h"
 
 using ::testing::_;
 using ::testing::Invoke;
 using ::testing::NotNull;
 using ::testing::Return;
+using ::testing::SaveArg;
+using ::testing::WithoutArgs;
+
+MATCHER(NotEmpty, "") {
+  return !arg.empty();
+}
 
 namespace media {
 
 MockPipelineClient::MockPipelineClient() {}
 MockPipelineClient::~MockPipelineClient() {}
 
 MockPipeline::MockPipeline() {}
 MockPipeline::~MockPipeline() {}
 
 void MockPipeline::Start(Demuxer* demuxer,
@@ skipping 120 matching lines @@
 }
 
 void MockCdmContext::set_cdm_id(int cdm_id) {
   cdm_id_ = cdm_id;
 }
 
 MockStreamParser::MockStreamParser() {}
 
 MockStreamParser::~MockStreamParser() {}
 
+MockCdmPromise::MockCdmPromise(bool expect_success) {
+  if (expect_success) {
+    EXPECT_CALL(*this, resolve())
+        .WillOnce(
+            WithoutArgs(Invoke(this, &MockCdmPromise::MarkPromiseSettled)));
+    EXPECT_CALL(*this, reject(_, _, _)).Times(0);
+  } else {
+    EXPECT_CALL(*this, resolve()).Times(0);
+    EXPECT_CALL(*this, reject(_, _, NotEmpty()))
+        .WillOnce(
+            WithoutArgs(Invoke(this, &MockCdmPromise::MarkPromiseSettled)));
+  }
+}
+
+MockCdmPromise::~MockCdmPromise() {}
+
+MockCdmSessionPromise::MockCdmSessionPromise(bool expect_success,
+                                             std::string* new_session_id) {
+  if (expect_success) {
+    EXPECT_CALL(*this, resolve(_))
+        .WillOnce(
+            DoAll(SaveArg<0>(new_session_id),
+                  WithoutArgs(Invoke(
+                      this, &MockCdmSessionPromise::MarkPromiseSettled))));
+    EXPECT_CALL(*this, reject(_, _, _)).Times(0);
+  } else {
+    EXPECT_CALL(*this, resolve(_)).Times(0);
+    EXPECT_CALL(*this, reject(_, _, NotEmpty()))
+        .WillOnce(WithoutArgs(
+            Invoke(this, &MockCdmSessionPromise::MarkPromiseSettled)));
+  }
+}
+
+MockCdmSessionPromise::~MockCdmSessionPromise() {}
+
 }  // namespace media