/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_smart.py

  • Committer: Martin
  • Date: 2010-05-25 17:27:52 UTC
  • mfrom: (5254 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5257.
  • Revision ID: gzlist@googlemail.com-20100525172752-amm089xcikv968sw
Merge bzr.dev to unite with similar changes already made

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_smart.py
1
 
# Copyright (C) 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2006-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
30
30
 
31
31
from bzrlib import (
32
32
    bencode,
 
33
    branch as _mod_branch,
33
34
    bzrdir,
34
35
    errors,
35
36
    pack,
36
 
    smart,
37
37
    tests,
 
38
    transport,
38
39
    urlutils,
39
40
    versionedfile,
40
41
    )
41
 
from bzrlib.branch import Branch, BranchReferenceFormat
42
 
import bzrlib.smart.branch
43
 
import bzrlib.smart.bzrdir, bzrlib.smart.bzrdir as smart_dir
44
 
import bzrlib.smart.packrepository
45
 
import bzrlib.smart.repository
46
 
from bzrlib.smart.request import (
47
 
    FailedSmartServerResponse,
48
 
    SmartServerRequest,
49
 
    SmartServerResponse,
50
 
    SuccessfulSmartServerResponse,
51
 
    )
52
 
from bzrlib.tests import (
53
 
    split_suite_by_re,
54
 
    )
55
 
from bzrlib.transport import chroot, get_transport
 
42
from bzrlib.smart import (
 
43
    branch as smart_branch,
 
44
    bzrdir as smart_dir,
 
45
    repository as smart_repo,
 
46
    packrepository as smart_packrepo,
 
47
    request as smart_req,
 
48
    server,
 
49
    vfs,
 
50
    )
 
51
from bzrlib.tests import test_server
 
52
from bzrlib.transport import (
 
53
    chroot,
 
54
    memory,
 
55
    )
56
56
 
57
57
 
58
58
def load_tests(standard_tests, module, loader):
59
59
    """Multiply tests version and protocol consistency."""
60
60
    # FindRepository tests.
61
 
    bzrdir_mod = bzrlib.smart.bzrdir
62
61
    scenarios = [
63
62
        ("find_repository", {
64
 
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV1}),
 
63
            "_request_class": smart_dir.SmartServerRequestFindRepositoryV1}),
65
64
        ("find_repositoryV2", {
66
 
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV2}),
 
65
            "_request_class": smart_dir.SmartServerRequestFindRepositoryV2}),
67
66
        ("find_repositoryV3", {
68
 
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV3}),
 
67
            "_request_class": smart_dir.SmartServerRequestFindRepositoryV3}),
69
68
        ]
70
 
    to_adapt, result = split_suite_by_re(standard_tests,
 
69
    to_adapt, result = tests.split_suite_by_re(standard_tests,
71
70
        "TestSmartServerRequestFindRepository")
72
 
    v2_only, v1_and_2 = split_suite_by_re(to_adapt,
 
71
    v2_only, v1_and_2 = tests.split_suite_by_re(to_adapt,
73
72
        "_v2")
74
73
    tests.multiply_tests(v1_and_2, scenarios, result)
75
74
    # The first scenario is only applicable to v1 protocols, it is deleted
81
80
class TestCaseWithChrootedTransport(tests.TestCaseWithTransport):
82
81
 
83
82
    def setUp(self):
 
83
        self.vfs_transport_factory = memory.MemoryServer
84
84
        tests.TestCaseWithTransport.setUp(self)
85
85
        self._chroot_server = None
86
86
 
89
89
            backing_transport = tests.TestCaseWithTransport.get_transport(self)
90
90
            self._chroot_server = chroot.ChrootServer(backing_transport)
91
91
            self.start_server(self._chroot_server)
92
 
        t = get_transport(self._chroot_server.get_url())
 
92
        t = transport.get_transport(self._chroot_server.get_url())
93
93
        if relpath is not None:
94
94
            t = t.clone(relpath)
95
95
        return t
96
96
 
97
97
 
98
 
class TestCaseWithSmartMedium(tests.TestCaseWithTransport):
 
98
class TestCaseWithSmartMedium(tests.TestCaseWithMemoryTransport):
99
99
 
100
100
    def setUp(self):
101
101
        super(TestCaseWithSmartMedium, self).setUp()
106
106
        self.transport_server = self.make_transport_server
107
107
 
108
108
    def make_transport_server(self):
109
 
        return smart.server.SmartTCPServer_for_testing('-' + self.id())
 
109
        return test_server.SmartTCPServer_for_testing('-' + self.id())
110
110
 
111
111
    def get_smart_medium(self):
112
112
        """Get a smart medium to use in tests."""
121
121
            None, 'foo')]),('text', [
122
122
            versionedfile.FulltextContentFactory(('k2',), None, None, 'bar')])]
123
123
        fmt = bzrdir.format_registry.get('pack-0.92')().repository_format
124
 
        bytes = smart.repository._stream_to_byte_stream(stream, fmt)
 
124
        bytes = smart_repo._stream_to_byte_stream(stream, fmt)
125
125
        streams = []
126
126
        # Iterate the resulting iterable; checking that we get only one stream
127
127
        # out.
128
 
        fmt, stream = smart.repository._byte_stream_to_stream(bytes)
 
128
        fmt, stream = smart_repo._byte_stream_to_stream(bytes)
129
129
        for kind, substream in stream:
130
130
            streams.append((kind, list(substream)))
131
131
        self.assertLength(1, streams)
135
135
class TestSmartServerResponse(tests.TestCase):
136
136
 
137
137
    def test__eq__(self):
138
 
        self.assertEqual(SmartServerResponse(('ok', )),
139
 
            SmartServerResponse(('ok', )))
140
 
        self.assertEqual(SmartServerResponse(('ok', ), 'body'),
141
 
            SmartServerResponse(('ok', ), 'body'))
142
 
        self.assertNotEqual(SmartServerResponse(('ok', )),
143
 
            SmartServerResponse(('notok', )))
144
 
        self.assertNotEqual(SmartServerResponse(('ok', ), 'body'),
145
 
            SmartServerResponse(('ok', )))
 
138
        self.assertEqual(smart_req.SmartServerResponse(('ok', )),
 
139
            smart_req.SmartServerResponse(('ok', )))
 
140
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
 
141
            smart_req.SmartServerResponse(('ok', ), 'body'))
 
142
        self.assertNotEqual(smart_req.SmartServerResponse(('ok', )),
 
143
            smart_req.SmartServerResponse(('notok', )))
 
144
        self.assertNotEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
 
145
            smart_req.SmartServerResponse(('ok', )))
146
146
        self.assertNotEqual(None,
147
 
            SmartServerResponse(('ok', )))
 
147
            smart_req.SmartServerResponse(('ok', )))
148
148
 
149
149
    def test__str__(self):
150
150
        """SmartServerResponses can be stringified."""
151
151
        self.assertEqual(
152
152
            "<SuccessfulSmartServerResponse args=('args',) body='body'>",
153
 
            str(SuccessfulSmartServerResponse(('args',), 'body')))
 
153
            str(smart_req.SuccessfulSmartServerResponse(('args',), 'body')))
154
154
        self.assertEqual(
155
155
            "<FailedSmartServerResponse args=('args',) body='body'>",
156
 
            str(FailedSmartServerResponse(('args',), 'body')))
 
156
            str(smart_req.FailedSmartServerResponse(('args',), 'body')))
157
157
 
158
158
 
159
159
class TestSmartServerRequest(tests.TestCaseWithMemoryTransport):
160
160
 
161
161
    def test_translate_client_path(self):
162
162
        transport = self.get_transport()
163
 
        request = SmartServerRequest(transport, 'foo/')
 
163
        request = smart_req.SmartServerRequest(transport, 'foo/')
164
164
        self.assertEqual('./', request.translate_client_path('foo/'))
165
165
        self.assertRaises(
166
166
            errors.InvalidURLJoin, request.translate_client_path, 'foo/..')
169
169
        self.assertRaises(
170
170
            errors.PathNotChild, request.translate_client_path, 'bar/')
171
171
        self.assertEqual('./baz', request.translate_client_path('foo/baz'))
 
172
        e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
 
173
        self.assertEqual('./' + urlutils.escape(e_acute),
 
174
                         request.translate_client_path('foo/' + e_acute))
 
175
 
 
176
    def test_translate_client_path_vfs(self):
 
177
        """VfsRequests receive escaped paths rather than raw UTF-8."""
 
178
        transport = self.get_transport()
 
179
        request = vfs.VfsRequest(transport, 'foo/')
 
180
        e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
 
181
        escaped = urlutils.escape('foo/' + e_acute)
 
182
        self.assertEqual('./' + urlutils.escape(e_acute),
 
183
                         request.translate_client_path(escaped))
172
184
 
173
185
    def test_transport_from_client_path(self):
174
186
        transport = self.get_transport()
175
 
        request = SmartServerRequest(transport, 'foo/')
 
187
        request = smart_req.SmartServerRequest(transport, 'foo/')
176
188
        self.assertEqual(
177
189
            transport.base,
178
190
            request.transport_from_client_path('foo/').base)
189
201
        local_result = dir.cloning_metadir()
190
202
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
191
203
        request = request_class(backing)
