/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
3407.2.2 by Martin Pool
Remove special case in RemoteBranchLockableFiles for branch.conf
1
# Copyright (C) 2006, 2007, 2008 Canonical Ltd
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
2
#
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
#
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
#
13
# You should have received a copy of the GNU General Public License
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
16
17
"""Tests for remote bzrdir/branch/repo/etc
18
19
These are proxy objects which act on remote objects by sending messages
20
through a smart client.  The proxies are to be created when attempting to open
21
the object given a transport that supports smartserver rpc operations. 
2018.18.7 by Martin Pool
(broken) Start addng client proxy test for Repository.tarball
22
23
These tests correspond to tests.test_smart, which exercises the server side.
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
24
"""
25
3211.5.2 by Robert Collins
Change RemoteRepository.get_parent_map to use bz2 not gzip for compression.
26
import bz2
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
27
from cStringIO import StringIO
28
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
29
from bzrlib import (
30
    errors,
3184.1.9 by Robert Collins
* ``Repository.get_data_stream`` is now deprecated in favour of
31
    graph,
2535.3.39 by Andrew Bennetts
Tidy some XXXs.
32
    pack,
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
33
    remote,
2018.18.9 by Martin Pool
remote Repository.tarball builds a temporary directory and tars that
34
    repository,
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
35
    tests,
36
    )
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
37
from bzrlib.branch import Branch
38
from bzrlib.bzrdir import BzrDir, BzrDirFormat
39
from bzrlib.remote import (
40
    RemoteBranch,
41
    RemoteBzrDir,
42
    RemoteBzrDirFormat,
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
43
    RemoteRepository,
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
44
    )
45
from bzrlib.revision import NULL_REVISION
2432.3.2 by Andrew Bennetts
Add test, and tidy implementation.
46
from bzrlib.smart import server, medium
2018.5.159 by Andrew Bennetts
Rename SmartClient to _SmartClient.
47
from bzrlib.smart.client import _SmartClient
3287.6.4 by Robert Collins
Fix up deprecation warnings for get_revision_graph.
48
from bzrlib.symbol_versioning import one_four
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
49
from bzrlib.transport import get_transport, http
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
50
from bzrlib.transport.memory import MemoryTransport
3431.3.1 by Andrew Bennetts
First rough cut of a fix for bug #230550, by adding .base to SmartClientMedia rather than relying on other objects to track this accurately while reusing client media.
51
from bzrlib.transport.remote import RemoteTransport, RemoteTCPTransport
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
52
2018.5.24 by Andrew Bennetts
Setting NO_SMART_VFS in environment will disable VFS methods in the smart server. (Robert Collins, John Arbash Meinel, Andrew Bennetts)
53
2018.5.42 by Robert Collins
Various hopefully improvements, but wsgi is broken, handing over to spiv :).
54
class BasicRemoteObjectTests(tests.TestCaseWithTransport):
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
55
56
    def setUp(self):
2018.5.95 by Andrew Bennetts
Add a Transport.is_readonly remote call, let {Branch,Repository}.lock_write remote call return UnlockableTransport, and miscellaneous test fixes.
57
        self.transport_server = server.SmartTCPServer_for_testing
2018.5.42 by Robert Collins
Various hopefully improvements, but wsgi is broken, handing over to spiv :).
58
        super(BasicRemoteObjectTests, self).setUp()
59
        self.transport = self.get_transport()
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
60
        # make a branch that can be opened over the smart transport
1752.2.31 by Martin Pool
[broken] some support for write operations over hpss
61
        self.local_wt = BzrDir.create_standalone_workingtree('.')
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
62
2018.5.171 by Andrew Bennetts
Disconnect RemoteTransports in some tests to avoid tripping up test_strace with leftover threads from previous tests.
63
    def tearDown(self):
64
        self.transport.disconnect()
65
        tests.TestCaseWithTransport.tearDown(self)
66
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
67
    def test_create_remote_bzrdir(self):
68
        b = remote.RemoteBzrDir(self.transport)
69
        self.assertIsInstance(b, BzrDir)
70
71
    def test_open_remote_branch(self):
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
72
        # open a standalone branch in the working directory
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
73
        b = remote.RemoteBzrDir(self.transport)
74
        branch = b.open_branch()
2018.5.163 by Andrew Bennetts
Deal with various review comments from Robert.
75
        self.assertIsInstance(branch, Branch)
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
76
1752.2.31 by Martin Pool
[broken] some support for write operations over hpss
77
    def test_remote_repository(self):
78
        b = BzrDir.open_from_transport(self.transport)
79
        repo = b.open_repository()
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
80
        revid = u'\xc823123123'.encode('utf8')
2018.5.40 by Robert Collins
Implement a remote Repository.has_revision method.
81
        self.assertFalse(repo.has_revision(revid))
82
        self.local_wt.commit(message='test commit', rev_id=revid)
83
        self.assertTrue(repo.has_revision(revid))
1752.2.31 by Martin Pool
[broken] some support for write operations over hpss
84
85
    def test_remote_branch_revision_history(self):
86
        b = BzrDir.open_from_transport(self.transport).open_branch()
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
87
        self.assertEqual([], b.revision_history())
88
        r1 = self.local_wt.commit('1st commit')
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
89
        r2 = self.local_wt.commit('1st commit', rev_id=u'\xc8'.encode('utf8'))
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
90
        self.assertEqual([r1, r2], b.revision_history())
1752.2.31 by Martin Pool
[broken] some support for write operations over hpss
91
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
92
    def test_find_correct_format(self):
2018.5.20 by Andrew Bennetts
Move bzrlib/transport/smart/_smart.py to bzrlib/transport/remote.py and rename SmartTransport to RemoteTransport (Robert Collins, Andrew Bennetts)
93
        """Should open a RemoteBzrDir over a RemoteTransport"""
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
94
        fmt = BzrDirFormat.find_format(self.transport)
2018.5.169 by Andrew Bennetts
Add a _server_formats flag to BzrDir.open_from_transport and BzrDirFormat.find_format, make RemoteBranch.control_files into a property.
95
        self.assertTrue(RemoteBzrDirFormat
96
                        in BzrDirFormat._control_server_formats)
1752.2.30 by Martin Pool
Start adding a RemoteBzrDir, etc
97
        self.assertIsInstance(fmt, remote.RemoteBzrDirFormat)
98
99
    def test_open_detected_smart_format(self):
100
        fmt = BzrDirFormat.find_format(self.transport)
101
        d = fmt.open(self.transport)
102
        self.assertIsInstance(d, BzrDir)
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
103
2477.1.1 by Martin Pool
Add RemoteBranch repr
104
    def test_remote_branch_repr(self):
105
        b = BzrDir.open_from_transport(self.transport).open_branch()
106
        self.assertStartsWith(str(b), 'RemoteBranch(')
107
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
108
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
109
class FakeProtocol(object):
110
    """Lookalike SmartClientRequestProtocolOne allowing body reading tests."""
111
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
112
    def __init__(self, body, fake_client):
2535.4.2 by Andrew Bennetts
Nasty hackery to make stream_knit_data_for_revisions response use streaming.
113
        self.body = body
114
        self._body_buffer = None
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
115
        self._fake_client = fake_client
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
116
117
    def read_body_bytes(self, count=-1):
2535.4.2 by Andrew Bennetts
Nasty hackery to make stream_knit_data_for_revisions response use streaming.
118
        if self._body_buffer is None:
119
            self._body_buffer = StringIO(self.body)
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
120
        bytes = self._body_buffer.read(count)
121
        if self._body_buffer.tell() == len(self._body_buffer.getvalue()):
122
            self._fake_client.expecting_body = False
123
        return bytes
124
125
    def cancel_read_body(self):
126
        self._fake_client.expecting_body = False
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
127
2535.4.2 by Andrew Bennetts
Nasty hackery to make stream_knit_data_for_revisions response use streaming.
128
    def read_streamed_body(self):
129
        return self.body
130
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
131
2018.5.159 by Andrew Bennetts
Rename SmartClient to _SmartClient.
132
class FakeClient(_SmartClient):
133
    """Lookalike for _SmartClient allowing testing."""
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
134
    
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
135
    def __init__(self, fake_medium_base='fake base'):
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
136
        """Create a FakeClient.
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
137
3104.4.2 by Andrew Bennetts
All tests passing.
138
        :param responses: A list of response-tuple, body-data pairs to be sent
3297.3.3 by Andrew Bennetts
SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.
139
            back to callers.  A special case is if the response-tuple is
140
            'unknown verb', then a UnknownSmartMethod will be raised for that
141
            call, using the second element of the tuple as the verb in the
142
            exception.
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
143
        """
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
144
        self.responses = []
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
145
        self._calls = []
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
146
        self.expecting_body = False
3431.3.2 by Andrew Bennetts
Remove 'base' from _SmartClient entirely, now that the medium has it.
147
        _SmartClient.__init__(self, FakeMedium(self._calls, fake_medium_base))
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
148
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
149
    def add_success_response(self, *args):
