~launchpad-pqm/launchpad/devel

# Copyright 2010-2011 Canonical Ltd.  This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).

import errno
import os
import shutil
import signal
import socket
import subprocess
import tempfile
import threading
import time

from bzrlib import (
    errors,
    osutils,
    tests,
    trace,
    )
from bzrlib.plugins import lpserve
from testtools import content

from canonical.config import config
from lp.codehosting import (
    get_bzr_path,
    get_BZR_PLUGIN_PATH_for_subprocess,
    )
from lp.testing.fakemethod import FakeMethod


class TestingLPForkingServiceInAThread(lpserve.LPForkingService):
    """A test-double to run a "forking service" in a thread.

    Note that we don't allow actually forking, but it does allow us to
    interact with the service for other operations.
    """

    # For testing we set the timeouts much lower, because we want the
    # tests to run quickly.
    WAIT_FOR_CHILDREN_TIMEOUT = 0.5
    SOCKET_TIMEOUT = 0.01
    SLEEP_FOR_CHILDREN_TIMEOUT = 0.01
    WAIT_FOR_REQUEST_TIMEOUT = 0.1

    # We're running in a thread as part of the test suite.  Blow up at
    # any attempt to fork.
    _fork_function = None

    def __init__(self, path, perms=None):
        self.service_started = threading.Event()
        self.service_stopped = threading.Event()
        self.this_thread = None
        self.fork_log = []
        super(TestingLPForkingServiceInAThread, self).__init__(
            path=path, perms=None)

    def _register_signals(self):
        # Don't register it for the test suite.
        pass

    def _unregister_signals(self):
        # We don't fork, and didn't register, so don't unregister.
        pass

    def _create_master_socket(self):
        super(TestingLPForkingServiceInAThread, self)._create_master_socket()
        self.service_started.set()

    def main_loop(self):
        self.service_stopped.clear()
        super(TestingLPForkingServiceInAThread, self).main_loop()
        self.service_stopped.set()

    def fork_one_request(self, conn, client_addr, command, env):
        # We intentionally don't allow the test suite to request a fork, as
        # threads + forks and everything else don't exactly play well together
        self.fork_log.append((command, env))
        conn.sendall('ok\nfake forking\n')
        conn.close()

    @staticmethod
    def start_service(test):
        """Start a new LPForkingService in a thread at a random path.

        This will block until the service has created its socket, and is ready
        to communicate.

        :return: A new TestingLPForkingServiceInAThread instance
        """
        fd, path = tempfile.mkstemp(prefix='tmp-lp-forking-service-',
                                    suffix='.sock')
        # We don't want a temp file, we want a temp socket
        os.close(fd)
        os.remove(path)
        new_service = TestingLPForkingServiceInAThread(path=path)
        thread = threading.Thread(target=new_service.main_loop,
                                  name='TestingLPForkingServiceInAThread')
        new_service.this_thread = thread
        # should we be doing thread.setDaemon(True) ?
        thread.start()
        new_service.service_started.wait(10.0)
        if not new_service.service_started.isSet():
            raise RuntimeError(
                'Failed to start the TestingLPForkingServiceInAThread')
        test.addCleanup(new_service.stop_service)
        # what about returning new_service._sockname ?
        return new_service

    def stop_service(self):
        """Stop the test-server thread. This can be called multiple times."""
        if self.this_thread is None:
            # We already stopped the process
            return
        self._should_terminate.set()
        self.service_stopped.wait(10.0)
        if not self.service_stopped.isSet():
            raise RuntimeError(
                'Failed to stop the TestingLPForkingServiceInAThread')
        self.this_thread.join()
        # Break any refcycles
        self.this_thread = None


