blob: 270e9697f082ca326cad17db775346881f70092e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Copyright 2014 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 <launch.h>
#include "base/mac/foundation_util.h"
#include "base/mac/launchd.h"
#include "base/mac/scoped_nsobject.h"
#include "base/test/test_timeouts.h"
#include "base/time/time.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/google_toolbox_for_mac/src/Foundation/GTMServiceManagement.h"
namespace {
// Returns the parameters needed to launch a really simple script from launchd.
NSDictionary* TestJobDictionary(NSString* label_ns) {
NSString* shell_script_ns = @"sleep 10; echo TestGTMSMJobSubmitRemove";
base::scoped_nsobject<NSMutableArray> job_arguments(
[[NSMutableArray alloc] init]);
[job_arguments addObject:@"/bin/sh"];
[job_arguments addObject:@"-c"];
[job_arguments addObject:shell_script_ns];
NSMutableDictionary* job_dictionary_ns = [NSMutableDictionary dictionary];
[job_dictionary_ns setObject:label_ns forKey:@LAUNCH_JOBKEY_LABEL];
[job_dictionary_ns setObject:[NSNumber numberWithBool:YES]
forKey:@LAUNCH_JOBKEY_RUNATLOAD];
[job_dictionary_ns setObject:job_arguments
forKey:@LAUNCH_JOBKEY_PROGRAMARGUMENTS];
return job_dictionary_ns;
};
} // namespace
TEST(ServiceProcessControlMac, TestGTMSMJobSubmitRemove) {
NSString* label_ns = @"com.chromium.ServiceProcessStateFileManipulationTest";
std::string label(label_ns.UTF8String);
CFStringRef label_cf = base::mac::NSToCFCast(label_ns);
// If the job is loaded or running, remove it.
pid_t pid = base::mac::PIDForJob(label);
CFErrorRef error = NULL;
if (pid >= 0)
ASSERT_TRUE(GTMSMJobRemove(label_cf, &error));
// The job should not be loaded or running.
pid = base::mac::PIDForJob(label);
EXPECT_LT(pid, 0);
// Submit a new job.
NSDictionary* job_dictionary_ns = TestJobDictionary(label_ns);
CFDictionaryRef job_dictionary_cf = base::mac::NSToCFCast(job_dictionary_ns);
ASSERT_TRUE(GTMSMJobSubmit(job_dictionary_cf, &error));
// The new job should be running.
pid = base::mac::PIDForJob(label);
EXPECT_GT(pid, 0);
// Remove the job.
ASSERT_TRUE(GTMSMJobRemove(label_cf, &error));
// Wait for the job to be killed.
base::TimeDelta timeout_in_ms = TestTimeouts::action_timeout();
base::Time start_time = base::Time::Now();
while (1) {
pid = base::mac::PIDForJob(label);
if (pid < 0)
break;
base::Time current_time = base::Time::Now();
if (current_time - start_time > timeout_in_ms)
break;
}
EXPECT_LT(pid, 0);
// Attempting to remove the job again should fail.
EXPECT_FALSE(GTMSMJobRemove(label_cf, &error));
}
|