192
 
        expected = SuccessfulSmartServerResponse(
 
204
        expected = smart_req.SuccessfulSmartServerResponse(
193
205
            (local_result.network_name(),
194
206
            local_result.repository_format.network_name(),
195
207
            ('branch', local_result.get_branch_format().network_name())))
201
213
        referenced_branch = self.make_branch('referenced')
202
214
        dir = self.make_bzrdir('.')
203
215
        local_result = dir.cloning_metadir()
204
 
        reference = BranchReferenceFormat().initialize(dir, referenced_branch)
205
 
        reference_url = BranchReferenceFormat().get_reference(dir)
 
216
        reference = _mod_branch.BranchReferenceFormat().initialize(
 
217
            dir, target_branch=referenced_branch)
 
218
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(dir)
206
219
        # The server shouldn't try to follow the branch reference, so it's fine
207
220
        # if the referenced branch isn't reachable.
208
221
        backing.rename('referenced', 'moved')
209
222
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
210
223
        request = request_class(backing)
211
 
        expected = FailedSmartServerResponse(('BranchReference',))
 
224
        expected = smart_req.FailedSmartServerResponse(('BranchReference',))
212
225
        self.assertEqual(expected, request.execute('', 'False'))
213
226
 
214
227
 
219
232
        """When there is a bzrdir present, the call succeeds."""
220
233
        backing = self.get_transport()
221
234
        self.make_bzrdir('.')
222
 
        request_class = bzrlib.smart.bzrdir.SmartServerRequestCreateRepository
 
235
        request_class = smart_dir.SmartServerRequestCreateRepository
223
236
        request = request_class(backing)
224
237
        reference_bzrdir_format = bzrdir.format_registry.get('pack-0.92')()
225
238
        reference_format = reference_bzrdir_format.repository_format
226
239
        network_name = reference_format.network_name()
227
 
        expected = SuccessfulSmartServerResponse(
 
240
        expected = smart_req.SuccessfulSmartServerResponse(
228
241
            ('ok', 'no', 'no', 'no', network_name))
229
242
        self.assertEqual(expected, request.execute('', network_name, 'True'))
230
243
 
237
250
        backing = self.get_transport()
238
251
        request = self._request_class(backing)
239
252
        self.make_bzrdir('.')
240
 
        self.assertEqual(SmartServerResponse(('norepository', )),
 
253
        self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
241
254
            request.execute(''))
242
255
 
243
256
    def test_nonshared_repository(self):
249
262
        result = self._make_repository_and_result()
250
263
        self.assertEqual(result, request.execute(''))
251
264
        self.make_bzrdir('subdir')
252
 
        self.assertEqual(SmartServerResponse(('norepository', )),
 
265
        self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
253
266
            request.execute('subdir'))
254
267
 
255
268
    def _make_repository_and_result(self, shared=False, format=None):
270
283
            external = 'yes'
271
284
        else:
272
285
            external = 'no'
273
 
        if (smart.bzrdir.SmartServerRequestFindRepositoryV3 ==
 
286
        if (smart_dir.SmartServerRequestFindRepositoryV3 ==
274
287
            self._request_class):
275
 
            return SuccessfulSmartServerResponse(
 
288
            return smart_req.SuccessfulSmartServerResponse(
276
289
                ('ok', '', rich_root, subtrees, external,
277
290
                 repo._format.network_name()))
278
 
        elif (smart.bzrdir.SmartServerRequestFindRepositoryV2 ==
 
291
        elif (smart_dir.SmartServerRequestFindRepositoryV2 ==
279
292
            self._request_class):
280
293
            # All tests so far are on formats, and for non-external
281
294
            # repositories.
282
 
            return SuccessfulSmartServerResponse(
 
295
            return smart_req.SuccessfulSmartServerResponse(
283
296
                ('ok', '', rich_root, subtrees, external))
284
297
        else:
285
 
            return SuccessfulSmartServerResponse(('ok', '', rich_root, subtrees))
 
298
            return smart_req.SuccessfulSmartServerResponse(
 
299
                ('ok', '', rich_root, subtrees))
286
300
 
287
301
    def test_shared_repository(self):
288
302
        """When there is a shared repository, we get 'ok', 'relpath-to-repo'."""
291
305
        result = self._make_repository_and_result(shared=True)
292
306
        self.assertEqual(result, request.execute(''))
293
307
        self.make_bzrdir('subdir')
294
 
        result2 = SmartServerResponse(result.args[0:1] + ('..', ) + result.args[2:])
 
308
        result2 = smart_req.SmartServerResponse(
 
309
            result.args[0:1] + ('..', ) + result.args[2:])
295
310
        self.assertEqual(result2,
296
311
            request.execute('subdir'))
297
312
        self.make_bzrdir('subdir/deeper')
298
 
        result3 = SmartServerResponse(result.args[0:1] + ('../..', ) + result.args[2:])
 
313
        result3 = smart_req.SmartServerResponse(
 
314
            result.args[0:1] + ('../..', ) + result.args[2:])
299
315
        self.assertEqual(result3,
300
316
            request.execute('subdir/deeper'))
301
317
 
303
319
        """Test for the format attributes for rich root and subtree support."""
304
320
        backing = self.get_transport()
305
321
        request = self._request_class(backing)
306
 
        result = self._make_repository_and_result(format='dirstate-with-subtree')
 
322
        result = self._make_repository_and_result(
 
323
            format='dirstate-with-subtree')
307
324
        # check the test will be valid
308
325
        self.assertEqual('yes', result.args[2])
309
326
        self.assertEqual('yes', result.args[3])
313
330
        """Test for the supports_external_lookups attribute."""
314
331
        backing = self.get_transport()
315
332
        request = self._request_class(backing)
316
 
        result = self._make_repository_and_result(format='dirstate-with-subtree')
 
333
        result = self._make_repository_and_result(
 
334
            format='dirstate-with-subtree')
317
335
        # check the test will be valid
318
336
        self.assertEqual('no', result.args[4])
319
337
        self.assertEqual(result, request.execute(''))
330
348
        local_result = dir._get_config()._get_config_file().read()
331
349
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
332
350
        request = request_class(backing)
333
 
        expected = SuccessfulSmartServerResponse((), local_result)
 
351
        expected = smart_req.SuccessfulSmartServerResponse((), local_result)
334
352
        self.assertEqual(expected, request.execute(''))
335
353
 
336
354
    def test_missing(self):
338
356
        dir = self.make_bzrdir('.')
339
357
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
340
358
        request = request_class(backing)
341
 
        expected = SuccessfulSmartServerResponse((), '')
 
359
        expected = smart_req.SuccessfulSmartServerResponse((), '')
342
360
        self.assertEqual(expected, request.execute(''))
343
361
 
344
362
 
347
365
    def test_empty_dir(self):
348
366
        """Initializing an empty dir should succeed and do it."""
349
367
        backing = self.get_transport()
350
 
        request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
351
 
        self.assertEqual(SmartServerResponse(('ok', )),
 
368
        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
 
369
        self.assertEqual(smart_req.SmartServerResponse(('ok', )),
352
370
            request.execute(''))
353
371
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
354
372
        # no branch, tree or repository is expected with the current
360
378
    def test_missing_dir(self):
361
379
        """Initializing a missing directory should fail like the bzrdir api."""
362
380
        backing = self.get_transport()
363
 
        request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
 
381
        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
364
382
        self.assertRaises(errors.NoSuchFile,
365
383
            request.execute, 'subdir')
366
384
 
367
385
    def test_initialized_dir(self):
368
386
        """Initializing an extant bzrdir should fail like the bzrdir api."""
369
387
        backing = self.get_transport()
370
 
        request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
 
388
        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
371
389
        self.make_bzrdir('subdir')
372
390
        self.assertRaises(errors.FileExists,
373
391
            request.execute, 'subdir')
374
392
 
375
393
 
376
 
class TestSmartServerRequestBzrDirInitializeEx(tests.TestCaseWithMemoryTransport):
 
394
class TestSmartServerRequestBzrDirInitializeEx(
 
395
    tests.TestCaseWithMemoryTransport):
377
396
    """Basic tests for BzrDir.initialize_ex_1.16 in the smart server.
378
397
 
379
398
    The main unit tests in test_bzrdir exercise the API comprehensively.
383
402
        """Initializing an empty dir should succeed and do it."""
384
403
        backing = self.get_transport()
385
404
        name = self.make_bzrdir('reference')._format.network_name()
386
 
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
387
 
        self.assertEqual(SmartServerResponse(('', '', '', '', '', '', name,
388
 
            'False', '', '', '')),
 
405
        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
 
406
        self.assertEqual(
 
407
            smart_req.SmartServerResponse(('', '', '', '', '', '', name,
 
408
                                           'False', '', '', '')),
389
409
            request.execute(name, '', 'True', 'False', 'False', '', '', '', '',
390
 
            'False'))
 
410
                            'False'))
391
411
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
392
412
        # no branch, tree or repository is expected with the current
393
413
        # default format.
399
419
        """Initializing a missing directory should fail like the bzrdir api."""
400
420
        backing = self.get_transport()
401
421
        name = self.make_bzrdir('reference')._format.network_name()
402
 
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
 
422
        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
403
423
        self.assertRaises(errors.NoSuchFile, request.execute, name,
404
424
            'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')
405
425
 
407
427
        """Initializing an extant directory should fail like the bzrdir api."""
408
428
        backing = self.get_transport()
409
429
        name = self.make_bzrdir('reference')._format.network_name()
410
 
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
 
430
        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
411
431
        self.make_bzrdir('subdir')
412
432
        self.assertRaises(errors.FileExists, request.execute, name, 'subdir',
413
433
            'False', 'False', 'False', '', '', '', '', 'False')
414
434
 
415
435
 
 
436
class TestSmartServerRequestOpenBzrDir(tests.TestCaseWithMemoryTransport):
 
437
 
 
438
    def test_no_directory(self):
 
439
        backing = self.get_transport()
 
440
        request = smart_dir.SmartServerRequestOpenBzrDir(backing)
 
441
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
442
            request.execute('does-not-exist'))
 
443
 
 
444
    def test_empty_directory(self):
 
445
        backing = self.get_transport()
 
446
        backing.mkdir('empty')
 
447
        request = smart_dir.SmartServerRequestOpenBzrDir(backing)
 
448
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
449
            request.execute('empty'))
 
450
 
 
451
    def test_outside_root_client_path(self):
 
452
        backing = self.get_transport()
 
453
        request = smart_dir.SmartServerRequestOpenBzrDir(backing,
 
454
            root_client_path='root')
 
455
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
456
            request.execute('not-root'))
 
457
 
 
458
 
 
459
class TestSmartServerRequestOpenBzrDir_2_1(tests.TestCaseWithMemoryTransport):
 
460
 
 
461
    def test_no_directory(self):
 
462
        backing = self.get_transport()
 
463
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
464
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
465
            request.execute('does-not-exist'))
 
466
 
 
467
    def test_empty_directory(self):
 
468
        backing = self.get_transport()
 
469
        backing.mkdir('empty')
 
470
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
471
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
472
            request.execute('empty'))
 
473
 
 
474
    def test_present_without_workingtree(self):
 
475
        backing = self.get_transport()
 
476
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
477
        self.make_bzrdir('.')
 
478
        self.assertEqual(smart_req.SmartServerResponse(('yes', 'no')),
 
479
            request.execute(''))
 
480
 
 
481
    def test_outside_root_client_path(self):
 
482
        backing = self.get_transport()
 
483
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing,
 
484
            root_client_path='root')
 
485
        self.assertEqual(smart_req.SmartServerResponse(('no',)),
 
486
            request.execute('not-root'))
 
487
 
 
488
 
 
489
class TestSmartServerRequestOpenBzrDir_2_1_disk(TestCaseWithChrootedTransport):
 
490
 
 
491
    def test_present_with_workingtree(self):
 
492
        self.vfs_transport_factory = test_server.LocalURLServer
 
493
        backing = self.get_transport()
 
494
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
495
        bd = self.make_bzrdir('.')
 
496
        bd.create_repository()
 
497
        bd.create_branch()
 
498
        bd.create_workingtree()
 
499
        self.assertEqual(smart_req.SmartServerResponse(('yes', 'yes')),
 
500
            request.execute(''))
 
501
 
 
502
 
416
503
class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):
417
504
 
418
505
    def test_no_branch(self):
419
506
        """When there is no branch, ('nobranch', ) is returned."""
420
507
        backing = self.get_transport()
421
 
        request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
 
508
        request = smart_dir.SmartServerRequestOpenBranch(backing)
422
509
        self.make_bzrdir('.')
423
 
        self.assertEqual(SmartServerResponse(('nobranch', )),
 
510
        self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
424
511
            request.execute(''))
425
512
 
426
513
    def test_branch(self):
427
514
        """When there is a branch, 'ok' is returned."""
428
515
        backing = self.get_transport()
429
 
        request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
 
516
        request = smart_dir.SmartServerRequestOpenBranch(backing)
430
517
        self.make_branch('.')
431
 
        self.assertEqual(SmartServerResponse(('ok', '')),
 
518
        self.assertEqual(smart_req.SmartServerResponse(('ok', '')),
432
519
            request.execute(''))
433
520
 
434
521
    def test_branch_reference(self):
435
522
        """When there is a branch reference, the reference URL is returned."""
 
523
        self.vfs_transport_factory = test_server.LocalURLServer
436
524
        backing = self.get_transport()
437
 
        request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
 
525
        request = smart_dir.SmartServerRequestOpenBranch(backing)
438
526
        branch = self.make_branch('branch')
439
527
        checkout = branch.create_checkout('reference',lightweight=True)
440
 
        reference_url = BranchReferenceFormat().get_reference(checkout.bzrdir)
 
528
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(
 
529
            checkout.bzrdir)
441
530
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
442
 
        self.assertEqual(SmartServerResponse(('ok', reference_url)),
 
531
        self.assertEqual(smart_req.SmartServerResponse(('ok', reference_url)),
443
532
            request.execute('reference'))
444
533
 
 
534
    def test_notification_on_branch_from_repository(self):
 
535
        """When there is a repository, the error should return details."""
 
536
        backing = self.get_transport()
 
537
        request = smart_dir.SmartServerRequestOpenBranch(backing)
 
538
        repo = self.make_repository('.')
 
539
        self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
 
540
            request.execute(''))
 
541
 
445
542
 
446
543
class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):
447
544
 
449
546
        """When there is no branch, ('nobranch', ) is returned."""
450
547
        backing = self.get_transport()
451
548
        self.make_bzrdir('.')
452
 
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
453
 
        self.assertEqual(SmartServerResponse(('nobranch', )),
454
 
            request.execute(''))
455
 
 
456
 
    def test_branch(self):
457
 
        """When there is a branch, 'ok' is returned."""
458
 
        backing = self.get_transport()
459
 
        expected = self.make_branch('.')._format.network_name()
460
 
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
461
 
        self.assertEqual(SuccessfulSmartServerResponse(('branch', expected)),
462
 
            request.execute(''))
463
 
 
464
 
    def test_branch_reference(self):
465
 
        """When there is a branch reference, the reference URL is returned."""
466
 
        backing = self.get_transport()
467
 
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
468
 
        branch = self.make_branch('branch')
469
 
        checkout = branch.create_checkout('reference',lightweight=True)
470
 
        reference_url = BranchReferenceFormat().get_reference(checkout.bzrdir)
471
 
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
472
 
        self.assertEqual(SuccessfulSmartServerResponse(('ref', reference_url)),
473
 
            request.execute('reference'))
474
 
 
475
 
    def test_stacked_branch(self):
476
 
        """Opening a stacked branch does not open the stacked-on branch."""
477
 
        trunk = self.make_branch('trunk')
478
 
        feature = self.make_branch('feature')
479
 
        feature.set_stacked_on_url(trunk.base)
480
 
        opened_branches = []
481
 
        Branch.hooks.install_named_hook('open', opened_branches.append, None)
482
 
        backing = self.get_transport()
483
 
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
484
 
        request.setup_jail()
485
 
        try:
486
 
            response = request.execute('feature')
487
 
        finally:
488
 
            request.teardown_jail()
489
 
        expected_format = feature._format.network_name()
490
 
        self.assertEqual(
491
 
            SuccessfulSmartServerResponse(('branch', expected_format)),
492
 
            response)
493
 
        self.assertLength(1, opened_branches)
 
549
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
550
        self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
 
551
            request.execute(''))
 
552
 
 
553
    def test_branch(self):
 
554
        """When there is a branch, 'ok' is returned."""
 
555
        backing = self.get_transport()
 
556
        expected = self.make_branch('.')._format.network_name()
 
557
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
558
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
559
                ('branch', expected)),
 
560
                         request.execute(''))
 
561
 
 
562
    def test_branch_reference(self):
 
563
        """When there is a branch reference, the reference URL is returned."""
 
564
        self.vfs_transport_factory = test_server.LocalURLServer
 
565
        backing = self.get_transport()
 
566
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
567
        branch = self.make_branch('branch')
 
568
        checkout = branch.create_checkout('reference',lightweight=True)
 
569
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(
 
570
            checkout.bzrdir)
 
571
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
 
572
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
573
                ('ref', reference_url)),
 
574
                         request.execute('reference'))
 
575
 
 
576
    def test_stacked_branch(self):
 
577
        """Opening a stacked branch does not open the stacked-on branch."""
 
578
        trunk = self.make_branch('trunk')
 
579
        feature = self.make_branch('feature')
 
580
        feature.set_stacked_on_url(trunk.base)
 
581
        opened_branches = []
 
582
        _mod_branch.Branch.hooks.install_named_hook(
 
583
            'open', opened_branches.append, None)
 
584
        backing = self.get_transport()
 
585
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
586
        request.setup_jail()
 
587
        try:
 
588
            response = request.execute('feature')
 
589
        finally:
 
590
            request.teardown_jail()
 
591
        expected_format = feature._format.network_name()
 
592
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
593
                ('branch', expected_format)),
 
594
                         response)
 
595
        self.assertLength(1, opened_branches)
 
596
 
 
597
    def test_notification_on_branch_from_repository(self):
 
598
        """When there is a repository, the error should return details."""
 
599
        backing = self.get_transport()
 
600
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
601
        repo = self.make_repository('.')
 
602
        self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
 
603
            request.execute(''))
 
604
 
 
605
 
 
606
class TestSmartServerRequestOpenBranchV3(TestCaseWithChrootedTransport):
 
607
 
 
608
    def test_no_branch(self):
 
609
        """When there is no branch, ('nobranch', ) is returned."""
 
610
        backing = self.get_transport()
 
611
        self.make_bzrdir('.')
 
612
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
613
        self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
 
614
            request.execute(''))
 
615
 
 
616
    def test_branch(self):
 
617
        """When there is a branch, 'ok' is returned."""
 
618
        backing = self.get_transport()
 
619
        expected = self.make_branch('.')._format.network_name()
 
620
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
621
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
622
                ('branch', expected)),
 
623
                         request.execute(''))
 
624
 
 
625
    def test_branch_reference(self):
 
626
        """When there is a branch reference, the reference URL is returned."""
 
627
        self.vfs_transport_factory = test_server.LocalURLServer
 
628
        backing = self.get_transport()
 
629
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
630
        branch = self.make_branch('branch')
 
631
        checkout = branch.create_checkout('reference',lightweight=True)
 
632
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(
 
633
            checkout.bzrdir)
 
634
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
 
635
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
636
                ('ref', reference_url)),
 
637
                         request.execute('reference'))
 
638
 
 
639
    def test_stacked_branch(self):
 
640
        """Opening a stacked branch does not open the stacked-on branch."""
 
641
        trunk = self.make_branch('trunk')
 
642
        feature = self.make_branch('feature')
 
643
        feature.set_stacked_on_url(trunk.base)
 
644
        opened_branches = []
 
645
        _mod_branch.Branch.hooks.install_named_hook(
 
646
            'open', opened_branches.append, None)
 
647
        backing = self.get_transport()
 
648
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
649
        request.setup_jail()
 
650
        try:
 
651
            response = request.execute('feature')
 
652
        finally:
 
653
            request.teardown_jail()
 
654
        expected_format = feature._format.network_name()
 
655
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
656
                ('branch', expected_format)),
 
657
                         response)
 
658
        self.assertLength(1, opened_branches)
 
659
 
 
660
    def test_notification_on_branch_from_repository(self):
 
661
        """When there is a repository, the error should return details."""
 
662
        backing = self.get_transport()
 
663
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
664
        repo = self.make_repository('.')
 
665
        self.assertEqual(smart_req.SmartServerResponse(
 
666
                ('nobranch', 'location is a repository')),
 
667
                         request.execute(''))
494
668
 
495
669
 
496
670
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithMemoryTransport):
498
672
    def test_empty(self):
499
673
        """For an empty branch, the body is empty."""
500
674
        backing = self.get_transport()
501
 
        request = smart.branch.SmartServerRequestRevisionHistory(backing)
 
675
        request = smart_branch.SmartServerRequestRevisionHistory(backing)
502
676
        self.make_branch('.')
503
 
        self.assertEqual(SmartServerResponse(('ok', ), ''),
 
677
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), ''),
504
678
            request.execute(''))
505
679
 
506
680
    def test_not_empty(self):
507
681
        """For a non-empty branch, the body is empty."""
508
682
        backing = self.get_transport()
509
 
        request = smart.branch.SmartServerRequestRevisionHistory(backing)
 
683
        request = smart_branch.SmartServerRequestRevisionHistory(backing)
510
684
        tree = self.make_branch_and_memory_tree('.')
511
685
        tree.lock_write()
512
686
        tree.add('')
514
688
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
515
689
        tree.unlock()
516
690
        self.assertEqual(
517
 
            SmartServerResponse(('ok', ), ('\x00'.join([r1, r2]))),
 
691
            smart_req.SmartServerResponse(('ok', ), ('\x00'.join([r1, r2]))),
518
692
            request.execute(''))
519
693
 
520
694
 
523
697
    def test_no_branch(self):
524
698
        """When there is a bzrdir and no branch, NotBranchError is raised."""
525
699
        backing = self.get_transport()
526
 
        request = smart.branch.SmartServerBranchRequest(backing)
 
700
        request = smart_branch.SmartServerBranchRequest(backing)
527
701
        self.make_bzrdir('.')
528
702
        self.assertRaises(errors.NotBranchError,
529
703
            request.execute, '')
531
705
    def test_branch_reference(self):
532
706
        """When there is a branch reference, NotBranchError is raised."""
533
707
        backing = self.get_transport()
534
 
        request = smart.branch.SmartServerBranchRequest(backing)
 
708
        request = smart_branch.SmartServerBranchRequest(backing)
535
709
        branch = self.make_branch('branch')
536
710
        checkout = branch.create_checkout('reference',lightweight=True)
537
711
        self.assertRaises(errors.NotBranchError,
538
712
            request.execute, 'checkout')
539
713
 
540
714
 
541
 
class TestSmartServerBranchRequestLastRevisionInfo(tests.TestCaseWithMemoryTransport):
 
715
class TestSmartServerBranchRequestLastRevisionInfo(
 
716
    tests.TestCaseWithMemoryTransport):
542
717
 
543
718
    def test_empty(self):
544
719
        """For an empty branch, the result is ('ok', '0', 'null:')."""
545
720
        backing = self.get_transport()
546
 
        request = smart.branch.SmartServerBranchRequestLastRevisionInfo(backing)
 
721
        request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
547
722
        self.make_branch('.')
548
 
        self.assertEqual(SmartServerResponse(('ok', '0', 'null:')),
 
723
        self.assertEqual(smart_req.SmartServerResponse(('ok', '0', 'null:')),
549
724
            request.execute(''))
550
725
 
551
726
    def test_not_empty(self):
552
727
        """For a non-empty branch, the result is ('ok', 'revno', 'revid')."""
553
728
        backing = self.get_transport()
554
 
        request = smart.branch.SmartServerBranchRequestLastRevisionInfo(backing)
 
729
        request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
555
730
        tree = self.make_branch_and_memory_tree('.')
556
731
        tree.lock_write()
557
732
        tree.add('')
560
735
        r2 = tree.commit('2nd commit', rev_id=rev_id_utf8)
561
736
        tree.unlock()
562
737
        self.assertEqual(
563
 
            SmartServerResponse(('ok', '2', rev_id_utf8)),
 
738
            smart_req.SmartServerResponse(('ok', '2', rev_id_utf8)),
564
739
            request.execute(''))
565
740
 
566
741
 
567
 
class TestSmartServerBranchRequestGetConfigFile(tests.TestCaseWithMemoryTransport):
 
742
class TestSmartServerBranchRequestGetConfigFile(
 
743
    tests.TestCaseWithMemoryTransport):
568
744
 
569
745
    def test_default(self):
570
746
        """With no file, we get empty content."""
571
747
        backing = self.get_transport()
572
 
        request = smart.branch.SmartServerBranchGetConfigFile(backing)
 
748
        request = smart_branch.SmartServerBranchGetConfigFile(backing)
573
749
        branch = self.make_branch('.')
574
750
        # there should be no file by default
575
751
        content = ''
576
 
        self.assertEqual(SmartServerResponse(('ok', ), content),
 
752
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), content),
577
753
            request.execute(''))
578
754
 
579
755
    def test_with_content(self):
581
757
        # branch.control_files.get('branch.conf') for now - in the future it may
582
758
        # perform more complex processing.
583
759
        backing = self.get_transport()
584
 
        request = smart.branch.SmartServerBranchGetConfigFile(backing)
 
760
        request = smart_branch.SmartServerBranchGetConfigFile(backing)
585
761
        branch = self.make_branch('.')
586
762
        branch._transport.put_bytes('branch.conf', 'foo bar baz')
587
 
        self.assertEqual(SmartServerResponse(('ok', ), 'foo bar baz'),
 
763
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'foo bar baz'),
588
764
            request.execute(''))
589
765
 
590
766
 
591
767
class TestLockedBranch(tests.TestCaseWithMemoryTransport):
592
768
 
593
769
    def get_lock_tokens(self, branch):
594
 
        branch_token = branch.lock_write()
595
 
        repo_token = branch.repository.lock_write()
 
770
        branch_token = branch.lock_write().branch_token
 
771
        repo_token = branch.repository.lock_write().repository_token
596
772
        branch.repository.unlock()
597
773
        return branch_token, repo_token
598
774
 
601
777
 
602
778
    def test_value_name(self):
603
779
        branch = self.make_branch('.')
604
 
        request = smart.branch.SmartServerBranchRequestSetConfigOption(
 
780
        request = smart_branch.SmartServerBranchRequestSetConfigOption(
605
781
            branch.bzrdir.root_transport)
606
782
        branch_token, repo_token = self.get_lock_tokens(branch)
607
783
        config = branch._get_config()
608
784
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
609
785
            '')
610
 
        self.assertEqual(SuccessfulSmartServerResponse(()), result)
 
786
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
611
787
        self.assertEqual('bar', config.get_option('foo'))
612
788
        # Cleanup
613
789
        branch.unlock()
614
790
 
615
791
    def test_value_name_section(self):
616
792
        branch = self.make_branch('.')
617
 
        request = smart.branch.SmartServerBranchRequestSetConfigOption(
 
793
        request = smart_branch.SmartServerBranchRequestSetConfigOption(
618
794
            branch.bzrdir.root_transport)
619
795
        branch_token, repo_token = self.get_lock_tokens(branch)
620
796
        config = branch._get_config()
621
797
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
622
798
            'gam')
623
 
        self.assertEqual(SuccessfulSmartServerResponse(()), result)
 
799
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
624
800
        self.assertEqual('bar', config.get_option('foo', 'gam'))
625
801
        # Cleanup
626
802
        branch.unlock()
627
803
 
628
804
 
 
805
class TestSmartServerBranchRequestSetConfigOptionDict(TestLockedBranch):
 
806
 
 
807
    def setUp(self):
 
808
        TestLockedBranch.setUp(self)
 
809
        # A dict with non-ascii keys and values to exercise unicode
 
810
        # roundtripping.
 
811
        self.encoded_value_dict = (
 
812
            'd5:ascii1:a11:unicode \xe2\x8c\x9a3:\xe2\x80\xbde')
 
813
        self.value_dict = {
 
814
            'ascii': 'a', u'unicode \N{WATCH}': u'\N{INTERROBANG}'}
 
815
 
 
816
    def test_value_name(self):
 
817
        branch = self.make_branch('.')
 
818
        request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
 
819
            branch.bzrdir.root_transport)
 
820
        branch_token, repo_token = self.get_lock_tokens(branch)
 
821
        config = branch._get_config()
 
822
        result = request.execute('', branch_token, repo_token,
 
823
            self.encoded_value_dict, 'foo', '')
 
824
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
 
825
        self.assertEqual(self.value_dict, config.get_option('foo'))
 
826
        # Cleanup
 
827
        branch.unlock()
 
828
 
 
829
    def test_value_name_section(self):
 
830
        branch = self.make_branch('.')
 
831
        request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
 
832
            branch.bzrdir.root_transport)
 
833
        branch_token, repo_token = self.get_lock_tokens(branch)
 
834
        config = branch._get_config()
 
835
        result = request.execute('', branch_token, repo_token,
 
836
            self.encoded_value_dict, 'foo', 'gam')
 
837
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
 
838
        self.assertEqual(self.value_dict, config.get_option('foo', 'gam'))
 
839
        # Cleanup
 
840
        branch.unlock()
 
841
 
 
842
 
629
843
class TestSmartServerBranchRequestSetTagsBytes(TestLockedBranch):
630
844
    # Only called when the branch format and tags match [yay factory
631
845
    # methods] so only need to test straight forward cases.
635
849
        tag_bytes = base_branch._get_tags_bytes()
636
850
        # get_lock_tokens takes out a lock.
637
851
        branch_token, repo_token = self.get_lock_tokens(base_branch)
638
 
        request = smart.branch.SmartServerBranchSetTagsBytes(
 
852
        request = smart_branch.SmartServerBranchSetTagsBytes(
639
853
            self.get_transport())
640
854
        response = request.execute('base', branch_token, repo_token)
641
855
        self.assertEqual(None, response)
643
857
        self.assertEqual(None, response)
644
858
        response = request.do_end()
645
859
        self.assertEquals(
646
 
            SuccessfulSmartServerResponse(()), response)
 
860
            smart_req.SuccessfulSmartServerResponse(()), response)
647
861
        base_branch.unlock()
648
862
 
649
863
    def test_lock_failed(self):
650
864
        base_branch = self.make_branch('base')
651
865
        base_branch.lock_write()
652
866
        tag_bytes = base_branch._get_tags_bytes()
653
 
        request = smart.branch.SmartServerBranchSetTagsBytes(
 
867
        request = smart_branch.SmartServerBranchSetTagsBytes(
654
868
            self.get_transport())
655
869
        self.assertRaises(errors.TokenMismatch, request.execute,
656
870
            'base', 'wrong token', 'wrong token')
687
901
 
688
902
    def assertRequestSucceeds(self, revision_id, revno):
689
903
        response = self.set_last_revision(revision_id, revno)
690
 
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)), response)
 
904
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
 
905
                         response)
691
906
 
692
907
 
693
908
class TestSetLastRevisionVerbMixin(object):
701
916
        """If the revision_id is not present, the verb returns NoSuchRevision.
702
917
        """
703
918
        revision_id = 'non-existent revision'
704
 
        self.assertEqual(
705
 
            FailedSmartServerResponse(('NoSuchRevision', revision_id)),
706
 
            self.set_last_revision(revision_id, 1))
 
919
        self.assertEqual(smart_req.FailedSmartServerResponse(('NoSuchRevision',
 
920
                                                              revision_id)),
 
921
                         self.set_last_revision(revision_id, 1))
707
922
 
708
923
    def make_tree_with_two_commits(self):
709
924
        self.tree.lock_write()
749
964
        rejection_message = u'rejection message\N{INTERROBANG}'
750
965
        def hook_that_rejects(params):
751
966
            raise errors.TipChangeRejected(rejection_message)
752
 
        Branch.hooks.install_named_hook(
 
967
        _mod_branch.Branch.hooks.install_named_hook(
753
968
            'pre_change_branch_tip', hook_that_rejects, None)
754
969
        self.assertEqual(
755
 
            FailedSmartServerResponse(
 
970
            smart_req.FailedSmartServerResponse(
756
971
                ('TipChangeRejected', rejection_message.encode('utf-8'))),
757
972
            self.set_last_revision('null:', 0))
758
973
 
761
976
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
762
977
    """Tests for Branch.set_last_revision verb."""
763
978
 
764
 
    request_class = smart.branch.SmartServerBranchRequestSetLastRevision
 
979
    request_class = smart_branch.SmartServerBranchRequestSetLastRevision
765
980
 
766
981
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
767
982
        return self.request.execute(
772
987
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
773
988
    """Tests for Branch.set_last_revision_info verb."""
774
989
 
775
 
    request_class = smart.branch.SmartServerBranchRequestSetLastRevisionInfo
 
990
    request_class = smart_branch.SmartServerBranchRequestSetLastRevisionInfo
776
991
 
777
992
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
778
993
        return self.request.execute(
789
1004
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
790
1005
    """Tests for Branch.set_last_revision_ex verb."""
791
1006
 
792
 
    request_class = smart.branch.SmartServerBranchRequestSetLastRevisionEx
 
1007
    request_class = smart_branch.SmartServerBranchRequestSetLastRevisionEx
793
1008
 
794
1009
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
795
1010
        return self.request.execute(
798
1013
    def assertRequestSucceeds(self, revision_id, revno):
799
1014
        response = self.set_last_revision(revision_id, revno)
800
1015
        self.assertEqual(
801
 
            SuccessfulSmartServerResponse(('ok', revno, revision_id)),
 
1016
            smart_req.SuccessfulSmartServerResponse(('ok', revno, revision_id)),
802
1017
            response)
803
1018
 
804
1019
    def test_branch_last_revision_info_rewind(self):
815
1030
        response = self.request.execute(
816
1031
            '', branch_token, repo_token, rev_id_utf8, 0, 0)
817
1032
        self.assertEqual(
818
 
            SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
 
1033
            smart_req.SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
819
1034
            response)
820
1035
        self.assertEqual(
821
1036
            (2, 'rev-2'), self.tree.branch.last_revision_info())
825
1040
        response = self.request.execute(
826
1041
            '', branch_token, repo_token, rev_id_utf8, 0, 1)
827
1042
        self.assertEqual(
828
 
            SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
 
1043
            smart_req.SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
829
1044
            response)
830
1045
        self.unlock_branch()
831
1046
        self.assertEqual(
856
1071
        """
857
1072
        self.make_branch_with_divergent_history()
858
1073
        self.assertEqual(
859
 
            FailedSmartServerResponse(('Diverged',)),
 
1074
            smart_req.FailedSmartServerResponse(('Diverged',)),
860
1075
            self.set_last_revision('child-1', 2))
861
1076
        # The branch tip was not changed.
862
1077
        self.assertEqual('child-2', self.tree.branch.last_revision())
870
1085
        response = self.request.execute(
871
1086
            '', branch_token, repo_token, 'child-1', 1, 0)
872
1087
        self.assertEqual(
873
 
            SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
 
1088
            smart_req.SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
874
1089
            response)
875
1090
        self.unlock_branch()
876
1091
        # The branch tip was changed.
881
1096
 
882
1097
    def test_get_parent_none(self):
883
1098
        base_branch = self.make_branch('base')
884
 
        request = smart.branch.SmartServerBranchGetParent(self.get_transport())
 
1099
        request = smart_branch.SmartServerBranchGetParent(self.get_transport())
885
1100
        response = request.execute('base')
886
1101
        self.assertEquals(
887
 
            SuccessfulSmartServerResponse(('',)), response)
 
1102
            smart_req.SuccessfulSmartServerResponse(('',)), response)
888
1103
 
889
1104
    def test_get_parent_something(self):
890
1105
        base_branch = self.make_branch('base')
891
1106
        base_branch.set_parent(self.get_url('foo'))
892
 
        request = smart.branch.SmartServerBranchGetParent(self.get_transport())
 
1107
        request = smart_branch.SmartServerBranchGetParent(self.get_transport())
893
1108
        response = request.execute('base')
894
1109
        self.assertEquals(
895
 
            SuccessfulSmartServerResponse(("../foo",)),
 
1110
            smart_req.SuccessfulSmartServerResponse(("../foo",)),
896
1111
            response)
897
1112
 
898
1113
 
899
 
class TestSmartServerBranchRequestSetParent(tests.TestCaseWithMemoryTransport):
 
1114
class TestSmartServerBranchRequestSetParent(TestLockedBranch):
900
1115
 
901
1116
    def test_set_parent_none(self):
902
1117
        branch = self.make_branch('base', format="1.9")
903
1118
        branch.lock_write()
904
1119
        branch._set_parent_location('foo')
905
1120
        branch.unlock()
906
 
        request = smart.branch.SmartServerBranchRequestSetParentLocation(
 
1121
        request = smart_branch.SmartServerBranchRequestSetParentLocation(
907
1122
            self.get_transport())
908
 
        branch_token = branch.lock_write()
909
 
        repo_token = branch.repository.lock_write()
 
1123
        branch_token, repo_token = self.get_lock_tokens(branch)
910
1124
        try:
911
1125
            response = request.execute('base', branch_token, repo_token, '')
912
1126
        finally:
913
 
            branch.repository.unlock()
914
1127
            branch.unlock()
915
 
        self.assertEqual(SuccessfulSmartServerResponse(()), response)
 
1128
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
916
1129
        self.assertEqual(None, branch.get_parent())
917
1130
 
918
1131
    def test_set_parent_something(self):
919
1132
        branch = self.make_branch('base', format="1.9")
920
 
        request = smart.branch.SmartServerBranchRequestSetParentLocation(
 
1133
        request = smart_branch.SmartServerBranchRequestSetParentLocation(
921
1134
            self.get_transport())
922
 
        branch_token = branch.lock_write()
923
 
        repo_token = branch.repository.lock_write()
 
1135
        branch_token, repo_token = self.get_lock_tokens(branch)
924
1136
        try:
925
1137
            response = request.execute('base', branch_token, repo_token,
926
1138
            'http://bar/')
927
1139
        finally:
928
 
            branch.repository.unlock()
929
1140
            branch.unlock()
930
 
        self.assertEqual(SuccessfulSmartServerResponse(()), response)
 
1141
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
931
1142
        self.assertEqual('http://bar/', branch.get_parent())
932
1143
 
933
1144
 
934
 
class TestSmartServerBranchRequestGetTagsBytes(tests.TestCaseWithMemoryTransport):
 
1145
class TestSmartServerBranchRequestGetTagsBytes(
 
1146
    tests.TestCaseWithMemoryTransport):
935
1147
    # Only called when the branch format and tags match [yay factory
936
1148
    # methods] so only need to test straight forward cases.
937
1149
 
938
1150
    def test_get_bytes(self):
939
1151
        base_branch = self.make_branch('base')
940
 
        request = smart.branch.SmartServerBranchGetTagsBytes(
 
1152
        request = smart_branch.SmartServerBranchGetTagsBytes(
941
1153
            self.get_transport())
942
1154
        response = request.execute('base')
943
1155
        self.assertEquals(
944
 
            SuccessfulSmartServerResponse(('',)), response)
 
1156
            smart_req.SuccessfulSmartServerResponse(('',)), response)
945
1157
 
946
1158
 
947
1159
class TestSmartServerBranchRequestGetStackedOnURL(tests.TestCaseWithMemoryTransport):
951
1163
        stacked_branch = self.make_branch('stacked', format='1.6')
952
1164
        # typically should be relative
953
1165
        stacked_branch.set_stacked_on_url('../base')
954
 
        request = smart.branch.SmartServerBranchRequestGetStackedOnURL(
 
1166
        request = smart_branch.SmartServerBranchRequestGetStackedOnURL(
955
1167
            self.get_transport())
956
1168
        response = request.execute('stacked')
957
1169
        self.assertEquals(
958
 
            SmartServerResponse(('ok', '../base')),
 
1170
            smart_req.SmartServerResponse(('ok', '../base')),
959
1171
            response)
960
1172
 
961
1173
 
962
 
class TestSmartServerBranchRequestLockWrite(tests.TestCaseWithMemoryTransport):
 
1174
class TestSmartServerBranchRequestLockWrite(TestLockedBranch):
963
1175
 
964
1176
    def setUp(self):
965
1177
        tests.TestCaseWithMemoryTransport.setUp(self)
966
1178
 
967
1179
    def test_lock_write_on_unlocked_branch(self):
968
1180
        backing = self.get_transport()
969
 
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
 
1181
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
970
1182
        branch = self.make_branch('.', format='knit')
971
1183
        repository = branch.repository
972
1184
        response = request.execute('')
973
1185
        branch_nonce = branch.control_files._lock.peek().get('nonce')
974
1186
        repository_nonce = repository.control_files._lock.peek().get('nonce')
975
 
        self.assertEqual(
976
 
            SmartServerResponse(('ok', branch_nonce, repository_nonce)),
977
 
            response)
 
1187
        self.assertEqual(smart_req.SmartServerResponse(
 
1188
                ('ok', branch_nonce, repository_nonce)),
 
1189
                         response)
978
1190
        # The branch (and associated repository) is now locked.  Verify that
979
1191
        # with a new branch object.
980
1192
        new_branch = repository.bzrdir.open_branch()
981
1193
        self.assertRaises(errors.LockContention, new_branch.lock_write)
982
1194
        # Cleanup
983
 
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
 
1195
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
984
1196
        response = request.execute('', branch_nonce, repository_nonce)
985
1197
 
986
1198
    def test_lock_write_on_locked_branch(self):
987
1199
        backing = self.get_transport()
988
 
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
 
1200
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
989
1201
        branch = self.make_branch('.')
990
 
        branch_token = branch.lock_write()
 
1202
        branch_token = branch.lock_write().branch_token
991
1203
        branch.leave_lock_in_place()
992
1204
        branch.unlock()
993
1205
        response = request.execute('')
994
1206
        self.assertEqual(
995
 
            SmartServerResponse(('LockContention',)), response)
 
1207
            smart_req.SmartServerResponse(('LockContention',)), response)
996
1208
        # Cleanup
997
1209
        branch.lock_write(branch_token)
998
1210
        branch.dont_leave_lock_in_place()
1000
1212
 
1001
1213
    def test_lock_write_with_tokens_on_locked_branch(self):
1002
1214
        backing = self.get_transport()
1003
 
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
 
1215
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1004
1216
        branch = self.make_branch('.', format='knit')
1005
 
        branch_token = branch.lock_write()
1006
 
        repo_token = branch.repository.lock_write()
1007
 
        branch.repository.unlock()
 
1217
        branch_token, repo_token = self.get_lock_tokens(branch)
1008
1218
        branch.leave_lock_in_place()
1009
1219
        branch.repository.leave_lock_in_place()
1010
1220
        branch.unlock()
1011
1221
        response = request.execute('',
1012
1222
                                   branch_token, repo_token)
1013
1223
        self.assertEqual(
1014
 
            SmartServerResponse(('ok', branch_token, repo_token)), response)
 
1224
            smart_req.SmartServerResponse(('ok', branch_token, repo_token)),
 
1225
            response)
1015
1226
        # Cleanup
1016
1227
        branch.repository.lock_write(repo_token)
1017
1228
        branch.repository.dont_leave_lock_in_place()
1022
1233
 
1023
1234
    def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
1024
1235
        backing = self.get_transport()
1025
 
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
 
1236
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1026
1237
        branch = self.make_branch('.', format='knit')
1027
 
        branch_token = branch.lock_write()
1028
 
        repo_token = branch.repository.lock_write()
1029
 
        branch.repository.unlock()
 
1238
        branch_token, repo_token = self.get_lock_tokens(branch)
1030
1239
        branch.leave_lock_in_place()
1031
1240
        branch.repository.leave_lock_in_place()
1032
1241
        branch.unlock()
1033
1242
        response = request.execute('',
1034
1243
                                   branch_token+'xxx', repo_token)
1035
1244
        self.assertEqual(
1036
 
            SmartServerResponse(('TokenMismatch',)), response)
 
1245
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
1037
1246
        # Cleanup
1038
1247
        branch.repository.lock_write(repo_token)
1039
1248
        branch.repository.dont_leave_lock_in_place()
1044
1253
 
1045
1254
    def test_lock_write_on_locked_repo(self):
1046
1255
        backing = self.get_transport()
1047
 
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
 
1256
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1048
1257
        branch = self.make_branch('.', format='knit')
1049
1258
        repo = branch.repository
1050
 
        repo_token = repo.lock_write()
 
1259
        repo_token = repo.lock_write().repository_token
1051
1260
        repo.leave_lock_in_place()
1052
1261
        repo.unlock()
1053
1262
        response = request.execute('')
1054
1263
        self.assertEqual(
1055
 
            SmartServerResponse(('LockContention',)), response)
 
1264
            smart_req.SmartServerResponse(('LockContention',)), response)
1056
1265
        # Cleanup
1057
1266
        repo.lock_write(repo_token)
1058
1267
        repo.dont_leave_lock_in_place()
1060
1269
 
1061
1270
    def test_lock_write_on_readonly_transport(self):
1062
1271
        backing = self.get_readonly_transport()
1063
 
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
 
1272
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1064
1273
        branch = self.make_branch('.')
1065
1274
        root = self.get_transport().clone('/')
1066
1275
        path = urlutils.relative_url(root.base, self.get_transport().base)
1070
1279
        self.assertEqual('LockFailed', error_name)
1071
1280
 
1072
1281
 
1073
 
class TestSmartServerBranchRequestUnlock(tests.TestCaseWithMemoryTransport):
 
1282
class TestSmartServerBranchRequestUnlock(TestLockedBranch):
1074
1283
 
1075
1284
    def setUp(self):
1076
1285
        tests.TestCaseWithMemoryTransport.setUp(self)
1077
1286
 
1078
1287
    def test_unlock_on_locked_branch_and_repo(self):
1079
1288
        backing = self.get_transport()
1080
 
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
 
1289
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
1081
1290
        branch = self.make_branch('.', format='knit')
1082
1291
        # Lock the branch
1083
 
        branch_token = branch.lock_write()
1084
 
        repo_token = branch.repository.lock_write()
1085
 
        branch.repository.unlock()
 
1292
        branch_token, repo_token = self.get_lock_tokens(branch)
1086
1293
        # Unlock the branch (and repo) object, leaving the physical locks
1087
1294
        # in place.
1088
1295
        branch.leave_lock_in_place()
1091
1298
        response = request.execute('',
1092
1299
                                   branch_token, repo_token)
1093
1300
        self.assertEqual(
1094
 
            SmartServerResponse(('ok',)), response)
 
1301
            smart_req.SmartServerResponse(('ok',)), response)
1095
1302
        # The branch is now unlocked.  Verify that with a new branch
1096
1303
        # object.
1097
1304
        new_branch = branch.bzrdir.open_branch()
1100
1307
 
1101
1308
    def test_unlock_on_unlocked_branch_unlocked_repo(self):
1102
1309
        backing = self.get_transport()
1103
 
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
 
1310
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
1104
1311
        branch = self.make_branch('.', format='knit')
1105
1312
        response = request.execute(
1106
1313
            '', 'branch token', 'repo token')
1107
1314
        self.assertEqual(
1108
 
            SmartServerResponse(('TokenMismatch',)), response)
 
1315
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
1109
1316
 
1110
1317
    def test_unlock_on_unlocked_branch_locked_repo(self):
1111
1318
        backing = self.get_transport()
1112
 
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
 
1319
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
1113
1320
        branch = self.make_branch('.', format='knit')
1114
1321
        # Lock the repository.
1115
 
        repo_token = branch.repository.lock_write()
 
1322
        repo_token = branch.repository.lock_write().repository_token
1116
1323
        branch.repository.leave_lock_in_place()
1117
1324
        branch.repository.unlock()
1118
1325
        # Issue branch lock_write request on the unlocked branch (with locked
1119
1326
        # repo).
1120
 
        response = request.execute(
1121
 
            '', 'branch token', repo_token)
 
1327
        response = request.execute('', 'branch token', repo_token)
1122
1328
        self.assertEqual(
1123
 
            SmartServerResponse(('TokenMismatch',)), response)
 
1329
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
1124
1330
        # Cleanup
1125
1331
        branch.repository.lock_write(repo_token)
1126
1332
        branch.repository.dont_leave_lock_in_place()
1136
1342
        # its the exact path being looked at and the server is not
1137
1343
        # searching.
1138
1344
        backing = self.get_transport()
1139
 
        request = smart.repository.SmartServerRepositoryRequest(backing)
 
1345
        request = smart_repo.SmartServerRepositoryRequest(backing)
1140
1346
        self.make_repository('.', shared=True)
1141
1347
        self.make_bzrdir('subdir')
1142
1348
        self.assertRaises(errors.NoRepositoryPresent,
1148
1354
    def test_trivial_bzipped(self):
1149
1355
        # This tests that the wire encoding is actually bzipped
1150
1356
        backing = self.get_transport()
1151
 
        request = smart.repository.SmartServerRepositoryGetParentMap(backing)
 
1357
        request = smart_repo.SmartServerRepositoryGetParentMap(backing)
1152
1358
        tree = self.make_branch_and_memory_tree('.')
1153
1359
 
1154
1360
        self.assertEqual(None,
1155
1361
            request.execute('', 'missing-id'))
1156
1362
        # Note that it returns a body that is bzipped.
1157
1363
        self.assertEqual(
1158
 
            SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
 
1364
            smart_req.SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
1159
1365
            request.do_body('\n\n0\n'))
1160
1366
 
1161
1367
    def test_trivial_include_missing(self):
1162
1368
        backing = self.get_transport()
1163
 
        request = smart.repository.SmartServerRepositoryGetParentMap(backing)
 
1369
        request = smart_repo.SmartServerRepositoryGetParentMap(backing)
1164
1370
        tree = self.make_branch_and_memory_tree('.')
1165
1371
 
1166
1372
        self.assertEqual(None,
1167
1373
            request.execute('', 'missing-id', 'include-missing:'))
1168
1374
        self.assertEqual(
1169
 
            SuccessfulSmartServerResponse(('ok', ),
 
1375
            smart_req.SuccessfulSmartServerResponse(('ok', ),
1170
1376
                bz2.compress('missing:missing-id')),
1171
1377
            request.do_body('\n\n0\n'))
1172
1378
 
1173
1379
 
1174
 
class TestSmartServerRepositoryGetRevisionGraph(tests.TestCaseWithMemoryTransport):
 
1380
class TestSmartServerRepositoryGetRevisionGraph(
 
1381
    tests.TestCaseWithMemoryTransport):
1175
1382
 
1176
1383
    def test_none_argument(self):
1177
1384
        backing = self.get_transport()
1178
 
        request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
 
1385
        request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1179
1386
        tree = self.make_branch_and_memory_tree('.')
1180
1387
        tree.lock_write()
1181
1388
        tree.add('')
1190
1397
        response.body = '\n'.join(sorted(response.body.split('\n')))
1191
1398
 
1192
1399
        self.assertEqual(
1193
 
            SmartServerResponse(('ok', ), '\n'.join(lines)), response)
 
1400
            smart_req.SmartServerResponse(('ok', ), '\n'.join(lines)), response)
1194
1401
 
1195
1402
    def test_specific_revision_argument(self):
1196
1403
        backing = self.get_transport()
1197
 
        request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
 
1404
        request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1198
1405
        tree = self.make_branch_and_memory_tree('.')
1199
1406
        tree.lock_write()
1200
1407
        tree.add('')
1203
1410
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1204
1411
        tree.unlock()
1205
1412
 
1206
 
        self.assertEqual(SmartServerResponse(('ok', ), rev_id_utf8),
 
1413
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), rev_id_utf8),
1207
1414
            request.execute('', rev_id_utf8))
1208
1415
 
1209
1416
    def test_no_such_revision(self):
1210
1417
        backing = self.get_transport()
1211
 
        request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
 
1418
        request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1212
1419
        tree = self.make_branch_and_memory_tree('.')
1213
1420
        tree.lock_write()
1214
1421
        tree.add('')
1216
1423
        tree.unlock()
1217
1424
 
1218
1425
        # Note that it still returns body (of zero bytes).
1219
 
        self.assertEqual(
1220
 
            SmartServerResponse(('nosuchrevision', 'missingrevision', ), ''),
1221
 
            request.execute('', 'missingrevision'))
1222
 
 
1223
 
 
1224
 
class TestSmartServerRepositoryGetRevIdForRevno(tests.TestCaseWithMemoryTransport):
 
1426
        self.assertEqual(smart_req.SmartServerResponse(
 
1427
                ('nosuchrevision', 'missingrevision', ), ''),
 
1428
                         request.execute('', 'missingrevision'))
 
1429
 
 
1430
 
 
1431
class TestSmartServerRepositoryGetRevIdForRevno(
 
1432
    tests.TestCaseWithMemoryTransport):
1225
1433
 
1226
1434
    def test_revno_found(self):
1227
1435
        backing = self.get_transport()
1228
 
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
 
1436
        request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1229
1437
        tree = self.make_branch_and_memory_tree('.')
1230
1438
        tree.lock_write()
1231
1439
        tree.add('')
1235
1443
        tree.commit('2nd commit', rev_id=rev2_id_utf8)
1236
1444
        tree.unlock()
1237
1445
 
1238
 
        self.assertEqual(SmartServerResponse(('ok', rev1_id_utf8)),
 
1446
        self.assertEqual(smart_req.SmartServerResponse(('ok', rev1_id_utf8)),
1239
1447
            request.execute('', 1, (2, rev2_id_utf8)))
1240
1448
 
1241
1449
    def test_known_revid_missing(self):
1242
1450
        backing = self.get_transport()
1243
 
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
 
1451
        request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1244
1452
        repo = self.make_repository('.')
1245
1453
        self.assertEqual(
1246
 
            FailedSmartServerResponse(('nosuchrevision', 'ghost')),
 
1454
            smart_req.FailedSmartServerResponse(('nosuchrevision', 'ghost')),
1247
1455
            request.execute('', 1, (2, 'ghost')))
1248
1456
 
1249
1457
    def test_history_incomplete(self):
1250
1458
        backing = self.get_transport()
1251
 
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
 
1459
        request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1252
1460
        parent = self.make_branch_and_memory_tree('parent', format='1.9')
1253
1461
        parent.lock_write()
1254
1462
        parent.add([''], ['TREE_ROOT'])
1262
1470
        local.branch.create_clone_on_transport(
1263
1471
            self.get_transport('stacked'), stacked_on=self.get_url('parent'))
1264
1472
        self.assertEqual(
1265
 
            SmartServerResponse(('history-incomplete', 2, r2)),
 
1473
            smart_req.SmartServerResponse(('history-incomplete', 2, r2)),
1266
1474
            request.execute('stacked', 1, (3, r3)))
1267
1475
 
1268
1476
 
1281
1489
    def test_ancestry_of(self):
1282
1490
        """The search argument may be a 'ancestry-of' some heads'."""
1283
1491
        backing = self.get_transport()
1284
 
        request = smart.repository.SmartServerRepositoryGetStream(backing)
 
1492
        request = smart_repo.SmartServerRepositoryGetStream(backing)
1285
1493
        repo, r1, r2 = self.make_two_commit_repo()
1286
1494
        fetch_spec = ['ancestry-of', r2]
1287
1495
        lines = '\n'.join(fetch_spec)
1294
1502
    def test_search(self):
1295
1503
        """The search argument may be a 'search' of some explicit keys."""
1296
1504
        backing = self.get_transport()
1297
 
        request = smart.repository.SmartServerRepositoryGetStream(backing)
 
1505
        request = smart_repo.SmartServerRepositoryGetStream(backing)
1298
1506
        repo, r1, r2 = self.make_two_commit_repo()
1299
1507
        fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
1300
1508
        lines = '\n'.join(fetch_spec)
1310
1518
    def test_missing_revision(self):
1311
1519
        """For a missing revision, ('no', ) is returned."""
1312
1520
        backing = self.get_transport()
1313
 
        request = smart.repository.SmartServerRequestHasRevision(backing)
 
1521
        request = smart_repo.SmartServerRequestHasRevision(backing)
1314
1522
        self.make_repository('.')
1315
 
        self.assertEqual(SmartServerResponse(('no', )),
 
1523
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
1316
1524
            request.execute('', 'revid'))
1317
1525
 
1318
1526
    def test_present_revision(self):
1319
1527
        """For a present revision, ('yes', ) is returned."""
1320
1528
        backing = self.get_transport()
1321
 
        request = smart.repository.SmartServerRequestHasRevision(backing)
 
1529
        request = smart_repo.SmartServerRequestHasRevision(backing)
1322
1530
        tree = self.make_branch_and_memory_tree('.')
1323
1531
        tree.lock_write()
1324
1532
        tree.add('')
1326
1534
        r1 = tree.commit('a commit', rev_id=rev_id_utf8)
1327
1535
        tree.unlock()
1328
1536
        self.assertTrue(tree.branch.repository.has_revision(rev_id_utf8))
1329
 
        self.assertEqual(SmartServerResponse(('yes', )),
 
1537
        self.assertEqual(smart_req.SmartServerResponse(('yes', )),
1330
1538
            request.execute('', rev_id_utf8))
1331
1539
 
1332
1540
 
1335
1543
    def test_empty_revid(self):
1336
1544
        """With an empty revid, we get only size an number and revisions"""
1337
1545
        backing = self.get_transport()
1338
 
        request = smart.repository.SmartServerRepositoryGatherStats(backing)
 
1546
        request = smart_repo.SmartServerRepositoryGatherStats(backing)
1339
1547
        repository = self.make_repository('.')
1340
1548
        stats = repository.gather_stats()
1341
1549
        expected_body = 'revisions: 0\n'
1342
 
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
 
1550
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1343
1551
                         request.execute('', '', 'no'))
1344
1552
 
1345
1553
    def test_revid_with_committers(self):
1346
1554
        """For a revid we get more infos."""
1347
1555
        backing = self.get_transport()
1348
1556
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1349
 
        request = smart.repository.SmartServerRepositoryGatherStats(backing)
 
1557
        request = smart_repo.SmartServerRepositoryGatherStats(backing)
1350
1558
        tree = self.make_branch_and_memory_tree('.')
1351
1559
        tree.lock_write()
1352
1560
        tree.add('')
1360
1568
        expected_body = ('firstrev: 123456.200 3600\n'
1361
1569
                         'latestrev: 654321.400 0\n'
1362
1570
                         'revisions: 2\n')
1363
 
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
 
1571
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1364
1572
                         request.execute('',
1365
1573
                                         rev_id_utf8, 'no'))
1366
1574
 
1368
1576
        """For a revid and requesting committers we get the whole thing."""
1369
1577
        backing = self.get_transport()
1370
1578
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1371
 
        request = smart.repository.SmartServerRepositoryGatherStats(backing)
 
1579
        request = smart_repo.SmartServerRepositoryGatherStats(backing)
1372
1580
        tree = self.make_branch_and_memory_tree('.')
1373
1581
        tree.lock_write()
1374
1582
        tree.add('')
1384
1592
                         'firstrev: 123456.200 3600\n'
1385
1593
                         'latestrev: 654321.400 0\n'
1386
1594
                         'revisions: 2\n')
1387
 
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
 
1595
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1388
1596
                         request.execute('',
1389
1597
                                         rev_id_utf8, 'yes'))
1390
1598
 
1394
1602
    def test_is_shared(self):
1395
1603
        """For a shared repository, ('yes', ) is returned."""
1396
1604
        backing = self.get_transport()
1397
 
        request = smart.repository.SmartServerRepositoryIsShared(backing)
 
1605
        request = smart_repo.SmartServerRepositoryIsShared(backing)
1398
1606
        self.make_repository('.', shared=True)
1399
 
        self.assertEqual(SmartServerResponse(('yes', )),
 
1607
        self.assertEqual(smart_req.SmartServerResponse(('yes', )),
1400
1608
            request.execute('', ))
1401
1609
 
1402
1610
    def test_is_not_shared(self):
1403
1611
        """For a shared repository, ('no', ) is returned."""
1404
1612
        backing = self.get_transport()
1405
 
        request = smart.repository.SmartServerRepositoryIsShared(backing)
 
1613
        request = smart_repo.SmartServerRepositoryIsShared(backing)
1406
1614
        self.make_repository('.', shared=False)
1407
 
        self.assertEqual(SmartServerResponse(('no', )),
 
1615
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
1408
1616
            request.execute('', ))
1409
1617
 
1410
1618
 
1412
1620
 
1413
1621
    def test_lock_write_on_unlocked_repo(self):
1414
1622
        backing = self.get_transport()
1415
 
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
 
1623
        request = smart_repo.SmartServerRepositoryLockWrite(backing)
1416
1624
        repository = self.make_repository('.', format='knit')
1417
1625
        response = request.execute('')
1418
1626
        nonce = repository.control_files._lock.peek().get('nonce')
1419
 
        self.assertEqual(SmartServerResponse(('ok', nonce)), response)
 
1627
        self.assertEqual(smart_req.SmartServerResponse(('ok', nonce)), response)
1420
1628
        # The repository is now locked.  Verify that with a new repository
1421
1629
        # object.
1422
1630
        new_repo = repository.bzrdir.open_repository()
1423
1631
        self.assertRaises(errors.LockContention, new_repo.lock_write)
1424
1632
        # Cleanup
1425
 
        request = smart.repository.SmartServerRepositoryUnlock(backing)
 
1633
        request = smart_repo.SmartServerRepositoryUnlock(backing)
1426
1634
        response = request.execute('', nonce)
1427
1635
 
1428
1636
    def test_lock_write_on_locked_repo(self):
1429
1637
        backing = self.get_transport()
1430
 
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
 
1638
        request = smart_repo.SmartServerRepositoryLockWrite(backing)
1431
1639
        repository = self.make_repository('.', format='knit')
1432
 
        repo_token = repository.lock_write()
 
1640
        repo_token = repository.lock_write().repository_token
1433
1641
        repository.leave_lock_in_place()
1434
1642
        repository.unlock()
1435
1643
        response = request.execute('')
1436
1644
        self.assertEqual(
1437
 
            SmartServerResponse(('LockContention',)), response)
 
1645
            smart_req.SmartServerResponse(('LockContention',)), response)
1438
1646
        # Cleanup
1439
1647
        repository.lock_write(repo_token)
1440
1648
        repository.dont_leave_lock_in_place()
1442
1650
 
1443
1651
    def test_lock_write_on_readonly_transport(self):
1444
1652
        backing = self.get_readonly_transport()
1445
 
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
 
1653
        request = smart_repo.SmartServerRepositoryLockWrite(backing)
1446
1654
        repository = self.make_repository('.', format='knit')
1447
1655
        response = request.execute('')
1448
1656
        self.assertFalse(response.is_successful())
1452
1660
class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
1453
1661
 
1454
1662
    def make_empty_byte_stream(self, repo):
1455
 
        byte_stream = smart.repository._stream_to_byte_stream([], repo._format)
 
1663
        byte_stream = smart_repo._stream_to_byte_stream([], repo._format)
1456
1664
        return ''.join(byte_stream)
1457
1665
 
1458
1666
 
1460
1668
 
1461
1669
    def test_insert_stream_empty(self):
1462
1670
        backing = self.get_transport()
1463
 
        request = smart.repository.SmartServerRepositoryInsertStream(backing)
 
1671
        request = smart_repo.SmartServerRepositoryInsertStream(backing)
1464
1672
        repository = self.make_repository('.')
1465
1673
        response = request.execute('', '')
1466
1674
        self.assertEqual(None, response)
1467
1675
        response = request.do_chunk(self.make_empty_byte_stream(repository))
1468
1676
        self.assertEqual(None, response)
1469
1677
        response = request.do_end()
1470
 
        self.assertEqual(SmartServerResponse(('ok', )), response)
1471
 
        
 
1678
        self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
 
1679
 
1472
1680
 
1473
1681
class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
1474
1682
 
1475
1683
    def test_insert_stream_empty(self):
1476
1684
        backing = self.get_transport()
1477
 
        request = smart.repository.SmartServerRepositoryInsertStreamLocked(
 
1685
        request = smart_repo.SmartServerRepositoryInsertStreamLocked(
1478
1686
            backing)
1479
1687
        repository = self.make_repository('.', format='knit')
1480
 
        lock_token = repository.lock_write()
 
1688
        lock_token = repository.lock_write().repository_token
1481
1689
        response = request.execute('', '', lock_token)
1482
1690
        self.assertEqual(None, response)
1483
1691
        response = request.do_chunk(self.make_empty_byte_stream(repository))
1484
1692
        self.assertEqual(None, response)
1485
1693
        response = request.do_end()
1486
 
        self.assertEqual(SmartServerResponse(('ok', )), response)
 
1694
        self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
1487
1695
        repository.unlock()
1488
1696
 
1489
1697
    def test_insert_stream_with_wrong_lock_token(self):
1490
1698
        backing = self.get_transport()
1491
 
        request = smart.repository.SmartServerRepositoryInsertStreamLocked(
 
1699
        request = smart_repo.SmartServerRepositoryInsertStreamLocked(
1492
1700
            backing)
1493
1701
        repository = self.make_repository('.', format='knit')
1494
 
        lock_token = repository.lock_write()
 
1702
        lock_token = repository.lock_write().repository_token
1495
1703
        self.assertRaises(
1496
1704
            errors.TokenMismatch, request.execute, '', '', 'wrong-token')
1497
1705
        repository.unlock()
1504
1712
 
1505
1713
    def test_unlock_on_locked_repo(self):
1506
1714
        backing = self.get_transport()
1507
 
        request = smart.repository.SmartServerRepositoryUnlock(backing)
 
1715
        request = smart_repo.SmartServerRepositoryUnlock(backing)
1508
1716
        repository = self.make_repository('.', format='knit')
1509
 
        token = repository.lock_write()
 
1717
        token = repository.lock_write().repository_token
1510
1718
        repository.leave_lock_in_place()
1511
1719
        repository.unlock()
1512
1720
        response = request.execute('', token)
1513
1721
        self.assertEqual(
1514
 
            SmartServerResponse(('ok',)), response)
 
1722
            smart_req.SmartServerResponse(('ok',)), response)
1515
1723
        # The repository is now unlocked.  Verify that with a new repository
1516
1724
        # object.
1517
1725
        new_repo = repository.bzrdir.open_repository()
1520
1728
 
1521
1729
    def test_unlock_on_unlocked_repo(self):
1522
1730
        backing = self.get_transport()
1523
 
        request = smart.repository.SmartServerRepositoryUnlock(backing)
 
1731
        request = smart_repo.SmartServerRepositoryUnlock(backing)
1524
1732
        repository = self.make_repository('.', format='knit')
1525
1733
        response = request.execute('', 'some token')
1526
1734
        self.assertEqual(
1527
 
            SmartServerResponse(('TokenMismatch',)), response)
 
1735
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
1528
1736
 
1529
1737
 
1530
1738
class TestSmartServerIsReadonly(tests.TestCaseWithMemoryTransport):
1531
1739
 
1532
1740
    def test_is_readonly_no(self):
1533
1741
        backing = self.get_transport()
1534
 
        request = smart.request.SmartServerIsReadonly(backing)
 
1742
        request = smart_req.SmartServerIsReadonly(backing)
1535
1743
        response = request.execute()
1536
1744
        self.assertEqual(
1537
 
            SmartServerResponse(('no',)), response)
 
1745
            smart_req.SmartServerResponse(('no',)), response)
1538
1746
 
1539
1747
    def test_is_readonly_yes(self):
1540
1748
        backing = self.get_readonly_transport()
1541
 
        request = smart.request.SmartServerIsReadonly(backing)
 
1749
        request = smart_req.SmartServerIsReadonly(backing)
1542
1750
        response = request.execute()
1543
1751
        self.assertEqual(
1544
 
            SmartServerResponse(('yes',)), response)
1545
 
 
1546
 
 
1547
 
class TestSmartServerRepositorySetMakeWorkingTrees(tests.TestCaseWithMemoryTransport):
 
1752
            smart_req.SmartServerResponse(('yes',)), response)
 
1753
 
 
1754
 
 
1755
class TestSmartServerRepositorySetMakeWorkingTrees(
 
1756
    tests.TestCaseWithMemoryTransport):
1548
1757
 
1549
1758
    def test_set_false(self):
1550
1759
        backing = self.get_transport()
1551
1760
        repo = self.make_repository('.', shared=True)
1552
1761
        repo.set_make_working_trees(True)
1553
 
        request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
 
1762
        request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
1554
1763
        request = request_class(backing)
1555
 
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
 
1764
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
1556
1765
            request.execute('', 'False'))
