Use Chromium BUILD to approximate Blimp protocol version, and check it.

blimp/common/create_blimp_message_unittest.cc

Index: blimp/common/create_blimp_message_unittest.cc
diff --git a/blimp/common/create_blimp_message_unittest.cc b/blimp/common/create_blimp_message_unittest.cc
index d2b57dc5488ad63b68cb4830e1867dd8735a1174..123cf98956617b69b39e08ac7615dbaaa9ff62e7 100644
--- a/blimp/common/create_blimp_message_unittest.cc
+++ b/blimp/common/create_blimp_message_unittest.cc
@@ -91,5 +91,29 @@ TEST(CreateBlimpMessageTest, StartConnectionMessage) {
             message->protocol_control().start_connection().protocol_version());
 }
 
+TEST(CreateBlimpMessageTest, EndConnectionMessage) {
+  std::unique_ptr<BlimpMessage> message =
+      CreateEndConnectionMessage(EndConnectionMessage::PROTOCOL_MISMATCH);
+  EXPECT_NE(nullptr, message);

[Kevin M, 2016/04/22 21:12:54]

ASSERT_NE for null checks?

+  EXPECT_EQ(BlimpMessage::PROTOCOL_CONTROL, message->type());
+  EXPECT_EQ(ProtocolControlMessage::END_CONNECTION,
+            message->protocol_control().type());
+  EXPECT_EQ(EndConnectionMessage::PROTOCOL_MISMATCH,
+            message->protocol_control().end_connection().reason());
+}
+
+TEST(CreateBlimpMessageTest, CheckpointAckMessage) {
+  const int64_t kTestCheckpointId = 1;
+
+  std::unique_ptr<BlimpMessage> message =
+      CreateCheckpointAckMessage(kTestCheckpointId);
+  EXPECT_NE(nullptr, message);
+  EXPECT_EQ(BlimpMessage::PROTOCOL_CONTROL, message->type());
+  EXPECT_EQ(ProtocolControlMessage::CHECKPOINT_ACK,
+            message->protocol_control().type());
+  EXPECT_EQ(kTestCheckpointId,
+            message->protocol_control().checkpoint_ack().checkpoint_id());
+}
+
 }  // namespace
 }  // namespace blimp