[balsa] Fix Http1ClientConnectionImplTest.HeadRequest. (#22105)

Fix Http1ClientConnectionImplTest.HeadRequest for balsa parser. Signed-off-by: Bence Béky <bnc@google.com>
envoyproxy · Jul 12, 2022 · d4bf478 · d4bf478
1 parent 0015180
commit d4bf478
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/source/common/http/http1/balsa_parser.cc b/source/common/http/http1/balsa_parser.cc
@@ -194,7 +194,11 @@ void BalsaParser::HeaderDone() {
     return;
   }
   headers_done_ = true;
-  status_ = convertResult(connection_->onHeadersComplete());
+  CallbackResult result = connection_->onHeadersComplete();
+  status_ = convertResult(result);
+  if (result == CallbackResult::NoBody || result == CallbackResult::NoBodyData) {
+    MessageDone();
+  }
 }
 
 void BalsaParser::ContinueHeaderDone() {}