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

ppapi/proxy/device_enumeration_resource_helper_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/basictypes.h"
 #include "base/compiler_specific.h"
 #include "ppapi/c/pp_errors.h"
 #include "ppapi/proxy/connection.h"
 #include "ppapi/proxy/device_enumeration_resource_helper.h"
+#include "ppapi/proxy/plugin_message_filter.h"
 #include "ppapi/proxy/plugin_resource.h"
 #include "ppapi/proxy/plugin_resource_tracker.h"
 #include "ppapi/proxy/plugin_var_tracker.h"
 #include "ppapi/proxy/ppapi_message_utils.h"
 #include "ppapi/proxy/ppapi_messages.h"
 #include "ppapi/proxy/ppapi_proxy_test.h"
 #include "ppapi/shared_impl/ppb_device_ref_shared.h"
 #include "ppapi/shared_impl/proxy_lock.h"
 #include "ppapi/shared_impl/var.h"
 #include "ppapi/thunk/enter.h"
@@ skipping 231 matching lines @@
   data_item.name = "name_1";
   data_item.id = "id_1";
   data.push_back(data_item);
   data_item.type = PP_DEVICETYPE_DEV_VIDEOCAPTURE;
   data_item.name = "name_2";
   data_item.id = "id_2";
   data.push_back(data_item);
 
   {
     ProxyAutoUnlock unlock;
-    ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-        PpapiPluginMsg_ResourceReply(
-            reply_params,
-            PpapiPluginMsg_DeviceEnumeration_EnumerateDevicesReply(data))));
+    PluginMessageFilter::DispatchResourceReplyForTest(
+        reply_params,
+        PpapiPluginMsg_DeviceEnumeration_EnumerateDevicesReply(data));
   }
   EXPECT_TRUE(callback.called());
   EXPECT_EQ(PP_OK, callback.result());
   EXPECT_EQ(2U, output.count());
   for (size_t i = 0; i < output.count(); ++i)
     EXPECT_TRUE(CompareDeviceRef(&var_tracker(), output.data()[i], data[i]));
 }
 
 TEST_F(DeviceEnumerationResourceHelperTest, MonitorDeviceChange) {
   ProxyAutoLock lock;
@@ skipping 22 matching lines @@
 
   ResourceMessageReplyParams reply_params(params.pp_resource(), 0);
   reply_params.set_result(PP_OK);
   std::vector<DeviceRefData> data;
 
   helper.SetExpectedResult(data);
 
   {
     ProxyAutoUnlock unlock;
     // Synthesize a response with no device.
-    ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-        PpapiPluginMsg_ResourceReply(
-            reply_params,
-            PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
-                callback_id, data))));
+    PluginMessageFilter::DispatchResourceReplyForTest(
+        reply_params,
+        PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
+            callback_id, data));
   }
   EXPECT_TRUE(helper.called() && helper.same_as_expected());
 
   DeviceRefData data_item;
   data_item.type = PP_DEVICETYPE_DEV_AUDIOCAPTURE;
   data_item.name = "name_1";
   data_item.id = "id_1";
   data.push_back(data_item);
   data_item.type = PP_DEVICETYPE_DEV_VIDEOCAPTURE;
   data_item.name = "name_2";
   data_item.id = "id_2";
   data.push_back(data_item);
 
   helper.SetExpectedResult(data);
 
   {
     ProxyAutoUnlock unlock;
     // Synthesize a response with some devices.
-    ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-        PpapiPluginMsg_ResourceReply(
-            reply_params,
-            PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
-                callback_id, data))));
+    PluginMessageFilter::DispatchResourceReplyForTest(
+        reply_params,
+        PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
+            callback_id, data));
   }
   EXPECT_TRUE(helper.called() && helper.same_as_expected());
 
   TestMonitorDeviceChange helper2(&var_tracker());
 
   result = device_enumeration.MonitorDeviceChange(
       &TestMonitorDeviceChange::MonitorDeviceChangeCallback, &helper2);
   ASSERT_EQ(PP_OK, result);
 
   // Should have sent another MonitorDeviceChange message.
   ResourceMessageCallParams params2;
   IPC::Message msg2;
   ASSERT_TRUE(sink().GetFirstResourceCallMatching(
       PpapiHostMsg_DeviceEnumeration_MonitorDeviceChange::ID, &params2, &msg2));
   sink().ClearMessages();
 
   uint32_t callback_id2 = 0;
   ASSERT_TRUE(UnpackMessage<PpapiHostMsg_DeviceEnumeration_MonitorDeviceChange>(
       msg2, &callback_id2));
 
   helper.SetExpectedResult(data);
   helper2.SetExpectedResult(data);
   {
     ProxyAutoUnlock unlock;
     // |helper2| should receive the result while |helper| shouldn't.
-    ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-        PpapiPluginMsg_ResourceReply(
-            reply_params,
-            PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
-                callback_id2, data))));
+    PluginMessageFilter::DispatchResourceReplyForTest(
+        reply_params,
+        PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
+            callback_id2, data));
   }
   EXPECT_TRUE(helper2.called() && helper2.same_as_expected());
   EXPECT_FALSE(helper.called());
 
   helper.SetExpectedResult(data);
   helper2.SetExpectedResult(data);
   {
     ProxyAutoUnlock unlock;
     // Even if a message with |callback_id| arrives. |helper| shouldn't receive
     // the result.
-    ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-        PpapiPluginMsg_ResourceReply(
-            reply_params,
-            PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
-                callback_id, data))));
+    PluginMessageFilter::DispatchResourceReplyForTest(
+        reply_params,
+        PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
+            callback_id, data));
   }
   EXPECT_FALSE(helper2.called());
   EXPECT_FALSE(helper.called());
 
   result = device_enumeration.MonitorDeviceChange(NULL, NULL);
   ASSERT_EQ(PP_OK, result);
 
   // Should have sent a StopMonitoringDeviceChange message.
   ResourceMessageCallParams params3;
   IPC::Message msg3;
   ASSERT_TRUE(sink().GetFirstResourceCallMatching(
       PpapiHostMsg_DeviceEnumeration_StopMonitoringDeviceChange::ID,
       &params3, &msg3));
   sink().ClearMessages();
 
   helper2.SetExpectedResult(data);
   {
     ProxyAutoUnlock unlock;
     // |helper2| shouldn't receive any result any more.
-    ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
-        PpapiPluginMsg_ResourceReply(
-            reply_params,
-            PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
-                callback_id2, data))));
+    PluginMessageFilter::DispatchResourceReplyForTest(
+        reply_params,
+        PpapiPluginMsg_DeviceEnumeration_NotifyDeviceChange(
+            callback_id2, data));
   }
   EXPECT_FALSE(helper2.called());
 }
 
 }  // namespace proxy
 }  // namespace ppapi