1557
1766
        repo = repo.bzrdir.open_repository()
1558
1767
        self.assertFalse(repo.make_working_trees())
1561
1770
        backing = self.get_transport()
1562
1771
        repo = self.make_repository('.', shared=True)
1563
1772
        repo.set_make_working_trees(False)
1564
 
        request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
 
1773
        request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
1565
1774
        request = request_class(backing)
1566
 
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
 
1775
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
1567
1776
            request.execute('', 'True'))
1568
1777
        repo = repo.bzrdir.open_repository()
1569
1778
        self.assertTrue(repo.make_working_trees())
1588
1797
        repo.lock_write()
1589
1798
        self.addCleanup(repo.unlock)
1590
1799
        backing = self.get_transport()
1591
 
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
 
1800
        request = smart_packrepo.SmartServerPackRepositoryAutopack(
1592
1801
            backing)
1593
1802
        response = request.execute('')
1594
 
        self.assertEqual(SmartServerResponse(('ok',)), response)
 
1803
        self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1595
1804
        repo._pack_collection.reload_pack_names()
1596
1805
        self.assertEqual(1, len(repo._pack_collection.names()))
1597
1806
 
1603
1812
        for x in range(9):
1604
1813
            tree.commit('commit %s' % x)
1605
1814
        backing = self.get_transport()