150
        self.responses.append(('success', args, None))
151
152
    def add_success_response_with_body(self, body, *args):
153
        self.responses.append(('success', args, body))
154
155
    def add_error_response(self, *args):
156
        self.responses.append(('error', args))
157
158
    def add_unknown_method_response(self, verb):
159
        self.responses.append(('unknown', verb))
160
3297.3.3 by Andrew Bennetts
SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.
161
    def _get_next_response(self):
162
        response_tuple = self.responses.pop(0)
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
163
        if response_tuple[0] == 'unknown':
164
            raise errors.UnknownSmartMethod(response_tuple[1])
165
        elif response_tuple[0] == 'error':
166
            raise errors.ErrorFromSmartServer(response_tuple[1])
3297.3.3 by Andrew Bennetts
SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.
167
        return response_tuple
168
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
169
    def call(self, method, *args):
170
        self._calls.append(('call', method, args))
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
171
        return self._get_next_response()[1]
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
172
2018.5.153 by Andrew Bennetts
Rename call2 to call_expecting_body, and other small changes prompted by review.
173
    def call_expecting_body(self, method, *args):
174
        self._calls.append(('call_expecting_body', method, args))
3297.3.3 by Andrew Bennetts
SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.
175
        result = self._get_next_response()
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
176
        self.expecting_body = True
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
177
        return result[1], FakeProtocol(result[2], self)
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
178
3184.1.10 by Robert Collins
Change the smart server verb for Repository.stream_revisions_chunked to use SearchResults as the request mechanism for downloads.
179
    def call_with_body_bytes_expecting_body(self, method, args, body):
180
        self._calls.append(('call_with_body_bytes_expecting_body', method,
181
            args, body))
3297.3.3 by Andrew Bennetts
SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.
182
        result = self._get_next_response()
3184.1.10 by Robert Collins
Change the smart server verb for Repository.stream_revisions_chunked to use SearchResults as the request mechanism for downloads.
183
        self.expecting_body = True
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
184
        return result[1], FakeProtocol(result[2], self)
3184.1.10 by Robert Collins
Change the smart server verb for Repository.stream_revisions_chunked to use SearchResults as the request mechanism for downloads.
185
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
186
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
187
class FakeMedium(medium.SmartClientMedium):
3104.4.2 by Andrew Bennetts
All tests passing.
188
3431.3.1 by Andrew Bennetts
First rough cut of a fix for bug #230550, by adding .base to SmartClientMedia rather than relying on other objects to track this accurately while reusing client media.
189
    def __init__(self, client_calls, base):
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
190
        medium.SmartClientMedium.__init__(self, base)
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
191
        self._client_calls = client_calls
3213.1.1 by Andrew Bennetts
Recover (by reconnecting) if the server turns out not to understand the new requests in 1.2 that send bodies.
192
193
    def disconnect(self):
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
194
        self._client_calls.append(('disconnect medium',))
3104.4.2 by Andrew Bennetts
All tests passing.
195
196
3192.2.1 by Andrew Bennetts
Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.
197
class TestVfsHas(tests.TestCase):
198
199
    def test_unicode_path(self):
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
200
        client = FakeClient('/')
201
        client.add_success_response('yes',)
3192.2.1 by Andrew Bennetts
Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.
202
        transport = RemoteTransport('bzr://localhost/', _client=client)
203
        filename = u'/hell\u00d8'.encode('utf8')
204
        result = transport.has(filename)
205
        self.assertEqual(
206
            [('call', 'has', (filename,))],
207
            client._calls)
208
        self.assertTrue(result)
209
210
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
211
class Test_ClientMedium_remote_path_from_transport(tests.TestCase):
212
    """Tests for the behaviour of client_medium.remote_path_from_transport."""
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
213
214
    def assertRemotePath(self, expected, client_base, transport_base):
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
215
        """Assert that the result of
216
        SmartClientMedium.remote_path_from_transport is the expected value for
217
        a given client_base and transport_base.
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
218
        """
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
219
        client_medium = medium.SmartClientMedium(client_base)
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
220
        transport = get_transport(transport_base)
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
221
        result = client_medium.remote_path_from_transport(transport)
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
222
        self.assertEqual(expected, result)
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
223
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
224
    def test_remote_path_from_transport(self):
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
225
        """SmartClientMedium.remote_path_from_transport calculates a URL for
226
        the given transport relative to the root of the client base URL.
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
227
        """
228
        self.assertRemotePath('xyz/', 'bzr://host/path', 'bzr://host/xyz')
229
        self.assertRemotePath(
230
            'path/xyz/', 'bzr://host/path', 'bzr://host/path/xyz')
231
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
232
    def assertRemotePathHTTP(self, expected, transport_base, relpath):
233
        """Assert that the result of
234
        HttpTransportBase.remote_path_from_transport is the expected value for
235
        a given transport_base and relpath of that transport.  (Note that
236
        HttpTransportBase is a subclass of SmartClientMedium)
237
        """
238
        base_transport = get_transport(transport_base)
239
        client_medium = base_transport.get_smart_medium()
240
        cloned_transport = base_transport.clone(relpath)
241
        result = client_medium.remote_path_from_transport(cloned_transport)
242
        self.assertEqual(expected, result)
243
        
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
244
    def test_remote_path_from_transport_http(self):
245
        """Remote paths for HTTP transports are calculated differently to other
246
        transports.  They are just relative to the client base, not the root
247
        directory of the host.
248
        """
249
        for scheme in ['http:', 'https:', 'bzr+http:', 'bzr+https:']:
3431.3.11 by Andrew Bennetts
Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.
250
            self.assertRemotePathHTTP(
251
                '../xyz/', scheme + '//host/path', '../xyz/')
252
            self.assertRemotePathHTTP(
253
                'xyz/', scheme + '//host/path', 'xyz/')
3313.3.3 by Andrew Bennetts
Add tests for _SmartClient.remote_path_for_transport.
254
255
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
256
class Test_ClientMedium_remote_is_at_least(tests.TestCase):
257
    """Tests for the behaviour of client_medium.remote_is_at_least."""
258
259
    def test_initially_unlimited(self):
260
        """A fresh medium assumes that the remote side supports all
261
        versions.
262
        """
263
        client_medium = medium.SmartClientMedium('dummy base')
264
        self.assertTrue(client_medium._is_remote_at_least((99, 99)))
265
    
3453.4.9 by Andrew Bennetts
Rename _remote_is_not to _remember_remote_is_before.
266
    def test__remember_remote_is_before(self):
267
        """Calling _remember_remote_is_before ratchets down the known remote
268
        version.
269
        """
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
270
        client_medium = medium.SmartClientMedium('dummy base')
271
        # Mark the remote side as being less than 1.6.  The remote side may
272
        # still be 1.5.
3453.4.9 by Andrew Bennetts
Rename _remote_is_not to _remember_remote_is_before.
273
        client_medium._remember_remote_is_before((1, 6))
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
274
        self.assertFalse(client_medium._is_remote_at_least((1, 6)))
275
        self.assertTrue(client_medium._is_remote_at_least((1, 5)))
3453.4.9 by Andrew Bennetts
Rename _remote_is_not to _remember_remote_is_before.
276
        # Calling _remember_remote_is_before again with a lower value works.
277
        client_medium._remember_remote_is_before((1, 5))
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
278
        self.assertFalse(client_medium._is_remote_at_least((1, 5)))
3453.4.9 by Andrew Bennetts
Rename _remote_is_not to _remember_remote_is_before.
279
        # You cannot call _remember_remote_is_before with a larger value.
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
280
        self.assertRaises(
3453.4.9 by Andrew Bennetts
Rename _remote_is_not to _remember_remote_is_before.
281
            AssertionError, client_medium._remember_remote_is_before, (1, 9))
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
282
283
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
284
class TestBzrDirOpenBranch(tests.TestCase):
285
286
    def test_branch_present(self):
287
        transport = MemoryTransport()
288
        transport.mkdir('quack')
289
        transport = transport.clone('quack')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
290
        client = FakeClient(transport.base)
291
        client.add_success_response('ok', '')
292
        client.add_success_response('ok', '', 'no', 'no', 'no')
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
293
        bzrdir = RemoteBzrDir(transport, _client=client)
294
        result = bzrdir.open_branch()
295
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
296
            [('call', 'BzrDir.open_branch', ('quack/',)),
3221.3.3 by Robert Collins
* Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so
297
             ('call', 'BzrDir.find_repositoryV2', ('quack/',))],
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
298
            client._calls)
299
        self.assertIsInstance(result, RemoteBranch)
300
        self.assertEqual(bzrdir, result.bzrdir)
301
302
    def test_branch_missing(self):
303
        transport = MemoryTransport()
304
        transport.mkdir('quack')
305
        transport = transport.clone('quack')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
306
        client = FakeClient(transport.base)
