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:
+  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;
+
+  queue_->DrainMessages(1, &messages);
+  ASSERT_TRUE(messages.empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestQueuedMessageSentWithNextFrame) {
+  std::vector<IPC::Message> messages;
+
+  queue_->DrainMessages(1, &messages);
+  queue_->QueueMessage(new IPC::Message(first_message_));
+  queue_->DrainMessages(2, &messages);
+  ASSERT_EQ(1u, messages.size());
+  ASSERT_EQ(first_message_.routing_id(), messages.front().routing_id());
+  messages.clear();
+
+  queue_->DrainMessages(2, &messages);
+  ASSERT_TRUE(messages.empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestQueueMessageSentWithCurrentFrame) {
+  std::vector<IPC::Message> messages;
+
+  queue_->DrainMessages(1, &messages);
+  queue_->QueueMessage(new IPC::Message(first_message_));
+  queue_->DrainMessages(1, &messages);
+  ASSERT_EQ(1u, messages.size());
+  ASSERT_EQ(first_message_.routing_id(), messages.front().routing_id());
+  messages.clear();
+
+  queue_->DrainMessages(1, &messages);
+  ASSERT_TRUE(messages.empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestTryQueueMessageForNextFrame) {
+  std::vector<IPC::Message> messages;
+
+  queue_->DrainMessages(1, &messages);
+  ASSERT_TRUE(queue_->TryQueueMessage(2, new IPC::Message(first_message_)));
+  queue_->DrainMessages(2, &messages);
+  ASSERT_EQ(1u, messages.size());
+  ASSERT_EQ(first_message_.routing_id(), messages.front().routing_id());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestTryQueueMessageForCurrentFrame) {
+  std::vector<IPC::Message> messages;
+
+  queue_->DrainMessages(2, &messages);
+  ASSERT_FALSE(queue_->TryQueueMessage(1, new IPC::Message(first_message_)));
+  queue_->DrainMessages(2, &messages);
+  ASSERT_TRUE(messages.empty());
+}
+
+TEST_F(FrameSwapMessageQueueTest, TestDrainsMessagesForCorrespondingFrames) {
+  std::vector<IPC::Message> messages;
+
+  ASSERT_TRUE(queue_->TryQueueMessage(1, new IPC::Message(first_message_)));
+  ASSERT_TRUE(queue_->TryQueueMessage(2, new IPC::Message(second_message_)));
+  ASSERT_TRUE(queue_->TryQueueMessage(3, new IPC::Message(third_message_)));
+  queue_->DrainMessages(0, &messages);
+  ASSERT_TRUE(messages.empty());
+
+  queue_->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();
+
+  queue_->DrainMessages(2, &messages);
+  ASSERT_TRUE(messages.empty());
+
+  queue_->DrainMessages(5, &messages);
+  ASSERT_EQ(1u, messages.size());
+  ASSERT_EQ(third_message_.routing_id(), messages.front().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;
+  queue_->QueueMessage(
+      new NotifiesDeletionMessage(&message_deleted, first_message_));
+  queue_ = NULL;
+  ASSERT_TRUE(message_deleted);
+}
+
+}  // namespace content