| Index: ipc/ipc_test_base.cc
|
| diff --git a/ipc/ipc_test_base.cc b/ipc/ipc_test_base.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..2d68a765b8c1f5c6582ec2b155e14a2301b7c761
|
| --- /dev/null
|
| +++ b/ipc/ipc_test_base.cc
|
| @@ -0,0 +1,131 @@
|
| +// 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 "build/build_config.h"
|
| +
|
| +#if defined(OS_WIN)
|
| +#include <windows.h>
|
| +#elif defined(OS_POSIX)
|
| +#include <sys/types.h>
|
| +#include <unistd.h>
|
| +#endif
|
| +
|
| +#include <stdio.h>
|
| +#include <string>
|
| +#include <utility>
|
| +
|
| +#include "ipc/ipc_test_base.h"
|
| +
|
| +#include "base/base_switches.h"
|
| +#include "base/command_line.h"
|
| +#include "base/debug/debug_on_start_win.h"
|
| +#include "base/perftimer.h"
|
| +#include "base/pickle.h"
|
| +#include "base/test/perf_test_suite.h"
|
| +#include "base/test/test_suite.h"
|
| +#include "base/threading/thread.h"
|
| +#include "base/time.h"
|
| +#include "ipc/ipc_descriptors.h"
|
| +#include "ipc/ipc_channel.h"
|
| +#include "ipc/ipc_channel_proxy.h"
|
| +#include "ipc/ipc_message_utils.h"
|
| +#include "ipc/ipc_multiprocess_test.h"
|
| +#include "ipc/ipc_sender.h"
|
| +#include "ipc/ipc_switches.h"
|
| +#include "testing/multiprocess_func_list.h"
|
| +
|
| +// Define to enable IPC performance testing instead of the regular unit tests
|
| +// #define PERFORMANCE_TEST
|
| +
|
| +const char kTestClientChannel[] = "T1";
|
| +const char kReflectorChannel[] = "T2";
|
| +const char kFuzzerChannel[] = "F3";
|
| +const char kSyncSocketChannel[] = "S4";
|
| +
|
| +void IPCTestBase::SetUp() {
|
| + MultiProcessTest::SetUp();
|
| +
|
| + // Construct a fresh IO Message loop for the duration of each test.
|
| + message_loop_ = new MessageLoopForIO();
|
| +}
|
| +
|
| +void IPCTestBase::TearDown() {
|
| + delete message_loop_;
|
| + message_loop_ = NULL;
|
| +
|
| + MultiProcessTest::TearDown();
|
| +}
|
| +
|
| +#if defined(OS_WIN)
|
| +base::ProcessHandle IPCTestBase::SpawnChild(IPCTestBase::ChildType child_type,
|
| + IPC::Channel *channel) {
|
| + // kDebugChildren support.
|
| + bool debug_on_start =
|
| + CommandLine::ForCurrentProcess()->HasSwitch(switches::kDebugChildren);
|
| +
|
| + switch (child_type) {
|
| + case TEST_CLIENT:
|
| + return MultiProcessTest::SpawnChild("RunTestClient", debug_on_start);
|
| + case TEST_REFLECTOR:
|
| + return MultiProcessTest::SpawnChild("RunReflector", debug_on_start);
|
| + case FUZZER_SERVER:
|
| + return MultiProcessTest::SpawnChild("RunFuzzServer", debug_on_start);
|
| + case SYNC_SOCKET_SERVER:
|
| + return MultiProcessTest::SpawnChild("RunSyncSocketServer", debug_on_start);
|
| + default:
|
| + return NULL;
|
| + }
|
| +}
|
| +#elif defined(OS_POSIX)
|
| +base::ProcessHandle IPCTestBase::SpawnChild(IPCTestBase::ChildType child_type,
|
| + IPC::Channel *channel) {
|
| + // kDebugChildren support.
|
| + bool debug_on_start =
|
| + CommandLine::ForCurrentProcess()->HasSwitch(switches::kDebugChildren);
|
| +
|
| + base::FileHandleMappingVector fds_to_map;
|
| + const int ipcfd = channel->GetClientFileDescriptor();
|
| + if (ipcfd > -1) {
|
| + fds_to_map.push_back(std::pair<int, int>(ipcfd, kPrimaryIPCChannel + 3));
|
| + }
|
| +
|
| + base::ProcessHandle ret = base::kNullProcessHandle;
|
| + switch (child_type) {
|
| + case TEST_CLIENT:
|
| + ret = MultiProcessTest::SpawnChild("RunTestClient",
|
| + fds_to_map,
|
| + debug_on_start);
|
| + break;
|
| + case TEST_DESCRIPTOR_CLIENT:
|
| + ret = MultiProcessTest::SpawnChild("RunTestDescriptorClient",
|
| + fds_to_map,
|
| + debug_on_start);
|
| + break;
|
| + case TEST_DESCRIPTOR_CLIENT_SANDBOXED:
|
| + ret = MultiProcessTest::SpawnChild("RunTestDescriptorClientSandboxed",
|
| + fds_to_map,
|
| + debug_on_start);
|
| + break;
|
| + case TEST_REFLECTOR:
|
| + ret = MultiProcessTest::SpawnChild("RunReflector",
|
| + fds_to_map,
|
| + debug_on_start);
|
| + break;
|
| + case FUZZER_SERVER:
|
| + ret = MultiProcessTest::SpawnChild("RunFuzzServer",
|
| + fds_to_map,
|
| + debug_on_start);
|
| + break;
|
| + case SYNC_SOCKET_SERVER:
|
| + ret = MultiProcessTest::SpawnChild("RunSyncSocketServer",
|
| + fds_to_map,
|
| + debug_on_start);
|
| + break;
|
| + default:
|
| + return base::kNullProcessHandle;
|
| + break;
|
| + }
|
| + return ret;
|
| +}
|
| +#endif // defined(OS_POSIX)
|
|
|