From eee7fed720b42286d18e9ab77543475eea73759e Mon Sep 17 00:00:00 2001
From: lganzzzo <lganzzzo@gmail.com>
Date: Sun, 13 Oct 2019 13:10:35 +0300
Subject: [PATCH] Tests. Fix response->send call

---
 test/oatpp-swagger/ControllerTest.cpp | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/test/oatpp-swagger/ControllerTest.cpp b/test/oatpp-swagger/ControllerTest.cpp
index 1c1183b..52c558d 100644
--- a/test/oatpp-swagger/ControllerTest.cpp
+++ b/test/oatpp-swagger/ControllerTest.cpp
@@ -8,6 +8,7 @@
 
 #include "oatpp-swagger/Controller.hpp"
 #include "oatpp/parser/json/mapping/ObjectMapper.hpp"
+#include "oatpp/core/data/stream/BufferStream.hpp"
 
 #include <iostream>
 
@@ -58,6 +59,8 @@ namespace {
 
 void ControllerTest::onRun() {
 
+  oatpp::data::stream::BufferOutputStream responseBuffer;
+
   // Create ObjectMapper
   auto objectMapper = oatpp::parser::json::mapping::ObjectMapper::createShared();
 
@@ -76,9 +79,10 @@ void ControllerTest::onRun() {
 
     // Get api json from swaggerController
     oatpp::data::stream::ChunkedBuffer stream;
+    stream.setOutputStreamIOMode(oatpp::data::stream::IOMode::BLOCKING);
 
     auto response = swaggerController->api();
-    response->send(&stream);
+    response->send(&stream, &responseBuffer);
 
     std::cout << stream.toString()->c_str() << "\n\n";
 
@@ -99,9 +103,10 @@ void ControllerTest::onRun() {
   { // index.html test
     // Get index.html from swaggerController
     oatpp::data::stream::ChunkedBuffer stream;
+    stream.setOutputStreamIOMode(oatpp::data::stream::IOMode::BLOCKING);
 
     auto response = swaggerController->getUIRoot();
-    response->send(&stream);
+    response->send(&stream, &responseBuffer);
 
     auto responseText = stream.toString();
     OATPP_LOGD(TAG, responseText->c_str());