← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/http_server.py

  • Committer: Martin Pool
  • Date: 2010-01-29 14:09:05 UTC
  • mto: This revision was merged to the branch mainline in revision 4992.
  • Revision ID: mbp@sourcefrog.net-20100129140905-2uiarb6p8di1ywsr
Correction to url

from review: https://code.edge.launchpad.net/~mbp/bzr/doc/+merge/18250

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/http_server.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
31
31
import urlparse
32
32
 
33
33
from bzrlib import transport
34
 
from bzrlib.tests import test_server
35
34
from bzrlib.transport import local
36
35
 
37
36
 
89
88
                                     errno.ECONNABORTED, errno.EBADF)):
90
89
                raise
91
90
 
92
 
    error_content_type = 'text/plain'
93
 
    error_message_format = '''\
94
 
Error code: %(code)s.
95
 
Message: %(message)s.
96
 
'''
97
 
 
98
 
    def send_error(self, code, message=None):
99
 
        """Send and log an error reply.
100
 
 
101
 
        We redefine the python-provided version to be able to set a 
102
 
        ``Content-Length`` header as some http/1.1 clients complain otherwise
103
 
        (see bug #568421).
104
 
 
105
 
        :param code: The HTTP error code.
106
 
 
107
 
        :param message: The explanation of the error code, Defaults to a short
108
 
             entry.
109
 
        """
110
 
 
111
 
        if message is None:
112
 
            try:
113
 
                message = self.responses[code][0]
114
 
            except KeyError:
115
 
                message = '???'
116
 
        self.log_error("code %d, message %s", code, message)
117
 
        content = (self.error_message_format %
118
 
                   {'code': code, 'message': message})
119
 
        self.send_response(code, message)
120
 
        self.send_header("Content-Type", self.error_content_type)
121
 
        self.send_header("Content-Length", "%d" % len(content))
122
 
        self.send_header('Connection', 'close')
123
 
        self.end_headers()
124
 
        if self.command != 'HEAD' and code >= 200 and code not in (204, 304):
125
 
            self.wfile.write(content)
126
 
 
127
91
    _range_regexp = re.compile(r'^(?P<start>\d+)-(?P<end>\d+)$')
128
92
    _tail_regexp = re.compile(r'^-(?P<tail>\d+)$')
129
93
 
566
530
        """
567
531
        # XXX: TODO: make the server back onto vfs_server rather than local
568
532
        # disk.
569
 
        if not (backing_transport_server is None
570
 
                or isinstance(backing_transport_server,
571
 
                              test_server.LocalURLServer)):
 
533
        if not (backing_transport_server is None or \
 
534
                isinstance(backing_transport_server, local.LocalURLServer)):
572
535
            raise AssertionError(
573
536
                "HTTPServer currently assumes local transport, got %s" % \
574
537
                backing_transport_server)