1606
 
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
 
1815
        request = smart_packrepo.SmartServerPackRepositoryAutopack(
1607
1816
            backing)
1608
1817
        response = request.execute('')
1609
 
        self.assertEqual(SmartServerResponse(('ok',)), response)
 
1818
        self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1610
1819
        repo._pack_collection.reload_pack_names()
1611
1820
        self.assertEqual(9, len(repo._pack_collection.names()))
1612
1821
 
1614
1823
        """A request to autopack a non-pack repo is a no-op."""
1615
1824
        repo = self.make_repository('.', format='knit')
1616
1825
        backing = self.get_transport()
1617
 
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
 
1826
        request = smart_packrepo.SmartServerPackRepositoryAutopack(
1618
1827
            backing)
1619
1828
        response = request.execute('')
1620
 
        self.assertEqual(SmartServerResponse(('ok',)), response)
 
1829
        self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
 
1830
 
 
1831
 
 
1832
class TestSmartServerVfsGet(tests.TestCaseWithMemoryTransport):
 
1833
 
 
1834
    def test_unicode_path(self):
 
1835
        """VFS requests expect unicode paths to be escaped."""
 
1836
        filename = u'foo\N{INTERROBANG}'
 
1837
        filename_escaped = urlutils.escape(filename)
 
1838
        backing = self.get_transport()
 
