← 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: Canonical.com Patch Queue Manager
  • Date: 2010-07-07 04:03:32 UTC
  • mfrom: (5335.2.3 doc)
  • Revision ID: pqm@pqm.ubuntu.com-20100707040332-we60v2hsd39rumlr
(mbp) developer docs about testing (Martin Pool)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/http_server.py
1
 
# Copyright (C) 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2006-2010 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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
import errno
18
18
import httplib
31
31
import urlparse
32
32
 
33
33
from bzrlib import transport
 
34
from bzrlib.tests import test_server
34
35
from bzrlib.transport import local
35
36
 
36
37
 
137
138
            # common)
138
139
            self.send_response(301)
139
140
            self.send_header("Location", self.path + "/")
140
 
            # Indicates that the body is empty for HTTP/1.1 clients 
 
141
            # Indicates that the body is empty for HTTP/1.1 clients
141
142
            self.send_header('Content-Length', '0')
142
143
            self.end_headers()
143
144
            return None
318
319
        self.test_case_server = test_case_server
319
320
        self._home_dir = test_case_server._home_dir
320
321
 
321
 
    def tearDown(self):
 
322
    def stop_server(self):
322
323
         """Called to clean-up the server.
323
 
 
 
324
 
324
325
         Since the server may be (surely is, even) in a blocking listen, we
325
326
         shutdown its socket before closing it.
326
327
         """
347
348
             # 'Socket is not connected' can also occur on OSX, with a
348
349
             # "regular" ENOTCONN (when something went wrong during test case
349
350
             # setup leading to self.setUp() *not* being called but
350
 
             # self.tearDown() still being called -- vila20081106
 
351
             # self.stop_server() still being called -- vila20081106
351
352
             if not len(e.args) or e.args[0] not in (errno.ENOTCONN, 10057):
352
353
                 raise
353
354
         # Let the server properly close the socket
382
383
        # lying around.
383
384
        self.daemon_threads = True
384
385
 
 
386
    def process_request_thread(self, request, client_address):
 
387
        SocketServer.ThreadingTCPServer.process_request_thread(
 
388
            self, request, client_address)
 
389
        # Under some circumstances (as in bug #383920), we need to force the
 
390
        # shutdown as python delays it until gc occur otherwise and the client
 
391
        # may hang.
 
392
        try:
 
393
            # The request process has been completed, the thread is about to
 
394
            # die, let's shutdown the socket if we can.
 
395
            request.shutdown(socket.SHUT_RDWR)
 
396
        except (socket.error, select.error), e:
 
397
            if e[0] in (errno.EBADF, errno.ENOTCONN):
 
398
                # Right, the socket is already down
 
399
                pass
 
400
            else:
 
401
                raise
 
402
 
385
403
 
386
404
class HttpServer(transport.Server):
387
405
    """A test server for http transports.
445
463
                raise httplib.UnknownProtocol(proto_vers)
446
464
            else:
447
465
                self._httpd = self.create_httpd(serv_cls, rhandler)
448
 
            host, self.port = self._httpd.socket.getsockname()
 
466
            self.host, self.port = self._httpd.socket.getsockname()
449
467
        return self._httpd
450
468
 
451
469
    def _http_start(self):
477
495
            except socket.timeout:
478
496
                pass
479
497
            except (socket.error, select.error), e:
480
 
               if e[0] == errno.EBADF:
481
 
                   # Starting with python-2.6, handle_request may raise socket
482
 
                   # or select exceptions when the server is shut down (as we
483
 
                   # do).
484
 
                   pass
485
 
               else:
486
 
                   raise
 
498
                if (e[0] == errno.EBADF
 
499
                    or (sys.platform == 'win32' and e[0] == 10038)):
 
500
                    # Starting with python-2.6, handle_request may raise socket
 
501
                    # or select exceptions when the server is shut down (as we
 
502
                    # do).
 
503
                    # 10038 = WSAENOTSOCK
 
504
                    # http://msdn.microsoft.com/en-us/library/ms740668%28VS.85%29.aspx
 
505
                    pass
 
506
                else:
 
507
                    raise
487
508
 
488
509
    def _get_remote_url(self, path):
489
510
        path_parts = path.split(os.path.sep)
501
522
        """Capture Server log output."""
502
523
        self.logs.append(format % args)
503
524
 
504
 
    def setUp(self, backing_transport_server=None):
505
 
        """See bzrlib.transport.Server.setUp.
506
 
        
 
525
    def start_server(self, backing_transport_server=None):
 
526
        """See bzrlib.transport.Server.start_server.
 
527
 
507
528
        :param backing_transport_server: The transport that requests over this
508
529
            protocol should be forwarded to. Note that this is currently not
509
530
            supported for HTTP.
510
531
        """
511
532
        # XXX: TODO: make the server back onto vfs_server rather than local
512
533
        # disk.
513
 
        if not (backing_transport_server is None or \
514
 
                isinstance(backing_transport_server, local.LocalURLServer)):
 
534
        if not (backing_transport_server is None
 
535
                or isinstance(backing_transport_server,
 
536
                              test_server.LocalURLServer)):
515
537
            raise AssertionError(
516
538
                "HTTPServer currently assumes local transport, got %s" % \
517
539
                backing_transport_server)
537
559
        self._http_starting.release()
538
560
        self.logs = []
539
561
 
540
 
    def tearDown(self):
541
 
        """See bzrlib.transport.Server.tearDown."""
542
 
        self._httpd.tearDown()
 
562
    def stop_server(self):
 
563
        self._httpd.stop_server()
543
564
        self._http_running = False
544
565
        # We don't need to 'self._http_thread.join()' here since the thread is
545
566
        # a daemonic one and will be garbage collected anyway. Joining just