/brz/remove-bazaar : contents of bzrlib/tests/test

: (revision 4562)

To get this branch, use:

bzr branch
http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

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