/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_smart_transport.py

  • Committer: Andrew Bennetts
  • Date: 2010-02-12 04:33:05 UTC
  • mfrom: (5031 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5032.
  • Revision ID: andrew.bennetts@canonical.com-20100212043305-ujdbsdoviql2t7i3
Merge lp:bzr

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_smart_transport.py
1
 
# Copyright (C) 2006, 2007, 2008, 2009 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
28
28
        errors,
29
29
        osutils,
30
30
        tests,
 
31
        transport,
31
32
        urlutils,
32
33
        )
33
34
from bzrlib.smart import (
39
40
        server,
40
41
        vfs,
41
42
)
42
 
from bzrlib.tests.test_smart import TestCaseWithSmartMedium
 
43
from bzrlib.tests import test_smart
43
44
from bzrlib.transport import (
44
 
        get_transport,
 
45
        http,
45
46
        local,
46
47
        memory,
47
48
        remote,
48
49
        )
49
 
from bzrlib.transport.http import SmartClientHTTPMediumRequest
50
50
 
51
51
 
52
52
class StringIOSSHVendor(object):
557
557
        self.assertRaises(errors.ReadingCompleted, request.read_bytes, None)
558
558
 
559
559
 
560
 
class RemoteTransportTests(TestCaseWithSmartMedium):
 
560
class RemoteTransportTests(test_smart.TestCaseWithSmartMedium):
561
561
 
562
562
    def test_plausible_url(self):
563
563
        self.assert_(self.get_url().startswith('bzr://'))
987
987
    All of these tests are run with a server running on another thread serving
988
988
    a MemoryTransport, and a connection to it already open.
989
989
 
990
 
    the server is obtained by calling self.setUpServer(readonly=False).
 
990
    the server is obtained by calling self.start_server(readonly=False).
991
991
    """
992
992
 
993
 
    def setUpServer(self, readonly=False, backing_transport=None):
 
993
    def start_server(self, readonly=False, backing_transport=None):
994
994
        """Setup the server.
995
995
 
996
996
        :param readonly: Create a readonly server.
1000
1000
        # self.vfs_transport_factory etc.
1001
1001
        if not backing_transport:
1002
1002
            mem_server = memory.MemoryServer()
1003
 
            mem_server.setUp()
1004
 
            self.addCleanup(mem_server.tearDown)
 
1003
            mem_server.start_server()
 
1004
            self.addCleanup(mem_server.stop_server)
1005
1005
            self.permit_url(mem_server.get_url())
1006
 
            self.backing_transport = get_transport(mem_server.get_url())
 
1006
            self.backing_transport = transport.get_transport(
 
1007
                mem_server.get_url())
1007
1008
        else:
1008
1009
            self.backing_transport = backing_transport
1009
1010
        if readonly:
1010
1011
            self.real_backing_transport = self.backing_transport
1011
 
            self.backing_transport = get_transport("readonly+" + self.backing_transport.abspath('.'))
 
1012
            self.backing_transport = transport.get_transport(
 
1013
                "readonly+" + self.backing_transport.abspath('.'))
1012
1014
        self.server = server.SmartTCPServer(self.backing_transport)
1013
1015
        self.server.start_background_thread('-' + self.id())
1014
1016
        self.transport = remote.RemoteTCPTransport(self.server.get_url())
1021
1023
            del self.transport
1022
1024
        if getattr(self, 'server', None):
1023
1025
            self.server.stop_background_thread()
 
1026
            # XXX: why not .stop_server() -- mbp 20100106
1024
1027
            del self.server
1025
1028
 
1026
1029
 
1028
1031
 
1029
1032
    def test_server_setup_teardown(self):
1030
1033
        """It should be safe to teardown the server with no requests."""
1031
 
        self.setUpServer()
 
1034
        self.start_server()
1032
1035
        server = self.server
1033
1036
        transport = remote.RemoteTCPTransport(self.server.get_url())
1034
1037
        self.tearDownServer()
1036
1039
 
1037
1040
    def test_server_closes_listening_sock_on_shutdown_after_request(self):
1038
1041
        """The server should close its listening socket when it's stopped."""
1039
 
        self.setUpServer()
 
1042
        self.start_server()
1040
1043
        server = self.server
1041
1044
        self.transport.has('.')
1042
1045
        self.tearDownServer()
1051
1054
 
1052
1055
    def setUp(self):
1053
1056
        super(WritableEndToEndTests, self).setUp()
1054
 
        self.setUpServer()
 
1057
        self.start_server()
1055
1058
 
1056
1059
    def test_start_tcp_server(self):
1057
1060
        url = self.server.get_url()
1130
1133
    def test_mkdir_error_readonly(self):
1131
1134
        """TransportNotPossible should be preserved from the backing transport."""
