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
3 // found in the LICENSE file.
4
5 #include "chrome/browser/mach_broker_mac.h"
6
7 #include "base/synchronization/lock.h"
8 #include "testing/gtest/include/gtest/gtest.h"
9
10 class MachBrokerTest : public testing::Test {
11 public:
12 // Helper function to acquire/release locks and call |PlaceholderForPid()|.
AddPlaceholderForPid(base::ProcessHandle pid)13 void AddPlaceholderForPid(base::ProcessHandle pid) {
14 base::AutoLock lock(broker_.GetLock());
15 broker_.AddPlaceholderForPid(pid);
16 }
17
18 // Helper function to acquire/release locks and call |FinalizePid()|.
FinalizePid(base::ProcessHandle pid,const MachBroker::MachInfo & mach_info)19 void FinalizePid(base::ProcessHandle pid,
20 const MachBroker::MachInfo& mach_info) {
21 base::AutoLock lock(broker_.GetLock());
22 broker_.FinalizePid(pid, mach_info);
23 }
24
25 protected:
26 MachBroker broker_;
27 };
28
TEST_F(MachBrokerTest,Locks)29 TEST_F(MachBrokerTest, Locks) {
30 // Acquire and release the locks. Nothing bad should happen.
31 base::AutoLock lock(broker_.GetLock());
32 }
33
TEST_F(MachBrokerTest,AddPlaceholderAndFinalize)34 TEST_F(MachBrokerTest, AddPlaceholderAndFinalize) {
35 // Add a placeholder for PID 1.
36 AddPlaceholderForPid(1);
37 EXPECT_EQ(0u, broker_.TaskForPid(1));
38
39 // Finalize PID 1.
40 FinalizePid(1, MachBroker::MachInfo().SetTask(100u));
41 EXPECT_EQ(100u, broker_.TaskForPid(1));
42
43 // Should be no entry for PID 2.
44 EXPECT_EQ(0u, broker_.TaskForPid(2));
45 }
46
TEST_F(MachBrokerTest,Invalidate)47 TEST_F(MachBrokerTest, Invalidate) {
48 AddPlaceholderForPid(1);
49 FinalizePid(1, MachBroker::MachInfo().SetTask(100u));
50
51 EXPECT_EQ(100u, broker_.TaskForPid(1));
52 broker_.InvalidatePid(1u);
53 EXPECT_EQ(0u, broker_.TaskForPid(1));
54 }
55
TEST_F(MachBrokerTest,FinalizeUnknownPid)56 TEST_F(MachBrokerTest, FinalizeUnknownPid) {
57 // Finalizing an entry for an unknown pid should not add it to the map.
58 FinalizePid(1u, MachBroker::MachInfo().SetTask(100u));
59 EXPECT_EQ(0u, broker_.TaskForPid(1u));
60 }
61