class TestTestingLPForkingServiceInAThread(tests.TestCaseWithTransport):

    def test_start_and_stop_service(self):
        service = TestingLPForkingServiceInAThread.start_service(self)
        service.stop_service()

    def test_multiple_stops(self):
        service = TestingLPForkingServiceInAThread.start_service(self)
        service.stop_service()
        # calling stop_service repeatedly is a no-op (and not an error)
        service.stop_service()

    def test_autostop(self):
        # We shouldn't leak a thread here, as it should be part of the test
        # case teardown.
        TestingLPForkingServiceInAThread.start_service(self)


class TestCaseWithLPForkingService(tests.TestCaseWithTransport):

    def setUp(self):
        super(TestCaseWithLPForkingService, self).setUp()
        self.service = TestingLPForkingServiceInAThread.start_service(self)

    def send_message_to_service(self, message, one_byte_at_a_time=False):
        client_sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        client_sock.connect(self.service.master_socket_path)
        if one_byte_at_a_time:
            for byte in message:
                client_sock.send(byte)
        else:
            client_sock.sendall(message)
        response = client_sock.recv(1024)
        return response


class TestLPForkingServiceCommandToArgv(tests.TestCase):

    def assertAsArgv(self, argv, command_str):
        self.assertEqual(argv,
            lpserve.LPForkingService.command_to_argv(command_str))

    def test_simple(self):
        self.assertAsArgv([u'foo'], 'foo')
        self.assertAsArgv([u'foo', u'bar'], 'foo bar')

    def test_quoted(self):
        self.assertAsArgv([u'foo'], 'foo')
        self.assertAsArgv([u'foo bar'], '"foo bar"')

    def test_unicode(self):
        self.assertAsArgv([u'command', u'\xe5'], 'command \xc3\xa5')


class TestLPForkingServiceParseEnv(tests.TestCase):

    def assertEnv(self, env, env_str):
        self.assertEqual(env, lpserve.LPForkingService.parse_env(env_str))

    def assertInvalid(self, env_str):
        self.assertRaises(ValueError, lpserve.LPForkingService.parse_env,
                                      env_str)

    def test_no_entries(self):
        self.assertEnv({}, 'end\n')

    def test_one_entries(self):
        self.assertEnv({'BZR_EMAIL': 'joe@foo.com'},
                       'BZR_EMAIL: joe@foo.com\n'
                       'end\n')

    def test_two_entries(self):
        self.assertEnv({'BZR_EMAIL': 'joe@foo.com', 'BAR': 'foo'},
                       'BZR_EMAIL: joe@foo.com\n'
                       'BAR: foo\n'
                       'end\n')

    def test_invalid_empty(self):
        self.assertInvalid('')

    def test_invalid_end(self):
        self.assertInvalid("BZR_EMAIL: joe@foo.com\n")

    def test_invalid_entry(self):
        self.assertInvalid("BZR_EMAIL joe@foo.com\nend\n")


