Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(244)

Side by Side Diff: content/browser/mach_broker_mac_unittest.cc

Issue 681733003: Removed the ProcessHandle from the RendererClosedDetails payload (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Used range-based loop Created 6 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2009 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 "content/browser/mach_broker_mac.h" 5 #include "content/browser/mach_broker_mac.h"
6 6
7 #include "base/synchronization/lock.h" 7 #include "base/synchronization/lock.h"
8 #include "testing/gtest/include/gtest/gtest.h" 8 #include "testing/gtest/include/gtest/gtest.h"
9 9
10 namespace content { 10 namespace content {
11 11
12 class MachBrokerTest : public testing::Test { 12 class MachBrokerTest : public testing::Test {
13 public: 13 public:
14 // Helper function to acquire/release locks and call |PlaceholderForPid()|. 14 // Helper function to acquire/release locks and call |PlaceholderForPid()|.
15 void AddPlaceholderForPid(base::ProcessHandle pid) { 15 void AddPlaceholderForPid(base::ProcessHandle pid, int child_process_id) {
16 base::AutoLock lock(broker_.GetLock()); 16 base::AutoLock lock(broker_.GetLock());
17 broker_.AddPlaceholderForPid(pid); 17 broker_.AddPlaceholderForPid(pid, child_process_id);
18 } 18 }
19 19
20 void InvalidatePid(base::ProcessHandle pid) { 20 void InvalidatePid(base::ProcessHandle pid) {
21 broker_.InvalidatePid(pid); 21 broker_.InvalidatePid(pid);
22 } 22 }
23 23
24 void InvalidateChildProcessId(int child_process_id) {
25 broker_.InvalidateChildProcessId(child_process_id);
26 }
27
28 int GetChildProcessCount(int child_process_id) {
29 int count = 0;
30 for (const auto& it : broker_.child_process_id_map_) {
31 if (it.second == child_process_id)
32 count++;
33 }
34 return count;
35 }
36
24 // Helper function to acquire/release locks and call |FinalizePid()|. 37 // Helper function to acquire/release locks and call |FinalizePid()|.
25 void FinalizePid(base::ProcessHandle pid, 38 void FinalizePid(base::ProcessHandle pid,
26 mach_port_t task_port) { 39 mach_port_t task_port) {
27 base::AutoLock lock(broker_.GetLock()); 40 base::AutoLock lock(broker_.GetLock());
28 broker_.FinalizePid(pid, task_port); 41 broker_.FinalizePid(pid, task_port);
29 } 42 }
30 43
31 protected: 44 protected:
32 MachBroker broker_; 45 MachBroker broker_;
33 }; 46 };
34 47
35 TEST_F(MachBrokerTest, Locks) { 48 TEST_F(MachBrokerTest, Locks) {
36 // Acquire and release the locks. Nothing bad should happen. 49 // Acquire and release the locks. Nothing bad should happen.
37 base::AutoLock lock(broker_.GetLock()); 50 base::AutoLock lock(broker_.GetLock());
38 } 51 }
39 52
40 TEST_F(MachBrokerTest, AddPlaceholderAndFinalize) { 53 TEST_F(MachBrokerTest, AddPlaceholderAndFinalize) {
41 // Add a placeholder for PID 1. 54 // Add a placeholder for PID 1.
42 AddPlaceholderForPid(1); 55 AddPlaceholderForPid(1, 1);
43 EXPECT_EQ(0u, broker_.TaskForPid(1)); 56 EXPECT_EQ(0u, broker_.TaskForPid(1));
44 57
45 // Finalize PID 1. 58 // Finalize PID 1.
46 FinalizePid(1, 100u); 59 FinalizePid(1, 100u);
47 EXPECT_EQ(100u, broker_.TaskForPid(1)); 60 EXPECT_EQ(100u, broker_.TaskForPid(1));
48 61
49 // Should be no entry for PID 2. 62 // Should be no entry for PID 2.
50 EXPECT_EQ(0u, broker_.TaskForPid(2)); 63 EXPECT_EQ(0u, broker_.TaskForPid(2));
51 } 64 }
52 65
53 TEST_F(MachBrokerTest, Invalidate) { 66 TEST_F(MachBrokerTest, Invalidate) {
54 AddPlaceholderForPid(1); 67 AddPlaceholderForPid(1, 1);
55 FinalizePid(1, 100u); 68 FinalizePid(1, 100u);
56 69
57 EXPECT_EQ(100u, broker_.TaskForPid(1)); 70 EXPECT_EQ(100u, broker_.TaskForPid(1));
58 InvalidatePid(1u); 71 InvalidatePid(1u);
59 EXPECT_EQ(0u, broker_.TaskForPid(1)); 72 EXPECT_EQ(0u, broker_.TaskForPid(1));
60 } 73 }
61 74
75 TEST_F(MachBrokerTest, InvalidateChildProcessId) {
76 // Add a placeholder for PID 1 and child process id 50.
77 AddPlaceholderForPid(1, 50);
78 FinalizePid(1, 100u);
79
80 EXPECT_EQ(100u, broker_.TaskForPid(1));
81 InvalidateChildProcessId(50);
82 EXPECT_EQ(0u, broker_.TaskForPid(1));
83 }
84
85 TEST_F(MachBrokerTest, ValidateChildProcessIdMap) {
86 // Add a placeholder for PID 1 and child process id 50.
87 AddPlaceholderForPid(1, 50);
88 FinalizePid(1, 100u);
89
90 EXPECT_EQ(1, GetChildProcessCount(50));
91 InvalidateChildProcessId(50);
92 EXPECT_EQ(0, GetChildProcessCount(50));
93 }
94
62 TEST_F(MachBrokerTest, FinalizeUnknownPid) { 95 TEST_F(MachBrokerTest, FinalizeUnknownPid) {
63 // Finalizing an entry for an unknown pid should not add it to the map. 96 // Finalizing an entry for an unknown pid should not add it to the map.
64 FinalizePid(1u, 100u); 97 FinalizePid(1u, 100u);
65 EXPECT_EQ(0u, broker_.TaskForPid(1u)); 98 EXPECT_EQ(0u, broker_.TaskForPid(1u));
66 } 99 }
67 100
68 } // namespace content 101 } // namespace content
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698