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

: (revision 4490.1.2)

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')
	1193	r1 = parent.commit(message='first commit')
	1194	r2 = parent.commit(message='second commit')
	1195	local = self.make_branch_and_memory_tree('local', format='1.9')
	1196	local.branch.pull(parent.branch)
	1197	local.set_parent_ids([r2])
	1198	r3 = local.commit(message='local commit')
	1199	local.branch.create_clone_on_transport(
	1200	self.get_transport('stacked'), stacked_on=self.get_url('parent'))
	1201	self.assertEqual(
	1202	SmartServerResponse(('history-incomplete', 2, r2)),
	1203	request.execute('stacked', 1, (3, r3)))
	1204
4070.9.14 by Andrew Bennetts Tweaks requested by Robert's review.	1205	class TestSmartServerRepositoryGetStream(tests.TestCaseWithMemoryTransport):
	1206
	1207	def make_two_commit_repo(self):
	1208	tree = self.make_branch_and_memory_tree('.')
	1209	tree.lock_write()
	1210	tree.add('')
	1211	r1 = tree.commit('1st commit')
	1212	r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
	1213	tree.unlock()
	1214	repo = tree.branch.repository
	1215	return repo, r1, r2
	1216
	1217	def test_ancestry_of(self):
	1218	"""The search argument may be a 'ancestry-of' some heads'."""
	1219	backing = self.get_transport()
	1220	request = smart.repository.SmartServerRepositoryGetStream(backing)
	1221	repo, r1, r2 = self.make_two_commit_repo()
	1222	fetch_spec = ['ancestry-of', r2]
	1223	lines = '\n'.join(fetch_spec)
	1224	request.execute('', repo._format.network_name())
	1225	response = request.do_body(lines)
	1226	self.assertEqual(('ok',), response.args)
	1227	stream_bytes = ''.join(response.body_stream)
	1228	self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
	1229
	1230	def test_search(self):
	1231	"""The search argument may be a 'search' of some explicit keys."""
	1232	backing = self.get_transport()
	1233	request = smart.repository.SmartServerRepositoryGetStream(backing)
	1234	repo, r1, r2 = self.make_two_commit_repo()
	1235	fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
	1236	lines = '\n'.join(fetch_spec)
	1237	request.execute('', repo._format.network_name())
	1238	response = request.do_body(lines)
	1239	self.assertEqual(('ok',), response.args)
	1240	stream_bytes = ''.join(response.body_stream)
	1241	self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
	1242
	1243
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1244	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).	1245
	1246	def test_missing_revision(self):
	1247	"""For a missing revision, ('no', ) is returned."""
	1248	backing = self.get_transport()
	1249	request = smart.repository.SmartServerRequestHasRevision(backing)
	1250	self.make_repository('.')
	1251	self.assertEqual(SmartServerResponse(('no', )),
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1252	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).	1253
	1254	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.	1255	"""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).	1256	backing = self.get_transport()
	1257	request = smart.repository.SmartServerRequestHasRevision(backing)
	1258	tree = self.make_branch_and_memory_tree('.')
	1259	tree.lock_write()
	1260	tree.add('')
2018.5.148 by Andrew Bennetts Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.	1261	rev_id_utf8 = u'\xc8abc'.encode('utf-8')
	1262	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).	1263	tree.unlock()
2018.5.148 by Andrew Bennetts Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.	1264	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.	1265	self.assertEqual(SmartServerResponse(('yes', )),
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1266	request.execute('', rev_id_utf8))
	1267
	1268
	1269	class TestSmartServerRepositoryGatherStats(tests.TestCaseWithMemoryTransport):
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1270
	1271	def test_empty_revid(self):
	1272	"""With an empty revid, we get only size an number and revisions"""
	1273	backing = self.get_transport()
	1274	request = smart.repository.SmartServerRepositoryGatherStats(backing)
	1275	repository = self.make_repository('.')
	1276	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.	1277	expected_body = 'revisions: 0\n'
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1278	self.assertEqual(SmartServerResponse(('ok', ), expected_body),
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1279	request.execute('', '', 'no'))
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1280
	1281	def test_revid_with_committers(self):
	1282	"""For a revid we get more infos."""
	1283	backing = self.get_transport()
2018.5.148 by Andrew Bennetts Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.	1284	rev_id_utf8 = u'\xc8abc'.encode('utf-8')
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1285	request = smart.repository.SmartServerRepositoryGatherStats(backing)
	1286	tree = self.make_branch_and_memory_tree('.')
	1287	tree.lock_write()
	1288	tree.add('')
	1289	# Let's build a predictable result
	1290	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.	1291	tree.commit('a commit', timestamp=654321.4, timezone=0,
	1292	rev_id=rev_id_utf8)
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1293	tree.unlock()
	1294
	1295	stats = tree.branch.repository.gather_stats()
	1296	expected_body = ('firstrev: 123456.200 3600\n'
	1297	'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.	1298	'revisions: 2\n')
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1299	self.assertEqual(SmartServerResponse(('ok', ), expected_body),
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1300	request.execute('',
2018.5.148 by Andrew Bennetts Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.	1301	rev_id_utf8, 'no'))
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1302
	1303	def test_not_empty_repository_with_committers(self):
	1304	"""For a revid and requesting committers we get the whole thing."""
	1305	backing = self.get_transport()
2018.5.148 by Andrew Bennetts Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.	1306	rev_id_utf8 = u'\xc8abc'.encode('utf-8')
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1307	request = smart.repository.SmartServerRepositoryGatherStats(backing)
	1308	tree = self.make_branch_and_memory_tree('.')
	1309	tree.lock_write()
	1310	tree.add('')
	1311	# Let's build a predictable result
	1312	tree.commit('a commit', timestamp=123456.2, timezone=3600,
	1313	committer='foo')
	1314	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.	1315	committer='bar', rev_id=rev_id_utf8)
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1316	tree.unlock()
	1317	stats = tree.branch.repository.gather_stats()
	1318
	1319	expected_body = ('committers: 2\n'
	1320	'firstrev: 123456.200 3600\n'
	1321	'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.	1322	'revisions: 2\n')
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1323	self.assertEqual(SmartServerResponse(('ok', ), expected_body),
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1324	request.execute('',
2018.5.148 by Andrew Bennetts Fix all the DeprecationWarnings in test_smart caused by unicode revision IDs.	1325	rev_id_utf8, 'yes'))
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1326
	1327
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1328	class TestSmartServerRepositoryIsShared(tests.TestCaseWithMemoryTransport):
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1329
	1330	def test_is_shared(self):
	1331	"""For a shared repository, ('yes', ) is returned."""
	1332	backing = self.get_transport()
	1333	request = smart.repository.SmartServerRepositoryIsShared(backing)
	1334	self.make_repository('.', shared=True)
	1335	self.assertEqual(SmartServerResponse(('yes', )),
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1336	request.execute('', ))
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1337
	1338	def test_is_not_shared(self):
2018.5.58 by Wouter van Heyst Small test fixes to reflect naming and documentation	1339	"""For a shared repository, ('no', ) is returned."""
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1340	backing = self.get_transport()
	1341	request = smart.repository.SmartServerRepositoryIsShared(backing)
	1342	self.make_repository('.', shared=False)
	1343	self.assertEqual(SmartServerResponse(('no', )),
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1344	request.execute('', ))
	1345
	1346
	1347	class TestSmartServerRepositoryLockWrite(tests.TestCaseWithMemoryTransport):
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1348
	1349	def test_lock_write_on_unlocked_repo(self):
	1350	backing = self.get_transport()
	1351	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.	1352	repository = self.make_repository('.', format='knit')
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1353	response = request.execute('')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1354	nonce = repository.control_files._lock.peek().get('nonce')
	1355	self.assertEqual(SmartServerResponse(('ok', nonce)), response)
	1356	# The repository is now locked. Verify that with a new repository
	1357	# object.
	1358	new_repo = repository.bzrdir.open_repository()
	1359	self.assertRaises(errors.LockContention, new_repo.lock_write)
4327.1.10 by Vincent Ladeuil Fix 10 more lock-related test failures.	1360	# Cleanup
	1361	request = smart.repository.SmartServerRepositoryUnlock(backing)
	1362	response = request.execute('', nonce)
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1363
	1364	def test_lock_write_on_locked_repo(self):
	1365	backing = self.get_transport()
	1366	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.	1367	repository = self.make_repository('.', format='knit')
4327.1.10 by Vincent Ladeuil Fix 10 more lock-related test failures.	1368	repo_token = repository.lock_write()
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1369	repository.leave_lock_in_place()
	1370	repository.unlock()
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1371	response = request.execute('')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1372	self.assertEqual(
	1373	SmartServerResponse(('LockContention',)), response)
4327.1.10 by Vincent Ladeuil Fix 10 more lock-related test failures.	1374	# Cleanup
	1375	repository.lock_write(repo_token)
	1376	repository.dont_leave_lock_in_place()
	1377	repository.unlock()
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1378
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.	1379	def test_lock_write_on_readonly_transport(self):
	1380	backing = self.get_readonly_transport()
	1381	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.	1382	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.	1383	response = request.execute('')
2872.5.3 by Martin Pool Pass back LockFailed from smart server lock methods	1384	self.assertFalse(response.is_successful())
	1385	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.	1386
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1387
4144.3.1 by Andrew Bennetts Add Repository.insert_stream_locked server-side implementation, plus tests for server-side _translate_error.	1388	class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
	1389
	1390	def make_empty_byte_stream(self, repo):
	1391	byte_stream = smart.repository._stream_to_byte_stream([], repo._format)
	1392	return ''.join(byte_stream)
	1393
	1394
	1395	class TestSmartServerRepositoryInsertStream(TestInsertStreamBase):
	1396
	1397	def test_insert_stream_empty(self):
	1398	backing = self.get_transport()
	1399	request = smart.repository.SmartServerRepositoryInsertStream(backing)
	1400	repository = self.make_repository('.')
	1401	response = request.execute('', '')
	1402	self.assertEqual(None, response)
	1403	response = request.do_chunk(self.make_empty_byte_stream(repository))
	1404	self.assertEqual(None, response)
	1405	response = request.do_end()
	1406	self.assertEqual(SmartServerResponse(('ok', )), response)
	1407
	1408
	1409	class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
	1410
	1411	def test_insert_stream_empty(self):
	1412	backing = self.get_transport()
	1413	request = smart.repository.SmartServerRepositoryInsertStreamLocked(
	1414	backing)
	1415	repository = self.make_repository('.', format='knit')
	1416	lock_token = repository.lock_write()
	1417	response = request.execute('', '', lock_token)
	1418	self.assertEqual(None, response)
	1419	response = request.do_chunk(self.make_empty_byte_stream(repository))
	1420	self.assertEqual(None, response)
	1421	response = request.do_end()
	1422	self.assertEqual(SmartServerResponse(('ok', )), response)
	1423	repository.unlock()
	1424
	1425	def test_insert_stream_with_wrong_lock_token(self):
	1426	backing = self.get_transport()
	1427	request = smart.repository.SmartServerRepositoryInsertStreamLocked(
	1428	backing)
	1429	repository = self.make_repository('.', format='knit')
	1430	lock_token = repository.lock_write()
	1431	self.assertRaises(
	1432	errors.TokenMismatch, request.execute, '', '', 'wrong-token')
	1433	repository.unlock()
	1434
	1435
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1436	class TestSmartServerRepositoryUnlock(tests.TestCaseWithMemoryTransport):
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1437
	1438	def setUp(self):
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1439	tests.TestCaseWithMemoryTransport.setUp(self)
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1440
	1441	def test_unlock_on_locked_repo(self):
	1442	backing = self.get_transport()
	1443	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.	1444	repository = self.make_repository('.', format='knit')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1445	token = repository.lock_write()
	1446	repository.leave_lock_in_place()
	1447	repository.unlock()
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1448	response = request.execute('', token)
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1449	self.assertEqual(
	1450	SmartServerResponse(('ok',)), response)
	1451	# The repository is now unlocked. Verify that with a new repository
	1452	# object.
	1453	new_repo = repository.bzrdir.open_repository()
	1454	new_repo.lock_write()
	1455	new_repo.unlock()
	1456
	1457	def test_unlock_on_unlocked_repo(self):
	1458	backing = self.get_transport()
	1459	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.	1460	repository = self.make_repository('.', format='knit')
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1461	response = request.execute('', 'some token')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1462	self.assertEqual(
	1463	SmartServerResponse(('TokenMismatch',)), response)
	1464
	1465
2692.1.1 by Andrew Bennetts Add translate_client_path method to SmartServerRequest.	1466	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.	1467
	1468	def test_is_readonly_no(self):
	1469	backing = self.get_transport()
	1470	request = smart.request.SmartServerIsReadonly(backing)
	1471	response = request.execute()
	1472	self.assertEqual(
	1473	SmartServerResponse(('no',)), response)
	1474
	1475	def test_is_readonly_yes(self):
	1476	backing = self.get_readonly_transport()
	1477	request = smart.request.SmartServerIsReadonly(backing)
	1478	response = request.execute()
	1479	self.assertEqual(
	1480	SmartServerResponse(('yes',)), response)
	1481
	1482
4017.3.4 by Robert Collins Create a verb for Repository.set_make_working_trees.	1483	class TestSmartServerRepositorySetMakeWorkingTrees(tests.TestCaseWithMemoryTransport):
	1484
	1485	def test_set_false(self):
	1486	backing = self.get_transport()
	1487	repo = self.make_repository('.', shared=True)
	1488	repo.set_make_working_trees(True)
	1489	request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
	1490	request = request_class(backing)
	1491	self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
	1492	request.execute('', 'False'))
	1493	repo = repo.bzrdir.open_repository()
	1494	self.assertFalse(repo.make_working_trees())
	1495
	1496	def test_set_true(self):
	1497	backing = self.get_transport()
	1498	repo = self.make_repository('.', shared=True)
	1499	repo.set_make_working_trees(False)
	1500	request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
	1501	request = request_class(backing)
	1502	self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
	1503	request.execute('', 'True'))
	1504	repo = repo.bzrdir.open_repository()
	1505	self.assertTrue(repo.make_working_trees())
	1506
	1507
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1508	class TestSmartServerPackRepositoryAutopack(tests.TestCaseWithTransport):
	1509
	1510	def make_repo_needing_autopacking(self, path='.'):
	1511	# Make a repo in need of autopacking.
	1512	tree = self.make_branch_and_tree('.', format='pack-0.92')
	1513	repo = tree.branch.repository
	1514	# monkey-patch the pack collection to disable autopacking
	1515	repo._pack_collection._max_pack_count = lambda count: count
	1516	for x in range(10):
	1517	tree.commit('commit %s' % x)
	1518	self.assertEqual(10, len(repo._pack_collection.names()))
	1519	del repo._pack_collection._max_pack_count
	1520	return repo
	1521
	1522	def test_autopack_needed(self):
	1523	repo = self.make_repo_needing_autopacking()
4145.1.6 by Robert Collins More test fallout, but all caught now.	1524	repo.lock_write()
	1525	self.addCleanup(repo.unlock)
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1526	backing = self.get_transport()
	1527	request = smart.packrepository.SmartServerPackRepositoryAutopack(
	1528	backing)
	1529	response = request.execute('')
	1530	self.assertEqual(SmartServerResponse(('ok',)), response)
	1531	repo._pack_collection.reload_pack_names()
	1532	self.assertEqual(1, len(repo._pack_collection.names()))
3943.8.1 by Marius Kruger remove all trailing whitespace from bzr source	1533
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1534	def test_autopack_not_needed(self):
	1535	tree = self.make_branch_and_tree('.', format='pack-0.92')
	1536	repo = tree.branch.repository
4145.1.6 by Robert Collins More test fallout, but all caught now.	1537	repo.lock_write()
	1538	self.addCleanup(repo.unlock)
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1539	for x in range(9):
	1540	tree.commit('commit %s' % x)
	1541	backing = self.get_transport()
	1542	request = smart.packrepository.SmartServerPackRepositoryAutopack(
	1543	backing)
	1544	response = request.execute('')
	1545	self.assertEqual(SmartServerResponse(('ok',)), response)
	1546	repo._pack_collection.reload_pack_names()
	1547	self.assertEqual(9, len(repo._pack_collection.names()))
3943.8.1 by Marius Kruger remove all trailing whitespace from bzr source	1548
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1549	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.	1550	"""A request to autopack a non-pack repo is a no-op."""
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1551	repo = self.make_repository('.', format='knit')
	1552	backing = self.get_transport()
	1553	request = smart.packrepository.SmartServerPackRepositoryAutopack(
	1554	backing)
	1555	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.	1556	self.assertEqual(SmartServerResponse(('ok',)), response)
3943.8.1 by Marius Kruger remove all trailing whitespace from bzr source	1557
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1558
2018.6.1 by Robert Collins Implement a BzrDir.open_branch smart server method for opening a branch without VFS.	1559	class TestHandlers(tests.TestCase):
	1560	"""Tests for the request.request_handlers object."""
	1561
3526.3.1 by Andrew Bennetts Remove registrations of defunct HPSS verbs.	1562	def test_all_registrations_exist(self):
	1563	"""All registered request_handlers can be found."""
	1564	# If there's a typo in a register_lazy call, this loop will fail with
	1565	# an AttributeError.
	1566	for key, item in smart.request.request_handlers.iteritems():
	1567	pass
	1568
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1569	def assertHandlerEqual(self, verb, handler):
	1570	self.assertEqual(smart.request.request_handlers.get(verb), handler)
	1571
2018.6.1 by Robert Collins Implement a BzrDir.open_branch smart server method for opening a branch without VFS.	1572	def test_registered_methods(self):
	1573	"""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()	1574	self.assertHandlerEqual('Branch.get_config_file',
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	1575	smart.branch.SmartServerBranchGetConfigFile)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1576	self.assertHandlerEqual('Branch.get_parent',
4078.2.1 by Robert Collins Add a Branch.get_parent remote call for RemoteBranch.	1577	smart.branch.SmartServerBranchGetParent)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1578	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.	1579	smart.branch.SmartServerBranchGetTagsBytes)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1580	self.assertHandlerEqual('Branch.lock_write',
2018.5.79 by Andrew Bennetts Implement RemoteBranch.lock_write/unlock as smart operations.	1581	smart.branch.SmartServerBranchRequestLockWrite)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1582	self.assertHandlerEqual('Branch.last_revision_info',
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	1583	smart.branch.SmartServerBranchRequestLastRevisionInfo)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1584	self.assertHandlerEqual('Branch.revision_history',
2018.5.38 by Robert Collins Implement RemoteBranch.revision_history().	1585	smart.branch.SmartServerRequestRevisionHistory)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1586	self.assertHandlerEqual('Branch.set_config_option',
4226.2.1 by Robert Collins Set branch config options via a smart method.	1587	smart.branch.SmartServerBranchRequestSetConfigOption)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1588	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	1589	smart.branch.SmartServerBranchRequestSetLastRevision)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1590	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.	1591	smart.branch.SmartServerBranchRequestSetLastRevisionInfo)
4288.1.7 by Robert Collins Add new remote server verb Branch.set_parent_location, dropping roundtrips further on push operations.	1592	self.assertHandlerEqual('Branch.set_last_revision_ex',
	1593	smart.branch.SmartServerBranchRequestSetLastRevisionEx)
	1594	self.assertHandlerEqual('Branch.set_parent_location',
	1595	smart.branch.SmartServerBranchRequestSetParentLocation)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1596	self.assertHandlerEqual('Branch.unlock',
2018.5.79 by Andrew Bennetts Implement RemoteBranch.lock_write/unlock as smart operations.	1597	smart.branch.SmartServerBranchRequestUnlock)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1598	self.assertHandlerEqual('BzrDir.find_repository',
3221.3.2 by Robert Collins * New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for	1599	smart.bzrdir.SmartServerRequestFindRepositoryV1)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1600	self.assertHandlerEqual('BzrDir.find_repositoryV2',
3221.3.2 by Robert Collins * New remote method ``RemoteBzrDir.find_repositoryV2`` adding support for	1601	smart.bzrdir.SmartServerRequestFindRepositoryV2)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1602	self.assertHandlerEqual('BzrDirFormat.initialize',
2018.5.42 by Robert Collins Various hopefully improvements, but wsgi is broken, handing over to spiv :).	1603	smart.bzrdir.SmartServerRequestInitializeBzrDir)
4436.1.1 by Andrew Bennetts Rename BzrDirFormat.initialize_ex verb to BzrDirFormat.initialize_ex_1.16.	1604	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.	1605	smart.bzrdir.SmartServerRequestBzrDirInitializeEx)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1606	self.assertHandlerEqual('BzrDir.cloning_metadir',
4070.2.3 by Robert Collins Get BzrDir.cloning_metadir working.	1607	smart.bzrdir.SmartServerBzrDirRequestCloningMetaDir)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1608	self.assertHandlerEqual('BzrDir.get_config_file',
	1609	smart.bzrdir.SmartServerBzrDirRequestConfigFile)
	1610	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.	1611	smart.bzrdir.SmartServerRequestOpenBranch)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1612	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.	1613	smart.bzrdir.SmartServerRequestOpenBranchV2)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1614	self.assertHandlerEqual('PackRepository.autopack',
3801.1.4 by Andrew Bennetts Add tests for autopack RPC.	1615	smart.packrepository.SmartServerPackRepositoryAutopack)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1616	self.assertHandlerEqual('Repository.gather_stats',
2018.10.2 by v.ladeuil+lp at free gather_stats server side and request registration	1617	smart.repository.SmartServerRepositoryGatherStats)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1618	self.assertHandlerEqual('Repository.get_parent_map',
3172.5.6 by Robert Collins Create new smart server verb Repository.get_parent_map.	1619	smart.repository.SmartServerRepositoryGetParentMap)
4419.2.6 by Andrew Bennetts Add tests for server-side logic, and fix the bugs exposed by those tests.	1620	self.assertHandlerEqual('Repository.get_rev_id_for_revno',
	1621	smart.repository.SmartServerRepositoryGetRevIdForRevno)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1622	self.assertHandlerEqual('Repository.get_revision_graph',
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1623	smart.repository.SmartServerRepositoryGetRevisionGraph)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1624	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.	1625	smart.repository.SmartServerRepositoryGetStream)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1626	self.assertHandlerEqual('Repository.has_revision',
2018.5.40 by Robert Collins Implement a remote Repository.has_revision method.	1627	smart.repository.SmartServerRequestHasRevision)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1628	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.	1629	smart.repository.SmartServerRepositoryInsertStream)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1630	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.	1631	smart.repository.SmartServerRepositoryInsertStreamLocked)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1632	self.assertHandlerEqual('Repository.is_shared',
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1633	smart.repository.SmartServerRepositoryIsShared)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1634	self.assertHandlerEqual('Repository.lock_write',
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1635	smart.repository.SmartServerRepositoryLockWrite)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1636	self.assertHandlerEqual('Repository.tarball',
2535.3.69 by Andrew Bennetts Add check for Repository.stream_knit_data_for_revisions to TestHandlers.test_registered_methods.	1637	smart.repository.SmartServerRepositoryTarball)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1638	self.assertHandlerEqual('Repository.unlock',
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1639	smart.repository.SmartServerRepositoryUnlock)
4288.1.2 by Robert Collins Create a server verb for doing BzrDir.get_config()	1640	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.	1641	smart.request.SmartServerIsReadonly)