Fix a DCHECK violation in ipc_perftests

ipc/ipc_test_base.h

 // Copyright (c) 2011 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.
 
 #ifndef IPC_IPC_TEST_BASE_H_
 #define IPC_IPC_TEST_BASE_H_
 
 #include <string>
 
 #include "base/basictypes.h"
@@ skipping 15 matching lines @@
  public:
   // The channel name is based on the client's name. This is a public static
   // helper to be used by the client-side code; server-side test code should
   // usually not use this (directly).
   static std::string GetChannelName(const std::string& test_client_name);
 
  protected:
   IPCTestBase();
   virtual ~IPCTestBase();
 
-  virtual void SetUp() OVERRIDE;
   virtual void TearDown() OVERRIDE;
 
-  // Initializes the test to use the given client.
+  // Initializes the test to use the given client and creates an IO message loop
+  // on the current thread.
   void Init(const std::string& test_client_name);
+  // Some tests create separate thread for IO message loop and run non-IO
+  // message loop on the main thread. As IPCTestBase creates IO message loop by
+  // default, such tests need to provide a custom message loop for the main
+  // thread.
+  void InitWithCustomMessageLoop(const std::string& test_client_name,
+                                 scoped_ptr<base::MessageLoop> message_loop);
 
   // Creates a channel with the given listener and connects to the channel
   // (returning true if successful), respectively. Use these to use a channel
   // directly. Since the listener must outlive the channel, you must destroy the
   // channel before the listener gets destroyed.
   void CreateChannel(IPC::Listener* listener);
   bool ConnectChannel();
   void DestroyChannel();
 
   // Releases or replaces existing channel.
@@ skipping 30 matching lines @@
     return channel_.get() ? static_cast<IPC::Sender*>(channel_.get()) :
                             static_cast<IPC::Sender*>(channel_proxy_.get());
   }
 
   IPC::Channel* channel() { return channel_.get(); }
   IPC::ChannelProxy* channel_proxy() { return channel_proxy_.get(); }
 
   const base::ProcessHandle& client_process() const { return client_process_; }
   scoped_refptr<base::TaskRunner> task_runner();
 
-  // Some tests creates separate thread for IO message loop and Run
-  // non-IO message loop on the main thread. As IPCTestBase creates IO
-  // message loop by default, such tests might want to replace it with
-  // non-IO one using set_message_loop().
-  //
-  // The replacement should be done at the beginning of the test.
-  // You should be careful not to replace "live" message loop that has
-  // started running.
-  void set_message_loop(scoped_ptr<base::MessageLoop> loop);
-
  private:
   std::string test_client_name_;
   scoped_ptr<base::MessageLoop> message_loop_;
 
   scoped_ptr<IPC::Channel> channel_;
   scoped_ptr<IPC::ChannelProxy> channel_proxy_;
 
   base::ProcessHandle client_process_;
 
   DISALLOW_COPY_AND_ASSIGN(IPCTestBase);
 };
 
 // Use this to declare the client side for tests using IPCTestBase.
 #define MULTIPROCESS_IPC_TEST_CLIENT_MAIN(test_client_name) \
     MULTIPROCESS_IPC_TEST_MAIN(test_client_name ## TestClientMain)
 
 #endif  // IPC_IPC_TEST_BASE_H_