307
        client.add_error_response('nobranch')
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
308
        bzrdir = RemoteBzrDir(transport, _client=client)
309
        self.assertRaises(errors.NotBranchError, bzrdir.open_branch)
310
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
311
            [('call', 'BzrDir.open_branch', ('quack/',))],
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
312
            client._calls)
313
3211.4.1 by Robert Collins
* ``RemoteBzrDir._get_tree_branch`` no longer triggers ``_ensure_real``,
314
    def test__get_tree_branch(self):
315
        # _get_tree_branch is a form of open_branch, but it should only ask for
316
        # branch opening, not any other network requests.
317
        calls = []
318
        def open_branch():
319
            calls.append("Called")
320
            return "a-branch"
321
        transport = MemoryTransport()
322
        # no requests on the network - catches other api calls being made.
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
323
        client = FakeClient(transport.base)
3211.4.1 by Robert Collins
* ``RemoteBzrDir._get_tree_branch`` no longer triggers ``_ensure_real``,
324
        bzrdir = RemoteBzrDir(transport, _client=client)
325
        # patch the open_branch call to record that it was called.
326
        bzrdir.open_branch = open_branch
327
        self.assertEqual((None, "a-branch"), bzrdir._get_tree_branch())
328
        self.assertEqual(["Called"], calls)
329
        self.assertEqual([], client._calls)
330
3192.2.1 by Andrew Bennetts
Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.
331
    def test_url_quoting_of_path(self):
332
        # Relpaths on the wire should not be URL-escaped.  So "~" should be
333
        # transmitted as "~", not "%7E".
3431.3.1 by Andrew Bennetts
First rough cut of a fix for bug #230550, by adding .base to SmartClientMedia rather than relying on other objects to track this accurately while reusing client media.
334
        transport = RemoteTCPTransport('bzr://localhost/~hello/')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
335
        client = FakeClient(transport.base)
336
        client.add_success_response('ok', '')
337
        client.add_success_response('ok', '', 'no', 'no', 'no')
3192.2.1 by Andrew Bennetts
Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.
338
        bzrdir = RemoteBzrDir(transport, _client=client)
339
        result = bzrdir.open_branch()
340
        self.assertEqual(
341
            [('call', 'BzrDir.open_branch', ('~hello/',)),
3221.3.3 by Robert Collins
* Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so
342
             ('call', 'BzrDir.find_repositoryV2', ('~hello/',))],
3192.2.1 by Andrew Bennetts
Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.
343
            client._calls)
344
3221.3.3 by Robert Collins
* Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so
345
    def check_open_repository(self, rich_root, subtrees, external_lookup='no'):
3104.4.2 by Andrew Bennetts
All tests passing.
346
        transport = MemoryTransport()
347
        transport.mkdir('quack')
348
        transport = transport.clone('quack')
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
349
        if rich_root:
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
350
            rich_response = 'yes'
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
351
        else:
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
352
            rich_response = 'no'
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
353
        if subtrees:
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
354
            subtree_response = 'yes'
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
355
        else:
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
356
            subtree_response = 'no'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
357
        client = FakeClient(transport.base)
358
        client.add_success_response(
359
            'ok', '', rich_response, subtree_response, external_lookup)
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
360
        bzrdir = RemoteBzrDir(transport, _client=client)
361
        result = bzrdir.open_repository()
362
        self.assertEqual(
3221.3.3 by Robert Collins
* Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so
363
            [('call', 'BzrDir.find_repositoryV2', ('quack/',))],
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
364
            client._calls)
365
        self.assertIsInstance(result, RemoteRepository)
366
        self.assertEqual(bzrdir, result.bzrdir)
367
        self.assertEqual(rich_root, result._format.rich_root_data)
2018.5.138 by Robert Collins
Merge bzr.dev.
368
        self.assertEqual(subtrees, result._format.supports_tree_reference)
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
369
370
    def test_open_repository_sets_format_attributes(self):
371
        self.check_open_repository(True, True)
372
        self.check_open_repository(False, True)
373
        self.check_open_repository(True, False)
374
        self.check_open_repository(False, False)
3221.3.3 by Robert Collins
* Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so
375
        self.check_open_repository(False, False, 'yes')
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
376
2432.3.2 by Andrew Bennetts
Add test, and tidy implementation.
377
    def test_old_server(self):
378
        """RemoteBzrDirFormat should fail to probe if the server version is too
379
        old.
380
        """
381
        self.assertRaises(errors.NotBranchError,
382
            RemoteBzrDirFormat.probe_transport, OldServerTransport())
383
384
3297.3.3 by Andrew Bennetts
SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.
385
class TestBzrDirOpenRepository(tests.TestCase):
386
387
    def test_backwards_compat_1_2(self):
388
        transport = MemoryTransport()
389
        transport.mkdir('quack')
390
        transport = transport.clone('quack')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
391
        client = FakeClient(transport.base)
392
        client.add_unknown_method_response('RemoteRepository.find_repositoryV2')
393
        client.add_success_response('ok', '', 'no', 'no')
3297.3.3 by Andrew Bennetts
SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.
394
        bzrdir = RemoteBzrDir(transport, _client=client)
395
        repo = bzrdir.open_repository()
396
        self.assertEqual(
397
            [('call', 'BzrDir.find_repositoryV2', ('quack/',)),
398
             ('call', 'BzrDir.find_repository', ('quack/',))],
399
            client._calls)
400
401
2432.3.2 by Andrew Bennetts
Add test, and tidy implementation.
402
class OldSmartClient(object):
403
    """A fake smart client for test_old_version that just returns a version one
404
    response to the 'hello' (query version) command.
405
    """
406
407
    def get_request(self):
408
        input_file = StringIO('ok\x011\n')
409
        output_file = StringIO()
410
        client_medium = medium.SmartSimplePipesClientMedium(
411
            input_file, output_file)
412
        return medium.SmartClientStreamMediumRequest(client_medium)
413
3241.1.1 by Andrew Bennetts
Shift protocol version querying from RemoteBzrDirFormat into SmartClientMedium.
414
    def protocol_version(self):
415
        return 1
416
2432.3.2 by Andrew Bennetts
Add test, and tidy implementation.
417
418
class OldServerTransport(object):
419
    """A fake transport for test_old_server that reports it's smart server
420
    protocol version as version one.
421
    """
422
423
    def __init__(self):
424
        self.base = 'fake:'
425
426
    def get_smart_client(self):
427
        return OldSmartClient()
428
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
429
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
430
class TestBranchLastRevisionInfo(tests.TestCase):
431
432
    def test_empty_branch(self):
433
        # in an empty branch we decode the response properly
434
        transport = MemoryTransport()
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
435
        client = FakeClient(transport.base)
436
        client.add_success_response('ok', '0', 'null:')
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
437
        transport.mkdir('quack')
438
        transport = transport.clone('quack')
439
        # we do not want bzrdir to make any remote calls
440
        bzrdir = RemoteBzrDir(transport, _client=False)
441
        branch = RemoteBranch(bzrdir, None, _client=client)
442
        result = branch.last_revision_info()
443
444
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
445
            [('call', 'Branch.last_revision_info', ('quack/',))],
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
446
            client._calls)
447
        self.assertEqual((0, NULL_REVISION), result)
448
449
    def test_non_empty_branch(self):
450
        # in a non-empty branch we also decode the response properly
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
451
        revid = u'\xc8'.encode('utf8')
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
452
        transport = MemoryTransport()
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
453
        client = FakeClient(transport.base)
454
        client.add_success_response('ok', '2', revid)
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
455
        transport.mkdir('kwaak')
456
        transport = transport.clone('kwaak')
457
        # we do not want bzrdir to make any remote calls
458
        bzrdir = RemoteBzrDir(transport, _client=False)
459
        branch = RemoteBranch(bzrdir, None, _client=client)
460
        result = branch.last_revision_info()
461
462
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
463
            [('call', 'Branch.last_revision_info', ('kwaak/',))],
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
464
            client._calls)
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
465
        self.assertEqual((2, revid), result)
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
466
467
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
468
class TestBranchSetLastRevision(tests.TestCase):
469
470
    def test_set_empty(self):
471
        # set_revision_history([]) is translated to calling
472
        # Branch.set_last_revision(path, '') on the wire.
3104.4.2 by Andrew Bennetts
All tests passing.
473
        transport = MemoryTransport()
474
        transport.mkdir('branch')
475
        transport = transport.clone('branch')
476
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
477
        client = FakeClient(transport.base)
478
        # lock_write
479
        client.add_success_response('ok', 'branch token', 'repo token')
480
        # set_last_revision
481
        client.add_success_response('ok')
482
        # unlock
483
        client.add_success_response('ok')
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
484
        bzrdir = RemoteBzrDir(transport, _client=False)
485
        branch = RemoteBranch(bzrdir, None, _client=client)
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
486
        # This is a hack to work around the problem that RemoteBranch currently
487
        # unnecessarily invokes _ensure_real upon a call to lock_write.