1839
        request = vfs.GetRequest(backing)
 
1840
        backing.put_bytes_non_atomic(filename_escaped, 'contents')
 
1841
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'contents'),
 
1842
            request.execute(filename_escaped))
1621
1843
 
1622
1844
 
1623
1845
class TestHandlers(tests.TestCase):
1627
1849
        """All registered request_handlers can be found."""
1628
1850
        # If there's a typo in a register_lazy call, this loop will fail with
1629
1851
        # an AttributeError.
1630
 
        for key, item in smart.request.request_handlers.iteritems():
 
1852
        for key, item in smart_req.request_handlers.iteritems():
1631
1853
            pass
1632
1854
 
1633
1855
    def assertHandlerEqual(self, verb, handler):
1634
 
        self.assertEqual(smart.request.request_handlers.get(verb), handler)
 
1856
        self.assertEqual(smart_req.request_handlers.get(verb), handler)
1635
1857
 
1636
1858
    def test_registered_methods(self):
1637
1859
        """Test that known methods are registered to the correct object."""
1638
1860
        self.assertHandlerEqual('Branch.get_config_file',
1639
 
            smart.branch.SmartServerBranchGetConfigFile)
 
1861
            smart_branch.SmartServerBranchGetConfigFile)
1640
1862
        self.assertHandlerEqual('Branch.get_parent',
1641
 
            smart.branch.SmartServerBranchGetParent)
 
