~bzr-pqm/bzr/bzr.dev

2018.18.1 by Martin Pool
Add stub Repository.tarball smart method
1
# Copyright (C) 2006, 2007 Canonical Ltd
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
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
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
16
2748.4.1 by Andrew Bennetts
Implement a ChunkedBodyDecoder.
17
"""Tests for the smart wire/domain protocol.
18
19
This module contains tests for the domain-level smart requests and responses,
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
20
such as the 'Branch.lock_write' request. Many of these use specific disk
21
formats to exercise calls that only make sense for formats with specific
22
properties.
2748.4.1 by Andrew Bennetts
Implement a ChunkedBodyDecoder.
23
24
Tests for low-level protocol encoding are found in test_smart_transport.
25
"""
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
26
3211.5.2 by Robert Collins
Change RemoteRepository.get_parent_map to use bz2 not gzip for compression.
27
import bz2
2692.1.18 by Andrew Bennetts
Merge from bzr.dev.
28
from cStringIO import StringIO
2018.18.2 by Martin Pool
smart method Repository.tarball actually returns the tarball
29
import tarfile
30
2692.1.2 by Andrew Bennetts
Merge from bzr.dev.
31
from bzrlib import (
2694.5.4 by Jelmer Vernooij
Move bzrlib.util.bencode to bzrlib._bencode_py.
32
    bencode,
2692.1.2 by Andrew Bennetts
Merge from bzr.dev.
33
    bzrdir,
34
    errors,
35
    pack,
36
    smart,
37
    tests,
38
    urlutils,
39
    )
3577.1.1 by Andrew Bennetts
Cherry-pick TipChangeRejected changes from pre-branch-tip-changed-hook loom.
40
from bzrlib.branch import Branch, BranchReferenceFormat
2692.1.22 by Andrew Bennetts
Reinstate imports needed to run test_smart alone.
41
import bzrlib.smart.branch
4070.2.3 by Robert Collins
Get BzrDir.cloning_metadir working.
42
import bzrlib.smart.bzrdir, bzrlib.smart.bzrdir as smart_dir
43
import bzrlib.smart.packrepository
2692.1.22 by Andrew Bennetts
Reinstate imports needed to run test_smart alone.
44
import bzrlib.smart.repository
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
45
from bzrlib.smart.request import (
46
    FailedSmartServerResponse,
47
    SmartServerRequest,
48
    SmartServerResponse,
49
    SuccessfulSmartServerResponse,
50
    )
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
51
from bzrlib.tests import (
52
    split_suite_by_re,
53
    )
2692.1.3 by Andrew Bennetts
Fix imports so that tests in test_smart.py can be run alone.
54
from bzrlib.transport import chroot, get_transport
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
55
56
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
57
def load_tests(standard_tests, module, loader):
58
    """Multiply tests version and protocol consistency."""
59
    # FindRepository tests.
60
    bzrdir_mod = bzrlib.smart.bzrdir
4084.5.1 by Robert Collins
Bulk update all test adaptation into a single approach, using multiply_tests rather than test adapters.
61
    scenarios = [
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
62
        ("find_repository", {
63
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV1}),
64
        ("find_repositoryV2", {
65
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV2}),
4053.1.4 by Robert Collins
Move the fetch control attributes from Repository to RepositoryFormat.
66
        ("find_repositoryV3", {
67
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV3}),
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
68
        ]
69
    to_adapt, result = split_suite_by_re(standard_tests,
70
        "TestSmartServerRequestFindRepository")
71
    v2_only, v1_and_2 = split_suite_by_re(to_adapt,
72
        "_v2")
4084.5.1 by Robert Collins
Bulk update all test adaptation into a single approach, using multiply_tests rather than test adapters.
73
    tests.multiply_tests(v1_and_2, scenarios, result)
74
    # The first scenario is only applicable to v1 protocols, it is deleted
75
    # since.
76
    tests.multiply_tests(v2_only, scenarios[1:], result)
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
77
    return result
78
79
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
80
class TestCaseWithChrootedTransport(tests.TestCaseWithTransport):
81
82
    def setUp(self):
83
        tests.TestCaseWithTransport.setUp(self)
84
        self._chroot_server = None
85
86
    def get_transport(self, relpath=None):
87
        if self._chroot_server is None:
88
            backing_transport = tests.TestCaseWithTransport.get_transport(self)
89
            self._chroot_server = chroot.ChrootServer(backing_transport)
90
            self._chroot_server.setUp()
91
            self.addCleanup(self._chroot_server.tearDown)
92
        t = get_transport(self._chroot_server.get_url())
93
        if relpath is not None:
94
            t = t.clone(relpath)
95
        return t
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
96
97
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
98
class TestCaseWithSmartMedium(tests.TestCaseWithTransport):
99
100
    def setUp(self):
101
        super(TestCaseWithSmartMedium, self).setUp()
102
        # We're allowed to set  the transport class here, so that we don't use
103
        # the default or a parameterized class, but rather use the
104
        # TestCaseWithTransport infrastructure to set up a smart server and
105
        # transport.
3245.4.28 by Andrew Bennetts
Remove another XXX, and include test ID in smart server thread names.
106
        self.transport_server = self.make_transport_server
107
108
    def make_transport_server(self):
109
        return smart.server.SmartTCPServer_for_testing('-' + self.id())
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
110
111
    def get_smart_medium(self):
112
        """Get a smart medium to use in tests."""
113
        return self.get_transport().get_smart_medium()
114
115
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
116
class TestSmartServerResponse(tests.TestCase):
117
118
    def test__eq__(self):
119
        self.assertEqual(SmartServerResponse(('ok', )),
120
            SmartServerResponse(('ok', )))
121
        self.assertEqual(SmartServerResponse(('ok', ), 'body'),
122
            SmartServerResponse(('ok', ), 'body'))
123
        self.assertNotEqual(SmartServerResponse(('ok', )),
124
            SmartServerResponse(('notok', )))
125
        self.assertNotEqual(SmartServerResponse(('ok', ), 'body'),
126
            SmartServerResponse(('ok', )))
2018.5.41 by Robert Collins
Fix SmartServerResponse.__eq__ to handle None.
127
        self.assertNotEqual(None,
128
            SmartServerResponse(('ok', )))
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
129
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
130
    def test__str__(self):
131
        """SmartServerResponses can be stringified."""
132
        self.assertEqual(
3691.2.6 by Martin Pool
Disable RemoteBranch stacking, but get get_stacked_on_url working, and passing back exceptions
133
            "<SuccessfulSmartServerResponse args=('args',) body='body'>",
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
134
            str(SuccessfulSmartServerResponse(('args',), 'body')))
135
        self.assertEqual(
3691.2.6 by Martin Pool
Disable RemoteBranch stacking, but get get_stacked_on_url working, and passing back exceptions
136
            "<FailedSmartServerResponse args=('args',) body='body'>",
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
137
            str(FailedSmartServerResponse(('args',), 'body')))
138
139
140
class TestSmartServerRequest(tests.TestCaseWithMemoryTransport):
141
142
    def test_translate_client_path(self):
143
        transport = self.get_transport()
144
        request = SmartServerRequest(transport, 'foo/')
145
        self.assertEqual('./', request.translate_client_path('foo/'))
146
        self.assertRaises(
147
            errors.InvalidURLJoin, request.translate_client_path, 'foo/..')
148
        self.assertRaises(
149
            errors.PathNotChild, request.translate_client_path, '/')
150
        self.assertRaises(
151
            errors.PathNotChild, request.translate_client_path, 'bar/')
152
        self.assertEqual('./baz', request.translate_client_path('foo/baz'))
153
154
    def test_transport_from_client_path(self):
155
        transport = self.get_transport()
156
        request = SmartServerRequest(transport, 'foo/')
157
        self.assertEqual(
158
            transport.base,
159
            request.transport_from_client_path('foo/').base)
160
161
4070.2.3 by Robert Collins
Get BzrDir.cloning_metadir working.
162
class TestSmartServerBzrDirRequestCloningMetaDir(
163
    tests.TestCaseWithMemoryTransport):
164
    """Tests for BzrDir.cloning_metadir."""
165
166
    def test_cloning_metadir(self):
167
        """When there is a bzrdir present, the call succeeds."""
168
        backing = self.get_transport()
169
        dir = self.make_bzrdir('.')
170
        local_result = dir.cloning_metadir()
171
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
172
        request = request_class(backing)
173
        expected = SuccessfulSmartServerResponse(
174
            (local_result.network_name(),
175
            local_result.repository_format.network_name(),
4084.2.2 by Robert Collins
Review feedback.
176
            ('branch', local_result.get_branch_format().network_name())))
4070.7.4 by Andrew Bennetts
Deal with branch references better in BzrDir.cloning_metadir RPC (changes protocol).
177
        self.assertEqual(expected, request.execute('', 'False'))
178
179
    def test_cloning_metadir_reference(self):
4160.2.9 by Andrew Bennetts
Fix BzrDir.cloning_metadir RPC to fail on branch references, and make
180
        """The request fails when bzrdir contains a branch reference."""
4070.7.4 by Andrew Bennetts
Deal with branch references better in BzrDir.cloning_metadir RPC (changes protocol).
181
        backing = self.get_transport()
182
        referenced_branch = self.make_branch('referenced')
183
        dir = self.make_bzrdir('.')
184
        local_result = dir.cloning_metadir()
185
        reference = BranchReferenceFormat().initialize(dir, referenced_branch)
186
        reference_url = BranchReferenceFormat().get_reference(dir)
187
        # The server shouldn't try to follow the branch reference, so it's fine
188
        # if the referenced branch isn't reachable.
189
        backing.rename('referenced', 'moved')
190
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
191
        request = request_class(backing)
4160.2.9 by Andrew Bennetts
Fix BzrDir.cloning_metadir RPC to fail on branch references, and make
192
        expected = FailedSmartServerResponse(('BranchReference',))
4070.2.3 by Robert Collins
Get BzrDir.cloning_metadir working.
193
        self.assertEqual(expected, request.execute('', 'False'))
194
195
4017.3.2 by Robert Collins
Reduce the number of round trips required to create a repository over the network.
196
class TestSmartServerRequestCreateRepository(tests.TestCaseWithMemoryTransport):
197
    """Tests for BzrDir.create_repository."""
198
199
    def test_makes_repository(self):
200
        """When there is a bzrdir present, the call succeeds."""
201
        backing = self.get_transport()
202
        self.make_bzrdir('.')
203
        request_class = bzrlib.smart.bzrdir.SmartServerRequestCreateRepository
204
        request = request_class(backing)
205
        reference_bzrdir_format = bzrdir.format_registry.get('default')()
206
        reference_format = reference_bzrdir_format.repository_format
207
        network_name = reference_format.network_name()
208
        expected = SuccessfulSmartServerResponse(
209
            ('ok', 'no', 'no', 'no', network_name))
210
        self.assertEqual(expected, request.execute('', network_name, 'True'))
211
212
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
213
class TestSmartServerRequestFindRepository(tests.TestCaseWithMemoryTransport):
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
214
    """Tests for BzrDir.find_repository."""
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
215
216
    def test_no_repository(self):
