test: fix RawConnectionDriver (envoyproxy#122)

Signed-off-by: Yuchen Dai <[email protected]>
billowqiu · May 4, 2020 · b6dccb7 · b6dccb7
1 parent 998b349
commit b6dccb7
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/echo2_integration_test.cc b/echo2_integration_test.cc
@@ -30,17 +30,15 @@ INSTANTIATE_TEST_SUITE_P(IpVersions, Echo2IntegrationTest,
                         testing::ValuesIn(TestEnvironment::getIpVersionsForTest()));
 
 TEST_P(Echo2IntegrationTest, Echo) {
-  Buffer::OwnedImpl buffer("hello");
   std::string response;
-  RawConnectionDriver connection(
-      lookupPort("listener_0"), buffer,
-      [&](Network::ClientConnection&, const Buffer::Instance& data) -> void {
+  auto connection = createConnectionDriver(
+      lookupPort("listener_0"), "hello",
+      [&](Network::ClientConnection& conn, const Buffer::Instance& data) -> void {
         response.append(data.toString());
-        connection.close();
-      },
-      GetParam());
+	conn.close(Network::ConnectionCloseType::FlushWrite);
+      });
 
-  connection.run();
+  connection->run();
   EXPECT_EQ("hello", response);
 }
 } // namespace Envoy