class TestLPForkingService(TestCaseWithLPForkingService):

    def test_send_quit_message(self):
        response = self.send_message_to_service('quit\n')
        self.assertEqual('ok\nquit command requested... exiting\n', response)
        self.service.service_stopped.wait(10.0)
        self.assertTrue(self.service.service_stopped.isSet())

    def test_send_invalid_message_fails(self):
        response = self.send_message_to_service('unknown\n')
        self.assertStartsWith(response, 'FAILURE')

    def test_send_hello_heartbeat(self):
        response = self.send_message_to_service('hello\n')
        self.assertEqual('ok\nyep, still alive\n', response)

    def test_send_simple_fork(self):
        response = self.send_message_to_service('fork rocks\n')
        self.assertEqual('ok\nfake forking\n', response)
        self.assertEqual([(['rocks'], {})], self.service.fork_log)

    def test_send_fork_env_with_empty_env(self):
        response = self.send_message_to_service(
            'fork-env rocks\n'
            'end\n')
        self.assertEqual('ok\nfake forking\n', response)
        self.assertEqual([(['rocks'], {})], self.service.fork_log)

    def test_send_fork_env_with_env(self):
        response = self.send_message_to_service(
            'fork-env rocks\n'
            'BZR_EMAIL: joe@example.com\n'
            'end\n')
        self.assertEqual('ok\nfake forking\n', response)
        self.assertEqual([(['rocks'], {'BZR_EMAIL': 'joe@example.com'})],
                         self.service.fork_log)

    def test_send_fork_env_slowly(self):
        response = self.send_message_to_service(
            'fork-env rocks\n'
            'BZR_EMAIL: joe@example.com\n'
            'end\n', one_byte_at_a_time=True)
        self.assertEqual('ok\nfake forking\n', response)
        self.assertEqual([(['rocks'], {'BZR_EMAIL': 'joe@example.com'})],
                         self.service.fork_log)

    def test_send_incomplete_fork_env_timeout(self):
        # We should get a failure message if we can't quickly read the whole
        # content
        response = self.send_message_to_service(
            'fork-env rocks\n'
            'BZR_EMAIL: joe@example.com\n',
            one_byte_at_a_time=True)
        # Note that we *don't* send a final 'end\n'
        self.assertStartsWith(response, 'FAILURE\n')

    def test_send_incomplete_request_timeout(self):
        # Requests end with '\n', send one without it
        response = self.send_message_to_service('hello',
                                                one_byte_at_a_time=True)
        self.assertStartsWith(response, 'FAILURE\n')

    def test_child_connection_timeout(self):
        self.assertEqual(self.service.CHILD_CONNECT_TIMEOUT,
                         self.service._child_connect_timeout)
        response = self.send_message_to_service('child_connect_timeout 1\n')
        self.assertEqual('ok\n', response)
        self.assertEqual(1, self.service._child_connect_timeout)

    def test_child_connection_timeout_bad_float(self):
        self.assertEqual(self.service.CHILD_CONNECT_TIMEOUT,
                         self.service._child_connect_timeout)
        response = self.send_message_to_service('child_connect_timeout 1.2\n')
        self.assertStartsWith(response, 'FAILURE:')

    def test_child_connection_timeout_no_val(self):
        response = self.send_message_to_service('child_connect_timeout \n')
        self.assertStartsWith(response, 'FAILURE:')

    def test_child_connection_timeout_bad_val(self):
        response = self.send_message_to_service('child_connect_timeout b\n')
        self.assertStartsWith(response, 'FAILURE:')

    def test__open_handles_will_timeout(self):
        # signal.alarm() has only 1-second granularity. :(
        self.service._child_connect_timeout = 1
        tempdir = tempfile.mkdtemp(prefix='testlpserve-')
        self.addCleanup(shutil.rmtree, tempdir, ignore_errors=True)
        os.mkfifo(os.path.join(tempdir, 'stdin'))
        os.mkfifo(os.path.join(tempdir, 'stdout'))
        os.mkfifo(os.path.join(tempdir, 'stderr'))

        # catch SIGALRM so we don't stop the test suite. It will still
        # interupt the blocking open() calls.
        signal.signal(signal.SIGALRM, FakeMethod())

        self.addCleanup(signal.signal, signal.SIGALRM, signal.SIG_DFL)
        e = self.assertRaises(errors.BzrError,
            self.service._open_handles, tempdir)
        self.assertContainsRe(str(e), r'After \d+.\d+s we failed to open.*')