217
        """When there is no repository to be found, ('norepository', ) is returned."""
218
        backing = self.get_transport()
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
219
        request = self._request_class(backing)
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
220
        self.make_bzrdir('.')
221
        self.assertEqual(SmartServerResponse(('norepository', )),
2692.1.19 by Andrew Bennetts
Tweak for consistency suggested by John's review.
222
            request.execute(''))
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
223
224
    def test_nonshared_repository(self):
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
225
        # nonshared repositorys only allow 'find' to return a handle when the
226
        # path the repository is being searched on is the same as that that
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
227
        # the repository is at.
228
        backing = self.get_transport()
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
229
        request = self._request_class(backing)
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
230
        result = self._make_repository_and_result()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
231
        self.assertEqual(result, request.execute(''))
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
232
        self.make_bzrdir('subdir')
233
        self.assertEqual(SmartServerResponse(('norepository', )),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
234
            request.execute('subdir'))
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
235
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
236
    def _make_repository_and_result(self, shared=False, format=None):
237
        """Convenience function to setup a repository.
238
239
        :result: The SmartServerResponse to expect when opening it.
240
        """
241
        repo = self.make_repository('.', shared=shared, format=format)
242
        if repo.supports_rich_root():
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
243
            rich_root = 'yes'
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
244
        else:
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
245
            rich_root = 'no'
2018.5.138 by Robert Collins
Merge bzr.dev.
246
        if repo._format.supports_tree_reference:
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
247
            subtrees = 'yes'
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
248
        else:
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
249
            subtrees = 'no'
4053.1.4 by Robert Collins
Move the fetch control attributes from Repository to RepositoryFormat.
250
        if (smart.bzrdir.SmartServerRequestFindRepositoryV3 ==
251
            self._request_class):
252
            return SuccessfulSmartServerResponse(
253
                ('ok', '', rich_root, subtrees, 'no',
254
                 repo._format.network_name()))
255
        elif (smart.bzrdir.SmartServerRequestFindRepositoryV2 ==
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
256
            self._request_class):
257
            # All tests so far are on formats, and for non-external
258
            # repositories.
259
            return SuccessfulSmartServerResponse(
260
                ('ok', '', rich_root, subtrees, 'no'))
261
        else:
262
            return SuccessfulSmartServerResponse(('ok', '', rich_root, subtrees))
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
263
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
264
    def test_shared_repository(self):
265
        """When there is a shared repository, we get 'ok', 'relpath-to-repo'."""
266
        backing = self.get_transport()
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
267
        request = self._request_class(backing)
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
268
        result = self._make_repository_and_result(shared=True)
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
269
        self.assertEqual(result, request.execute(''))
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
270
        self.make_bzrdir('subdir')
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
271
        result2 = SmartServerResponse(result.args[0:1] + ('..', ) + result.args[2:])
272
        self.assertEqual(result2,
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
273
            request.execute('subdir'))
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
274
        self.make_bzrdir('subdir/deeper')
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
275
        result3 = SmartServerResponse(result.args[0:1] + ('../..', ) + result.args[2:])
276
        self.assertEqual(result3,
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
277
            request.execute('subdir/deeper'))
2018.5.34 by Robert Collins
Get test_remote.BasicRemoteObjectTests.test_open_remote_branch passing by implementing a remote method BzrDir.find_repository.
278
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
279
    def test_rich_root_and_subtree_encoding(self):
280
        """Test for the format attributes for rich root and subtree support."""
281
        backing = self.get_transport()
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
282
        request = self._request_class(backing)
2018.5.118 by Robert Collins
Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.
283
        result = self._make_repository_and_result(format='dirstate-with-subtree')
284
        # check the test will be valid
2018.5.166 by Andrew Bennetts
Small changes in response to Aaron's review.
285
        self.assertEqual('yes', result.args[2])
286
        self.assertEqual('yes', result.args[3])
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
287
        self.assertEqual(result, request.execute(''))
288
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
289
    def test_supports_external_lookups_no_v2(self):
290
        """Test for the supports_external_lookups attribute."""
291
        backing = self.get_transport()
292
        request = self._request_class(backing)
293
        result = self._make_repository_and_result(format='dirstate-with-subtree')
294
        # check the test will be valid
295
        self.assertEqual('no', result.args[4])
2692.1.24 by Andrew Bennetts
Merge from bzr.dev.
296
        self.assertEqual(result, request.execute(''))
297
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
298
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
299
class TestSmartServerBzrDirRequestGetConfigFile(
300
    tests.TestCaseWithMemoryTransport):
301
    """Tests for BzrDir.get_config_file."""
302
303
    def test_present(self):
304
        backing = self.get_transport()
305
        dir = self.make_bzrdir('.')
306
        dir.get_config().set_default_stack_on("/")
307
        local_result = dir._get_config()._get_config_file().read()
308
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
309
        request = request_class(backing)
310
        expected = SuccessfulSmartServerResponse((), local_result)
311
        self.assertEqual(expected, request.execute(''))
312
313
    def test_missing(self):
314
        backing = self.get_transport()
315
        dir = self.make_bzrdir('.')
316
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
317
        request = request_class(backing)
318
        expected = SuccessfulSmartServerResponse((), '')
319
        self.assertEqual(expected, request.execute(''))
320
321
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
322
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithMemoryTransport):
2018.5.42 by Robert Collins
Various hopefully improvements, but wsgi is broken, handing over to spiv :).
323
324
    def test_empty_dir(self):
325
        """Initializing an empty dir should succeed and do it."""
326
        backing = self.get_transport()
327
        request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
328
        self.assertEqual(SmartServerResponse(('ok', )),
2692.1.20 by Andrew Bennetts
Tweak for consistency suggested by John's review.
329
            request.execute(''))
2018.5.42 by Robert Collins
Various hopefully improvements, but wsgi is broken, handing over to spiv :).
330
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
331
        # no branch, tree or repository is expected with the current
2018.5.42 by Robert Collins
Various hopefully improvements, but wsgi is broken, handing over to spiv :).
332
        # default formart.
333
        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
334
        self.assertRaises(errors.NotBranchError, made_dir.open_branch)
335
        self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
336
337
    def test_missing_dir(self):
338
        """Initializing a missing directory should fail like the bzrdir api."""
339
        backing = self.get_transport()
340
        request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
341
        self.assertRaises(errors.NoSuchFile,
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
342
            request.execute, 'subdir')
2018.5.42 by Robert Collins
Various hopefully improvements, but wsgi is broken, handing over to spiv :).
343
344
    def test_initialized_dir(self):
345
        """Initializing an extant bzrdir should fail like the bzrdir api."""
346
        backing = self.get_transport()
347
        request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
348
        self.make_bzrdir('subdir')
349
        self.assertRaises(errors.FileExists,
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
350
            request.execute, 'subdir')
351
352
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
353
class TestSmartServerRequestBzrDirInitializeEx(tests.TestCaseWithMemoryTransport):
4436.1.1 by Andrew Bennetts
Rename BzrDirFormat.initialize_ex verb to BzrDirFormat.initialize_ex_1.16.
354
    """Basic tests for BzrDir.initialize_ex_1.16 in the smart server.
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
355
4294.2.10 by Robert Collins
Review feedback.
356
    The main unit tests in test_bzrdir exercise the API comprehensively.
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
357
    """
358
359
    def test_empty_dir(self):
360
        """Initializing an empty dir should succeed and do it."""
361
        backing = self.get_transport()
4294.2.8 by Robert Collins
Reduce round trips pushing new branches substantially.
362
        name = self.make_bzrdir('reference')._format.network_name()
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
363
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
4294.2.8 by Robert Collins
Reduce round trips pushing new branches substantially.
364
        self.assertEqual(SmartServerResponse(('', '', '', '', '', '', name,
4307.2.2 by Robert Collins
Lock repositories created by BzrDirFormat.initialize_on_transport_ex.
365
            'False', '', '', '')),
4294.2.8 by Robert Collins
Reduce round trips pushing new branches substantially.
366
            request.execute(name, '', 'True', 'False', 'False', '', '', '', '',
367
            'False'))
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
368
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
369
        # no branch, tree or repository is expected with the current
4294.2.10 by Robert Collins
Review feedback.
370
        # default format.
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
371
        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
372
        self.assertRaises(errors.NotBranchError, made_dir.open_branch)
373
        self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
374
375
    def test_missing_dir(self):
376
        """Initializing a missing directory should fail like the bzrdir api."""
377
        backing = self.get_transport()
4294.2.8 by Robert Collins
Reduce round trips pushing new branches substantially.
378
        name = self.make_bzrdir('reference')._format.network_name()
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
379
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
4294.2.8 by Robert Collins
Reduce round trips pushing new branches substantially.
380
        self.assertRaises(errors.NoSuchFile, request.execute, name,
381
            'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
382
383
    def test_initialized_dir(self):
4416.3.4 by Jonathan Lange
Fix a typo.
384
        """Initializing an extant directory should fail like the bzrdir api."""
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
385
        backing = self.get_transport()
4294.2.8 by Robert Collins
Reduce round trips pushing new branches substantially.
386
        name = self.make_bzrdir('reference')._format.network_name()
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
387
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
388
        self.make_bzrdir('subdir')
4294.2.8 by Robert Collins
Reduce round trips pushing new branches substantially.
389
        self.assertRaises(errors.FileExists, request.execute, name, 'subdir',
390
            'False', 'False', 'False', '', '', '', '', 'False')
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
391
392
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
393
class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
394
395
    def test_no_branch(self):
396
        """When there is no branch, ('nobranch', ) is returned."""
397
        backing = self.get_transport()
398
        request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
399
        self.make_bzrdir('.')
400
        self.assertEqual(SmartServerResponse(('nobranch', )),
2692.1.20 by Andrew Bennetts
Tweak for consistency suggested by John's review.
401
            request.execute(''))
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
402
403
    def test_branch(self):
404
        """When there is a branch, 'ok' is returned."""
405
        backing = self.get_transport()
406
        request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
407
        self.make_branch('.')
408
        self.assertEqual(SmartServerResponse(('ok', '')),
2692.1.20 by Andrew Bennetts
Tweak for consistency suggested by John's review.
409
            request.execute(''))
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
410
411
    def test_branch_reference(self):
412
        """When there is a branch reference, the reference URL is returned."""
413
        backing = self.get_transport()
414
        request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
415
        branch = self.make_branch('branch')
416
        checkout = branch.create_checkout('reference',lightweight=True)
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
417
        reference_url = BranchReferenceFormat().get_reference(checkout.bzrdir)
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
418
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
419
        self.assertEqual(SmartServerResponse(('ok', reference_url)),
2692.1.20 by Andrew Bennetts
Tweak for consistency suggested by John's review.
420
            request.execute('reference'))
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
421
422
4084.2.1 by Robert Collins
Make accessing a branch.tags.get_tag_dict use a smart[er] method rather than VFS calls and real objects.
423
class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):
424
425
    def test_no_branch(self):
426
        """When there is no branch, ('nobranch', ) is returned."""