1863
            smart_branch.SmartServerBranchGetParent)
1642
1864
        self.assertHandlerEqual('Branch.get_tags_bytes',
1643
 
            smart.branch.SmartServerBranchGetTagsBytes)
 
1865
            smart_branch.SmartServerBranchGetTagsBytes)
1644
1866
        self.assertHandlerEqual('Branch.lock_write',
1645
 
            smart.branch.SmartServerBranchRequestLockWrite)
 
1867
            smart_branch.SmartServerBranchRequestLockWrite)
1646
1868
        self.assertHandlerEqual('Branch.last_revision_info',
1647
 
            smart.branch.SmartServerBranchRequestLastRevisionInfo)
 
1869
            smart_branch.SmartServerBranchRequestLastRevisionInfo)
1648
1870
        self.assertHandlerEqual('Branch.revision_history',
1649
 
            smart.branch.SmartServerRequestRevisionHistory)
 
1871
            smart_branch.SmartServerRequestRevisionHistory)
1650
1872
        self.assertHandlerEqual('Branch.set_config_option',
1651
 
            smart.branch.SmartServerBranchRequestSetConfigOption)
 
1873
            smart_branch.SmartServerBranchRequestSetConfigOption)
1652
1874
        self.assertHandlerEqual('Branch.set_last_revision',
1653
 
            smart.branch.SmartServerBranchRequestSetLastRevision)
 
