diff options
Diffstat (limited to 'src/test')
-rw-r--r-- | src/test/cpp/util/file_test.cc | 19 |
1 files changed, 13 insertions, 6 deletions
diff --git a/src/test/cpp/util/file_test.cc b/src/test/cpp/util/file_test.cc index 0602dcb20c..c941619244 100644 --- a/src/test/cpp/util/file_test.cc +++ b/src/test/cpp/util/file_test.cc @@ -32,10 +32,13 @@ TEST(FileTest, TestSingleThreadedPipe) { std::unique_ptr<IPipe> pipe(CreatePipe()); char buffer[50] = {0}; ASSERT_TRUE(pipe.get()->Send("hello", 5)); - ASSERT_EQ(3, pipe.get()->Receive(buffer, 3)); + int error = -1; + ASSERT_EQ(3, pipe.get()->Receive(buffer, 3, &error)); ASSERT_TRUE(pipe.get()->Send(" world", 6)); - ASSERT_EQ(5, pipe.get()->Receive(buffer + 3, 5)); - ASSERT_EQ(3, pipe.get()->Receive(buffer + 8, 40)); + ASSERT_EQ(5, pipe.get()->Receive(buffer + 3, 5, &error)); + ASSERT_EQ(IPipe::SUCCESS, error); + ASSERT_EQ(3, pipe.get()->Receive(buffer + 8, 40, &error)); + ASSERT_EQ(IPipe::SUCCESS, error); ASSERT_EQ(0, strncmp(buffer, "hello world", 11)); } @@ -50,9 +53,13 @@ TEST(FileTest, TestMultiThreadedPipe) { // Wait for all data to be fully written to the pipe. writer_thread.join(); - ASSERT_EQ(3, pipe.get()->Receive(buffer, 3)); - ASSERT_EQ(5, pipe.get()->Receive(buffer + 3, 5)); - ASSERT_EQ(3, pipe.get()->Receive(buffer + 8, 40)); + int error = -1; + ASSERT_EQ(3, pipe.get()->Receive(buffer, 3, &error)); + ASSERT_EQ(IPipe::SUCCESS, error); + ASSERT_EQ(5, pipe.get()->Receive(buffer + 3, 5, &error)); + ASSERT_EQ(IPipe::SUCCESS, error); + ASSERT_EQ(3, pipe.get()->Receive(buffer + 8, 40, &error)); + ASSERT_EQ(IPipe::SUCCESS, error); ASSERT_EQ(0, strncmp(buffer, "hello world", 11)); } |