427
        backing = self.get_transport()
428
        self.make_bzrdir('.')
429
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
430
        self.assertEqual(SmartServerResponse(('nobranch', )),
431
            request.execute(''))
432
433
    def test_branch(self):
434
        """When there is a branch, 'ok' is returned."""
435
        backing = self.get_transport()
436
        expected = self.make_branch('.')._format.network_name()
437
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
438
        self.assertEqual(SuccessfulSmartServerResponse(('branch', expected)),
439
            request.execute(''))
440
441
    def test_branch_reference(self):
442
        """When there is a branch reference, the reference URL is returned."""
443
        backing = self.get_transport()
444
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
445
        branch = self.make_branch('branch')
446
        checkout = branch.create_checkout('reference',lightweight=True)
447
        reference_url = BranchReferenceFormat().get_reference(checkout.bzrdir)
448
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
449
        self.assertEqual(SuccessfulSmartServerResponse(('ref', reference_url)),
450
            request.execute('reference'))
451
4160.2.1 by Andrew Bennetts
Failing test for BzrDir.open_branchV2 RPC not opening stacked-on branch.
452
    def test_stacked_branch(self):
453
        """Opening a stacked branch does not open the stacked-on branch."""
454
        trunk = self.make_branch('trunk')
455
        feature = self.make_branch('feature', format='1.9')
456
        feature.set_stacked_on_url(trunk.base)
457
        opened_branches = []
458
        Branch.hooks.install_named_hook('open', opened_branches.append, None)
459
        backing = self.get_transport()
460
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
4160.2.4 by Andrew Bennetts
Use BzrDir pre_open hook to jail request code from accessing transports other than the backing transport.
461
        request.setup_jail()
462
        try:
463
            response = request.execute('feature')
464
        finally:
465
            request.teardown_jail()
4160.2.1 by Andrew Bennetts
Failing test for BzrDir.open_branchV2 RPC not opening stacked-on branch.
466
        expected_format = feature._format.network_name()
467
        self.assertEqual(
468
            SuccessfulSmartServerResponse(('branch', expected_format)),
469
            response)
470
        self.assertLength(1, opened_branches)
471
4084.2.1 by Robert Collins
Make accessing a branch.tags.get_tag_dict use a smart[er] method rather than VFS calls and real objects.
472
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
473
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithMemoryTransport):
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
474
475
    def test_empty(self):
476
        """For an empty branch, the body is empty."""
477
        backing = self.get_transport()
478
        request = smart.branch.SmartServerRequestRevisionHistory(backing)
479
        self.make_branch('.')
480
        self.assertEqual(SmartServerResponse(('ok', ), ''),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
481
            request.execute(''))
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
482
483
    def test_not_empty(self):
484
        """For a non-empty branch, the body is empty."""
485
        backing = self.get_transport()
486
        request = smart.branch.SmartServerRequestRevisionHistory(backing)
487
        tree = self.make_branch_and_memory_tree('.')
488
        tree.lock_write()
489
        tree.add('')
490
        r1 = tree.commit('1st commit')
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
491
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
492
        tree.unlock()
2018.5.83 by Andrew Bennetts
Fix some test failures caused by the switch from unicode to UTF-8-encoded strs for revision IDs.
493
        self.assertEqual(
494
            SmartServerResponse(('ok', ), ('\x00'.join([r1, r2]))),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
495
            request.execute(''))
496
497
498
class TestSmartServerBranchRequest(tests.TestCaseWithMemoryTransport):
2018.5.49 by Wouter van Heyst
Refactor SmartServerBranchRequest out from SmartServerRequestRevisionHistory to
499
500
    def test_no_branch(self):
501
        """When there is a bzrdir and no branch, NotBranchError is raised."""
502
        backing = self.get_transport()
503
        request = smart.branch.SmartServerBranchRequest(backing)
504
        self.make_bzrdir('.')
505
        self.assertRaises(errors.NotBranchError,
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
506
            request.execute, '')
2018.5.49 by Wouter van Heyst
Refactor SmartServerBranchRequest out from SmartServerRequestRevisionHistory to
507
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
508
    def test_branch_reference(self):
509
        """When there is a branch reference, NotBranchError is raised."""
510
        backing = self.get_transport()
2018.5.49 by Wouter van Heyst
Refactor SmartServerBranchRequest out from SmartServerRequestRevisionHistory to
511
        request = smart.branch.SmartServerBranchRequest(backing)
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
512
        branch = self.make_branch('branch')
513
        checkout = branch.create_checkout('reference',lightweight=True)
514
        self.assertRaises(errors.NotBranchError,
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
515
            request.execute, 'checkout')
516
517
518
class TestSmartServerBranchRequestLastRevisionInfo(tests.TestCaseWithMemoryTransport):
2018.5.50 by Wouter van Heyst
Add SmartServerBranchRequestLastRevisionInfo method.
519
520
    def test_empty(self):
2018.5.170 by Andrew Bennetts
Use 'null:' instead of '' to mean NULL_REVISION on the wire.
521
        """For an empty branch, the result is ('ok', '0', 'null:')."""
2018.5.50 by Wouter van Heyst
Add SmartServerBranchRequestLastRevisionInfo method.
522
        backing = self.get_transport()
523
        request = smart.branch.SmartServerBranchRequestLastRevisionInfo(backing)
524
        self.make_branch('.')
2018.5.170 by Andrew Bennetts
Use 'null:' instead of '' to mean NULL_REVISION on the wire.
525
        self.assertEqual(SmartServerResponse(('ok', '0', 'null:')),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
526
            request.execute(''))
2018.5.50 by Wouter van Heyst
Add SmartServerBranchRequestLastRevisionInfo method.
527
528
    def test_not_empty(self):
529
        """For a non-empty branch, the result is ('ok', 'revno', 'revid')."""
530
        backing = self.get_transport()
531
        request = smart.branch.SmartServerBranchRequestLastRevisionInfo(backing)
532
        tree = self.make_branch_and_memory_tree('.')
533
        tree.lock_write()
534
        tree.add('')
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
535
        rev_id_utf8 = u'\xc8'.encode('utf-8')
2018.5.50 by Wouter van Heyst
Add SmartServerBranchRequestLastRevisionInfo method.
536
        r1 = tree.commit('1st commit')
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
537
        r2 = tree.commit('2nd commit', rev_id=rev_id_utf8)
2018.5.50 by Wouter van Heyst
Add SmartServerBranchRequestLastRevisionInfo method.
538
        tree.unlock()
539
        self.assertEqual(
2018.5.83 by Andrew Bennetts
Fix some test failures caused by the switch from unicode to UTF-8-encoded strs for revision IDs.
540
            SmartServerResponse(('ok', '2', rev_id_utf8)),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
541
            request.execute(''))
542
543
544
class TestSmartServerBranchRequestGetConfigFile(tests.TestCaseWithMemoryTransport):
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
545
546
    def test_default(self):
547
        """With no file, we get empty content."""
548
        backing = self.get_transport()
549
        request = smart.branch.SmartServerBranchGetConfigFile(backing)
550
        branch = self.make_branch('.')
551
        # there should be no file by default
552
        content = ''
553
        self.assertEqual(SmartServerResponse(('ok', ), content),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
554
            request.execute(''))
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
555
556
    def test_with_content(self):
557
        # SmartServerBranchGetConfigFile should return the content from
558
        # branch.control_files.get('branch.conf') for now - in the future it may
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
559
        # perform more complex processing.
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
560
        backing = self.get_transport()
561
        request = smart.branch.SmartServerBranchGetConfigFile(backing)
562
        branch = self.make_branch('.')
3407.2.5 by Martin Pool
Deprecate LockableFiles.put_utf8
563
        branch._transport.put_bytes('branch.conf', 'foo bar baz')
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
564
        self.assertEqual(SmartServerResponse(('ok', ), 'foo bar baz'),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
565
            request.execute(''))
566
567
4226.2.1 by Robert Collins
Set branch config options via a smart method.
568
class TestLockedBranch(tests.TestCaseWithMemoryTransport):
569
570
    def get_lock_tokens(self, branch):
571
        branch_token = branch.lock_write()
572
        repo_token = branch.repository.lock_write()
573
        branch.repository.unlock()
574
        return branch_token, repo_token
575
576
577
class TestSmartServerBranchRequestSetConfigOption(TestLockedBranch):
578
579
    def test_value_name(self):
580
        branch = self.make_branch('.')
581
        request = smart.branch.SmartServerBranchRequestSetConfigOption(
582
            branch.bzrdir.root_transport)
583
        branch_token, repo_token = self.get_lock_tokens(branch)
584
        config = branch._get_config()
585
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
586
            '')
587
        self.assertEqual(SuccessfulSmartServerResponse(()), result)
588
        self.assertEqual('bar', config.get_option('foo'))
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
589
        # Cleanup
590
        branch.unlock()
4226.2.1 by Robert Collins
Set branch config options via a smart method.
591
592
    def test_value_name_section(self):
593
        branch = self.make_branch('.')
594
        request = smart.branch.SmartServerBranchRequestSetConfigOption(
595
            branch.bzrdir.root_transport)
596
        branch_token, repo_token = self.get_lock_tokens(branch)
597
        config = branch._get_config()
598
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
599
            'gam')
600
        self.assertEqual(SuccessfulSmartServerResponse(()), result)
601
        self.assertEqual('bar', config.get_option('foo', 'gam'))
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
602
        # Cleanup
603
        branch.unlock()
4226.2.1 by Robert Collins
Set branch config options via a smart method.
604
605
4556.2.2 by Andrew Bennetts
Handle failures more gracefully.
606
class TestSmartServerBranchRequestSetTagsBytes(TestLockedBranch):
607
    # Only called when the branch format and tags match [yay factory
608
    # methods] so only need to test straight forward cases.
609
610
    def test_set_bytes(self):
611
        base_branch = self.make_branch('base')
612
        tag_bytes = base_branch._get_tags_bytes()
613
        # get_lock_tokens takes out a lock.
614
        branch_token, repo_token = self.get_lock_tokens(base_branch)
615
        request = smart.branch.SmartServerBranchSetTagsBytes(
616
            self.get_transport())
617
        response = request.execute('base', branch_token, repo_token)
618
        self.assertEqual(None, response)
619
        response = request.do_chunk(tag_bytes)
620
        self.assertEqual(None, response)
621
        response = request.do_end()
622
        self.assertEquals(
623
            SuccessfulSmartServerResponse(()), response)
624
        base_branch.unlock()
625
626
    def test_lock_failed(self):
627
        base_branch = self.make_branch('base')
628
        base_branch.lock_write()
629
        tag_bytes = base_branch._get_tags_bytes()
630
        request = smart.branch.SmartServerBranchSetTagsBytes(
631
            self.get_transport())
632
        self.assertRaises(errors.TokenMismatch, request.execute,
633
            'base', 'wrong token', 'wrong token')
634
        # The request handler will keep processing the message parts, so even
635
        # if the request fails immediately do_chunk and do_end are still
636
        # called.
637
        request.do_chunk(tag_bytes)
638
        request.do_end()