488
        branch._ensure_real = lambda: None
489
        branch.lock_write()
490
        client._calls = []
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
491
        result = branch.set_revision_history([])
492
        self.assertEqual(
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
493
            [('call', 'Branch.set_last_revision',
3104.4.2 by Andrew Bennetts
All tests passing.
494
                ('branch/', 'branch token', 'repo token', 'null:'))],
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
495
            client._calls)
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
496
        branch.unlock()
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
497
        self.assertEqual(None, result)
498
499
    def test_set_nonempty(self):
500
        # set_revision_history([rev-id1, ..., rev-idN]) is translated to calling
501
        # Branch.set_last_revision(path, rev-idN) on the wire.
3104.4.2 by Andrew Bennetts
All tests passing.
502
        transport = MemoryTransport()
503
        transport.mkdir('branch')
504
        transport = transport.clone('branch')
505
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
506
        client = FakeClient(transport.base)
507
        # lock_write
508
        client.add_success_response('ok', 'branch token', 'repo token')
509
        # set_last_revision
510
        client.add_success_response('ok')
511
        # unlock
512
        client.add_success_response('ok')
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
513
        bzrdir = RemoteBzrDir(transport, _client=False)
514
        branch = RemoteBranch(bzrdir, None, _client=client)
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
515
        # This is a hack to work around the problem that RemoteBranch currently
516
        # unnecessarily invokes _ensure_real upon a call to lock_write.
517
        branch._ensure_real = lambda: None
518
        # Lock the branch, reset the record of remote calls.
519
        branch.lock_write()
520
        client._calls = []
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
521
522
        result = branch.set_revision_history(['rev-id1', 'rev-id2'])
523
        self.assertEqual(
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
524
            [('call', 'Branch.set_last_revision',
3104.4.2 by Andrew Bennetts
All tests passing.
525
                ('branch/', 'branch token', 'repo token', 'rev-id2'))],
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
526
            client._calls)
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
527
        branch.unlock()
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
528
        self.assertEqual(None, result)
529
530
    def test_no_such_revision(self):
531
        transport = MemoryTransport()
532
        transport.mkdir('branch')
533
        transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
534
        # A response of 'NoSuchRevision' is translated into an exception.
535
        client = FakeClient(transport.base)
536
        # lock_write
537
        client.add_success_response('ok', 'branch token', 'repo token')
538
        # set_last_revision
539
        client.add_error_response('NoSuchRevision', 'rev-id')
540
        # unlock
541
        client.add_success_response('ok')
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
542
543
        bzrdir = RemoteBzrDir(transport, _client=False)
544
        branch = RemoteBranch(bzrdir, None, _client=client)
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
545
        branch._ensure_real = lambda: None
546
        branch.lock_write()
547
        client._calls = []
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
548
549
        self.assertRaises(
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
550
            errors.NoSuchRevision, branch.set_revision_history, ['rev-id'])
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
551
        branch.unlock()
2018.12.3 by Andrew Bennetts
Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.
552
553
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
554
class TestBranchSetLastRevisionInfo(tests.TestCase):
555
3297.4.2 by Andrew Bennetts
Add backwards compatibility for servers older than 1.4.
556
    def test_set_last_revision_info(self):
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
557
        # set_last_revision_info(num, 'rev-id') is translated to calling
558
        # Branch.set_last_revision_info(num, 'rev-id') on the wire.
3297.4.1 by Andrew Bennetts
Merge 'Add Branch.set_last_revision_info smart method'.
559
        transport = MemoryTransport()
560
        transport.mkdir('branch')
561
        transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
562
        client = FakeClient(transport.base)
563
        # lock_write
564
        client.add_success_response('ok', 'branch token', 'repo token')
565
        # set_last_revision
566
        client.add_success_response('ok')
567
        # unlock
568
        client.add_success_response('ok')
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
569
570
        bzrdir = RemoteBzrDir(transport, _client=False)
571
        branch = RemoteBranch(bzrdir, None, _client=client)
572
        # This is a hack to work around the problem that RemoteBranch currently
573
        # unnecessarily invokes _ensure_real upon a call to lock_write.
574
        branch._ensure_real = lambda: None
575
        # Lock the branch, reset the record of remote calls.
576
        branch.lock_write()
577
        client._calls = []
578
        result = branch.set_last_revision_info(1234, 'a-revision-id')
579
        self.assertEqual(
580
            [('call', 'Branch.set_last_revision_info',
3297.4.1 by Andrew Bennetts
Merge 'Add Branch.set_last_revision_info smart method'.
581
                ('branch/', 'branch token', 'repo token',
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
582
                 '1234', 'a-revision-id'))],
583
            client._calls)
584
        self.assertEqual(None, result)
585
586
    def test_no_such_revision(self):
587
        # A response of 'NoSuchRevision' is translated into an exception.
588
        transport = MemoryTransport()
589
        transport.mkdir('branch')
590
        transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
591
        client = FakeClient(transport.base)
592
        # lock_write
593
        client.add_success_response('ok', 'branch token', 'repo token')
594
        # set_last_revision
595
        client.add_error_response('NoSuchRevision', 'revid')
596
        # unlock
597
        client.add_success_response('ok')
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
598
599
        bzrdir = RemoteBzrDir(transport, _client=False)
600
        branch = RemoteBranch(bzrdir, None, _client=client)
601
        # This is a hack to work around the problem that RemoteBranch currently
602
        # unnecessarily invokes _ensure_real upon a call to lock_write.
603
        branch._ensure_real = lambda: None
604
        # Lock the branch, reset the record of remote calls.
605
        branch.lock_write()
606
        client._calls = []
607
608
        self.assertRaises(
609
            errors.NoSuchRevision, branch.set_last_revision_info, 123, 'revid')
610
        branch.unlock()
611
3297.4.2 by Andrew Bennetts
Add backwards compatibility for servers older than 1.4.
612
    def lock_remote_branch(self, branch):
613
        """Trick a RemoteBranch into thinking it is locked."""
614
        branch._lock_mode = 'w'
615
        branch._lock_count = 2
616
        branch._lock_token = 'branch token'
617
        branch._repo_lock_token = 'repo token'
618
619
    def test_backwards_compatibility(self):
620
        """If the server does not support the Branch.set_last_revision_info
621
        verb (which is new in 1.4), then the client falls back to VFS methods.
622
        """
623
        # This test is a little messy.  Unlike most tests in this file, it
624
        # doesn't purely test what a Remote* object sends over the wire, and
625
        # how it reacts to responses from the wire.  It instead relies partly
626
        # on asserting that the RemoteBranch will call
627
        # self._real_branch.set_last_revision_info(...).
628
629
        # First, set up our RemoteBranch with a FakeClient that raises
630
        # UnknownSmartMethod, and a StubRealBranch that logs how it is called.
631
        transport = MemoryTransport()
632
        transport.mkdir('branch')
633
        transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
634
        client = FakeClient(transport.base)
635
        client.add_unknown_method_response('Branch.set_last_revision_info')
3297.4.2 by Andrew Bennetts
Add backwards compatibility for servers older than 1.4.
636
        bzrdir = RemoteBzrDir(transport, _client=False)
637
        branch = RemoteBranch(bzrdir, None, _client=client)
638
        class StubRealBranch(object):
639
            def __init__(self):
640
                self.calls = []
641
            def set_last_revision_info(self, revno, revision_id):
642
                self.calls.append(
643
                    ('set_last_revision_info', revno, revision_id))
644
        real_branch = StubRealBranch()
645
        branch._real_branch = real_branch
646
        self.lock_remote_branch(branch)
647
648
        # Call set_last_revision_info, and verify it behaved as expected.
649
        result = branch.set_last_revision_info(1234, 'a-revision-id')
650
        self.assertEqual(
651
            [('call', 'Branch.set_last_revision_info',
652
                ('branch/', 'branch token', 'repo token',
653
                 '1234', 'a-revision-id')),],
654
            client._calls)
655
        self.assertEqual(
656
            [('set_last_revision_info', 1234, 'a-revision-id')],
657
            real_branch.calls)
658
3245.4.53 by Andrew Bennetts
Add some missing 'raise' statements to test_remote.
659
    def test_unexpected_error(self):
660
        # A response of 'NoSuchRevision' is translated into an exception.
661
        transport = MemoryTransport()
662
        transport.mkdir('branch')
663
        transport = transport.clone('branch')
664
        client = FakeClient(transport.base)
665
        # lock_write
666
        client.add_success_response('ok', 'branch token', 'repo token')
667
        # set_last_revision
668
        client.add_error_response('UnexpectedError')
669
        # unlock
670
        client.add_success_response('ok')
671
672
        bzrdir = RemoteBzrDir(transport, _client=False)
673
        branch = RemoteBranch(bzrdir, None, _client=client)
674
        # This is a hack to work around the problem that RemoteBranch currently
