Make MyServer as a stack variable

23c8ad85 · Yu Yang · fae5d82c · 23c8ad85
隐藏空白更改
内联并排

Showing with 2 addition and 9 deletion

paddle/pserver/test/test_ProtoServer.cpp paddle/pserver/test/test_ProtoServer.cpp +2 -9

未找到文件。
--- a/paddle/pserver/test/test_ProtoServer.cpp
+++ b/paddle/pserver/test/test_ProtoServer.cpp
@@ -161,15 +161,8 @@ TEST(ProtoServer, extended) {
 int main(int argc, char** argv) {
  paddle::initMain(argc, argv);
  testing::InitGoogleTest(&argc, argv);
+  MyServer server(FLAGS_port, FLAGS_rdma_tcp == "rdma" ? 0 : -1);
-  std::unique_ptr<MyServer> server;
+  server.start();
-  if (FLAGS_rdma_tcp == "rdma") {
-    server.reset(new MyServer(FLAGS_port, 0));
-  } else {
-    server.reset(new MyServer(FLAGS_port));
-  }
-  server->start();
  usleep(10000);
  return RUN_ALL_TESTS();