639
        base_branch.unlock()
640
641
642
4226.2.1 by Robert Collins
Set branch config options via a smart method.
643
class SetLastRevisionTestBase(TestLockedBranch):
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
644
    """Base test case for verbs that implement set_last_revision."""
645
646
    def setUp(self):
647
        tests.TestCaseWithMemoryTransport.setUp(self)
648
        backing_transport = self.get_transport()
649
        self.request = self.request_class(backing_transport)
650
        self.tree = self.make_branch_and_memory_tree('.')
651
652
    def lock_branch(self):
4226.2.1 by Robert Collins
Set branch config options via a smart method.
653
        return self.get_lock_tokens(self.tree.branch)
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
654
655
    def unlock_branch(self):
656
        self.tree.branch.unlock()
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
657
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
658
    def set_last_revision(self, revision_id, revno):
659
        branch_token, repo_token = self.lock_branch()
660
        response = self._set_last_revision(
661
            revision_id, revno, branch_token, repo_token)
662
        self.unlock_branch()
663
        return response
664
665
    def assertRequestSucceeds(self, revision_id, revno):
666
        response = self.set_last_revision(revision_id, revno)
667
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)), response)
668
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
669
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
670
class TestSetLastRevisionVerbMixin(object):
671
    """Mixin test case for verbs that implement set_last_revision."""
672
673
    def test_set_null_to_null(self):
674
        """An empty branch can have its last revision set to 'null:'."""
675
        self.assertRequestSucceeds('null:', 0)
676
677
    def test_NoSuchRevision(self):
678
        """If the revision_id is not present, the verb returns NoSuchRevision.
679
        """
680
        revision_id = 'non-existent revision'
681
        self.assertEqual(
682
            FailedSmartServerResponse(('NoSuchRevision', revision_id)),
683
            self.set_last_revision(revision_id, 1))
684
685
    def make_tree_with_two_commits(self):
686
        self.tree.lock_write()
687
        self.tree.add('')
688
        rev_id_utf8 = u'\xc8'.encode('utf-8')
689
        r1 = self.tree.commit('1st commit', rev_id=rev_id_utf8)
690
        r2 = self.tree.commit('2nd commit', rev_id='rev-2')
691
        self.tree.unlock()
692
693
    def test_branch_last_revision_info_is_updated(self):
694
        """A branch's tip can be set to a revision that is present in its
695
        repository.
696
        """
697
        # Make a branch with an empty revision history, but two revisions in
698
        # its repository.
699
        self.make_tree_with_two_commits()
700
        rev_id_utf8 = u'\xc8'.encode('utf-8')
701
        self.tree.branch.set_revision_history([])
702
        self.assertEqual(
703
            (0, 'null:'), self.tree.branch.last_revision_info())
704
        # We can update the branch to a revision that is present in the
705
        # repository.
706
        self.assertRequestSucceeds(rev_id_utf8, 1)