675
        # unnecessarily invokes _ensure_real upon a call to lock_write.
676
        branch._ensure_real = lambda: None
677
        # Lock the branch, reset the record of remote calls.
678
        branch.lock_write()
679
        client._calls = []
680
681
        err = self.assertRaises(
682
            errors.ErrorFromSmartServer,
683
            branch.set_last_revision_info, 123, 'revid')
684
        self.assertEqual(('UnexpectedError',), err.error_tuple)
685
        branch.unlock()
686
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
687
2018.5.169 by Andrew Bennetts
Add a _server_formats flag to BzrDir.open_from_transport and BzrDirFormat.find_format, make RemoteBranch.control_files into a property.
688
class TestBranchControlGetBranchConf(tests.TestCaseWithMemoryTransport):
3408.3.1 by Martin Pool
Remove erroneous handling of branch.conf for RemoteBranch
689
    """Getting the branch configuration should use an abstract method not vfs.
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
690
    """
691
692
    def test_get_branch_conf(self):
3408.3.1 by Martin Pool
Remove erroneous handling of branch.conf for RemoteBranch
693
        raise tests.KnownFailure('branch.conf is not retrieved by get_config_file')
3407.2.10 by Martin Pool
Merge trunk
694
        ## # We should see that branch.get_config() does a single rpc to get the
695
        ## # remote configuration file, abstracting away where that is stored on
696
        ## # the server.  However at the moment it always falls back to using the
697
        ## # vfs, and this would need some changes in config.py.
3408.3.1 by Martin Pool
Remove erroneous handling of branch.conf for RemoteBranch
698
3407.2.10 by Martin Pool
Merge trunk
699
        ## # in an empty branch we decode the response properly
700
        ## client = FakeClient([(('ok', ), '# config file body')], self.get_url())
701
        ## # we need to make a real branch because the remote_branch.control_files
702
        ## # will trigger _ensure_real.
703
        ## branch = self.make_branch('quack')
704
        ## transport = branch.bzrdir.root_transport
705
        ## # we do not want bzrdir to make any remote calls
706
        ## bzrdir = RemoteBzrDir(transport, _client=False)
707
        ## branch = RemoteBranch(bzrdir, None, _client=client)
708
        ## config = branch.get_config()
709
        ## self.assertEqual(
710
        ##     [('call_expecting_body', 'Branch.get_config_file', ('quack/',))],
711
        ##     client._calls)
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
712
713
2018.5.95 by Andrew Bennetts
Add a Transport.is_readonly remote call, let {Branch,Repository}.lock_write remote call return UnlockableTransport, and miscellaneous test fixes.
714
class TestBranchLockWrite(tests.TestCase):
715
716
    def test_lock_write_unlockable(self):
717
        transport = MemoryTransport()
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
718
        client = FakeClient(transport.base)
719
        client.add_error_response('UnlockableTransport')
2018.5.95 by Andrew Bennetts
Add a Transport.is_readonly remote call, let {Branch,Repository}.lock_write remote call return UnlockableTransport, and miscellaneous test fixes.
720
        transport.mkdir('quack')
721
        transport = transport.clone('quack')
722
        # we do not want bzrdir to make any remote calls
723
        bzrdir = RemoteBzrDir(transport, _client=False)
724
        branch = RemoteBranch(bzrdir, None, _client=client)
725
        self.assertRaises(errors.UnlockableTransport, branch.lock_write)
726
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
727
            [('call', 'Branch.lock_write', ('quack/', '', ''))],
2018.5.95 by Andrew Bennetts
Add a Transport.is_readonly remote call, let {Branch,Repository}.lock_write remote call return UnlockableTransport, and miscellaneous test fixes.
728
            client._calls)
729
730
2466.2.2 by Andrew Bennetts
Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.
731
class TestTransportIsReadonly(tests.TestCase):
732
733
    def test_true(self):
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
734
        client = FakeClient()
735
        client.add_success_response('yes')
2466.2.2 by Andrew Bennetts
Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.
736
        transport = RemoteTransport('bzr://example.com/', medium=False,
737
                                    _client=client)
738
        self.assertEqual(True, transport.is_readonly())
739
        self.assertEqual(
740
            [('call', 'Transport.is_readonly', ())],
741
            client._calls)
742
743
    def test_false(self):
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
744
        client = FakeClient()
745
        client.add_success_response('no')
2466.2.2 by Andrew Bennetts
Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.
746
        transport = RemoteTransport('bzr://example.com/', medium=False,
747
                                    _client=client)
748
        self.assertEqual(False, transport.is_readonly())
749
        self.assertEqual(
750
            [('call', 'Transport.is_readonly', ())],
751
            client._calls)
752
753
    def test_error_from_old_server(self):
754
        """bzr 0.15 and earlier servers don't recognise the is_readonly verb.
755
        
756
        Clients should treat it as a "no" response, because is_readonly is only
757
        advisory anyway (a transport could be read-write, but then the
758
        underlying filesystem could be readonly anyway).
759
        """
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
760
        client = FakeClient()
761
        client.add_unknown_method_response('Transport.is_readonly')
2471.2.1 by Andrew Bennetts
Fix trivial incompatibility with bzr 0.11 servers, which give a slightly different error to bzr 0.15 servers.
762
        transport = RemoteTransport('bzr://example.com/', medium=False,
763
                                    _client=client)
764
        self.assertEqual(False, transport.is_readonly())
765
        self.assertEqual(
766
            [('call', 'Transport.is_readonly', ())],
767
            client._calls)
768
2466.2.2 by Andrew Bennetts
Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.
769
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
770
class TestRemoteRepository(tests.TestCase):
2018.18.9 by Martin Pool
remote Repository.tarball builds a temporary directory and tars that
771
    """Base for testing RemoteRepository protocol usage.
772
    
773
    These tests contain frozen requests and responses.  We want any changes to 
774
    what is sent or expected to be require a thoughtful update to these tests
775
    because they might break compatibility with different-versioned servers.
776
    """
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
777
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
778
    def setup_fake_client_and_repository(self, transport_path):
2018.18.7 by Martin Pool
(broken) Start addng client proxy test for Repository.tarball
779
        """Create the fake client and repository for testing with.
780
        
781
        There's no real server here; we just have canned responses sent
782
        back one by one.
783
        
784
        :param transport_path: Path below the root of the MemoryTransport
785
            where the repository will be created.
786
        """
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
787
        transport = MemoryTransport()
788
        transport.mkdir(transport_path)
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
789
        client = FakeClient(transport.base)
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
790
        transport = transport.clone(transport_path)
791
        # we do not want bzrdir to make any remote calls
792
        bzrdir = RemoteBzrDir(transport, _client=False)
793
        repo = RemoteRepository(bzrdir, None, _client=client)
794
        return repo, client
795
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
796
2018.12.2 by Andrew Bennetts
Remove some duplicate code in test_remote
797
class TestRepositoryGatherStats(TestRemoteRepository):
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
798
799
    def test_revid_none(self):
800
        # ('ok',), body with revisions and size
801
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
802
        repo, client = self.setup_fake_client_and_repository(transport_path)
803
        client.add_success_response_with_body(
804
            'revisions: 2\nsize: 18\n', 'ok')
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
805
        result = repo.gather_stats(None)
806
        self.assertEqual(
2018.5.153 by Andrew Bennetts
Rename call2 to call_expecting_body, and other small changes prompted by review.
807
            [('call_expecting_body', 'Repository.gather_stats',
3104.4.2 by Andrew Bennetts
All tests passing.
808
             ('quack/','','no'))],
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
809
            client._calls)
810
        self.assertEqual({'revisions': 2, 'size': 18}, result)
811
812
    def test_revid_no_committers(self):
813
        # ('ok',), body without committers
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
814
        body = ('firstrev: 123456.300 3600\n'
815
                'latestrev: 654231.400 0\n'
816
                'revisions: 2\n'
817
                'size: 18\n')
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
818
        transport_path = 'quick'
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
819
        revid = u'\xc8'.encode('utf8')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
820
        repo, client = self.setup_fake_client_and_repository(transport_path)
821
        client.add_success_response_with_body(body, 'ok')
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
822
        result = repo.gather_stats(revid)
823
        self.assertEqual(
2018.5.153 by Andrew Bennetts
Rename call2 to call_expecting_body, and other small changes prompted by review.
824
            [('call_expecting_body', 'Repository.gather_stats',
3104.4.2 by Andrew Bennetts
All tests passing.
825
              ('quick/', revid, 'no'))],
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
826
            client._calls)
827
        self.assertEqual({'revisions': 2, 'size': 18,
828
                          'firstrev': (123456.300, 3600),
829
                          'latestrev': (654231.400, 0),},
830
                         result)
831
832
    def test_revid_with_committers(self):
