MachMessageDeadline WIP

util/mach/mach_message_test.cc

 // Copyright 2014 The Crashpad Authors. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-#include "util/mach/mach_message_util.h"
+#include "util/mach/mach_message.h"
 
 #include "base/basictypes.h"
 #include "gtest/gtest.h"
 #include "util/mach/mach_extensions.h"
 
 namespace crashpad {
 namespace test {
 namespace {
 
-TEST(MachMessageUtil, PrepareMIGReplyFromRequest_SetMIGReplyError) {
+TEST(MachMessage, MachMessageDeadlineFromTimeout) {
+  MachMessageDeadline deadline_0 = MachMessageDeadlineFromTimeout(0);
+  EXPECT_EQ(kMachMessageNonblocking, deadline_0);
+
+  deadline_0 = MachMessageDeadlineFromTimeout(1);
+  MachMessageDeadline deadline_1 = MachMessageDeadlineFromTimeout(100);
+
+  EXPECT_NE(kMachMessageNonblocking, deadline_0);
+  EXPECT_NE(kMachMessageWaitIndefinitely, deadline_0);
+  EXPECT_NE(kMachMessageNonblocking, deadline_1);
+  EXPECT_NE(kMachMessageWaitIndefinitely, deadline_1);
+  EXPECT_GE(deadline_1, deadline_0);
+}
+
+TEST(MachMessage, PrepareMIGReplyFromRequest_SetMIGReplyError) {
   mach_msg_header_t request;
   request.msgh_bits =
       MACH_MSGH_BITS_COMPLEX |
       MACH_MSGH_BITS(MACH_MSG_TYPE_PORT_SEND_ONCE, MACH_MSG_TYPE_PORT_SEND);
   request.msgh_size = 64;
   request.msgh_remote_port = 0x01234567;
   request.msgh_local_port = 0x89abcdef;
   request.msgh_reserved = 0xa5a5a5a5;
   request.msgh_id = 1011;
 
@@ skipping 20 matching lines @@
   EXPECT_EQ(NDR_record.char_rep, reply.NDR.char_rep);
   EXPECT_EQ(NDR_record.float_rep, reply.NDR.float_rep);
   EXPECT_EQ(NDR_record.reserved2, reply.NDR.reserved2);
   EXPECT_EQ(MIG_TYPE_ERROR, reply.RetCode);
 
   SetMIGReplyError(&reply.Head, MIG_BAD_ID);
 
   EXPECT_EQ(MIG_BAD_ID, reply.RetCode);
 }
 
-TEST(MachMessageUtil, MachMessageTrailerFromHeader) {
+TEST(MachMessage, MachMessageTrailerFromHeader) {
   mach_msg_empty_t empty;
   empty.send.header.msgh_size = sizeof(mach_msg_empty_send_t);
   EXPECT_EQ(&empty.rcv.trailer,
             MachMessageTrailerFromHeader(&empty.rcv.header));
 
   struct TestSendMessage : public mach_msg_header_t {
     uint8_t data[126];
   };
   struct TestReceiveMessage : public TestSendMessage {
     mach_msg_trailer_t trailer;
   };
   union TestMessage {
     TestSendMessage send;
     TestReceiveMessage receive;
   };
 
   TestMessage test;
   test.send.msgh_size = sizeof(TestSendMessage);
-  EXPECT_EQ(&test.receive.trailer,
-            MachMessageTrailerFromHeader(&test.receive));
+  EXPECT_EQ(&test.receive.trailer, MachMessageTrailerFromHeader(&test.receive));
 }
 
 }  // namespace
 }  // namespace test
 }  // namespace crashpad