1 files changed, 9 insertions, 2 deletions
diff --git a/chromium/third_party/blink/renderer/core/fetch/bytes_consumer_test_util.cc b/chromium/third_party/blink/renderer/core/fetch/bytes_consumer_test_util.cc
index 93bdfddbd84..9ff3bb84b76 100644
--- a/chromium/third_party/blink/renderer/core/fetch/bytes_consumer_test_util.cc
+++ b/chromium/third_party/blink/renderer/core/fetch/bytes_consumer_test_util.cc
@@ -56,6 +56,7 @@ Result BytesConsumerTestUtil::ReplayingBytesConsumer::BeginRead(
   }
   const Command& command = commands_[0];
   switch (command.GetName()) {
+    case Command::kDataAndDone:
     case Command::kData:
       DCHECK_LE(offset_, command.Body().size());
       *buffer = command.Body().data() + offset_;
@@ -87,7 +88,8 @@ Result BytesConsumerTestUtil::ReplayingBytesConsumer::BeginRead(
 Result BytesConsumerTestUtil::ReplayingBytesConsumer::EndRead(size_t read) {
   DCHECK(!commands_.IsEmpty());
   const Command& command = commands_[0];
-  DCHECK_EQ(Command::kData, command.GetName());
+  const auto name = command.GetName();
+  DCHECK(name == Command::kData || name == Command::kDataAndDone);
   offset_ += read;
   DCHECK_LE(offset_, command.Body().size());
   if (offset_ < command.Body().size())
@@ -95,7 +97,12 @@ Result BytesConsumerTestUtil::ReplayingBytesConsumer::EndRead(size_t read) {
 
   offset_ = 0;
   commands_.pop_front();
-  return Result::kOk;
+
+  if (name == Command::kData)
+    return Result::kOk;
+
+  Close();
+  return Result::kDone;
 }
 
 void BytesConsumerTestUtil::ReplayingBytesConsumer::SetClient(Client* client) {