/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_http.py

  • Committer: Jelmer Vernooij
  • Date: 2010-03-21 21:39:33 UTC
  • mfrom: (5102 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5143.
  • Revision ID: jelmer@samba.org-20100321213933-fexeh9zcoz8oaju2
merge bzr.dev.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_http.py
1
 
# Copyright (C) 2005, 2006, 2007, 2008, 2009 Canonical Ltd
 
1
# Copyright (C) 2005-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
48
48
    deprecated_in,
49
49
    )
50
50
from bzrlib.tests import (
 
51
    features,
51
52
    http_server,
52
53
    http_utils,
53
54
    )
61
62
    )
62
63
 
63
64
 
64
 
try:
 
65
if features.pycurl.available():
65
66
    from bzrlib.transport.http._pycurl import PyCurlTransport
66
 
    pycurl_present = True
67
 
except errors.DependencyNotPresent:
68
 
    pycurl_present = False
69
67
 
70
68
 
71
69
def load_tests(standard_tests, module, loader):
84
82
                        _server=http_server.HttpServer_urllib,
85
83
                        _qualified_prefix='http+urllib',)),
86
84
        ]
87
 
    if pycurl_present:
 
85
    if features.pycurl.available():
88
86
        transport_scenarios.append(
89
87
            ('pycurl', dict(_transport=PyCurlTransport,
90
88
                            _server=http_server.HttpServer_PyCurl,
156
154
        activity_scenarios.append(
157
155
            ('urllib,https', dict(_activity_server=ActivityHTTPSServer,
158
156
                                  _transport=_urllib.HttpTransport_urllib,)),)
159
 
    if pycurl_present:
 
157
    if features.pycurl.available():
160
158
        activity_scenarios.append(
161
159
            ('pycurl,http', dict(_activity_server=ActivityHTTPServer,
162
160
                                 _transport=PyCurlTransport,)),)
220
218
    def get_url(self):
221
219
        return '%s://%s:%s/' % (self.scheme, self.host, self.port)
222
220
 
223
 
    def setUp(self):
 
221
    def start_server(self):
224
222
        self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
225
223
        self._sock.bind(('127.0.0.1', 0))
226
224
        self.host, self.port = self._sock.getsockname()
249
247
            # The client may have already closed the socket.
250
248
            pass
251
249
 
252
 
    def tearDown(self):
 
250
    def stop_server(self):
253
251
        try:
254
252
            self._sock.close()
255
253
        except socket.error:
308
306
 
309
307
        server = http_server.HttpServer(BogusRequestHandler)
310
308
        try:
311
 
            self.assertRaises(httplib.UnknownProtocol, server.setUp)
 
309
            self.assertRaises(httplib.UnknownProtocol, server.start_server)
312
310
        except:
313
 
            server.tearDown()
 
311
            server.stop_server()
314
312
            self.fail('HTTP Server creation did not raise UnknownProtocol')
315
313
 
316
314
    def test_force_invalid_protocol(self):
317
315
        server = http_server.HttpServer(protocol_version='HTTP/0.1')
318
316
        try:
319
 
            self.assertRaises(httplib.UnknownProtocol, server.setUp)
 
317
            self.assertRaises(httplib.UnknownProtocol, server.start_server)
320
318
        except:
321
 
            server.tearDown()
 
319
            server.stop_server()
322
320
            self.fail('HTTP Server creation did not raise UnknownProtocol')
323
321
 
324
322
    def test_server_start_and_stop(self):
325
323
        server = http_server.HttpServer()
326
 
        server.setUp()
 
324
        server.start_server()
327
325
        try:
328
326
            self.assertTrue(server._http_running)
329
327
        finally:
330
 
            server.tearDown()
 
328
            server.stop_server()
331
329
        self.assertFalse(server._http_running)
332
330
 
333
331
    def test_create_http_server_one_zero(self):
376
374
    """Test case to inherit from if pycurl is present"""
377
375
 
378
376
    def _get_pycurl_maybe(self):
379
 
        try:
380
 
            from bzrlib.transport.http._pycurl import PyCurlTransport
381
 
            return PyCurlTransport
382
 
        except errors.DependencyNotPresent:
383
 
            raise tests.TestSkipped('pycurl not present')
 
377
        self.requireFeature(features.pycurl)
 
378
        return PyCurlTransport
384
379
 
385
380
    _transport = property(_get_pycurl_maybe)
386
381
 
396
391
        self.assertEqual('http://example.com', url)
397
392
        self.assertEqual(0, len(f.credentials))
398
393
        url = http.extract_auth(
399
 
            'http://user:pass@www.bazaar-vcs.org/bzr/bzr.dev', f)
400
 
        self.assertEqual('http://www.bazaar-vcs.org/bzr/bzr.dev', url)
 
394
            'http://user:pass@example.com/bzr/bzr.dev', f)
 
395
        self.assertEqual('http://example.com/bzr/bzr.dev', url)
401
396
        self.assertEqual(1, len(f.credentials))
402
 
        self.assertEqual([None, 'www.bazaar-vcs.org', 'user', 'pass'],
 
397
        self.assertEqual([None, 'example.com', 'user', 'pass'],
403
398
                         f.credentials[0])
404
399
 
405
400
 
433
428
    def test_http_impl_urls(self):
434
429
        """There are servers which ask for particular clients to connect"""
435
430
        server = self._server()
436
 
        server.setUp()
 
431
        server.start_server()
437
432
        try:
438
433
            url = server.get_url()
439
434
            self.assertTrue(url.startswith('%s://' % self._qualified_prefix))
440
435
        finally:
441
 
            server.tearDown()
 
436
            server.stop_server()
442
437
 
443
438
 
444
439
class TestHttps_pycurl(TestWithTransport_pycurl, tests.TestCase):
453
448
        https by supplying a fake version_info that do not
454
449
        support it.
455
450
        """
456
 
        try:
457
 
            import pycurl
458
 
        except ImportError:
459
 
            raise tests.TestSkipped('pycurl not present')
 
451
        self.requireFeature(features.pycurl)
 
452
        # Import the module locally now that we now it's available.
 
453
        pycurl = features.pycurl.module
460
454
 
461
 
        version_info_orig = pycurl.version_info
462
 
        try:
463
 
            # Now that we have pycurl imported, we can fake its version_info
464
 
            # This was taken from a windows pycurl without SSL
465
 
            # (thanks to bialix)
466
 
            pycurl.version_info = lambda : (2,
467
 
                                            '7.13.2',
468
 
                                            462082,
469
 
                                            'i386-pc-win32',
470
 
                                            2576,
471
 
                                            None,
472
 
                                            0,
473
 
                                            None,
474
 
                                            ('ftp', 'gopher', 'telnet',
475
 
                                             'dict', 'ldap', 'http', 'file'),
476
 
                                            None,
477
 
                                            0,
478
 
                                            None)
479
 
            self.assertRaises(errors.DependencyNotPresent, self._transport,
480
 
                              'https://launchpad.net')
481
 
        finally:
482
 
            # Restore the right function
483
 
            pycurl.version_info = version_info_orig
 
455
        self.overrideAttr(pycurl, 'version_info',
 
456
                          # Fake the pycurl version_info This was taken from
 
457
                          # a windows pycurl without SSL (thanks to bialix)
 
458
                          lambda : (2,
 
459
                                    '7.13.2',
 
460
                                    462082,
 
461
                                    'i386-pc-win32',
 
462
                                    2576,
 
463
                                    None,
 
464
                                    0,
 
465
                                    None,
 
466
                                    ('ftp', 'gopher', 'telnet',
 
467
                                     'dict', 'ldap', 'http', 'file'),
 
468
                                    None,
 
469
                                    0,
 
470
                                    None))
 
471
        self.assertRaises(errors.DependencyNotPresent, self._transport,
 
472
                          'https://launchpad.net')
484
473
 
485
474
 
486
475
class TestHTTPConnections(http_utils.TestCaseWithWebserver):
603
592
                                      protocol_version=self._protocol_version)
604
593
 
605
594
    def _testing_pycurl(self):
606
 
        return pycurl_present and self._transport == PyCurlTransport
 
595
        # TODO: This is duplicated for lots of the classes in this file
 
596
        return (features.pycurl.available()
 
597
                and self._transport == PyCurlTransport)
607
598
 
608
599
 
609
600
class WallRequestHandler(http_server.TestingHTTPRequestHandler):
718
709
    _req_handler_class = BadProtocolRequestHandler
719
710
 
720
711
    def setUp(self):
721
 
        if pycurl_present and self._transport == PyCurlTransport:
 
712
        if self._testing_pycurl():
722
713
            raise tests.TestNotApplicable(
723
714
                "pycurl doesn't check the protocol version")
724
715
        super(TestBadProtocolServer, self).setUp()
768
759
        self.assertEqual(None, server.host)
769
760
        self.assertEqual(None, server.port)
770
761
 
771
 
    def test_setUp_and_tearDown(self):
 
762
    def test_setUp_and_stop(self):
772
763
        server = RecordingServer(expect_body_tail=None)
773
 
        server.setUp()
 
764
        server.start_server()
774
765
        try:
775
766
            self.assertNotEqual(None, server.host)
776
767
            self.assertNotEqual(None, server.port)
777
768
        finally:
778
 
            server.tearDown()
 
769
            server.stop_server()
779
770
        self.assertEqual(None, server.host)
780
771
        self.assertEqual(None, server.port)
781
772
 
1187
1178
        self._old_env = {}
1188
1179
 
1189
1180
    def _testing_pycurl(self):
1190
 
        return pycurl_present and self._transport == PyCurlTransport
 
1181
        # TODO: This is duplicated for lots of the classes in this file
 
1182
        return (features.pycurl.available()
 
1183
                and self._transport == PyCurlTransport)
1191
1184
 
1192
1185
    def create_transport_secondary_server(self):
1193
1186
        """Creates an http server that will serve files with
1367
1360
 
1368
1361
 
1369
1362
def install_redirected_request(test):
1370
 
    test.original_class = _urllib2_wrappers.Request
1371
 
    def restore():
1372
 
        _urllib2_wrappers.Request = test.original_class
1373
 
    _urllib2_wrappers.Request = RedirectedRequest
1374
 
    test.addCleanup(restore)
 
1363
    test.overrideAttr(_urllib2_wrappers, 'Request', RedirectedRequest)
1375
1364
 
1376
1365
 
1377
1366
class TestHTTPSilentRedirections(http_utils.TestCaseWithRedirectedWebserver):
1389
1378
    """
1390
1379
 
1391
1380
    def setUp(self):
1392
 
        if pycurl_present and self._transport == PyCurlTransport:
 
1381
        if (features.pycurl.available()
 
1382
            and self._transport == PyCurlTransport):
1393
1383
            raise tests.TestNotApplicable(
1394
1384
                "pycurl doesn't redirect silently annymore")
1395
1385
        super(TestHTTPSilentRedirections, self).setUp()
1507
1497
        return self._auth_server(protocol_version=self._protocol_version)
1508
1498
 
1509
1499
    def _testing_pycurl(self):
1510
 
        return pycurl_present and self._transport == PyCurlTransport
 
1500
        # TODO: This is duplicated for lots of the classes in this file
 
1501
        return (features.pycurl.available()
 
1502
                and self._transport == PyCurlTransport)
1511
1503
 
1512
1504
    def get_user_url(self, user, password):
1513
1505
        """Build an url embedding user and password"""
1964
1956
    def setUp(self):
1965
1957
        tests.TestCase.setUp(self)
1966
1958
        self.server = self._activity_server(self._protocol_version)
1967
 
        self.server.setUp()
 
1959
        self.server.start_server()
1968
1960
        self.activities = {}
1969
1961
        def report_activity(t, bytes, direction):
1970
1962
            count = self.activities.get(direction, 0)
1979
1971
 
1980
1972
    def tearDown(self):
1981
1973
        self._transport._report_activity = self.orig_report_activity
1982
 
        self.server.tearDown()
 
1974
        self.server.stop_server()
1983
1975
        tests.TestCase.tearDown(self)
1984
1976
 
1985
1977
    def get_transport(self):
2105
2097
    def setUp(self):
2106
2098
        tests.TestCase.setUp(self)
2107
2099
        self.server = self._activity_server(self._protocol_version)
2108
 
        self.server.setUp()
 
2100
        self.server.start_server()
2109
2101
        self.activities = {}
2110
2102
        def report_activity(t, bytes, direction):
2111
2103
            count = self.activities.get(direction, 0)
2120
2112
 
2121
2113
    def tearDown(self):
2122
2114
        self._transport._report_activity = self.orig_report_activity
2123
 
        self.server.tearDown()
 
2115
        self.server.stop_server()
2124
2116
        tests.TestCase.tearDown(self)
2125
2117
 
2126
2118
 
2136
2128
        self.server = ActivityHTTPServer('HTTP/1.1')
2137
2129
        self._transport=_urllib.HttpTransport_urllib
2138
2130
 
2139
 
        self.server.setUp()
 
2131
        self.server.start_server()
2140
2132
 
2141
2133
        # We override at class level because constructors may propagate the
2142
2134
        # bound method and render instance overriding ineffective (an
2146
2138
 
2147
2139
    def tearDown(self):
2148
2140
        self._transport._report_activity = self.orig_report_activity
2149
 
        self.server.tearDown()
 
2141
        self.server.stop_server()
2150
2142
        tests.TestCase.tearDown(self)
2151
2143
 
2152
2144
    def assertActivitiesMatch(self):