[eme] Reject CDM calls after connection error

media/base/mock_filters.cc

Index: media/base/mock_filters.cc
diff --git a/media/base/mock_filters.cc b/media/base/mock_filters.cc
index 19ace24e3953090427e6cd63f0bc1111decc7c61..5d0bae3c9023137cd728d9711d8c4e082ac7db7a 100644
--- a/media/base/mock_filters.cc
+++ b/media/base/mock_filters.cc
@@ -10,6 +10,12 @@ using ::testing::_;
 using ::testing::Invoke;
 using ::testing::NotNull;
 using ::testing::Return;
+using ::testing::SaveArg;
+using ::testing::WithoutArgs;
+
+MATCHER(NotEmpty, "") {
+  return !arg.empty();
+}
 
 namespace media {
 
@@ -150,4 +156,39 @@ 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