| Index: chrome/test/pyautolib/mock_pref_pane.py
|
| ===================================================================
|
| --- chrome/test/pyautolib/mock_pref_pane.py (revision 261231)
|
| +++ chrome/test/pyautolib/mock_pref_pane.py (working copy)
|
| @@ -1,124 +0,0 @@
|
| -# Copyright (c) 2012 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.
|
| -
|
| -"""Mock pref pane for testing purpose on Mac."""
|
| -
|
| -import Foundation
|
| -import os
|
| -import signal
|
| -import subprocess
|
| -import sys
|
| -import tempfile
|
| -import time
|
| -
|
| -
|
| -class MockPrefPane(object):
|
| - """Mock Pref Pane to enable/disable/changepin without system prompt.
|
| -
|
| - This only applies to Mac.
|
| - """
|
| -
|
| - def __init__(self):
|
| - self._service_name = 'org.chromium.chromoting'
|
| - self._real_user_id = os.getuid()
|
| - self._config_file = os.path.join(tempfile.gettempdir(),
|
| - '%s.json' % self._service_name)
|
| - self._tool_script = '/Library/PrivilegedHelperTools/%s.me2me.sh' % \
|
| - self._service_name
|
| -
|
| - def _GetJobPid(self):
|
| - """Gets the org.chromium.chromoting job id."""
|
| - process = subprocess.Popen(['launchctl', 'list'], stdout=subprocess.PIPE)
|
| - pid = None
|
| - for line in process.stdout:
|
| - # Format is:
|
| - # 12345 - my.job (if my.job is running, number is job's PID)
|
| - # - 0 my.other.job (if my.other.job is not running)
|
| - fields = line.strip().split('\t')
|
| - if fields[2] == self._service_name and fields[0] != "-":
|
| - pid = fields[0]
|
| - break
|
| - process.wait()
|
| - return pid
|
| -
|
| - def Enable(self):
|
| - """Handles what pref pane does for enabling connection."""
|
| - # Elevate privileges, otherwise tool_script executes with EUID != 0.
|
| - os.setuid(0)
|
| - subprocess.call([self._tool_script, '--enable'],
|
| - stdin=open(self._config_file))
|
| -
|
| - # Drop privileges, start the launchd job as the logged-in user.
|
| - os.setuid(self._real_user_id)
|
| - subprocess.call(['launchctl', 'start', self._service_name])
|
| -
|
| - # Starting a launchd job is an asynchronous operation that typically takes
|
| - # a couple of seconds, so poll until the job has started.
|
| - for _ in range(1, 10):
|
| - if self._GetJobPid():
|
| - print '*** org.chromium.chromoting is running ***'
|
| - break
|
| - time.sleep(2)
|
| -
|
| - def Disable(self):
|
| - """Handles what pref pane does for disabling connection."""
|
| - # Elevate privileges, otherwise tool_script executes with EUID != 0.
|
| - os.setuid(0)
|
| - subprocess.call([self._tool_script, '--disable'],
|
| - stdin=open(self._config_file))
|
| -
|
| - # Drop privileges, stop the launchd job as the logged-in user.
|
| - os.setuid(self._real_user_id)
|
| - subprocess.call(['launchctl', 'stop', self._service_name])
|
| -
|
| - # Stopping a launchd job is an asynchronous operation that typically takes
|
| - # a couple of seconds, so poll until the job has stopped.
|
| - for _ in range(1, 10):
|
| - if not self._GetJobPid():
|
| - print '*** org.chromium.chromoting is not running ***'
|
| - break
|
| - time.sleep(2)
|
| -
|
| - def ChangePin(self):
|
| - """Handles what pref pane does for changing pin."""
|
| - # Elevate privileges, otherwise tool_script executes with EUID != 0.
|
| - os.setuid(0)
|
| - subprocess.call([self._tool_script, '--save-config'],
|
| - stdin=open(self._config_file))
|
| -
|
| - # Drop privileges and send SIGHUP to org.chromium.chromoting
|
| - os.setuid(self._real_user_id)
|
| - os.kill(int(self._GetJobPid()), signal.SIGHUP)
|
| -
|
| - def NotifyWebapp(self):
|
| - """Notifies the web app that pref pane operation is done."""
|
| - notif_center = Foundation.NSDistributedNotificationCenter.defaultCenter()
|
| - notif_center.postNotificationName_object_userInfo_(
|
| - self._service_name + '.update_succeeded', None, None)
|
| -
|
| -
|
| -def Main():
|
| - """Handles the mock pref pane actions."""
|
| - assert sys.platform.startswith('darwin')
|
| -
|
| - print '*** Started mock pref pane ***'
|
| - print '*** EUID=%d, UID=%d ***' % (os.geteuid(), os.getuid())
|
| -
|
| - pref_pane = MockPrefPane()
|
| -
|
| - if sys.argv[1] == 'enable':
|
| - pref_pane.Enable()
|
| - elif sys.argv[1] == 'disable':
|
| - pref_pane.Disable()
|
| - elif sys.argv[1] == 'changepin':
|
| - pref_pane.ChangePin()
|
| - else:
|
| - print >>sys.stderr, 'Invalid syntax'
|
| - return
|
| -
|
| - pref_pane.NotifyWebapp()
|
| -
|
| -
|
| -if __name__ == '__main__':
|
| - Main()
|
|
|