Deliver IPC messages together with SwapCompositorFrame

content/renderer/gpu/frame_swap_message_queue_unittest.cc

+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "content/renderer/gpu/frame_swap_message_queue.h"
+#include "ipc/ipc_message.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+class FrameSwapMessageQueueTest : public testing::Test {
+ public:
+  FrameSwapMessageQueueTest()
+      : first_message_(41, 0, IPC::Message::PRIORITY_NORMAL),
+        second_message_(42, 0, IPC::Message::PRIORITY_NORMAL),
+        third_message_(43, 0, IPC::Message::PRIORITY_NORMAL),
+        queue_(new FrameSwapMessageQueue()) {}
+
+ protected:
+  void QueueMessage(IPC::Message* msg) {
+    queue_->QueueMessage(msg);
+  }
+  void QueueMessage(int source_frame_number, IPC::Message* msg) {
+    queue_->QueueMessage(source_frame_number, msg, NULL);
+  }
+  void QueueMessage(int source_frame_number, IPC::Message* msg, bool* first) {
+    queue_->QueueMessage(source_frame_number, msg, first);
+  }
+  void DrainMessages(int source_frame_number,
+                     std::vector<IPC::Message>* messages) {
+    queue_->AdvanceToFrame(source_frame_number);
+    scoped_ptr<FrameSwapMessageQueue::SendMessageScope> send_message_scope =
+        queue_->AcquireSendMessageScope();
+    queue_->DrainMessages(messages);
+  }
+  void DrainMessages(std::vector<IPC::Message>* messages) {
+    scoped_ptr<FrameSwapMessageQueue::SendMessageScope> send_message_scope =
+        queue_->AcquireSendMessageScope();
+    queue_->DrainMessages(messages);
+  }
+  bool HasMessageForId(const std::vector<IPC::Message>& messages,
+                       int routing_id) {
+    for (std::vector<IPC::Message>::const_iterator i = messages.begin();
+         i != messages.end();
+         ++i) {
+      if (i->routing_id() == routing_id)
+        return true;
+    }
+    return false;
+  }
+
+  IPC::Message first_message_;
+  IPC::Message second_message_;
+  IPC::Message third_message_;
+  scoped_refptr<FrameSwapMessageQueue> queue_;
+};
+
+TEST_F(FrameSwapMessageQueueTest, TestEmptyQueueDrain) {
+  std::vector<IPC::Message> messages;
+
+  DrainMessages(&messages);
+  ASSERT_TRUE(messages.empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestEmpty) {
+  std::vector<IPC::Message> messages;
+  ASSERT_TRUE(queue_->Empty());
+  QueueMessage(new IPC::Message(first_message_));
+  ASSERT_FALSE(queue_->Empty());
+  DrainMessages(&messages);
+  ASSERT_TRUE(queue_->Empty());
+  QueueMessage(1, new IPC::Message(first_message_));
+  ASSERT_FALSE(queue_->Empty());
+  queue_->AdvanceToFrame(1);
+  ASSERT_FALSE(queue_->Empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestQueueMessageFirst) {
+  std::vector<IPC::Message> messages;
+  bool first = false;
+
+  QueueMessage(1, new IPC::Message(first_message_), &first);
+  ASSERT_TRUE(first);
+  QueueMessage(1, new IPC::Message(second_message_), &first);
+  ASSERT_FALSE(first);
+  QueueMessage(2, new IPC::Message(first_message_), &first);
+  ASSERT_TRUE(first);
+
+  DrainMessages(4, &messages);
+  QueueMessage(4, new IPC::Message(first_message_), &first);
+  ASSERT_TRUE(first);
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestQueuedMessageSentWithNextFrame) {
+  std::vector<IPC::Message> messages;
+
+  DrainMessages(1, &messages);
+  QueueMessage(new IPC::Message(first_message_));
+  DrainMessages(2, &messages);
+  ASSERT_EQ(1u, messages.size());
+  ASSERT_EQ(first_message_.routing_id(), messages.front().routing_id());
+  messages.clear();
+
+  DrainMessages(2, &messages);
+  ASSERT_TRUE(messages.empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestQueueMessageSentWithCurrentFrame) {
+  std::vector<IPC::Message> messages;
+
+  DrainMessages(1, &messages);
+  QueueMessage(new IPC::Message(first_message_));
+  DrainMessages(1, &messages);
+  ASSERT_EQ(1u, messages.size());
+  ASSERT_EQ(first_message_.routing_id(), messages.front().routing_id());
+  messages.clear();
+
+  DrainMessages(1, &messages);
+  ASSERT_TRUE(messages.empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestDrainsMessagesForCorrespondingFrames) {
+  std::vector<IPC::Message> messages;
+
+  QueueMessage(1, new IPC::Message(first_message_));
+  QueueMessage(2, new IPC::Message(second_message_));
+  QueueMessage(3, new IPC::Message(third_message_));
+  DrainMessages(0, &messages);
+  ASSERT_TRUE(messages.empty());
+
+  DrainMessages(2, &messages);
+  ASSERT_EQ(2u, messages.size());
+  ASSERT_TRUE(HasMessageForId(messages, first_message_.routing_id()));
+  ASSERT_TRUE(HasMessageForId(messages, second_message_.routing_id()));
+  messages.clear();
+
+  DrainMessages(2, &messages);
+  ASSERT_TRUE(messages.empty());
+
+  DrainMessages(5, &messages);
+  ASSERT_EQ(1u, messages.size());
+  ASSERT_EQ(third_message_.routing_id(), messages.front().routing_id());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestQueueMessagePreservesFifoOrdering) {
+  std::vector<IPC::Message> messages;
+
+  QueueMessage(new IPC::Message(first_message_));
+  QueueMessage(new IPC::Message(second_message_));
+  DrainMessages(1, &messages);
+  ASSERT_EQ(2u, messages.size());
+  ASSERT_EQ(first_message_.routing_id(), messages[0].routing_id());
+  ASSERT_EQ(second_message_.routing_id(), messages[1].routing_id());
+}
+
+class NotifiesDeletionMessage : public IPC::Message {
+ public:
+  NotifiesDeletionMessage(bool* deleted, const IPC::Message& other)
+      : IPC::Message(other), deleted_(deleted) {}
+  virtual ~NotifiesDeletionMessage() { *deleted_ = true; }
+
+ private:
+  bool* deleted_;
+};
+
+TEST_F(FrameSwapMessageQueueTest, TestDeletesMessage) {
+  bool message_deleted = false;
+  QueueMessage(new NotifiesDeletionMessage(&message_deleted, first_message_));
+  queue_ = NULL;
+  ASSERT_TRUE(message_deleted);
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestDeletesMessageQueuedForFrame) {
+  bool message_deleted = false;
+  QueueMessage(1,
+               new NotifiesDeletionMessage(&message_deleted, first_message_));
+  queue_ = NULL;
+  ASSERT_TRUE(message_deleted);
+}
+
+}  // namespace content