1132
1135
        self._captureVar('BZR_NO_SMART_VFS', None)
1133
 
        self.setUpServer(readonly=True)
 
1136
        self.start_server(readonly=True)
1134
1137
        self.assertRaises(errors.TransportNotPossible, self.transport.mkdir,
1135
1138
            'foo')
1136
1139
 
1146
1149
        self.hook_calls = []
1147
1150
        server.SmartTCPServer.hooks.install_named_hook('server_started',
1148
1151
            self.capture_server_call, None)
1149
 
        self.setUpServer()
 
1152
        self.start_server()
1150
1153
        # at this point, the server will be starting a thread up.
1151
1154
        # there is no indicator at the moment, so bodge it by doing a request.
1152
1155
        self.transport.has('.')
1160
1163
        self.hook_calls = []
1161
1164
        server.SmartTCPServer.hooks.install_named_hook('server_started',
1162
1165
            self.capture_server_call, None)
1163
 
        self.setUpServer(backing_transport=get_transport("."))
 
1166
        self.start_server(backing_transport=transport.get_transport("."))
1164
1167
        # at this point, the server will be starting a thread up.
1165
1168
        # there is no indicator at the moment, so bodge it by doing a request.
1166
1169
        self.transport.has('.')
1176
1179
        self.hook_calls = []
1177
1180
        server.SmartTCPServer.hooks.install_named_hook('server_stopped',
1178
1181
            self.capture_server_call, None)
1179
 
        self.setUpServer()
 
1182
        self.start_server()
1180
1183
        result = [([self.backing_transport.base], self.transport.base)]
1181
1184
        # check the stopping message isn't emitted up front.
1182
1185
        self.assertEqual([], self.hook_calls)
1193
1196
        self.hook_calls = []
1194
1197
        server.SmartTCPServer.hooks.install_named_hook('server_stopped',
1195
1198
            self.capture_server_call, None)
1196
 
        self.setUpServer(backing_transport=get_transport("."))
 
1199
        self.start_server(backing_transport=transport.get_transport("."))
1197
1200
        result = [(
1198
1201
            [self.backing_transport.base, self.backing_transport.external_url()]
1199
1202
            , self.transport.base)]
1338
1341
class RemoteTransportRegistration(tests.TestCase):
1339
1342
 
1340
1343
    def test_registration(self):
1341
 
        t = get_transport('bzr+ssh://example.com/path')
 
1344
        t = transport.get_transport('bzr+ssh://example.com/path')
1342
1345
        self.assertIsInstance(t, remote.RemoteSSHTransport)
1343
1346
        self.assertEqual('example.com', t._host)
1344
1347
 
1345
1348
    def test_bzr_https(self):
1346
1349
        # https://bugs.launchpad.net/bzr/+bug/128456
1347
 
        t = get_transport('bzr+https://example.com/path')
 
1350
        t = transport.get_transport('bzr+https://example.com/path')
1348
1351
        self.assertIsInstance(t, remote.RemoteHTTPTransport)
1349
1352
        self.assertStartsWith(
1350
1353
            t._http_transport.base,
2892
2895
        self.assertWriteCount(1)
2893
2896
 
2894
2897
    def test_send_response_with_body_stream_flushes_buffers_sometimes(self):
2895
 
        """When there are many chunks (>100), multiple writes will occur rather
 
2898
        """When there are many bytes (>1MB), multiple writes will occur rather
2896
2899
        than buffering indefinitely.
2897
2900
        """
2898
 
        # Construct a response with stream with 40 chunks in it.  Every chunk
2899
 
        # triggers 3 buffered writes, so we expect > 100 buffered writes, but <
2900
 
        # 200.
2901
 
        body_stream = ['chunk %d' % count for count in range(40)]
 
2901
        # Construct a response with stream with ~1.5MB in it. This should
 
2902
        # trigger 2 writes, but not 3
 
2903
        onekib = '12345678' * 128
 
2904
        body_stream = [onekib] * (1024 + 512)
2902
2905
        response = _mod_request.SuccessfulSmartServerResponse(
2903
2906
            ('arg', 'arg'), body_stream=body_stream)
2904
2907
        self.responder.send_response(response)
2905
 
        # The write buffer is flushed every 100 buffered writes, so we expect 2
2906
 
        # actual writes.
2907
2908
        self.assertWriteCount(2)
2908
2909
 
2909
2910
 
3517
3518
 
3518
3519
    def test_smart_http_medium_request_accept_bytes(self):
3519
3520
        medium = FakeHTTPMedium()
3520
 
        request = SmartClientHTTPMediumRequest(medium)
 
3521
        request = http.SmartClientHTTPMediumRequest(medium)
3521
3522
        request.accept_bytes('abc')
3522
3523
        request.accept_bytes('def')
3523
3524
        self.assertEqual(None, medium.written_request)