← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_smart.py

  • Committer: Robert Collins
  • Date: 2010-05-06 23:41:35 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506234135-yivbzczw1sejxnxc
Lock methods on ``Tree``, ``Branch`` and ``Repository`` are now
expected to return an object which can be used to unlock them. This reduces
duplicate code when using cleanups. The previous 'tokens's returned by
``Branch.lock_write`` and ``Repository.lock_write`` are now attributes
on the result of the lock_write. ``repository.RepositoryWriteLockResult``
and ``branch.BranchWriteLockResult`` document this. (Robert Collins)

``log._get_info_for_log_files`` now takes an add_cleanup callable.
(Robert Collins)

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