833
        # ('ok',), body with committers
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
834
        body = ('committers: 128\n'
835
                'firstrev: 123456.300 3600\n'
836
                'latestrev: 654231.400 0\n'
837
                'revisions: 2\n'
838
                'size: 18\n')
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
839
        transport_path = 'buick'
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
840
        revid = u'\xc8'.encode('utf8')
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
841
        repo, client = self.setup_fake_client_and_repository(transport_path)
842
        client.add_success_response_with_body(body, 'ok')
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
843
        result = repo.gather_stats(revid, True)
844
        self.assertEqual(
2018.5.153 by Andrew Bennetts
Rename call2 to call_expecting_body, and other small changes prompted by review.
845
            [('call_expecting_body', 'Repository.gather_stats',
3104.4.2 by Andrew Bennetts
All tests passing.
846
              ('buick/', revid, 'yes'))],
2018.10.3 by v.ladeuil+lp at free
more tests for gather_stats
847
            client._calls)
848
        self.assertEqual({'revisions': 2, 'size': 18,
849
                          'committers': 128,
850
                          'firstrev': (123456.300, 3600),
851
                          'latestrev': (654231.400, 0),},
852
                         result)
853
854
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
855
class TestRepositoryGetGraph(TestRemoteRepository):
856
857
    def test_get_graph(self):
3172.5.8 by Robert Collins
Review feedback.
858
        # get_graph returns a graph with the repository as the
859
        # parents_provider.
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
860
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
861
        repo, client = self.setup_fake_client_and_repository(transport_path)
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
862
        graph = repo.get_graph()
863
        self.assertEqual(graph._parents_provider, repo)
864
865
866
class TestRepositoryGetParentMap(TestRemoteRepository):
867
868
    def test_get_parent_map_caching(self):
869
        # get_parent_map returns from cache until unlock()
870
        # setup a reponse with two revisions
871
        r1 = u'\u0e33'.encode('utf8')
872
        r2 = u'\u0dab'.encode('utf8')
873
        lines = [' '.join([r2, r1]), r1]
3211.5.2 by Robert Collins
Change RemoteRepository.get_parent_map to use bz2 not gzip for compression.
874
        encoded_body = bz2.compress('\n'.join(lines))
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
875
876
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
877
        repo, client = self.setup_fake_client_and_repository(transport_path)
878
        client.add_success_response_with_body(encoded_body, 'ok')
879
        client.add_success_response_with_body(encoded_body, 'ok')
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
880
        repo.lock_read()
881
        graph = repo.get_graph()
882
        parents = graph.get_parent_map([r2])
883
        self.assertEqual({r2: (r1,)}, parents)
884
        # locking and unlocking deeper should not reset
885
        repo.lock_read()
886
        repo.unlock()
887
        parents = graph.get_parent_map([r1])
3172.5.6 by Robert Collins
Create new smart server verb Repository.get_parent_map.
888
        self.assertEqual({r1: (NULL_REVISION,)}, parents)
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
889
        self.assertEqual(
3211.5.1 by Robert Collins
Change the smart server get_parents method to take a graph search to exclude already recieved parents from. This prevents history shortcuts causing huge numbers of duplicates.
890
            [('call_with_body_bytes_expecting_body',
891
              'Repository.get_parent_map', ('quack/', r2), '\n\n0')],
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
892
            client._calls)
893
        repo.unlock()
894
        # now we call again, and it should use the second response.
895
        repo.lock_read()
896
        graph = repo.get_graph()
3172.5.6 by Robert Collins
Create new smart server verb Repository.get_parent_map.
897
        parents = graph.get_parent_map([r1])
898
        self.assertEqual({r1: (NULL_REVISION,)}, parents)
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
899
        self.assertEqual(
3211.5.1 by Robert Collins
Change the smart server get_parents method to take a graph search to exclude already recieved parents from. This prevents history shortcuts causing huge numbers of duplicates.
900
            [('call_with_body_bytes_expecting_body',
901
              'Repository.get_parent_map', ('quack/', r2), '\n\n0'),
902
             ('call_with_body_bytes_expecting_body',
903
              'Repository.get_parent_map', ('quack/', r1), '\n\n0'),
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
904
            ],
905
            client._calls)
906
        repo.unlock()
907
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
908
    def test_get_parent_map_reconnects_if_unknown_method(self):
909
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
910
        repo, client = self.setup_fake_client_and_repository(transport_path)
911
        client.add_unknown_method_response('Repository,get_parent_map')
912
        client.add_success_response_with_body('', 'ok')
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
913
        self.assertTrue(client._medium._is_remote_at_least((1, 2)))
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
914
        rev_id = 'revision-id'
3297.3.5 by Andrew Bennetts
Suppress a deprecation warning.
915
        expected_deprecations = [
916
            'bzrlib.remote.RemoteRepository.get_revision_graph was deprecated '
917
            'in version 1.4.']
918
        parents = self.callDeprecated(
919
            expected_deprecations, repo.get_parent_map, [rev_id])
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
920
        self.assertEqual(
3213.1.8 by Andrew Bennetts
Merge from bzr.dev.
921
            [('call_with_body_bytes_expecting_body',
922
              'Repository.get_parent_map', ('quack/', rev_id), '\n\n0'),
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
923
             ('disconnect medium',),
924
             ('call_expecting_body', 'Repository.get_revision_graph',
925
              ('quack/', ''))],
926
            client._calls)
3389.1.2 by Andrew Bennetts
Add test for the bug John found.
927
        # The medium is now marked as being connected to an older server
3453.4.1 by Andrew Bennetts
Better infrastructure on SmartClientMedium for tracking the remote version.
928
        self.assertFalse(client._medium._is_remote_at_least((1, 2)))
3389.1.2 by Andrew Bennetts
Add test for the bug John found.
929
930
    def test_get_parent_map_fallback_parentless_node(self):
931
        """get_parent_map falls back to get_revision_graph on old servers.  The
932
        results from get_revision_graph are tweaked to match the get_parent_map
933
        API.
934
3389.1.3 by Andrew Bennetts
Remove XXX from test description.
935
        Specifically, a {key: ()} result from get_revision_graph means "no
3389.1.2 by Andrew Bennetts
Add test for the bug John found.
936
        parents" for that key, which in get_parent_map results should be
3389.1.3 by Andrew Bennetts
Remove XXX from test description.
937
        represented as {key: ('null:',)}.
3389.1.2 by Andrew Bennetts
Add test for the bug John found.
938
939
        This is the test for https://bugs.launchpad.net/bzr/+bug/214894
940
        """
941
        rev_id = 'revision-id'
942
        transport_path = 'quack'
3245.4.40 by Andrew Bennetts
Merge from bzr.dev.
943
        repo, client = self.setup_fake_client_and_repository(transport_path)
944
        client.add_success_response_with_body(rev_id, 'ok')
3453.4.9 by Andrew Bennetts
Rename _remote_is_not to _remember_remote_is_before.
945
        client._medium._remember_remote_is_before((1, 2))
3389.1.2 by Andrew Bennetts
Add test for the bug John found.
946
        expected_deprecations = [
947
            'bzrlib.remote.RemoteRepository.get_revision_graph was deprecated '
948
            'in version 1.4.']
949
        parents = self.callDeprecated(
950
            expected_deprecations, repo.get_parent_map, [rev_id])
951
        self.assertEqual(
952
            [('call_expecting_body', 'Repository.get_revision_graph',
953
             ('quack/', ''))],
954
            client._calls)
955
        self.assertEqual({rev_id: ('null:',)}, parents)
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
956
3297.2.3 by Andrew Bennetts
Test the code path that the typo is on.
957
    def test_get_parent_map_unexpected_response(self):
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
958
        repo, client = self.setup_fake_client_and_repository('path')
959
        client.add_success_response('something unexpected!')
3297.2.3 by Andrew Bennetts
Test the code path that the typo is on.
960
        self.assertRaises(
961
            errors.UnexpectedSmartServerResponse,
962
            repo.get_parent_map, ['a-revision-id'])
3213.1.2 by Andrew Bennetts
Add test for reconnection if get_parent_map is unknown by the server.
963
3172.5.4 by Robert Collins
Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.
964
2018.5.68 by Wouter van Heyst
Merge RemoteRepository.gather_stats.
965
class TestRepositoryGetRevisionGraph(TestRemoteRepository):
966
    
967
    def test_null_revision(self):
968
        # a null revision has the predictable result {}, we should have no wire
969
        # traffic when calling it with this argument
970
        transport_path = 'empty'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
971
        repo, client = self.setup_fake_client_and_repository(transport_path)
972
        client.add_success_response('notused')
3287.6.4 by Robert Collins
Fix up deprecation warnings for get_revision_graph.
973
        result = self.applyDeprecated(one_four, repo.get_revision_graph,
974
            NULL_REVISION)
2018.5.68 by Wouter van Heyst
Merge RemoteRepository.gather_stats.
975
        self.assertEqual([], client._calls)
976
        self.assertEqual({}, result)
977
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
978
    def test_none_revision(self):
979
        # with none we want the entire graph
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
980
        r1 = u'\u0e33'.encode('utf8')
