mac: Fix launchd test on 10.10

chrome/browser/service_process/service_process_control_mac_unittest.mm

+// 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/launchd.h"
+#include "base/mac/mac_util.h"
+#include "base/mac/scoped_nsobject.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));
+  pid = base::mac::PIDForJob(label);
+  EXPECT_LT(pid, 0);
+
+  // Attempting to remove the job again should fail.
+  EXPECT_FALSE(GTMSMJobRemove(label_cf, &error));
+}