OLD | NEW |
1 // Copyright (c) 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/browser_watcher/exit_code_watcher_win.h" | 5 #include "components/browser_watcher/exit_code_watcher_win.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include <utility> | 9 #include <utility> |
10 | 10 |
(...skipping 23 matching lines...) Expand all Loading... |
34 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(INFINITE)); | 34 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(INFINITE)); |
35 return 1; | 35 return 1; |
36 } | 36 } |
37 | 37 |
38 class ScopedSleeperProcess { | 38 class ScopedSleeperProcess { |
39 public: | 39 public: |
40 ScopedSleeperProcess() : is_killed_(false) { | 40 ScopedSleeperProcess() : is_killed_(false) { |
41 } | 41 } |
42 | 42 |
43 ~ScopedSleeperProcess() { | 43 ~ScopedSleeperProcess() { |
44 if (process_.IsValid()) { | 44 if (spawn_child_.process.IsValid()) { |
45 process_.Terminate(-1, false); | 45 spawn_child_.process.Terminate(-1, false); |
46 EXPECT_TRUE(process_.WaitForExit(nullptr)); | 46 EXPECT_TRUE(spawn_child_.process.WaitForExit(nullptr)); |
47 } | 47 } |
48 } | 48 } |
49 | 49 |
50 void Launch() { | 50 void Launch() { |
51 ASSERT_FALSE(process_.IsValid()); | 51 ASSERT_FALSE(spawn_child_.process.IsValid()); |
52 | 52 |
53 base::CommandLine cmd_line(base::GetMultiProcessTestChildBaseCommandLine()); | 53 base::CommandLine cmd_line(base::GetMultiProcessTestChildBaseCommandLine()); |
54 base::LaunchOptions options; | 54 base::LaunchOptions options; |
55 options.start_hidden = true; | 55 options.start_hidden = true; |
56 process_ = base::SpawnMultiProcessTestChild("Sleeper", cmd_line, options); | 56 spawn_child_ = |
57 ASSERT_TRUE(process_.IsValid()); | 57 base::SpawnMultiProcessTestChild("Sleeper", cmd_line, options); |
| 58 ASSERT_TRUE(spawn_child_.process.IsValid()); |
58 } | 59 } |
59 | 60 |
60 void Kill(int exit_code, bool wait) { | 61 void Kill(int exit_code, bool wait) { |
61 ASSERT_TRUE(process_.IsValid()); | 62 ASSERT_TRUE(spawn_child_.process.IsValid()); |
62 ASSERT_FALSE(is_killed_); | 63 ASSERT_FALSE(is_killed_); |
63 process_.Terminate(exit_code, false); | 64 spawn_child_.process.Terminate(exit_code, false); |
64 int seen_exit_code = 0; | 65 int seen_exit_code = 0; |
65 EXPECT_TRUE(process_.WaitForExit(&seen_exit_code)); | 66 EXPECT_TRUE(spawn_child_.process.WaitForExit(&seen_exit_code)); |
66 EXPECT_EQ(exit_code, seen_exit_code); | 67 EXPECT_EQ(exit_code, seen_exit_code); |
67 is_killed_ = true; | 68 is_killed_ = true; |
68 } | 69 } |
69 | 70 |
70 const base::Process& process() const { return process_; } | 71 const base::Process& process() const { return spawn_child_.process; } |
71 | 72 |
72 private: | 73 private: |
73 base::Process process_; | 74 base::SpawnChildResult spawn_child_; |
74 bool is_killed_; | 75 bool is_killed_; |
75 }; | 76 }; |
76 | 77 |
77 class ExitCodeWatcherTest : public testing::Test { | 78 class ExitCodeWatcherTest : public testing::Test { |
78 public: | 79 public: |
79 typedef testing::Test Super; | 80 typedef testing::Test Super; |
80 | 81 |
81 static const int kExitCode = 0xCAFEBABE; | 82 static const int kExitCode = 0xCAFEBABE; |
82 | 83 |
83 ExitCodeWatcherTest() : cmd_line_(base::CommandLine::NO_PROGRAM) {} | 84 ExitCodeWatcherTest() : cmd_line_(base::CommandLine::NO_PROGRAM) {} |
(...skipping 82 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
166 // Kill the sleeper, and make sure it's exited before we continue. | 167 // Kill the sleeper, and make sure it's exited before we continue. |
167 ASSERT_NO_FATAL_FAILURE(sleeper.Kill(kExitCode, true)); | 168 ASSERT_NO_FATAL_FAILURE(sleeper.Kill(kExitCode, true)); |
168 | 169 |
169 watcher.WaitForExit(); | 170 watcher.WaitForExit(); |
170 EXPECT_EQ(kExitCode, watcher.exit_code()); | 171 EXPECT_EQ(kExitCode, watcher.exit_code()); |
171 | 172 |
172 VerifyWroteExitCode(sleeper.process().Pid(), kExitCode); | 173 VerifyWroteExitCode(sleeper.process().Pid(), kExitCode); |
173 } | 174 } |
174 | 175 |
175 } // namespace browser_watcher | 176 } // namespace browser_watcher |
OLD | NEW |