981
        r2 = u'\u0dab'.encode('utf8')
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
982
        lines = [' '.join([r2, r1]), r1]
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
983
        encoded_body = '\n'.join(lines)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
984
985
        transport_path = 'sinhala'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
986
        repo, client = self.setup_fake_client_and_repository(transport_path)
987
        client.add_success_response_with_body(encoded_body, 'ok')
3287.6.4 by Robert Collins
Fix up deprecation warnings for get_revision_graph.
988
        result = self.applyDeprecated(one_four, repo.get_revision_graph)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
989
        self.assertEqual(
2018.5.153 by Andrew Bennetts
Rename call2 to call_expecting_body, and other small changes prompted by review.
990
            [('call_expecting_body', 'Repository.get_revision_graph',
3104.4.2 by Andrew Bennetts
All tests passing.
991
             ('sinhala/', ''))],
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
992
            client._calls)
2625.8.1 by Robert Collins
LIBRARY API BREAKS:
993
        self.assertEqual({r1: (), r2: (r1, )}, result)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
994
995
    def test_specific_revision(self):
996
        # with a specific revision we want the graph for that
997
        # with none we want the entire graph
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
998
        r11 = u'\u0e33'.encode('utf8')
999
        r12 = u'\xc9'.encode('utf8')
1000
        r2 = u'\u0dab'.encode('utf8')
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1001
        lines = [' '.join([r2, r11, r12]), r11, r12]
2018.5.106 by Andrew Bennetts
Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.
1002
        encoded_body = '\n'.join(lines)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1003
1004
        transport_path = 'sinhala'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1005
        repo, client = self.setup_fake_client_and_repository(transport_path)
1006
        client.add_success_response_with_body(encoded_body, 'ok')
3287.6.4 by Robert Collins
Fix up deprecation warnings for get_revision_graph.
1007
        result = self.applyDeprecated(one_four, repo.get_revision_graph, r2)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1008
        self.assertEqual(
2018.5.153 by Andrew Bennetts
Rename call2 to call_expecting_body, and other small changes prompted by review.
1009
            [('call_expecting_body', 'Repository.get_revision_graph',
3104.4.2 by Andrew Bennetts
All tests passing.
1010
             ('sinhala/', r2))],
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1011
            client._calls)
2625.8.1 by Robert Collins
LIBRARY API BREAKS:
1012
        self.assertEqual({r11: (), r12: (), r2: (r11, r12), }, result)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1013
1014
    def test_no_such_revision(self):
1015
        revid = '123'
1016
        transport_path = 'sinhala'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1017
        repo, client = self.setup_fake_client_and_repository(transport_path)
1018
        client.add_error_response('nosuchrevision', revid)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1019
        # also check that the right revision is reported in the error
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1020
        self.assertRaises(errors.NoSuchRevision,
3287.6.4 by Robert Collins
Fix up deprecation warnings for get_revision_graph.
1021
            self.applyDeprecated, one_four, repo.get_revision_graph, revid)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1022
        self.assertEqual(
2018.5.153 by Andrew Bennetts
Rename call2 to call_expecting_body, and other small changes prompted by review.
1023
            [('call_expecting_body', 'Repository.get_revision_graph',
3104.4.2 by Andrew Bennetts
All tests passing.
1024
             ('sinhala/', revid))],
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1025
            client._calls)
1026
3245.4.53 by Andrew Bennetts
Add some missing 'raise' statements to test_remote.
1027
    def test_unexpected_error(self):
1028
        revid = '123'
1029
        transport_path = 'sinhala'
1030
        repo, client = self.setup_fake_client_and_repository(transport_path)
1031
        client.add_error_response('AnUnexpectedError')
1032
        e = self.assertRaises(errors.ErrorFromSmartServer,
1033
            self.applyDeprecated, one_four, repo.get_revision_graph, revid)
1034
        self.assertEqual(('AnUnexpectedError',), e.error_tuple)
1035
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1036
        
1037
class TestRepositoryIsShared(TestRemoteRepository):
1038
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1039
    def test_is_shared(self):
1040
        # ('yes', ) for Repository.is_shared -> 'True'.
1041
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1042
        repo, client = self.setup_fake_client_and_repository(transport_path)
1043
        client.add_success_response('yes')
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1044
        result = repo.is_shared()
1045
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
1046
            [('call', 'Repository.is_shared', ('quack/',))],
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1047
            client._calls)
1048
        self.assertEqual(True, result)
1049
1050
    def test_is_not_shared(self):
1051
        # ('no', ) for Repository.is_shared -> 'False'.
1052
        transport_path = 'qwack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1053
        repo, client = self.setup_fake_client_and_repository(transport_path)
1054
        client.add_success_response('no')
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1055
        result = repo.is_shared()
1056
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
1057
            [('call', 'Repository.is_shared', ('qwack/',))],
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1058
            client._calls)
1059
        self.assertEqual(False, result)
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1060
1061
1062
class TestRepositoryLockWrite(TestRemoteRepository):
1063
1064
    def test_lock_write(self):
1065
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1066
        repo, client = self.setup_fake_client_and_repository(transport_path)
1067
        client.add_success_response('ok', 'a token')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1068
        result = repo.lock_write()
1069
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
1070
            [('call', 'Repository.lock_write', ('quack/', ''))],
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1071
            client._calls)
1072
        self.assertEqual('a token', result)
1073
1074
    def test_lock_write_already_locked(self):
1075
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1076
        repo, client = self.setup_fake_client_and_repository(transport_path)
1077
        client.add_error_response('LockContention')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1078
        self.assertRaises(errors.LockContention, repo.lock_write)
1079
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
1080
            [('call', 'Repository.lock_write', ('quack/', ''))],
2018.5.95 by Andrew Bennetts
Add a Transport.is_readonly remote call, let {Branch,Repository}.lock_write remote call return UnlockableTransport, and miscellaneous test fixes.
1081
            client._calls)
1082
1083
    def test_lock_write_unlockable(self):
1084
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1085
        repo, client = self.setup_fake_client_and_repository(transport_path)
1086
        client.add_error_response('UnlockableTransport')
2018.5.95 by Andrew Bennetts
Add a Transport.is_readonly remote call, let {Branch,Repository}.lock_write remote call return UnlockableTransport, and miscellaneous test fixes.
1087
        self.assertRaises(errors.UnlockableTransport, repo.lock_write)
1088
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
1089
            [('call', 'Repository.lock_write', ('quack/', ''))],
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1090
            client._calls)
1091
1092
1093
class TestRepositoryUnlock(TestRemoteRepository):
1094
1095
    def test_unlock(self):
1096
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1097
        repo, client = self.setup_fake_client_and_repository(transport_path)
1098
        client.add_success_response('ok', 'a token')
1099
        client.add_success_response('ok')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1100
        repo.lock_write()
1101
        repo.unlock()
1102
        self.assertEqual(
3104.4.2 by Andrew Bennetts
All tests passing.
1103
            [('call', 'Repository.lock_write', ('quack/', '')),
1104
             ('call', 'Repository.unlock', ('quack/', 'a token'))],
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1105
            client._calls)
1106
1107
    def test_unlock_wrong_token(self):
1108
        # If somehow the token is wrong, unlock will raise TokenMismatch.
1109
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1110
        repo, client = self.setup_fake_client_and_repository(transport_path)
1111
        client.add_success_response('ok', 'a token')
1112
        client.add_error_response('TokenMismatch')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1113
        repo.lock_write()
1114
        self.assertRaises(errors.TokenMismatch, repo.unlock)
1115
1116
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
1117
class TestRepositoryHasRevision(TestRemoteRepository):
1118
1119
    def test_none(self):
1120
        # repo.has_revision(None) should not cause any traffic.
1121
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1122
        repo, client = self.setup_fake_client_and_repository(transport_path)
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
1123
1124
        # The null revision is always there, so has_revision(None) == True.
3172.3.3 by Robert Collins
Missed one occurence of None -> NULL_REVISION.
1125
        self.assertEqual(True, repo.has_revision(NULL_REVISION))
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
1126
1127
        # The remote repo shouldn't be accessed.
1128
        self.assertEqual([], client._calls)
2018.18.7 by Martin Pool
(broken) Start addng client proxy test for Repository.tarball
1129
1130
1131
class TestRepositoryTarball(TestRemoteRepository):
1132
2018.18.9 by Martin Pool
remote Repository.tarball builds a temporary directory and tars that
1133
    # This is a canned tarball reponse we can validate against