707
        self.assertEqual(
708
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
709
710
    def test_branch_last_revision_info_rewind(self):
711
        """A branch's tip can be set to a revision that is an ancestor of the
712
        current tip.
713
        """
714
        self.make_tree_with_two_commits()
715
        rev_id_utf8 = u'\xc8'.encode('utf-8')
716
        self.assertEqual(
717
            (2, 'rev-2'), self.tree.branch.last_revision_info())
718
        self.assertRequestSucceeds(rev_id_utf8, 1)
719
        self.assertEqual(
720
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
721
3577.1.1 by Andrew Bennetts
Cherry-pick TipChangeRejected changes from pre-branch-tip-changed-hook loom.
722
    def test_TipChangeRejected(self):
723
        """If a pre_change_branch_tip hook raises TipChangeRejected, the verb
724
        returns TipChangeRejected.
725
        """
726
        rejection_message = u'rejection message\N{INTERROBANG}'
727
        def hook_that_rejects(params):
728
            raise errors.TipChangeRejected(rejection_message)
729
        Branch.hooks.install_named_hook(
730
            'pre_change_branch_tip', hook_that_rejects, None)
731
        self.assertEqual(
732
            FailedSmartServerResponse(
733
                ('TipChangeRejected', rejection_message.encode('utf-8'))),
734
            self.set_last_revision('null:', 0))
735
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
736
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
737
class TestSmartServerBranchRequestSetLastRevision(
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
738
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
739
    """Tests for Branch.set_last_revision verb."""
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
740
741
    request_class = smart.branch.SmartServerBranchRequestSetLastRevision
742
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
743
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
744
        return self.request.execute(
745
            '', branch_token, repo_token, revision_id)
746
747
748
class TestSmartServerBranchRequestSetLastRevisionInfo(
749
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
750
    """Tests for Branch.set_last_revision_info verb."""
751
752
    request_class = smart.branch.SmartServerBranchRequestSetLastRevisionInfo
753
754
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
755
        return self.request.execute(
756
            '', branch_token, repo_token, revno, revision_id)
757
758
    def test_NoSuchRevision(self):
759
        """Branch.set_last_revision_info does not have to return
760
        NoSuchRevision if the revision_id is absent.
761
        """
762
        raise tests.TestNotApplicable()
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
763
764
3441.5.25 by Andrew Bennetts
Rename Branch.set_last_revision_descendant verb to Branch.set_last_revision_ex. It's a cop out, but at least it's not misleading.
765
class TestSmartServerBranchRequestSetLastRevisionEx(
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
766
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
767
    """Tests for Branch.set_last_revision_ex verb."""
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
768
3441.5.25 by Andrew Bennetts
Rename Branch.set_last_revision_descendant verb to Branch.set_last_revision_ex. It's a cop out, but at least it's not misleading.
769
    request_class = smart.branch.SmartServerBranchRequestSetLastRevisionEx
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
770
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
771
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
772
        return self.request.execute(
773
            '', branch_token, repo_token, revision_id, 0, 0)
774
775
    def assertRequestSucceeds(self, revision_id, revno):
776
        response = self.set_last_revision(revision_id, revno)
777
        self.assertEqual(
778
            SuccessfulSmartServerResponse(('ok', revno, revision_id)),
779
            response)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
780
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
781
    def test_branch_last_revision_info_rewind(self):
782
        """A branch's tip can be set to a revision that is an ancestor of the
783
        current tip, but only if allow_overwrite_descendant is passed.
784
        """
785
        self.make_tree_with_two_commits()
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
786
        rev_id_utf8 = u'\xc8'.encode('utf-8')
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
787
        self.assertEqual(
788
            (2, 'rev-2'), self.tree.branch.last_revision_info())
789
        # If allow_overwrite_descendant flag is 0, then trying to set the tip
790
        # to an older revision ID has no effect.
791
        branch_token, repo_token = self.lock_branch()
792
        response = self.request.execute(
793
            '', branch_token, repo_token, rev_id_utf8, 0, 0)
794
        self.assertEqual(
795
            SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
796
            response)
797
        self.assertEqual(
798
            (2, 'rev-2'), self.tree.branch.last_revision_info())
799
800
        # If allow_overwrite_descendant flag is 1, then setting the tip to an
801
        # ancestor works.
802
        response = self.request.execute(
803
            '', branch_token, repo_token, rev_id_utf8, 0, 1)
804
        self.assertEqual(
805
            SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
806
            response)
807
        self.unlock_branch()
808
        self.assertEqual(
809
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
810
3441.5.31 by Andrew Bennetts
Add test for allow_diverged flag.
811
    def make_branch_with_divergent_history(self):
812
        """Make a branch with divergent history in its repo.
813
814
        The branch's tip will be 'child-2', and the repo will also contain
815
        'child-1', which diverges from a common base revision.
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
816
        """
817
        self.tree.lock_write()
818
        self.tree.add('')
819
        r1 = self.tree.commit('1st commit')
820
        revno_1, revid_1 = self.tree.branch.last_revision_info()
821
        r2 = self.tree.commit('2nd commit', rev_id='child-1')
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
822
        # Undo the second commit
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
823
        self.tree.branch.set_last_revision_info(revno_1, revid_1)
824
        self.tree.set_parent_ids([revid_1])
3441.5.6 by Andrew Bennetts
Greatly simplify RemoteBranch.update_revisions. Still needs more tests.
825
        # Make a new second commit, child-2.  child-2 has diverged from
826
        # child-1.
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
827
        new_r2 = self.tree.commit('2nd commit', rev_id='child-2')
828
        self.tree.unlock()
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
829
3441.5.31 by Andrew Bennetts
Add test for allow_diverged flag.
830
    def test_not_allow_diverged(self):
831
        """If allow_diverged is not passed, then setting a divergent history
832
        returns a Diverged error.
833
        """
834
        self.make_branch_with_divergent_history()
3297.4.3 by Andrew Bennetts
Add more tests, handle NoSuchRevision in case the remote branch's format can raise it.
835
        self.assertEqual(
3441.5.30 by Andrew Bennetts
Improve tests for all Branch.set_last_revision* verbs.
836
            FailedSmartServerResponse(('Diverged',)),
837
            self.set_last_revision('child-1', 2))
838
        # The branch tip was not changed.
839
        self.assertEqual('child-2', self.tree.branch.last_revision())
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
840
3441.5.31 by Andrew Bennetts
Add test for allow_diverged flag.
841
    def test_allow_diverged(self):
842
        """If allow_diverged is passed, then setting a divergent history
843
        succeeds.
844
        """
845
        self.make_branch_with_divergent_history()
846
        branch_token, repo_token = self.lock_branch()
847
        response = self.request.execute(
848
            '', branch_token, repo_token, 'child-1', 1, 0)
849
        self.assertEqual(
850
            SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
851
            response)
852
        self.unlock_branch()
853
        # The branch tip was changed.
854
        self.assertEqual('child-1', self.tree.branch.last_revision())
855
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
856
4078.2.1 by Robert Collins
Add a Branch.get_parent remote call for RemoteBranch.
857
class TestSmartServerBranchRequestGetParent(tests.TestCaseWithMemoryTransport):
858
859
    def test_get_parent_none(self):
860
        base_branch = self.make_branch('base')
861
        request = smart.branch.SmartServerBranchGetParent(self.get_transport())
862
        response = request.execute('base')
863
        self.assertEquals(
4083.1.7 by Andrew Bennetts
Fix same trivial bug [(x) != (x,)] in test_remote and test_smart.
864
            SuccessfulSmartServerResponse(('',)), response)
4078.2.1 by Robert Collins
Add a Branch.get_parent remote call for RemoteBranch.
865
866
    def test_get_parent_something(self):
867
        base_branch = self.make_branch('base')
868
        base_branch.set_parent(self.get_url('foo'))
869
        request = smart.branch.SmartServerBranchGetParent(self.get_transport())
870
        response = request.execute('base')
871
        self.assertEquals(
4083.1.7 by Andrew Bennetts
Fix same trivial bug [(x) != (x,)] in test_remote and test_smart.
872
            SuccessfulSmartServerResponse(("../foo",)),
4078.2.1 by Robert Collins
Add a Branch.get_parent remote call for RemoteBranch.
873
            response)
874
875
4288.1.7 by Robert Collins
Add new remote server verb Branch.set_parent_location, dropping roundtrips further on push operations.
876
class TestSmartServerBranchRequestSetParent(tests.TestCaseWithMemoryTransport):
877
878
    def test_set_parent_none(self):
879
        branch = self.make_branch('base', format="1.9")
4288.1.9 by Robert Collins
Fix up test usable of _set_parent_location on unlocked branches.
880
        branch.lock_write()
4288.1.7 by Robert Collins
Add new remote server verb Branch.set_parent_location, dropping roundtrips further on push operations.
881
        branch._set_parent_location('foo')
4288.1.9 by Robert Collins
Fix up test usable of _set_parent_location on unlocked branches.
882
        branch.unlock()
4288.1.7 by Robert Collins
Add new remote server verb Branch.set_parent_location, dropping roundtrips further on push operations.
883
        request = smart.branch.SmartServerBranchRequestSetParentLocation(
884
            self.get_transport())
885
        branch_token = branch.lock_write()
886
        repo_token = branch.repository.lock_write()
887
        try:
888
            response = request.execute('base', branch_token, repo_token, '')
889
        finally:
890
            branch.repository.unlock()
891
            branch.unlock()
892
        self.assertEqual(SuccessfulSmartServerResponse(()), response)
893
        self.assertEqual(None, branch.get_parent())
894
895
    def test_set_parent_something(self):
896
        branch = self.make_branch('base', format="1.9")
897
        request = smart.branch.SmartServerBranchRequestSetParentLocation(
898
            self.get_transport())
899
        branch_token = branch.lock_write()
900
        repo_token = branch.repository.lock_write()
901
        try:
902
            response = request.execute('base', branch_token, repo_token,
903
            'http://bar/')
904
        finally:
905
            branch.repository.unlock()
906
            branch.unlock()
907
        self.assertEqual(SuccessfulSmartServerResponse(()), response)
908
        self.assertEqual('http://bar/', branch.get_parent())
909
910
4084.2.1 by Robert Collins
Make accessing a branch.tags.get_tag_dict use a smart[er] method rather than VFS calls and real objects.
911
class TestSmartServerBranchRequestGetTagsBytes(tests.TestCaseWithMemoryTransport):
4556.2.2 by Andrew Bennetts
Handle failures more gracefully.
912
    # Only called when the branch format and tags match [yay factory
913
    # methods] so only need to test straight forward cases.
4084.2.1 by Robert Collins
Make accessing a branch.tags.get_tag_dict use a smart[er] method rather than VFS calls and real objects.
914
915
    def test_get_bytes(self):
916
        base_branch = self.make_branch('base')
917
        request = smart.branch.SmartServerBranchGetTagsBytes(
918
            self.get_transport())
919
        response = request.execute('base')
920
        self.assertEquals(
921
            SuccessfulSmartServerResponse(('',)), response)
922
923
3691.2.5 by Martin Pool
Add Branch.get_stacked_on_url rpc and tests for same
924
class TestSmartServerBranchRequestGetStackedOnURL(tests.TestCaseWithMemoryTransport):
925
926
    def test_get_stacked_on_url(self):
927
        base_branch = self.make_branch('base', format='1.6')
928
        stacked_branch = self.make_branch('stacked', format='1.6')
929
        # typically should be relative
930
        stacked_branch.set_stacked_on_url('../base')
931
        request = smart.branch.SmartServerBranchRequestGetStackedOnURL(
932
            self.get_transport())
933
        response = request.execute('stacked')
934
        self.assertEquals(
935
            SmartServerResponse(('ok', '../base')),
936
            response)
937
938
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
939
class TestSmartServerBranchRequestLockWrite(tests.TestCaseWithMemoryTransport):
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
940
941
    def setUp(self):
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
942
        tests.TestCaseWithMemoryTransport.setUp(self)
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
943
944
    def test_lock_write_on_unlocked_branch(self):
945
        backing = self.get_transport()
946
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
947
        branch = self.make_branch('.', format='knit')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
948
        repository = branch.repository
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
949
        response = request.execute('')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
950
        branch_nonce = branch.control_files._lock.peek().get('nonce')
951
        repository_nonce = repository.control_files._lock.peek().get('nonce')
952
        self.assertEqual(
953
            SmartServerResponse(('ok', branch_nonce, repository_nonce)),
954
            response)
955
        # The branch (and associated repository) is now locked.  Verify that
956
        # with a new branch object.
957
        new_branch = repository.bzrdir.open_branch()
958
        self.assertRaises(errors.LockContention, new_branch.lock_write)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
959
        # Cleanup
960
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
961
        response = request.execute('', branch_nonce, repository_nonce)
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
962
963
    def test_lock_write_on_locked_branch(self):
964
        backing = self.get_transport()
965
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
966
        branch = self.make_branch('.')
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
967
        branch_token = branch.lock_write()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
968
        branch.leave_lock_in_place()
969
        branch.unlock()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
970
        response = request.execute('')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
971
        self.assertEqual(
972
            SmartServerResponse(('LockContention',)), response)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
973
        # Cleanup
974
        branch.lock_write(branch_token)
975
        branch.dont_leave_lock_in_place()
976
        branch.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
977
978
    def test_lock_write_with_tokens_on_locked_branch(self):
979
        backing = self.get_transport()
980
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
981
        branch = self.make_branch('.', format='knit')
2018.5.142 by Andrew Bennetts
Change Branch.lock_token to only accept and receive the branch lock token (rather than the branch and repo lock tokens).
982
        branch_token = branch.lock_write()
983
        repo_token = branch.repository.lock_write()
984
        branch.repository.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
985
        branch.leave_lock_in_place()
986
        branch.repository.leave_lock_in_place()
987
        branch.unlock()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
988
        response = request.execute('',
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
989
                                   branch_token, repo_token)
990
        self.assertEqual(
991
            SmartServerResponse(('ok', branch_token, repo_token)), response)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
992
        # Cleanup
993
        branch.repository.lock_write(repo_token)
994
        branch.repository.dont_leave_lock_in_place()
995
        branch.repository.unlock()
996
        branch.lock_write(branch_token)
997
        branch.dont_leave_lock_in_place()
998
        branch.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
999
1000
    def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
1001
        backing = self.get_transport()
1002
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1003
        branch = self.make_branch('.', format='knit')
2018.5.142 by Andrew Bennetts
Change Branch.lock_token to only accept and receive the branch lock token (rather than the branch and repo lock tokens).
1004
        branch_token = branch.lock_write()
1005
        repo_token = branch.repository.lock_write()
1006
        branch.repository.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1007
        branch.leave_lock_in_place()
1008
        branch.repository.leave_lock_in_place()
1009
        branch.unlock()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1010
        response = request.execute('',
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1011
                                   branch_token+'xxx', repo_token)
1012
        self.assertEqual(
1013
            SmartServerResponse(('TokenMismatch',)), response)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
1014
        # Cleanup
1015
        branch.repository.lock_write(repo_token)
1016
        branch.repository.dont_leave_lock_in_place()
1017
        branch.repository.unlock()
1018
        branch.lock_write(branch_token)
1019
        branch.dont_leave_lock_in_place()
1020
        branch.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1021
1022
    def test_lock_write_on_locked_repo(self):
1023
        backing = self.get_transport()
1024
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1025
        branch = self.make_branch('.', format='knit')
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
1026
        repo = branch.repository
1027
        repo_token = repo.lock_write()
1028
        repo.leave_lock_in_place()
1029
        repo.unlock()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1030
        response = request.execute('')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1031
        self.assertEqual(
1032
            SmartServerResponse(('LockContention',)), response)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
1033
        # Cleanup
1034
        repo.lock_write(repo_token)
1035
        repo.dont_leave_lock_in_place()
1036
        repo.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1037
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.
1038
    def test_lock_write_on_readonly_transport(self):
1039
        backing = self.get_readonly_transport()
1040
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
1041
        branch = self.make_branch('.')
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1042
        root = self.get_transport().clone('/')
1043
        path = urlutils.relative_url(root.base, self.get_transport().base)
1044
        response = request.execute(path)
2872.5.3 by Martin Pool
Pass back LockFailed from smart server lock methods
1045
        error_name, lock_str, why_str = response.args
1046
        self.assertFalse(response.is_successful())
1047
        self.assertEqual('LockFailed', error_name)
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.
1048
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1049
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1050
class TestSmartServerBranchRequestUnlock(tests.TestCaseWithMemoryTransport):
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1051
1052
    def setUp(self):
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1053
        tests.TestCaseWithMemoryTransport.setUp(self)
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1054
1055
    def test_unlock_on_locked_branch_and_repo(self):
1056
        backing = self.get_transport()
1057
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1058
        branch = self.make_branch('.', format='knit')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1059
        # Lock the branch
2018.5.142 by Andrew Bennetts
Change Branch.lock_token to only accept and receive the branch lock token (rather than the branch and repo lock tokens).
1060
        branch_token = branch.lock_write()
1061
        repo_token = branch.repository.lock_write()
1062
        branch.repository.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1063
        # Unlock the branch (and repo) object, leaving the physical locks
1064
        # in place.
1065
        branch.leave_lock_in_place()
1066
        branch.repository.leave_lock_in_place()
1067
        branch.unlock()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1068
        response = request.execute('',
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1069
                                   branch_token, repo_token)
1070
        self.assertEqual(
1071
            SmartServerResponse(('ok',)), response)
1072
        # The branch is now unlocked.  Verify that with a new branch
1073
        # object.
1074
        new_branch = branch.bzrdir.open_branch()
1075
        new_branch.lock_write()
1076
        new_branch.unlock()
1077
1078
    def test_unlock_on_unlocked_branch_unlocked_repo(self):
1079
        backing = self.get_transport()
1080
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1081
        branch = self.make_branch('.', format='knit')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1082
        response = request.execute(
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1083
            '', 'branch token', 'repo token')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1084
        self.assertEqual(
1085
            SmartServerResponse(('TokenMismatch',)), response)
1086
1087
    def test_unlock_on_unlocked_branch_locked_repo(self):
1088
        backing = self.get_transport()
1089
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1090
        branch = self.make_branch('.', format='knit')
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1091
        # Lock the repository.
1092
        repo_token = branch.repository.lock_write()
1093
        branch.repository.leave_lock_in_place()
1094
        branch.repository.unlock()
1095
        # Issue branch lock_write request on the unlocked branch (with locked
1096
        # repo).
1097
        response = request.execute(
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1098
            '', 'branch token', repo_token)
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1099
        self.assertEqual(
1100
            SmartServerResponse(('TokenMismatch',)), response)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
1101
        # Cleanup
1102
        branch.repository.lock_write(repo_token)
1103
        branch.repository.dont_leave_lock_in_place()
1104
        branch.repository.unlock()
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1105
1106
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1107
class TestSmartServerRepositoryRequest(tests.TestCaseWithMemoryTransport):
2018.5.56 by Robert Collins
Factor out code we expect to be common in SmartServerRequestHasRevision to SmartServerRepositoryRequest (Robert Collins, Vincent Ladeuil).
1108
1109
    def test_no_repository(self):
1110
        """Raise NoRepositoryPresent when there is a bzrdir and no repo."""
1111
        # we test this using a shared repository above the named path,
1112
        # thus checking the right search logic is used - that is, that
1113
        # its the exact path being looked at and the server is not
1114
        # searching.
1115
        backing = self.get_transport()
2018.5.58 by Wouter van Heyst
Small test fixes to reflect naming and documentation
1116
        request = smart.repository.SmartServerRepositoryRequest(backing)
2018.5.56 by Robert Collins
Factor out code we expect to be common in SmartServerRequestHasRevision to SmartServerRepositoryRequest (Robert Collins, Vincent Ladeuil).
1117
        self.make_repository('.', shared=True)
1118
        self.make_bzrdir('subdir')
1119
        self.assertRaises(errors.NoRepositoryPresent,
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1120
            request.execute, 'subdir')
1121
1122
3441.5.4 by Andrew Bennetts
Fix test failures, and add some tests for the remote graph heads RPC.
1123
class TestSmartServerRepositoryGetParentMap(tests.TestCaseWithMemoryTransport):
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.
1124
3211.5.3 by Robert Collins
Adjust size of batch and change gzip comments to bzip2.
1125
    def test_trivial_bzipped(self):
1126
        # This tests that the wire encoding is actually bzipped
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.
1127
        backing = self.get_transport()
1128
        request = smart.repository.SmartServerRepositoryGetParentMap(backing)
1129
        tree = self.make_branch_and_memory_tree('.')
1130
1131
        self.assertEqual(None,
2692.1.24 by Andrew Bennetts
Merge from bzr.dev.
1132
            request.execute('', 'missing-id'))
4190.1.3 by Robert Collins
Allow optional inclusion of ghost data in server get_parent_map calls.
1133
        # Note that it returns a body that is bzipped.
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.
1134
        self.assertEqual(
3211.5.2 by Robert Collins
Change RemoteRepository.get_parent_map to use bz2 not gzip for compression.
1135
            SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
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.
1136
            request.do_body('\n\n0\n'))
1137
4190.1.3 by Robert Collins
Allow optional inclusion of ghost data in server get_parent_map calls.
1138
    def test_trivial_include_missing(self):
1139
        backing = self.get_transport()
1140
        request = smart.repository.SmartServerRepositoryGetParentMap(backing)
1141
        tree = self.make_branch_and_memory_tree('.')
1142
1143
        self.assertEqual(None,
1144
            request.execute('', 'missing-id', 'include-missing:'))
1145
        self.assertEqual(
1146
            SuccessfulSmartServerResponse(('ok', ),
1147
                bz2.compress('missing:missing-id')),
1148
            request.do_body('\n\n0\n'))
1149
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.
1150
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1151
class TestSmartServerRepositoryGetRevisionGraph(tests.TestCaseWithMemoryTransport):
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1152
1153
    def test_none_argument(self):
1154
        backing = self.get_transport()
1155
        request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
1156
        tree = self.make_branch_and_memory_tree('.')
1157
        tree.lock_write()
1158
        tree.add('')
1159
        r1 = tree.commit('1st commit')
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1160
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1161
        tree.unlock()
1162
1163
        # the lines of revision_id->revision_parent_list has no guaranteed
1164
        # order coming out of a dict, so sort both our test and response
1165
        lines = sorted([' '.join([r2, r1]), r1])
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1166
        response = request.execute('', '')
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1167
        response.body = '\n'.join(sorted(response.body.split('\n')))
1168
2018.5.83 by Andrew Bennetts
Fix some test failures caused by the switch from unicode to UTF-8-encoded strs for revision IDs.
1169
        self.assertEqual(
1170
            SmartServerResponse(('ok', ), '\n'.join(lines)), response)
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1171
1172
    def test_specific_revision_argument(self):
1173
        backing = self.get_transport()
1174
        request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
1175
        tree = self.make_branch_and_memory_tree('.')
1176
        tree.lock_write()
1177
        tree.add('')
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1178
        rev_id_utf8 = u'\xc9'.encode('utf-8')
1179
        r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
1180
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1181
        tree.unlock()
1182
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1183
        self.assertEqual(SmartServerResponse(('ok', ), rev_id_utf8),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1184
            request.execute('', rev_id_utf8))
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1185
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1186
    def test_no_such_revision(self):
1187
        backing = self.get_transport()
1188
        request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
1189
        tree = self.make_branch_and_memory_tree('.')
1190
        tree.lock_write()
1191
        tree.add('')
1192
        r1 = tree.commit('1st commit')
1193
        tree.unlock()
1194
2018.14.1 by Andrew Bennetts
Update to current hpss branch? Fix lots of test failures.
1195
        # Note that it still returns body (of zero bytes).
1196
        self.assertEqual(
1197
            SmartServerResponse(('nosuchrevision', 'missingrevision', ), ''),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1198
            request.execute('', 'missingrevision'))
1199
1200
4419.2.6 by Andrew Bennetts
Add tests for server-side logic, and fix the bugs exposed by those tests.
1201
class TestSmartServerRepositoryGetRevIdForRevno(tests.TestCaseWithMemoryTransport):
1202
1203
    def test_revno_found(self):
1204
        backing = self.get_transport()
1205
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
1206
        tree = self.make_branch_and_memory_tree('.')
1207
        tree.lock_write()
1208
        tree.add('')
1209
        rev1_id_utf8 = u'\xc8'.encode('utf-8')
1210
        rev2_id_utf8 = u'\xc9'.encode('utf-8')
1211
        tree.commit('1st commit', rev_id=rev1_id_utf8)
1212
        tree.commit('2nd commit', rev_id=rev2_id_utf8)
1213
        tree.unlock()
1214
1215
        self.assertEqual(SmartServerResponse(('ok', rev1_id_utf8)),
1216
            request.execute('', 1, (2, rev2_id_utf8)))
1217
1218
    def test_known_revid_missing(self):
1219
        backing = self.get_transport()
1220
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
1221
        repo = self.make_repository('.')
1222
        self.assertEqual(
1223
            FailedSmartServerResponse(('nosuchrevision', 'ghost')),
1224
            request.execute('', 1, (2, 'ghost')))
1225
1226
    def test_history_incomplete(self):
1227
        backing = self.get_transport()
1228
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
1229
        parent = self.make_branch_and_memory_tree('parent', format='1.9')
4526.9.21 by Robert Collins
Fix test_smart's test_history_incomplete to generate a good tree before committing.
1230
        parent.lock_write()
1231
        parent.add([''], ['TREE_ROOT'])
4419.2.6 by Andrew Bennetts
Add tests for server-side logic, and fix the bugs exposed by those tests.
1232
        r1 = parent.commit(message='first commit')
1233
        r2 = parent.commit(message='second commit')
4526.9.21 by Robert Collins
Fix test_smart's test_history_incomplete to generate a good tree before committing.
1234
        parent.unlock()
4419.2.6 by Andrew Bennetts
Add tests for server-side logic, and fix the bugs exposed by those tests.
1235
        local = self.make_branch_and_memory_tree('local', format='1.9')
1236
        local.branch.pull(parent.branch)
1237
        local.set_parent_ids([r2])
1238
        r3 = local.commit(message='local commit')
1239
        local.branch.create_clone_on_transport(
1240
            self.get_transport('stacked'), stacked_on=self.get_url('parent'))
1241
        self.assertEqual(
1242
            SmartServerResponse(('history-incomplete', 2, r2)),
1243
            request.execute('stacked', 1, (3, r3)))
1244
4476.3.68 by Andrew Bennetts
Review comments from John.
1245
4070.9.14 by Andrew Bennetts
Tweaks requested by Robert's review.
1246
class TestSmartServerRepositoryGetStream(tests.TestCaseWithMemoryTransport):
1247
1248
    def make_two_commit_repo(self):
1249
        tree = self.make_branch_and_memory_tree('.')
1250
        tree.lock_write()
1251
        tree.add('')
1252
        r1 = tree.commit('1st commit')
1253
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1254
        tree.unlock()
1255
        repo = tree.branch.repository
1256
        return repo, r1, r2
1257
1258
    def test_ancestry_of(self):
1259
        """The search argument may be a 'ancestry-of' some heads'."""
1260
        backing = self.get_transport()
1261
        request = smart.repository.SmartServerRepositoryGetStream(backing)
1262
        repo, r1, r2 = self.make_two_commit_repo()
1263
        fetch_spec = ['ancestry-of', r2]
1264
        lines = '\n'.join(fetch_spec)
1265
        request.execute('', repo._format.network_name())
1266
        response = request.do_body(lines)
1267
        self.assertEqual(('ok',), response.args)
1268
        stream_bytes = ''.join(response.body_stream)
1269
        self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1270
1271
    def test_search(self):
1272
        """The search argument may be a 'search' of some explicit keys."""
1273
        backing = self.get_transport()
1274
        request = smart.repository.SmartServerRepositoryGetStream(backing)
1275
        repo, r1, r2 = self.make_two_commit_repo()
1276
        fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
1277
        lines = '\n'.join(fetch_spec)
1278
        request.execute('', repo._format.network_name())
1279
        response = request.do_body(lines)
1280
        self.assertEqual(('ok',), response.args)
1281
        stream_bytes = ''.join(response.body_stream)
1282
        self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1283
1284
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1285
class TestSmartServerRequestHasRevision(tests.TestCaseWithMemoryTransport):
2018.5.56 by Robert Collins
Factor out code we expect to be common in SmartServerRequestHasRevision to SmartServerRepositoryRequest (Robert Collins, Vincent Ladeuil).
1286
1287
    def test_missing_revision(self):
1288
        """For a missing revision, ('no', ) is returned."""
1289
        backing = self.get_transport()
1290
        request = smart.repository.SmartServerRequestHasRevision(backing)
1291
        self.make_repository('.')
1292
        self.assertEqual(SmartServerResponse(('no', )),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1293
            request.execute('', 'revid'))
2018.5.56 by Robert Collins
Factor out code we expect to be common in SmartServerRequestHasRevision to SmartServerRepositoryRequest (Robert Collins, Vincent Ladeuil).
1294
1295
    def test_present_revision(self):
2018.5.158 by Andrew Bennetts
Return 'yes'/'no' rather than 'ok'/'no' from the Repository.has_revision smart command.
1296
        """For a present revision, ('yes', ) is returned."""
2018.5.56 by Robert Collins
Factor out code we expect to be common in SmartServerRequestHasRevision to SmartServerRepositoryRequest (Robert Collins, Vincent Ladeuil).
1297
        backing = self.get_transport()
1298
        request = smart.repository.SmartServerRequestHasRevision(backing)
1299
        tree = self.make_branch_and_memory_tree('.')
1300
        tree.lock_write()
1301
        tree.add('')
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1302
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1303
        r1 = tree.commit('a commit', rev_id=rev_id_utf8)
2018.5.56 by Robert Collins
Factor out code we expect to be common in SmartServerRequestHasRevision to SmartServerRepositoryRequest (Robert Collins, Vincent Ladeuil).
1304
        tree.unlock()
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1305
        self.assertTrue(tree.branch.repository.has_revision(rev_id_utf8))
2018.5.158 by Andrew Bennetts
Return 'yes'/'no' rather than 'ok'/'no' from the Repository.has_revision smart command.
1306
        self.assertEqual(SmartServerResponse(('yes', )),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1307
            request.execute('', rev_id_utf8))
1308
1309
1310
class TestSmartServerRepositoryGatherStats(tests.TestCaseWithMemoryTransport):
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1311
1312
    def test_empty_revid(self):
1313
        """With an empty revid, we get only size an number and revisions"""
1314
        backing = self.get_transport()
1315
        request = smart.repository.SmartServerRepositoryGatherStats(backing)
1316
        repository = self.make_repository('.')
1317
        stats = repository.gather_stats()
3350.6.4 by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores.
1318
        expected_body = 'revisions: 0\n'
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1319
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1320
                         request.execute('', '', 'no'))
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1321
1322
    def test_revid_with_committers(self):
1323
        """For a revid we get more infos."""
1324
        backing = self.get_transport()
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1325
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1326
        request = smart.repository.SmartServerRepositoryGatherStats(backing)
1327
        tree = self.make_branch_and_memory_tree('.')
1328
        tree.lock_write()
1329
        tree.add('')
1330
        # Let's build a predictable result
1331
        tree.commit('a commit', timestamp=123456.2, timezone=3600)
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1332
        tree.commit('a commit', timestamp=654321.4, timezone=0,
1333
                    rev_id=rev_id_utf8)
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1334
        tree.unlock()
1335
1336
        stats = tree.branch.repository.gather_stats()
1337
        expected_body = ('firstrev: 123456.200 3600\n'
1338
                         'latestrev: 654321.400 0\n'
3350.6.4 by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores.
1339
                         'revisions: 2\n')
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1340
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1341
                         request.execute('',
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1342
                                         rev_id_utf8, 'no'))
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1343
1344
    def test_not_empty_repository_with_committers(self):
1345
        """For a revid and requesting committers we get the whole thing."""
1346
        backing = self.get_transport()
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1347
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1348
        request = smart.repository.SmartServerRepositoryGatherStats(backing)
1349
        tree = self.make_branch_and_memory_tree('.')
1350
        tree.lock_write()
1351
        tree.add('')
1352
        # Let's build a predictable result
1353
        tree.commit('a commit', timestamp=123456.2, timezone=3600,
1354
                    committer='foo')
1355
        tree.commit('a commit', timestamp=654321.4, timezone=0,
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1356
                    committer='bar', rev_id=rev_id_utf8)
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1357
        tree.unlock()
1358
        stats = tree.branch.repository.gather_stats()
1359
1360
        expected_body = ('committers: 2\n'
1361
                         'firstrev: 123456.200 3600\n'
1362
                         'latestrev: 654321.400 0\n'
3350.6.4 by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores.
1363
                         'revisions: 2\n')
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1364
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1365
                         request.execute('',
2018.5.148 by Andrew Bennetts
Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.
1366
                                         rev_id_utf8, 'yes'))
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1367
1368
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1369
class TestSmartServerRepositoryIsShared(tests.TestCaseWithMemoryTransport):
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1370
1371
    def test_is_shared(self):
1372
        """For a shared repository, ('yes', ) is returned."""
1373
        backing = self.get_transport()
1374
        request = smart.repository.SmartServerRepositoryIsShared(backing)
1375
        self.make_repository('.', shared=True)
1376
        self.assertEqual(SmartServerResponse(('yes', )),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1377
            request.execute('', ))
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1378
1379
    def test_is_not_shared(self):
2018.5.58 by Wouter van Heyst
Small test fixes to reflect naming and documentation
1380
        """For a shared repository, ('no', ) is returned."""
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1381
        backing = self.get_transport()
1382
        request = smart.repository.SmartServerRepositoryIsShared(backing)
1383
        self.make_repository('.', shared=False)
1384
        self.assertEqual(SmartServerResponse(('no', )),
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1385
            request.execute('', ))
1386
1387
1388
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithMemoryTransport):
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1389
1390
    def test_lock_write_on_unlocked_repo(self):
1391
        backing = self.get_transport()
1392
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1393
        repository = self.make_repository('.', format='knit')
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1394
        response = request.execute('')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1395
        nonce = repository.control_files._lock.peek().get('nonce')
1396
        self.assertEqual(SmartServerResponse(('ok', nonce)), response)
1397
        # The repository is now locked.  Verify that with a new repository
1398
        # object.
1399
        new_repo = repository.bzrdir.open_repository()
1400
        self.assertRaises(errors.LockContention, new_repo.lock_write)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
1401
        # Cleanup
1402
        request = smart.repository.SmartServerRepositoryUnlock(backing)
1403
        response = request.execute('', nonce)
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1404
1405
    def test_lock_write_on_locked_repo(self):
1406
        backing = self.get_transport()
1407
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1408
        repository = self.make_repository('.', format='knit')
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
1409
        repo_token = repository.lock_write()
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1410
        repository.leave_lock_in_place()
1411
        repository.unlock()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1412
        response = request.execute('')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1413
        self.assertEqual(
1414
            SmartServerResponse(('LockContention',)), response)
4327.1.10 by Vincent Ladeuil
Fix 10 more lock-related test failures.
1415
        # Cleanup
1416
        repository.lock_write(repo_token)
1417
        repository.dont_leave_lock_in_place()
1418
        repository.unlock()
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1419
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.
1420
    def test_lock_write_on_readonly_transport(self):
1421
        backing = self.get_readonly_transport()
1422
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1423
        repository = self.make_repository('.', format='knit')
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.
1424
        response = request.execute('')
2872.5.3 by Martin Pool
Pass back LockFailed from smart server lock methods
1425
        self.assertFalse(response.is_successful())
1426
        self.assertEqual('LockFailed', response.args[0])
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.
1427
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1428
4144.3.1 by Andrew Bennetts
Add Repository.insert_stream_locked server-side implementation, plus tests for server-side _translate_error.
1429
class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
1430
1431
    def make_empty_byte_stream(self, repo):
1432
        byte_stream = smart.repository._stream_to_byte_stream([], repo._format)
1433
        return ''.join(byte_stream)
1434
1435
1436
class TestSmartServerRepositoryInsertStream(TestInsertStreamBase):
1437
1438
    def test_insert_stream_empty(self):
1439
        backing = self.get_transport()
1440
        request = smart.repository.SmartServerRepositoryInsertStream(backing)
1441
        repository = self.make_repository('.')
1442
        response = request.execute('', '')
1443
        self.assertEqual(None, response)
1444
        response = request.do_chunk(self.make_empty_byte_stream(repository))
1445
        self.assertEqual(None, response)
1446
        response = request.do_end()
1447
        self.assertEqual(SmartServerResponse(('ok', )), response)
1448
        
1449
1450
class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
1451
1452
    def test_insert_stream_empty(self):
1453
        backing = self.get_transport()
1454
        request = smart.repository.SmartServerRepositoryInsertStreamLocked(
1455
            backing)
1456
        repository = self.make_repository('.', format='knit')
1457
        lock_token = repository.lock_write()
1458
        response = request.execute('', '', lock_token)
1459
        self.assertEqual(None, response)
1460
        response = request.do_chunk(self.make_empty_byte_stream(repository))
1461
        self.assertEqual(None, response)
1462
        response = request.do_end()
1463
        self.assertEqual(SmartServerResponse(('ok', )), response)
1464
        repository.unlock()
1465
1466
    def test_insert_stream_with_wrong_lock_token(self):
1467
        backing = self.get_transport()
1468
        request = smart.repository.SmartServerRepositoryInsertStreamLocked(
1469
            backing)
1470
        repository = self.make_repository('.', format='knit')
1471
        lock_token = repository.lock_write()
1472
        self.assertRaises(
1473
            errors.TokenMismatch, request.execute, '', '', 'wrong-token')
1474
        repository.unlock()
1475
1476
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1477
class TestSmartServerRepositoryUnlock(tests.TestCaseWithMemoryTransport):
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1478
1479
    def setUp(self):
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1480
        tests.TestCaseWithMemoryTransport.setUp(self)
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1481
1482
    def test_unlock_on_locked_repo(self):
1483
        backing = self.get_transport()
1484
        request = smart.repository.SmartServerRepositoryUnlock(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1485
        repository = self.make_repository('.', format='knit')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1486
        token = repository.lock_write()
1487
        repository.leave_lock_in_place()
1488
        repository.unlock()
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1489
        response = request.execute('', token)
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1490
        self.assertEqual(
1491
            SmartServerResponse(('ok',)), response)
1492
        # The repository is now unlocked.  Verify that with a new repository
1493
        # object.
1494
        new_repo = repository.bzrdir.open_repository()
1495
        new_repo.lock_write()
1496
        new_repo.unlock()
1497
1498
    def test_unlock_on_unlocked_repo(self):
1499
        backing = self.get_transport()
1500
        request = smart.repository.SmartServerRepositoryUnlock(backing)
3015.2.12 by Robert Collins
Make test_smart use specific formats as needed to exercise locked and unlocked repositories.
1501
        repository = self.make_repository('.', format='knit')
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1502
        response = request.execute('', 'some token')
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1503
        self.assertEqual(
1504
            SmartServerResponse(('TokenMismatch',)), response)
1505
1506
2692.1.1 by Andrew Bennetts
Add translate_client_path method to SmartServerRequest.
1507
class TestSmartServerIsReadonly(tests.TestCaseWithMemoryTransport):
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.
1508
1509
    def test_is_readonly_no(self):
1510
        backing = self.get_transport()
1511
        request = smart.request.SmartServerIsReadonly(backing)
1512
        response = request.execute()
1513
        self.assertEqual(
1514
            SmartServerResponse(('no',)), response)
1515
1516
    def test_is_readonly_yes(self):
1517
        backing = self.get_readonly_transport()
1518
        request = smart.request.SmartServerIsReadonly(backing)
1519
        response = request.execute()
1520
        self.assertEqual(
1521
            SmartServerResponse(('yes',)), response)
1522
1523
4017.3.4 by Robert Collins
Create a verb for Repository.set_make_working_trees.
1524
class TestSmartServerRepositorySetMakeWorkingTrees(tests.TestCaseWithMemoryTransport):
1525
1526
    def test_set_false(self):
1527
        backing = self.get_transport()
1528
        repo = self.make_repository('.', shared=True)
1529
        repo.set_make_working_trees(True)
1530
        request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
1531
        request = request_class(backing)
1532
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
1533
            request.execute('', 'False'))
1534
        repo = repo.bzrdir.open_repository()
1535
        self.assertFalse(repo.make_working_trees())
1536
1537
    def test_set_true(self):
1538
        backing = self.get_transport()
1539
        repo = self.make_repository('.', shared=True)
1540
        repo.set_make_working_trees(False)
1541
        request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
1542
        request = request_class(backing)
1543
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
1544
            request.execute('', 'True'))
1545
        repo = repo.bzrdir.open_repository()
1546
        self.assertTrue(repo.make_working_trees())
1547
1548
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1549
class TestSmartServerPackRepositoryAutopack(tests.TestCaseWithTransport):
1550
1551
    def make_repo_needing_autopacking(self, path='.'):
1552
        # Make a repo in need of autopacking.
1553
        tree = self.make_branch_and_tree('.', format='pack-0.92')
1554
        repo = tree.branch.repository
1555
        # monkey-patch the pack collection to disable autopacking
1556
        repo._pack_collection._max_pack_count = lambda count: count
1557
        for x in range(10):
1558
            tree.commit('commit %s' % x)
1559
        self.assertEqual(10, len(repo._pack_collection.names()))
1560
        del repo._pack_collection._max_pack_count
1561
        return repo
1562
1563
    def test_autopack_needed(self):
1564
        repo = self.make_repo_needing_autopacking()
4145.1.6 by Robert Collins
More test fallout, but all caught now.
1565
        repo.lock_write()
1566
        self.addCleanup(repo.unlock)
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1567
        backing = self.get_transport()
1568
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
1569
            backing)