1875
            smart_branch.SmartServerBranchRequestSetLastRevision)
1654
1876
        self.assertHandlerEqual('Branch.set_last_revision_info',
1655
 
            smart.branch.SmartServerBranchRequestSetLastRevisionInfo)
 
1877
            smart_branch.SmartServerBranchRequestSetLastRevisionInfo)
1656
1878
        self.assertHandlerEqual('Branch.set_last_revision_ex',
1657
 
            smart.branch.SmartServerBranchRequestSetLastRevisionEx)
 
1879
            smart_branch.SmartServerBranchRequestSetLastRevisionEx)
1658
1880
        self.assertHandlerEqual('Branch.set_parent_location',
1659
 
            smart.branch.SmartServerBranchRequestSetParentLocation)
 
1881
            smart_branch.SmartServerBranchRequestSetParentLocation)
1660
1882
        self.assertHandlerEqual('Branch.unlock',
1661
 
            smart.branch.SmartServerBranchRequestUnlock)
 
1883
            smart_branch.SmartServerBranchRequestUnlock)
1662
1884
        self.assertHandlerEqual('BzrDir.find_repository',
1663
 
            smart.bzrdir.SmartServerRequestFindRepositoryV1)
 
1885
            smart_dir.SmartServerRequestFindRepositoryV1)
1664
1886
        self.assertHandlerEqual('BzrDir.find_repositoryV2',
1665
 
            smart.bzrdir.SmartServerRequestFindRepositoryV2)
 
1887
            smart_dir.SmartServerRequestFindRepositoryV2)
1666
1888
        self.assertHandlerEqual('BzrDirFormat.initialize',
1667
 
            smart.bzrdir.SmartServerRequestInitializeBzrDir)
 
1889
            smart_dir.SmartServerRequestInitializeBzrDir)
1668
1890
        self.assertHandlerEqual('BzrDirFormat.initialize_ex_1.16',
1669
 
            smart.bzrdir.SmartServerRequestBzrDirInitializeEx)
 
1891
            smart_dir.SmartServerRequestBzrDirInitializeEx)
1670
1892
        self.assertHandlerEqual('BzrDir.cloning_metadir',
1671
 
            smart.bzrdir.SmartServerBzrDirRequestCloningMetaDir)
 
1893
            smart_dir.SmartServerBzrDirRequestCloningMetaDir)
1672
1894
        self.assertHandlerEqual('BzrDir.get_config_file',
1673
 
            smart.bzrdir.SmartServerBzrDirRequestConfigFile)
 
1895
            smart_dir.SmartServerBzrDirRequestConfigFile)
1674
1896
        self.assertHandlerEqual('BzrDir.open_branch',
1675
 
            smart.bzrdir.SmartServerRequestOpenBranch)
 
1897
            smart_dir.SmartServerRequestOpenBranch)
1676
1898
        self.assertHandlerEqual('BzrDir.open_branchV2',
1677
 
            smart.bzrdir.SmartServerRequestOpenBranchV2)
 
1899
            smart_dir.SmartServerRequestOpenBranchV2)
 
1900
        self.assertHandlerEqual('BzrDir.open_branchV3',
 
1901
            smart_dir.SmartServerRequestOpenBranchV3)
1678
1902
        self.assertHandlerEqual('PackRepository.autopack',
1679
 
            smart.packrepository.SmartServerPackRepositoryAutopack)
 
1903
            smart_packrepo.SmartServerPackRepositoryAutopack)
1680
1904
        self.assertHandlerEqual('Repository.gather_stats',
1681
 
            smart.repository.SmartServerRepositoryGatherStats)
 
1905
            smart_repo.SmartServerRepositoryGatherStats)
1682
1906
        self.assertHandlerEqual('Repository.get_parent_map',
1683
 
            smart.repository.SmartServerRepositoryGetParentMap)
 
1907
            smart_repo.SmartServerRepositoryGetParentMap)
1684
1908
        self.assertHandlerEqual('Repository.get_rev_id_for_revno',
1685
 
            smart.repository.SmartServerRepositoryGetRevIdForRevno)
 
1909
            smart_repo.SmartServerRepositoryGetRevIdForRevno)
1686
1910
        self.assertHandlerEqual('Repository.get_revision_graph',
1687
 
            smart.repository.SmartServerRepositoryGetRevisionGraph)
 
1911
            smart_repo.SmartServerRepositoryGetRevisionGraph)
1688
1912
        self.assertHandlerEqual('Repository.get_stream',
1689
 
            smart.repository.SmartServerRepositoryGetStream)
 
1913
            smart_repo.SmartServerRepositoryGetStream)
1690
1914
        self.assertHandlerEqual('Repository.has_revision',
1691
 
            smart.repository.SmartServerRequestHasRevision)
 
1915
            smart_repo.SmartServerRequestHasRevision)
1692
1916
        self.assertHandlerEqual('Repository.insert_stream',
1693
 
            smart.repository.SmartServerRepositoryInsertStream)
 
1917
            smart_repo.SmartServerRepositoryInsertStream)
1694
1918
        self.assertHandlerEqual('Repository.insert_stream_locked',
1695
 
            smart.repository.SmartServerRepositoryInsertStreamLocked)
 
1919
            smart_repo.SmartServerRepositoryInsertStreamLocked)
1696
1920
        self.assertHandlerEqual('Repository.is_shared',
1697
 
            smart.repository.SmartServerRepositoryIsShared)
 
1921
            smart_repo.SmartServerRepositoryIsShared)
1698
1922
        self.assertHandlerEqual('Repository.lock_write',
1699
 
            smart.repository.SmartServerRepositoryLockWrite)
 
1923
            smart_repo.SmartServerRepositoryLockWrite)
1700
1924
        self.assertHandlerEqual('Repository.tarball',
1701
 
            smart.repository.SmartServerRepositoryTarball)
 
1925
            smart_repo.SmartServerRepositoryTarball)
1702
1926
        self.assertHandlerEqual('Repository.unlock',
1703
 
            smart.repository.SmartServerRepositoryUnlock)
 
1927
            smart_repo.SmartServerRepositoryUnlock)
1704
1928
        self.assertHandlerEqual('Transport.is_readonly',
1705
 
            smart.request.SmartServerIsReadonly)
 
1929
            smart_req.SmartServerIsReadonly)