2018.18.18 by Martin Pool
reformat
1134
    tarball_content = (
2018.18.23 by Martin Pool
review cleanups
1135
        'QlpoOTFBWSZTWdGkj3wAAWF/k8aQACBIB//A9+8cIX/v33AACEAYABAECEACNz'
1136
        'JqsgJJFPTSnk1A3qh6mTQAAAANPUHkagkSTEkaA09QaNAAAGgAAAcwCYCZGAEY'
1137
        'mJhMJghpiaYBUkKammSHqNMZQ0NABkNAeo0AGneAevnlwQoGzEzNVzaYxp/1Uk'
1138
        'xXzA1CQX0BJMZZLcPBrluJir5SQyijWHYZ6ZUtVqqlYDdB2QoCwa9GyWwGYDMA'
1139
        'OQYhkpLt/OKFnnlT8E0PmO8+ZNSo2WWqeCzGB5fBXZ3IvV7uNJVE7DYnWj6qwB'
1140
        'k5DJDIrQ5OQHHIjkS9KqwG3mc3t+F1+iujb89ufyBNIKCgeZBWrl5cXxbMGoMs'
1141
        'c9JuUkg5YsiVcaZJurc6KLi6yKOkgCUOlIlOpOoXyrTJjK8ZgbklReDdwGmFgt'
1142
        'dkVsAIslSVCd4AtACSLbyhLHryfb14PKegrVDba+U8OL6KQtzdM5HLjAc8/p6n'
1143
        '0lgaWU8skgO7xupPTkyuwheSckejFLK5T4ZOo0Gda9viaIhpD1Qn7JqqlKAJqC'
1144
        'QplPKp2nqBWAfwBGaOwVrz3y1T+UZZNismXHsb2Jq18T+VaD9k4P8DqE3g70qV'
1145
        'JLurpnDI6VS5oqDDPVbtVjMxMxMg4rzQVipn2Bv1fVNK0iq3Gl0hhnnHKm/egy'
1146
        'nWQ7QH/F3JFOFCQ0aSPfA='
1147
        ).decode('base64')
2018.18.9 by Martin Pool
remote Repository.tarball builds a temporary directory and tars that
1148
2018.18.7 by Martin Pool
(broken) Start addng client proxy test for Repository.tarball
1149
    def test_repository_tarball(self):
2018.18.9 by Martin Pool
remote Repository.tarball builds a temporary directory and tars that
1150
        # Test that Repository.tarball generates the right operations
2018.18.7 by Martin Pool
(broken) Start addng client proxy test for Repository.tarball
1151
        transport_path = 'repo'
2018.18.14 by Martin Pool
merge hpss again; restore incorrectly removed RemoteRepository.break_lock
1152
        expected_calls = [('call_expecting_body', 'Repository.tarball',
3104.4.2 by Andrew Bennetts
All tests passing.
1153
                           ('repo/', 'bz2',),),
2018.18.7 by Martin Pool
(broken) Start addng client proxy test for Repository.tarball
1154
            ]
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1155
        repo, client = self.setup_fake_client_and_repository(transport_path)
1156
        client.add_success_response_with_body(self.tarball_content, 'ok')
2018.18.7 by Martin Pool
(broken) Start addng client proxy test for Repository.tarball
1157
        # Now actually ask for the tarball
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1158
        tarball_file = repo._get_tarball('bz2')
2018.18.25 by Martin Pool
Repository.tarball fixes for python2.4
1159
        try:
1160
            self.assertEqual(expected_calls, client._calls)
1161
            self.assertEqual(self.tarball_content, tarball_file.read())
1162
        finally:
1163
            tarball_file.close()
2018.18.9 by Martin Pool
remote Repository.tarball builds a temporary directory and tars that
1164
1165
1166
class TestRemoteRepositoryCopyContent(tests.TestCaseWithTransport):
1167
    """RemoteRepository.copy_content_into optimizations"""
1168
2018.18.10 by Martin Pool
copy_content_into from Remote repositories by using temporary directories on both ends.
1169
    def test_copy_content_remote_to_local(self):
1170
        self.transport_server = server.SmartTCPServer_for_testing
1171
        src_repo = self.make_repository('repo1')
1172
        src_repo = repository.Repository.open(self.get_url('repo1'))
1173
        # At the moment the tarball-based copy_content_into can't write back
1174
        # into a smart server.  It would be good if it could upload the
1175
        # tarball; once that works we'd have to create repositories of
1176
        # different formats. -- mbp 20070410
1177
        dest_url = self.get_vfs_only_url('repo2')
1178
        dest_bzrdir = BzrDir.create(dest_url)
1179
        dest_repo = dest_bzrdir.create_repository()
1180
        self.assertFalse(isinstance(dest_repo, RemoteRepository))
1181
        self.assertTrue(isinstance(src_repo, RemoteRepository))
1182
        src_repo.copy_content_into(dest_repo)
2535.3.39 by Andrew Bennetts
Tidy some XXXs.
1183
1184
2535.3.49 by Andrew Bennetts
Rename 'Repository.fetch_revisions' smart request to 'Repository.stream_knit_data_for_revisions'.
1185
class TestRepositoryStreamKnitData(TestRemoteRepository):
2535.3.39 by Andrew Bennetts
Tidy some XXXs.
1186
1187
    def make_pack_file(self, records):
1188
        pack_file = StringIO()
1189
        pack_writer = pack.ContainerWriter(pack_file.write)
1190
        pack_writer.begin()
1191
        for bytes, names in records:
1192
            pack_writer.add_bytes_record(bytes, names)
1193
        pack_writer.end()
1194
        pack_file.seek(0)
1195
        return pack_file
1196
2535.4.2 by Andrew Bennetts
Nasty hackery to make stream_knit_data_for_revisions response use streaming.
1197
    def make_pack_stream(self, records):
2535.4.18 by Andrew Bennetts
Use pack.ContainerSerialiser to remove some nasty cruft.
1198
        pack_serialiser = pack.ContainerSerialiser()
1199
        yield pack_serialiser.begin()
2535.4.2 by Andrew Bennetts
Nasty hackery to make stream_knit_data_for_revisions response use streaming.
1200
        for bytes, names in records:
2535.4.18 by Andrew Bennetts
Use pack.ContainerSerialiser to remove some nasty cruft.
1201
            yield pack_serialiser.bytes_record(bytes, names)
1202
        yield pack_serialiser.end()
2535.4.2 by Andrew Bennetts
Nasty hackery to make stream_knit_data_for_revisions response use streaming.
1203
2535.3.39 by Andrew Bennetts
Tidy some XXXs.
1204
    def test_bad_pack_from_server(self):
2535.3.50 by Andrew Bennetts
Use tuple names in data streams rather than concatenated strings.
1205
        """A response with invalid data (e.g. it has a record with multiple
1206
        names) triggers an exception.
1207
        
1208
        Not all possible errors will be caught at this stage, but obviously
1209
        malformed data should be.
1210
        """
1211
        record = ('bytes', [('name1',), ('name2',)])
2535.4.2 by Andrew Bennetts
Nasty hackery to make stream_knit_data_for_revisions response use streaming.
1212
        pack_stream = self.make_pack_stream([record])
2535.3.39 by Andrew Bennetts
Tidy some XXXs.
1213
        transport_path = 'quack'
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1214
        repo, client = self.setup_fake_client_and_repository(transport_path)
1215
        client.add_success_response_with_body(pack_stream, 'ok')
3184.1.9 by Robert Collins
* ``Repository.get_data_stream`` is now deprecated in favour of
1216
        search = graph.SearchResult(set(['revid']), set(), 1, set(['revid']))
1217
        stream = repo.get_data_stream_for_search(search)
2535.3.39 by Andrew Bennetts
Tidy some XXXs.
1218
        self.assertRaises(errors.SmartProtocolError, list, stream)
1219
    
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
1220
    def test_backwards_compatibility(self):
1221
        """If the server doesn't recognise this request, fallback to VFS."""
3245.4.24 by Andrew Bennetts
Consistently raise errors from the server as ErrorFromSmartServer exceptions.
1222
        repo, client = self.setup_fake_client_and_repository('path')
1223
        client.add_unknown_method_response(
1224
            'Repository.stream_revisions_chunked')
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
1225
        self.mock_called = False
1226
        repo._real_repository = MockRealRepository(self)
3184.1.9 by Robert Collins
* ``Repository.get_data_stream`` is now deprecated in favour of
1227
        search = graph.SearchResult(set(['revid']), set(), 1, set(['revid']))
1228
        repo.get_data_stream_for_search(search)
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
1229
        self.assertTrue(self.mock_called)
1230
        self.failIf(client.expecting_body,
1231
            "The protocol has been left in an unclean state that will cause "
1232
            "TooManyConcurrentRequests errors.")
1233
1234
1235
class MockRealRepository(object):
1236
    """Helper class for TestRepositoryStreamKnitData.test_unknown_method."""
1237
1238
    def __init__(self, test):
1239
        self.test = test
1240
3184.1.9 by Robert Collins
* ``Repository.get_data_stream`` is now deprecated in favour of
1241
    def get_data_stream_for_search(self, search):
1242
        self.test.assertEqual(set(['revid']), search.get_keys())
2535.3.68 by Andrew Bennetts
Backwards compatibility for new smart method.
1243
        self.test.mock_called = True
1244
1245