ipc/ipc_perftest_support.h

   1 // Copyright (c) 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef IPC_IPC_PERFTEST_SUPPORT_H_
   6 #define IPC_IPC_PERFTEST_SUPPORT_H_
   7
   8 #include <vector>
   9
  10 #include "ipc/ipc_test_base.h"
  11
  12 namespace IPC {
  13 namespace test {
  14
  15 class ChannelReflectorListener;
  16
  17 class PingPongTestParams {
  18  public:
  19   PingPongTestParams(size_t size, int count)
  20       : message_size_(size), message_count_(count) {
  21   }
  22
  23   size_t message_size() const { return message_size_; }
  24   int message_count() const { return message_count_; }
  25
  26  private:
  27   size_t message_size_;
  28   int message_count_;
  29 };
  30
  31 class IPCChannelPerfTestBase : public IPCTestBase {
  32  public:
  33   static std::vector<PingPongTestParams> GetDefaultTestParams();
  34
  35   void RunTestChannelPingPong(
  36       const std::vector<PingPongTestParams>& params_list);
  37   void RunTestChannelProxyPingPong(
  38       const std::vector<PingPongTestParams>& params_list);
  39 };
  40
  41 class PingPongTestClient {
  42  public:
  43   PingPongTestClient();
  44   virtual ~PingPongTestClient();
  45
  46   virtual scoped_ptr<Channel> CreateChannel(Listener* listener);
  47   int RunMain();
  48   scoped_refptr<base::TaskRunner> task_runner();
  49
  50  private:
  51   base::MessageLoopForIO main_message_loop_;
  52   scoped_ptr<ChannelReflectorListener> listener_;
  53   scoped_ptr<Channel> channel_;
  54 };
  55
  56 }
  57 }
  58
  59 #endif  // IPC_IPC_PERFTEST_SUPPORT_H_