Support using TrackedCallbacks as hints to determine the handling thread of resource reply messages

ppapi/proxy/websocket_resource_unittest.cc

 // Copyright (c) 2012 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 "base/memory/ref_counted.h"
 #include "base/message_loop/message_loop.h"
 #include "ppapi/c/pp_errors.h"
 #include "ppapi/c/ppb_var.h"
 #include "ppapi/c/ppb_websocket.h"
 #include "ppapi/proxy/locking_resource_releaser.h"
+#include "ppapi/proxy/plugin_message_filter.h"
 #include "ppapi/proxy/ppapi_messages.h"
 #include "ppapi/proxy/ppapi_proxy_test.h"
 #include "ppapi/proxy/websocket_resource.h"
 #include "ppapi/shared_impl/ppapi_globals.h"
 #include "ppapi/shared_impl/ppb_var_shared.h"
 #include "ppapi/shared_impl/proxy_lock.h"
 #include "ppapi/shared_impl/resource_tracker.h"
 #include "ppapi/shared_impl/scoped_pp_resource.h"
 #include "ppapi/shared_impl/scoped_pp_var.h"
 #include "ppapi/shared_impl/tracked_callback.h"
@@ skipping 56 matching lines @@
   PpapiHostMsg_WebSocket_Connect::Schema::Param p;
   PpapiHostMsg_WebSocket_Connect::Read(&msg, &p);
   EXPECT_EQ(url, p.a);
   EXPECT_EQ(protocol0, p.b[0]);
   EXPECT_EQ(protocol1, p.b[1]);
 
   // Synthesize a response.
   ResourceMessageReplyParams reply_params(params.pp_resource(),
                                           params.sequence());
   reply_params.set_result(PP_OK);
-  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-      PpapiPluginMsg_ResourceReply(reply_params,
-          PpapiPluginMsg_WebSocket_ConnectReply(url, protocol1))));
+  PluginMessageFilter::DispatchResourceReplyForTest(
+      reply_params, PpapiPluginMsg_WebSocket_ConnectReply(url, protocol1));
 
   EXPECT_EQ(PP_OK, g_callback_result);
   EXPECT_EQ(true, g_callback_called);
 }
 
 // Does a test for unsolicited replies.
 TEST_F(WebSocketResourceTest, UnsolicitedReplies) {
   const PPB_WebSocket_1_0* websocket_iface =
       thunk::GetPPB_WebSocket_1_0_Thunk();
 
   LockingResourceReleaser res(websocket_iface->Create(pp_instance()));
 
   // Check if BufferedAmountReply is handled.
   ResourceMessageReplyParams reply_params(res.get(), 0);
   reply_params.set_result(PP_OK);
-  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-      PpapiPluginMsg_ResourceReply(
-          reply_params,
-          PpapiPluginMsg_WebSocket_BufferedAmountReply(19760227u))));
+  PluginMessageFilter::DispatchResourceReplyForTest(
+      reply_params, PpapiPluginMsg_WebSocket_BufferedAmountReply(19760227u));
 
   uint64_t amount = websocket_iface->GetBufferedAmount(res.get());
   EXPECT_EQ(19760227u, amount);
 
   // Check if StateReply is handled.
-  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-      PpapiPluginMsg_ResourceReply(
-          reply_params,
-          PpapiPluginMsg_WebSocket_StateReply(
-              static_cast<int32_t>(PP_WEBSOCKETREADYSTATE_CLOSING)))));
+  PluginMessageFilter::DispatchResourceReplyForTest(
+      reply_params,
+      PpapiPluginMsg_WebSocket_StateReply(
+          static_cast<int32_t>(PP_WEBSOCKETREADYSTATE_CLOSING)));
 
   PP_WebSocketReadyState state = websocket_iface->GetReadyState(res.get());
   EXPECT_EQ(PP_WEBSOCKETREADYSTATE_CLOSING, state);
 }
 
 TEST_F(WebSocketResourceTest, MessageError) {
   const PPB_WebSocket_1_0* websocket_iface =
       thunk::GetPPB_WebSocket_1_0_Thunk();
 
   std::string url("ws://ws.google.com");
   PP_Var url_var = MakeStringVar(url);
 
   LockingResourceReleaser res(websocket_iface->Create(pp_instance()));
 
   // Establish the connection virtually.
   int32_t result =
       websocket_iface->Connect(res.get(), url_var, NULL, 0, MakeCallback());
   ASSERT_EQ(PP_OK_COMPLETIONPENDING, result);
 
   ResourceMessageCallParams params;
   IPC::Message msg;
   ASSERT_TRUE(sink().GetFirstResourceCallMatching(
       PpapiHostMsg_WebSocket_Connect::ID, &params, &msg));
 
   ResourceMessageReplyParams connect_reply_params(params.pp_resource(),
                                                   params.sequence());
   connect_reply_params.set_result(PP_OK);
-  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-      PpapiPluginMsg_ResourceReply(connect_reply_params,
-          PpapiPluginMsg_WebSocket_ConnectReply(url, std::string()))));
+  PluginMessageFilter::DispatchResourceReplyForTest(
+      connect_reply_params,
+      PpapiPluginMsg_WebSocket_ConnectReply(url, std::string()));
 
   EXPECT_EQ(PP_OK, g_callback_result);
   EXPECT_TRUE(g_callback_called);
 
   PP_Var message;
   result = websocket_iface->ReceiveMessage(res.get(), &message, MakeCallback());
   EXPECT_FALSE(g_callback_called);
 
   // Synthesize a WebSocket_ErrorReply message.
   ResourceMessageReplyParams error_reply_params(res.get(), 0);
   error_reply_params.set_result(PP_OK);
-  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-      PpapiPluginMsg_ResourceReply(error_reply_params,
-          PpapiPluginMsg_WebSocket_ErrorReply())));
+  PluginMessageFilter::DispatchResourceReplyForTest(
+      error_reply_params, PpapiPluginMsg_WebSocket_ErrorReply());
 
   EXPECT_EQ(PP_ERROR_FAILED, g_callback_result);
   EXPECT_TRUE(g_callback_called);
 }
 
 }  // namespace proxy
 }  // namespace ppapi