1570
        response = request.execute('')
1571
        self.assertEqual(SmartServerResponse(('ok',)), response)
1572
        repo._pack_collection.reload_pack_names()
1573
        self.assertEqual(1, len(repo._pack_collection.names()))
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1574
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1575
    def test_autopack_not_needed(self):
1576
        tree = self.make_branch_and_tree('.', format='pack-0.92')
1577
        repo = tree.branch.repository
4145.1.6 by Robert Collins
More test fallout, but all caught now.
1578
        repo.lock_write()
1579
        self.addCleanup(repo.unlock)
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1580
        for x in range(9):
1581
            tree.commit('commit %s' % x)
1582
        backing = self.get_transport()
1583
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
1584
            backing)
1585
        response = request.execute('')
1586
        self.assertEqual(SmartServerResponse(('ok',)), response)
1587
        repo._pack_collection.reload_pack_names()
1588
        self.assertEqual(9, len(repo._pack_collection.names()))
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1589
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1590
    def test_autopack_on_nonpack_format(self):
3801.1.20 by Andrew Bennetts
Return ('ok',) rather than an error the autopack RPC is used on a non-pack repo.
1591
        """A request to autopack a non-pack repo is a no-op."""
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1592
        repo = self.make_repository('.', format='knit')
1593
        backing = self.get_transport()