class TestCaseWithSubprocess(tests.TestCaseWithTransport):
    """Override the bzr start_bzr_subprocess command.

    The launchpad infrastructure requires a fair amount of configuration to
    get paths, etc correct. This provides a "start_bzr_subprocess" command
    that has all of those paths appropriately set, but otherwise functions the
    same as the bzrlib.tests.TestCase version.
    """

    def get_python_path(self):
        """Return the path to the Python interpreter."""
        return '%s/bin/py' % config.root

    def start_bzr_subprocess(self, process_args, env_changes=None,
                             working_dir=None):
        """Start bzr in a subprocess for testing.

        Copied and modified from `bzrlib.tests.TestCase.start_bzr_subprocess`.
        This version removes some of the skipping stuff, some of the
        irrelevant comments (e.g. about win32) and uses Launchpad's own
        mechanisms for getting the path to 'bzr'.

        Comments starting with 'LAUNCHPAD' are comments about our
        modifications.
        """
        if env_changes is None:
            env_changes = {}
        env_changes['BZR_PLUGIN_PATH'] = get_BZR_PLUGIN_PATH_for_subprocess()
        old_env = {}

        def cleanup_environment():
            for env_var, value in env_changes.iteritems():
                old_env[env_var] = osutils.set_or_unset_env(env_var, value)

        def restore_environment():
            for env_var, value in old_env.iteritems():
                osutils.set_or_unset_env(env_var, value)

        cwd = None
        if working_dir is not None:
            cwd = osutils.getcwd()
            os.chdir(working_dir)

        # LAUNCHPAD: Because of buildout, we need to get a custom Python
        # binary, not sys.executable.
        python_path = self.get_python_path()
        # LAUNCHPAD: We can't use self.get_bzr_path(), since it'll find
        # lib/bzrlib, rather than the path to sourcecode/bzr/bzr.
        bzr_path = get_bzr_path()
        try:
            cleanup_environment()
            command = [python_path, bzr_path]
            command.extend(process_args)
            process = self._popen(
                command, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
                stderr=subprocess.PIPE)
        finally:
            restore_environment()
            if cwd is not None:
                os.chdir(cwd)

        return process


class TestCaseWithLPForkingServiceSubprocess(TestCaseWithSubprocess):
    """Tests will get a separate process to communicate to.

    The number of these tests should be small, because it is expensive to
    start and stop the daemon.

    TODO: This should probably use testresources, or layers somehow...
    """

    def setUp(self):
        super(TestCaseWithLPForkingServiceSubprocess, self).setUp()
        (self.service_process,
         self.service_path) = self.start_service_subprocess()
        self.addCleanup(self.stop_service)

    def start_conversation(self, message, one_byte_at_a_time=False):
        """Start talking to the service, and get the initial response."""
        client_sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        trace.mutter('sending %r to socket %s' % (message, self.service_path))
        client_sock.connect(self.service_path)
        if one_byte_at_a_time:
            for byte in message:
                client_sock.send(byte)
        else:
            client_sock.sendall(message)
        response = client_sock.recv(1024)
        trace.mutter('response: %r' % (response,))
        if response.startswith("FAILURE"):
            raise RuntimeError('Failed to send message: %r' % (response,))
        return response, client_sock

    def send_message_to_service(self, message, one_byte_at_a_time=False):
        response, client_sock = self.start_conversation(message,
            one_byte_at_a_time=one_byte_at_a_time)
        client_sock.close()
        return response

    def send_fork_request(self, command, env=None):
        if env is not None:
            request_lines = ['fork-env %s\n' % (command,)]
            for key, value in env.iteritems():
                request_lines.append('%s: %s\n' % (key, value))
            request_lines.append('end\n')
            request = ''.join(request_lines)
        else:
            request = 'fork %s\n' % (command,)
        response, sock = self.start_conversation(request)
        ok, pid, path, tail = response.split('\n')
        self.assertEqual('ok', ok)
        self.assertEqual('', tail)
        # Don't really care what it is, but should be an integer
        pid = int(pid)
        path = path.strip()
        self.assertContainsRe(path, '/lp-forking-service-child-')
        return path, pid, sock

    def _start_subprocess(self, path, env_changes):
        proc = self.start_bzr_subprocess(
            ['lp-service', '--path', path, '--no-preload',
             '--children-timeout=1'],
            env_changes=env_changes)
        trace.mutter('started lp-service subprocess')
        expected = 'Listening on socket: %s\n' % (path,)
        path_line = proc.stderr.readline()
        trace.mutter(path_line)
        self.assertEqual(expected, path_line)
        return proc

    def start_service_subprocess(self):
        # Make sure this plugin is exposed to the subprocess
        # SLOOWWW (~2 seconds, which is why we are doing the work anyway)
        fd, tempname = tempfile.mkstemp(prefix='tmp-log-bzr-lp-forking-')
        # I'm not 100% sure about when cleanup runs versus addDetail, but I
        # think this will work.
        self.addCleanup(os.remove, tempname)

        def read_log():
            f = os.fdopen(fd)
            f.seek(0)
            content = f.read()
            f.close()
            return [content]
        self.addDetail('server-log', content.Content(
            content.ContentType('text', 'plain', {"charset": "utf8"}),
            read_log))
        service_fd, path = tempfile.mkstemp(prefix='tmp-lp-service-',
                                            suffix='.sock')
        os.close(service_fd)
        # The service wants to create this file as a socket.
        os.remove(path)
        env_changes = {
            'BZR_PLUGIN_PATH': lpserve.__path__[0],
            'BZR_LOG': tempname,
            }
        proc = self._start_subprocess(path, env_changes)
        return proc, path

    def stop_service(self):
        if self.service_process is None:
            # Already stopped.
            return
        # First, try to stop the service gracefully, by sending a 'quit'
        # message.
        try:
            response = self.send_message_to_service('quit\n')
        except socket.error:
            # Ignore a failure to connect; the service must be
            # stopping/stopped already.
            response = None
        tend = time.time() + 10.0
        while self.service_process.poll() is None:
            if time.time() > tend:
                self.finish_bzr_subprocess(process=self.service_process,
                    send_signal=signal.SIGINT, retcode=3)
                self.fail('Failed to quit gracefully after 10.0 seconds')
            time.sleep(0.1)
        if response is not None:
            self.assertEqual('ok\nquit command requested... exiting\n',
                             response)

    def _get_fork_handles(self, path):
        trace.mutter('getting handles for: %s' % (path,))
        stdin_path = os.path.join(path, 'stdin')
        stdout_path = os.path.join(path, 'stdout')
        stderr_path = os.path.join(path, 'stderr')
        # The ordering must match the ordering of the service or we get a
        # deadlock.
        child_stdin = open(stdin_path, 'wb', 0)
        child_stdout = open(stdout_path, 'rb', 0)
        child_stderr = open(stderr_path, 'rb', 0)
        return child_stdin, child_stdout, child_stderr

    def communicate_with_fork(self, path, stdin=None):
        child_stdin, child_stdout, child_stderr = self._get_fork_handles(path)
        if stdin is not None:
            child_stdin.write(stdin)
        child_stdin.close()
        stdout_content = child_stdout.read()
        stderr_content = child_stderr.read()
        return stdout_content, stderr_content

    def assertReturnCode(self, expected_code, sock):
        """Assert that we get the expected return code as a message."""
        response = sock.recv(1024)
        self.assertStartsWith(response, 'exited\n')
        code = int(response.split('\n', 1)[1])
        self.assertEqual(expected_code, code)


class TestLPServiceInSubprocess(TestCaseWithLPForkingServiceSubprocess):

    def test_fork_lp_serve_hello(self):
        path, _, sock = self.send_fork_request('lp-serve --inet 2')
        stdout_content, stderr_content = self.communicate_with_fork(path,
            'hello\n')
        self.assertEqual('ok\x012\n', stdout_content)
        self.assertEqual('', stderr_content)
        self.assertReturnCode(0, sock)

    def DONT_test_fork_lp_serve_multiple_hello(self):
        # This ensures that the fifos are all set to blocking mode
        # We can't actually run this test, because by default 'bzr serve
        # --inet' does not flush after each message. So we end up blocking
        # forever waiting for the server to finish responding to the first
        # request.
        path, _, sock = self.send_fork_request('lp-serve --inet 2')
        child_stdin, child_stdout, child_stderr = self._get_fork_handles(path)
        child_stdin.write('hello\n')
        child_stdin.flush()
        self.assertEqual('ok\x012\n', child_stdout.read())
        child_stdin.write('hello\n')
        self.assertEqual('ok\x012\n', child_stdout.read())
        child_stdin.close()
        self.assertEqual('', child_stderr.read())
        child_stdout.close()
        child_stderr.close()

    def test_fork_replay(self):
        path, _, sock = self.send_fork_request('launchpad-replay')
        stdout_content, stderr_content = self.communicate_with_fork(path,
            '1 hello\n2 goodbye\n1 maybe\n')
        self.assertEqualDiff('hello\nmaybe\n', stdout_content)
        self.assertEqualDiff('goodbye\n', stderr_content)
        self.assertReturnCode(0, sock)

    def test_just_run_service(self):
        # Start and stop are defined in setUp()
        pass

    def test_fork_multiple_children(self):
        paths = []
        for idx in range(4):
            paths.append(self.send_fork_request('launchpad-replay'))
        # Do them out of order, as order shouldn't matter.
        for idx in [3, 2, 0, 1]:
            p, pid, sock = paths[idx]
            stdout_msg = 'hello %d\n' % (idx,)
            stderr_msg = 'goodbye %d\n' % (idx + 1,)
            stdout, stderr = self.communicate_with_fork(p,
                '1 %s2 %s' % (stdout_msg, stderr_msg))
            self.assertEqualDiff(stdout_msg, stdout)
            self.assertEqualDiff(stderr_msg, stderr)
            self.assertReturnCode(0, sock)

    def test_fork_respects_env_vars(self):
        path, pid, sock = self.send_fork_request('whoami',
            env={'BZR_EMAIL': 'this_test@example.com'})
        stdout_content, stderr_content = self.communicate_with_fork(path)
        self.assertEqual('', stderr_content)
        self.assertEqual('this_test@example.com\n', stdout_content)

    def _check_exits_nicely(self, sig_id):
        path, _, sock = self.send_fork_request('rocks')
        self.assertEqual(None, self.service_process.poll())
        # Now when we send SIGTERM, it should wait for the child to exit,
        # before it tries to exit itself.
        # In python2.6+ we could use self.service_process.terminate()
        os.kill(self.service_process.pid, sig_id)
        self.assertEqual(None, self.service_process.poll())
        # Now talk to the child, so the service can close
        stdout_content, stderr_content = self.communicate_with_fork(path)
        self.assertEqual('It sure does!\n', stdout_content)
        self.assertEqual('', stderr_content)
        self.assertReturnCode(0, sock)
        # And the process should exit cleanly
        self.assertEqual(0, self.service_process.wait())

    def test_sigterm_exits_nicely(self):
        self._check_exits_nicely(signal.SIGTERM)

    def test_sigint_exits_nicely(self):
        self._check_exits_nicely(signal.SIGINT)

    def test_child_exits_eventually(self):
        # We won't ever bind to the socket the child wants, and after some
        # time, the child should exit cleanly.
        # First, tell the subprocess that we want children to exit quickly.
        # *sigh* signal.alarm only has 1s resolution, so this test is slow.
        response = self.send_message_to_service('child_connect_timeout 1\n')
        self.assertEqual('ok\n', response)
        # Now request a fork.
        path, pid, sock = self.send_fork_request('rocks')
        # We started opening the child, but stop before we get all handles
        # open. After 1 second, the child should get signaled and die.
        # The master process should notice, and tell us the status of the
        # exited child.
        val = sock.recv(4096)
        self.assertEqual('exited\n%s\n' % (signal.SIGALRM,), val)
        # The master process should clean up after the now deceased child.
        self.assertPathDoesNotExist(path)


class TestCaseWithLPForkingServiceDaemon(
    TestCaseWithLPForkingServiceSubprocess):
    """Test LPForkingService interaction, when run in daemon mode."""

    def _cleanup_daemon(self, pid, pid_filename):
        try:
            os.kill(pid, signal.SIGKILL)
        except (OSError, IOError), e:
            trace.mutter('failed to kill pid %d, might be already dead: %s'
                         % (pid, e))
        try:
            os.remove(pid_filename)
        except (OSError, IOError), e:
            if e.errno != errno.ENOENT:
                trace.mutter('failed to remove %r: %s'
                             % (pid_filename, e))

    def _start_subprocess(self, path, env_changes):
        fd, pid_filename = tempfile.mkstemp(prefix='tmp-lp-forking-service-',
                                            suffix='.pid')
        self.service_pid_filename = pid_filename
        os.close(fd)
        proc = self.start_bzr_subprocess(
            ['lp-service', '--path', path, '--no-preload',
             '--children-timeout=1', '--pid-file', pid_filename],
            env_changes=env_changes)
        trace.mutter('started lp-service daemon')
        # We wait for the spawned process to exit, expecting it to report the
        # final pid into the pid_filename.
        tnow = time.time()
        tstop_waiting = tnow + 1.0
        # When this returns, the first fork has completed and the parent has
        # exited.
        proc.wait()
        while tnow < tstop_waiting:
            # Wait for the socket to become available
            if os.path.exists(path):
                # The service has created the socket for us to communicate
                break
            time.sleep(0.1)
            tnow = time.time()

        with open(pid_filename, 'rb') as f:
            pid = f.read()
            trace.mutter('found pid: %r' % (pid,))
        pid = int(pid.strip())
        # This is now the pid of the final daemon
        trace.mutter('lp-forking-service daemon at pid %s' % (pid,))
        # Because nothing else will clean this up, add this final handler to
        # clean up if all else fails.
        self.addCleanup(self._cleanup_daemon, pid, pid_filename)
        # self.service_process will now be the pid of the daemon,
        # rather than a Popen object.
        return pid

    def stop_service(self):
        if self.service_process is None:
            # Already stopped
            return
        # First, try to stop the service gracefully, by sending a 'quit'
        # message
        try:
            response = self.send_message_to_service('quit\n')
        except socket.error, e:
            # Ignore a failure to connect; the service must be
            # stopping/stopped already.
            response = None
        if response is not None:
            self.assertEqual('ok\nquit command requested... exiting\n',
                             response)
        # Wait for the process to actually exit, or force it if necessary.
        tnow = time.time()
        tend = tnow + 2.0
        # We'll be nice a couple of times, and then get mean
        attempts = [None, None, None, signal.SIGTERM, signal.SIGKILL]
        stopped = False
        unclean = False
        while tnow < tend:
            try:
                os.kill(self.service_process, 0)
            except (OSError, IOError), e:
                if e.errno == errno.ESRCH:
                    # The process has successfully exited
                    stopped = True
                    break
                raise
            else:
                # The process has not exited yet
                time.sleep(0.1)
                if attempts:
                    sig = attempts.pop(0)
                    if sig is not None:
                        unclean = True
                        try:
                            os.kill(self.service_process, sig)
                        except (OSError, IOError), e:
                            if e.errno == errno.ESRCH:
                                stopped = True
                                break
                            raise
        if not stopped:
            self.fail('Unable to stop the daemon process (pid %s) after 2.0s'
                      % (self.service_process,))
        elif unclean:
            self.fail('Process (pid %s) had to be shut-down'
                      % (self.service_process,))
        self.service_process = None

    def test_simple_start_and_stop(self):
        # All the work is done in setUp().
        pass

    def test_starts_and_cleans_up(self):
        # The service should be up and responsive.
        response = self.send_message_to_service('hello\n')
        self.assertEqual('ok\nyep, still alive\n', response)
        self.failUnless(os.path.isfile(self.service_pid_filename))
        with open(self.service_pid_filename, 'rb') as f:
            content = f.read()
        self.assertEqualDiff('%d\n' % (self.service_process,), content)
        # We're done.  Shut it down.
        self.stop_service()
        self.failIf(os.path.isfile(self.service_pid_filename))