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));
 }