1594
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
1595
            backing)
1596
        response = request.execute('')
3801.1.20 by Andrew Bennetts
Return ('ok',) rather than an error the autopack RPC is used on a non-pack repo.
1597
        self.assertEqual(SmartServerResponse(('ok',)), response)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1598
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1599
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
1600
class TestHandlers(tests.TestCase):
1601
    """Tests for the request.request_handlers object."""
1602
3526.3.1 by Andrew Bennetts
Remove registrations of defunct HPSS verbs.
1603
    def test_all_registrations_exist(self):
1604
        """All registered request_handlers can be found."""
1605
        # If there's a typo in a register_lazy call, this loop will fail with
1606
        # an AttributeError.
1607
        for key, item in smart.request.request_handlers.iteritems():
1608
            pass
1609
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1610
    def assertHandlerEqual(self, verb, handler):
1611
        self.assertEqual(smart.request.request_handlers.get(verb), handler)
1612
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
1613
    def test_registered_methods(self):
1614
        """Test that known methods are registered to the correct object."""
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1615
        self.assertHandlerEqual('Branch.get_config_file',
2018.5.59 by Robert Collins
Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).
1616
            smart.branch.SmartServerBranchGetConfigFile)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1617
        self.assertHandlerEqual('Branch.get_parent',
4078.2.1 by Robert Collins
Add a Branch.get_parent remote call for RemoteBranch.
1618
            smart.branch.SmartServerBranchGetParent)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1619
        self.assertHandlerEqual('Branch.get_tags_bytes',
4084.2.1 by Robert Collins
Make accessing a branch.tags.get_tag_dict use a smart[er] method rather than VFS calls and real objects.
1620
            smart.branch.SmartServerBranchGetTagsBytes)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1621
        self.assertHandlerEqual('Branch.lock_write',
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1622
            smart.branch.SmartServerBranchRequestLockWrite)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1623
        self.assertHandlerEqual('Branch.last_revision_info',
2018.5.51 by Wouter van Heyst
Test and implement RemoteBranch.last_revision_info()
1624
            smart.branch.SmartServerBranchRequestLastRevisionInfo)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1625
        self.assertHandlerEqual('Branch.revision_history',
2018.5.38 by Robert Collins
Implement RemoteBranch.revision_history().
1626
            smart.branch.SmartServerRequestRevisionHistory)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1627
        self.assertHandlerEqual('Branch.set_config_option',
4226.2.1 by Robert Collins
Set branch config options via a smart method.
1628
            smart.branch.SmartServerBranchRequestSetConfigOption)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1629
        self.assertHandlerEqual('Branch.set_last_revision',
2018.5.77 by Wouter van Heyst
Fix typo in request_handlers registration of Branch.set_last_revision, and test that registration
1630
            smart.branch.SmartServerBranchRequestSetLastRevision)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1631
        self.assertHandlerEqual('Branch.set_last_revision_info',
2892.2.1 by Andrew Bennetts
Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.
1632
            smart.branch.SmartServerBranchRequestSetLastRevisionInfo)
4288.1.7 by Robert Collins
Add new remote server verb Branch.set_parent_location, dropping roundtrips further on push operations.
1633
        self.assertHandlerEqual('Branch.set_last_revision_ex',
1634
            smart.branch.SmartServerBranchRequestSetLastRevisionEx)
1635
        self.assertHandlerEqual('Branch.set_parent_location',
1636
            smart.branch.SmartServerBranchRequestSetParentLocation)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1637
        self.assertHandlerEqual('Branch.unlock',
2018.5.79 by Andrew Bennetts
Implement RemoteBranch.lock_write/unlock as smart operations.
1638
            smart.branch.SmartServerBranchRequestUnlock)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1639
        self.assertHandlerEqual('BzrDir.find_repository',
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
1640
            smart.bzrdir.SmartServerRequestFindRepositoryV1)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1641
        self.assertHandlerEqual('BzrDir.find_repositoryV2',
3221.3.2 by Robert Collins
* New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for
1642
            smart.bzrdir.SmartServerRequestFindRepositoryV2)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1643
        self.assertHandlerEqual('BzrDirFormat.initialize',
2018.5.42 by Robert Collins
Various hopefully improvements, but wsgi is broken, handing over to spiv :).
1644
            smart.bzrdir.SmartServerRequestInitializeBzrDir)
4436.1.1 by Andrew Bennetts
Rename BzrDirFormat.initialize_ex verb to BzrDirFormat.initialize_ex_1.16.
1645
        self.assertHandlerEqual('BzrDirFormat.initialize_ex_1.16',
4294.2.7 by Robert Collins
Start building up a BzrDir.initialize_ex verb for the smart server.
1646
            smart.bzrdir.SmartServerRequestBzrDirInitializeEx)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1647
        self.assertHandlerEqual('BzrDir.cloning_metadir',
4070.2.3 by Robert Collins
Get BzrDir.cloning_metadir working.
1648
            smart.bzrdir.SmartServerBzrDirRequestCloningMetaDir)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1649
        self.assertHandlerEqual('BzrDir.get_config_file',
1650
            smart.bzrdir.SmartServerBzrDirRequestConfigFile)
1651
        self.assertHandlerEqual('BzrDir.open_branch',
2018.6.1 by Robert Collins
Implement a BzrDir.open_branch smart server method for opening a branch without VFS.
1652
            smart.bzrdir.SmartServerRequestOpenBranch)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1653
        self.assertHandlerEqual('BzrDir.open_branchV2',
4084.2.1 by Robert Collins
Make accessing a branch.tags.get_tag_dict use a smart[er] method rather than VFS calls and real objects.
1654
            smart.bzrdir.SmartServerRequestOpenBranchV2)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1655
        self.assertHandlerEqual('PackRepository.autopack',
3801.1.4 by Andrew Bennetts
Add tests for autopack RPC.
1656
            smart.packrepository.SmartServerPackRepositoryAutopack)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1657
        self.assertHandlerEqual('Repository.gather_stats',
2018.10.2 by v.ladeuil+lp at free
gather_stats server side and request registration
1658
            smart.repository.SmartServerRepositoryGatherStats)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1659
        self.assertHandlerEqual('Repository.get_parent_map',
3172.5.6 by Robert Collins
Create new smart server verb Repository.get_parent_map.
1660
            smart.repository.SmartServerRepositoryGetParentMap)
4419.2.6 by Andrew Bennetts
Add tests for server-side logic, and fix the bugs exposed by those tests.
1661
        self.assertHandlerEqual('Repository.get_rev_id_for_revno',
1662
            smart.repository.SmartServerRepositoryGetRevIdForRevno)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1663
        self.assertHandlerEqual('Repository.get_revision_graph',
2018.5.67 by Wouter van Heyst
Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)
1664
            smart.repository.SmartServerRepositoryGetRevisionGraph)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1665
        self.assertHandlerEqual('Repository.get_stream',
4144.3.1 by Andrew Bennetts
Add Repository.insert_stream_locked server-side implementation, plus tests for server-side _translate_error.
1666
            smart.repository.SmartServerRepositoryGetStream)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1667
        self.assertHandlerEqual('Repository.has_revision',
2018.5.40 by Robert Collins
Implement a remote Repository.has_revision method.
1668
            smart.repository.SmartServerRequestHasRevision)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1669
        self.assertHandlerEqual('Repository.insert_stream',
4144.3.1 by Andrew Bennetts
Add Repository.insert_stream_locked server-side implementation, plus tests for server-side _translate_error.
1670
            smart.repository.SmartServerRepositoryInsertStream)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1671
        self.assertHandlerEqual('Repository.insert_stream_locked',
4144.3.1 by Andrew Bennetts
Add Repository.insert_stream_locked server-side implementation, plus tests for server-side _translate_error.
1672
            smart.repository.SmartServerRepositoryInsertStreamLocked)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1673
        self.assertHandlerEqual('Repository.is_shared',
2018.5.57 by Robert Collins
Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).
1674
            smart.repository.SmartServerRepositoryIsShared)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1675
        self.assertHandlerEqual('Repository.lock_write',
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1676
            smart.repository.SmartServerRepositoryLockWrite)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1677
        self.assertHandlerEqual('Repository.tarball',
2535.3.69 by Andrew Bennetts
Add check for Repository.stream_knit_data_for_revisions to TestHandlers.test_registered_methods.
1678
            smart.repository.SmartServerRepositoryTarball)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1679
        self.assertHandlerEqual('Repository.unlock',
2018.5.78 by Andrew Bennetts
Implement RemoteRepository.lock_write/unlock to expect and send tokens over the
1680
            smart.repository.SmartServerRepositoryUnlock)
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1681
        self.assertHandlerEqual('Transport.is_readonly',
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.
1682
            smart.request.SmartServerIsReadonly)