Unverified Commit 3a11d261 authored by Chao Ma's avatar Chao Ma Committed by GitHub
Browse files

update (#593)

parent 01a4cc5b
...@@ -21,19 +21,13 @@ bool start_server(); ...@@ -21,19 +21,13 @@ bool start_server();
#include <unistd.h> #include <unistd.h>
TEST(SocketCommunicatorTest, SendAndRecv) { TEST(SocketCommunicatorTest, SendAndRecv) {
int pid = fork(); std::thread client_thread(start_client);
ASSERT_GE(pid, 0); start_server();
if (pid > 0) { client_thread.join();
EXPECT_TRUE(start_server());
} else {
start_client();
}
} }
#else // WIN32 #else // WIN32
// Win32 doesn't have a fork() equivalent so use threads instead.
#include <windows.h> #include <windows.h>
#include <winsock2.h> #include <winsock2.h>
...@@ -79,12 +73,12 @@ TEST(SocketCommunicatorTest, SendAndRecv) { ...@@ -79,12 +73,12 @@ TEST(SocketCommunicatorTest, SendAndRecv) {
#endif // WIN32 #endif // WIN32
void start_client() { void start_client() {
const char * msg = "0123456789"; const char * msg = "123456789";
sleep(1); sleep(1);
SocketSender sender; SocketSender sender;
sender.AddReceiver("127.0.0.1", 2049, 0); sender.AddReceiver("127.0.0.1", 2049, 0);
sender.Connect(); sender.Connect();
sender.Send(msg, 10, 0); sender.Send(msg, 9, 0);
sender.Finalize(); sender.Finalize();
} }
...@@ -92,8 +86,8 @@ bool start_server() { ...@@ -92,8 +86,8 @@ bool start_server() {
char serbuff[10]; char serbuff[10];
memset(serbuff, '\0', 10); memset(serbuff, '\0', 10);
SocketReceiver receiver; SocketReceiver receiver;
receiver.Wait("127.0.0.1", 2049, 1, 500); receiver.Wait("127.0.0.1", 2049, 1, 500 * 1024);
receiver.Recv(serbuff, 10); receiver.Recv(serbuff, 9);
receiver.Finalize(); receiver.Finalize();
return string("0123456789") == string(serbuff); return string("123456789") == string(serbuff);
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment