← Back to branch summary

~launchpad-pqm/launchpad/devel

« back to all changes in this revision

Viewing changes to lib/lp/poppy/tests/helpers.py

Fix lint.

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/lp/poppy/tests/helpers.py
 
1
# Copyright 2009 Canonical Ltd.  This software is licensed under the
 
2
# GNU Affero General Public License version 3 (see the file LICENSE).
 
3
 
 
4
__metaclass__ = type
 
5
 
 
6
import os
 
7
import select
 
8
import subprocess
 
9
import sys
 
10
import time
 
11
 
 
12
from canonical.config import config
 
13
from lp.services.osutils import two_stage_kill
 
14
 
 
15
 
 
16
class SoyuzUploadError(Exception):
 
17
    """Used in the soyuz-upload test."""
 
18
 
 
19
 
 
20
class PoppyTestSetup:
 
21
    """Provides a setup and teardown mechanism for a poppy subprocess instance.
 
22
 
 
23
    Use this like you would LibrarianTestSetup or similar.
 
24
    """
 
25
 
 
26
    def __init__(self, fsroot,
 
27
                 user='anonymous',
 
28
                 cmd='echo @distro@; ls @fsroot@',
 
29
                 port=3421):
 
30
        self.fsroot = fsroot
 
31
        self.user = user
 
32
        self.cmd = cmd
 
33
        self.port = str(port)
 
34
        self.running = False
 
35
 
 
36
    def startPoppy(self):
 
37
        """Start the poppy instance."""
 
38
        script = os.path.join(config.root, "daemons/poppy-upload.py")
 
39
        self.process = subprocess.Popen(
 
40
            [sys.executable, script, "--allow-user", self.user,
 
41
             "--cmd", self.cmd, self.fsroot, self.port],
 
42
            stdout=subprocess.PIPE)
 
43
        self.running = True
 
44
 
 
45
    def killPoppy(self):
 
46
        """Kill the poppy instance dead."""
 
47
        if not self.alive:
 
48
            return
 
49
        two_stage_kill(self.process.pid)
 
50
        self.running = False
 
51
 
 
52
    @property
 
53
    def alive(self):
 
54
        """Whether or not the poppy instance is still alive."""
 
55
        if not self.running:
 
56
            return False
 
57
        return self.process.poll() is None
 
58
 
 
59
    def verify_output(self, expected):
 
60
        """Verify that poppy writes the expected output."""
 
61
        timelimit = time.time() + 60
 
62
        buffer = ""
 
63
        while True:
 
64
            rlist, wlist, xlist = select.select(
 
65
                [self.process.stdout.fileno()], [], [], timelimit - time.time())
 
66
            if len(rlist) == 0:
 
67
                if self.process.poll() is not None:
 
68
                    raise SoyuzUploadError("Poppy died unexpectedly")
 
69
                # Try and kill poppy too?
 
70
                raise SoyuzUploadError(
 
71
                    "FTP server timed out. The following was expected:\n%r\n"
 
72
                    "But the following was received:\n%r\n"
 
73
                    % (expected, buffer))
 
74
            else:
 
75
                # reset the time limit
 
76
                timelimit = time.time() + 60
 
77
                chunk = os.read(self.process.stdout.fileno(), 4096)
 
78
                buffer += chunk
 
79
                # XXX: jamesh 2007-02-20:
 
80
                # We might have an incomplete line at the end of the
 
81
                # buffer.  This doesn't seem to be a problem for the
 
82
                # amount of data being written in the tests though.
 
83
                lines = buffer.splitlines()
 
84
                for line in lines:
 
85
                    if line == self.user:
 
86
                        continue
 
87
                    if line not in expected:
 
88
                        raise SoyuzUploadError("Unexpected line found in "
 
89
                                               "poppy output: %r" % line)
 
90
                    expected.remove(line)
 
91
                buffer = ""
 
92
                if not expected:
 
93
                    break
 
94
                # stdout was closed before we read all the expected lines
 
95
                if chunk == "":
 
96
                    raise SoyuzUploadError("FTP server exited before the "
 
97
                                           "expected data was received: %r"
 
98
                                           % expected)
 
99
        else:
 
100
            raise SoyuzUploadError(
 
101
                "FTP server timed out.\n"
 
102
                "The following was expected:\n%r\n"
 
103
                "But the following was received:\n%r\n"
 
104
                % (expected, buffer))
 
105