~bzr-pqm/bzr/bzr.dev : contents of bzrlib/tests/test

~bzr-pqm/bzr/bzr.dev : (revision 3692.1.6)

3407.2.2 by Martin Pool Remove special case in RemoteBranchLockableFiles for branch.conf	1	# Copyright (C) 2006, 2007, 2008 Canonical Ltd
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	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
	15	# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	16
	17	"""Tests for remote bzrdir/branch/repo/etc
	18
	19	These are proxy objects which act on remote objects by sending messages
	20	through a smart client. The proxies are to be created when attempting to open
	21	the object given a transport that supports smartserver rpc operations.
2018.18.7 by Martin Pool (broken) Start addng client proxy test for Repository.tarball	22
	23	These tests correspond to tests.test_smart, which exercises the server side.
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	24	"""
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	25
3211.5.2 by Robert Collins Change RemoteRepository.get_parent_map to use bz2 not gzip for compression.	26	import bz2
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	27	from cStringIO import StringIO
	28
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	29	from bzrlib import (
	30	errors,
3184.1.9 by Robert Collins * ``Repository.get_data_stream`` is now deprecated in favour of	31	graph,
2535.3.39 by Andrew Bennetts Tidy some XXXs.	32	pack,
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	33	remote,
2018.18.9 by Martin Pool remote Repository.tarball builds a temporary directory and tars that	34	repository,
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	35	tests,
	36	)
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	37	from bzrlib.branch import Branch
	38	from bzrlib.bzrdir import BzrDir, BzrDirFormat
	39	from bzrlib.remote import (
	40	RemoteBranch,
	41	RemoteBzrDir,
	42	RemoteBzrDirFormat,
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	43	RemoteRepository,
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	44	)
	45	from bzrlib.revision import NULL_REVISION
2432.3.2 by Andrew Bennetts Add test, and tidy implementation.	46	from bzrlib.smart import server, medium
2018.5.159 by Andrew Bennetts Rename SmartClient to _SmartClient.	47	from bzrlib.smart.client import _SmartClient
3287.6.4 by Robert Collins Fix up deprecation warnings for get_revision_graph.	48	from bzrlib.symbol_versioning import one_four
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	49	from bzrlib.transport import get_transport, http
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	50	from bzrlib.transport.memory import MemoryTransport
3431.3.1 by Andrew Bennetts First rough cut of a fix for bug #230550, by adding .base to SmartClientMedia rather than relying on other objects to track this accurately while reusing client media.	51	from bzrlib.transport.remote import RemoteTransport, RemoteTCPTransport
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	52
2018.5.24 by Andrew Bennetts Setting NO_SMART_VFS in environment will disable VFS methods in the smart server. (Robert Collins, John Arbash Meinel, Andrew Bennetts)	53
2018.5.42 by Robert Collins Various hopefully improvements, but wsgi is broken, handing over to spiv :).	54	class BasicRemoteObjectTests(tests.TestCaseWithTransport):
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	55
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	56	def setUp(self):
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.	57	self.transport_server = server.SmartTCPServer_for_testing
2018.5.42 by Robert Collins Various hopefully improvements, but wsgi is broken, handing over to spiv :).	58	super(BasicRemoteObjectTests, self).setUp()
	59	self.transport = self.get_transport()
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	60	# make a branch that can be opened over the smart transport
1752.2.31 by Martin Pool [broken] some support for write operations over hpss	61	self.local_wt = BzrDir.create_standalone_workingtree('.')
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	62
2018.5.171 by Andrew Bennetts Disconnect RemoteTransports in some tests to avoid tripping up test_strace with leftover threads from previous tests.	63	def tearDown(self):
	64	self.transport.disconnect()
	65	tests.TestCaseWithTransport.tearDown(self)
	66
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	67	def test_create_remote_bzrdir(self):
	68	b = remote.RemoteBzrDir(self.transport)
	69	self.assertIsInstance(b, BzrDir)
	70
	71	def test_open_remote_branch(self):
2018.6.1 by Robert Collins Implement a BzrDir.open_branch smart server method for opening a branch without VFS.	72	# open a standalone branch in the working directory
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	73	b = remote.RemoteBzrDir(self.transport)
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	74	branch = b.open_branch()
2018.5.163 by Andrew Bennetts Deal with various review comments from Robert.	75	self.assertIsInstance(branch, Branch)
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	76
1752.2.31 by Martin Pool [broken] some support for write operations over hpss	77	def test_remote_repository(self):
	78	b = BzrDir.open_from_transport(self.transport)
	79	repo = b.open_repository()
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	80	revid = u'\xc823123123'.encode('utf8')
2018.5.40 by Robert Collins Implement a remote Repository.has_revision method.	81	self.assertFalse(repo.has_revision(revid))
	82	self.local_wt.commit(message='test commit', rev_id=revid)
	83	self.assertTrue(repo.has_revision(revid))
1752.2.31 by Martin Pool [broken] some support for write operations over hpss	84
	85	def test_remote_branch_revision_history(self):
	86	b = BzrDir.open_from_transport(self.transport).open_branch()
2018.5.38 by Robert Collins Implement RemoteBranch.revision_history().	87	self.assertEqual([], b.revision_history())
	88	r1 = self.local_wt.commit('1st commit')
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	89	r2 = self.local_wt.commit('1st commit', rev_id=u'\xc8'.encode('utf8'))
2018.5.38 by Robert Collins Implement RemoteBranch.revision_history().	90	self.assertEqual([r1, r2], b.revision_history())
1752.2.31 by Martin Pool [broken] some support for write operations over hpss	91
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	92	def test_find_correct_format(self):
2018.5.20 by Andrew Bennetts Move bzrlib/transport/smart/_smart.py to bzrlib/transport/remote.py and rename SmartTransport to RemoteTransport (Robert Collins, Andrew Bennetts)	93	"""Should open a RemoteBzrDir over a RemoteTransport"""
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	94	fmt = BzrDirFormat.find_format(self.transport)
2018.5.169 by Andrew Bennetts Add a _server_formats flag to BzrDir.open_from_transport and BzrDirFormat.find_format, make RemoteBranch.control_files into a property.	95	self.assertTrue(RemoteBzrDirFormat
	96	in BzrDirFormat._control_server_formats)
1752.2.30 by Martin Pool Start adding a RemoteBzrDir, etc	97	self.assertIsInstance(fmt, remote.RemoteBzrDirFormat)
	98
	99	def test_open_detected_smart_format(self):
	100	fmt = BzrDirFormat.find_format(self.transport)
	101	d = fmt.open(self.transport)
	102	self.assertIsInstance(d, BzrDir)
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	103
2477.1.1 by Martin Pool Add RemoteBranch repr	104	def test_remote_branch_repr(self):
	105	b = BzrDir.open_from_transport(self.transport).open_branch()
	106	self.assertStartsWith(str(b), 'RemoteBranch(')
	107
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	108
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	109	class FakeProtocol(object):
	110	"""Lookalike SmartClientRequestProtocolOne allowing body reading tests."""
	111
2535.3.68 by Andrew Bennetts Backwards compatibility for new smart method.	112	def __init__(self, body, fake_client):
2535.4.2 by Andrew Bennetts Nasty hackery to make stream_knit_data_for_revisions response use streaming.	113	self.body = body
	114	self._body_buffer = None
2535.3.68 by Andrew Bennetts Backwards compatibility for new smart method.	115	self._fake_client = fake_client
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	116
	117	def read_body_bytes(self, count=-1):
2535.4.2 by Andrew Bennetts Nasty hackery to make stream_knit_data_for_revisions response use streaming.	118	if self._body_buffer is None:
	119	self._body_buffer = StringIO(self.body)
2535.3.68 by Andrew Bennetts Backwards compatibility for new smart method.	120	bytes = self._body_buffer.read(count)
	121	if self._body_buffer.tell() == len(self._body_buffer.getvalue()):
	122	self._fake_client.expecting_body = False
	123	return bytes
	124
	125	def cancel_read_body(self):
	126	self._fake_client.expecting_body = False
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	127
2535.4.2 by Andrew Bennetts Nasty hackery to make stream_knit_data_for_revisions response use streaming.	128	def read_streamed_body(self):
	129	return self.body
	130
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	131
2018.5.159 by Andrew Bennetts Rename SmartClient to _SmartClient.	132	class FakeClient(_SmartClient):
	133	"""Lookalike for _SmartClient allowing testing."""
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	134
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	135	def __init__(self, fake_medium_base='fake base'):
2535.3.68 by Andrew Bennetts Backwards compatibility for new smart method.	136	"""Create a FakeClient.
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	137
3104.4.2 by Andrew Bennetts All tests passing.	138	:param responses: A list of response-tuple, body-data pairs to be sent
3297.3.3 by Andrew Bennetts SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.	139	back to callers. A special case is if the response-tuple is
	140	'unknown verb', then a UnknownSmartMethod will be raised for that
	141	call, using the second element of the tuple as the verb in the
	142	exception.
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	143	"""
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	144	self.responses = []
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	145	self._calls = []
2535.3.68 by Andrew Bennetts Backwards compatibility for new smart method.	146	self.expecting_body = False
3431.3.2 by Andrew Bennetts Remove 'base' from _SmartClient entirely, now that the medium has it.	147	_SmartClient.__init__(self, FakeMedium(self._calls, fake_medium_base))
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	148
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	149	def add_success_response(self, *args):
	150	self.responses.append(('success', args, None))
	151
	152	def add_success_response_with_body(self, body, *args):
	153	self.responses.append(('success', args, body))
	154
	155	def add_error_response(self, *args):
	156	self.responses.append(('error', args))
	157
	158	def add_unknown_method_response(self, verb):
	159	self.responses.append(('unknown', verb))
	160
3297.3.3 by Andrew Bennetts SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.	161	def _get_next_response(self):
	162	response_tuple = self.responses.pop(0)
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	163	if response_tuple[0] == 'unknown':
	164	raise errors.UnknownSmartMethod(response_tuple[1])
	165	elif response_tuple[0] == 'error':
	166	raise errors.ErrorFromSmartServer(response_tuple[1])
3297.3.3 by Andrew Bennetts SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.	167	return response_tuple
	168
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	169	def call(self, method, *args):
	170	self._calls.append(('call', method, args))
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	171	return self._get_next_response()[1]
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	172
2018.5.153 by Andrew Bennetts Rename call2 to call_expecting_body, and other small changes prompted by review.	173	def call_expecting_body(self, method, *args):
	174	self._calls.append(('call_expecting_body', method, args))
3297.3.3 by Andrew Bennetts SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.	175	result = self._get_next_response()
2535.3.68 by Andrew Bennetts Backwards compatibility for new smart method.	176	self.expecting_body = True
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	177	return result[1], FakeProtocol(result[2], self)
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	178
3184.1.10 by Robert Collins Change the smart server verb for Repository.stream_revisions_chunked to use SearchResults as the request mechanism for downloads.	179	def call_with_body_bytes_expecting_body(self, method, args, body):
	180	self._calls.append(('call_with_body_bytes_expecting_body', method,
	181	args, body))
3297.3.3 by Andrew Bennetts SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.	182	result = self._get_next_response()
3184.1.10 by Robert Collins Change the smart server verb for Repository.stream_revisions_chunked to use SearchResults as the request mechanism for downloads.	183	self.expecting_body = True
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	184	return result[1], FakeProtocol(result[2], self)
3184.1.10 by Robert Collins Change the smart server verb for Repository.stream_revisions_chunked to use SearchResults as the request mechanism for downloads.	185
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	186
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	187	class FakeMedium(medium.SmartClientMedium):
3104.4.2 by Andrew Bennetts All tests passing.	188
3431.3.1 by Andrew Bennetts First rough cut of a fix for bug #230550, by adding .base to SmartClientMedia rather than relying on other objects to track this accurately while reusing client media.	189	def __init__(self, client_calls, base):
3453.4.1 by Andrew Bennetts Better infrastructure on SmartClientMedium for tracking the remote version.	190	medium.SmartClientMedium.__init__(self, base)
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	191	self._client_calls = client_calls
3213.1.1 by Andrew Bennetts Recover (by reconnecting) if the server turns out not to understand the new requests in 1.2 that send bodies.	192
	193	def disconnect(self):
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	194	self._client_calls.append(('disconnect medium',))
3104.4.2 by Andrew Bennetts All tests passing.	195
3104.4.2 by Andrew Bennetts All tests passing.	196
3192.2.1 by Andrew Bennetts Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.	197	class TestVfsHas(tests.TestCase):
	198
	199	def test_unicode_path(self):
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	200	client = FakeClient('/')
	201	client.add_success_response('yes',)
3192.2.1 by Andrew Bennetts Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.	202	transport = RemoteTransport('bzr://localhost/', _client=client)
	203	filename = u'/hell\u00d8'.encode('utf8')
	204	result = transport.has(filename)
	205	self.assertEqual(
	206	[('call', 'has', (filename,))],
	207	client._calls)
	208	self.assertTrue(result)
	209
	210
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	211	class Test_ClientMedium_remote_path_from_transport(tests.TestCase):
	212	"""Tests for the behaviour of client_medium.remote_path_from_transport."""
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	213
	214	def assertRemotePath(self, expected, client_base, transport_base):
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	215	"""Assert that the result of
	216	SmartClientMedium.remote_path_from_transport is the expected value for
	217	a given client_base and transport_base.
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	218	"""
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	219	client_medium = medium.SmartClientMedium(client_base)
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	220	transport = get_transport(transport_base)
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	221	result = client_medium.remote_path_from_transport(transport)
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	222	self.assertEqual(expected, result)
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	223
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	224	def test_remote_path_from_transport(self):
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	225	"""SmartClientMedium.remote_path_from_transport calculates a URL for
	226	the given transport relative to the root of the client base URL.
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	227	"""
	228	self.assertRemotePath('xyz/', 'bzr://host/path', 'bzr://host/xyz')
	229	self.assertRemotePath(
	230	'path/xyz/', 'bzr://host/path', 'bzr://host/path/xyz')
	231
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	232	def assertRemotePathHTTP(self, expected, transport_base, relpath):
	233	"""Assert that the result of
	234	HttpTransportBase.remote_path_from_transport is the expected value for
	235	a given transport_base and relpath of that transport. (Note that
	236	HttpTransportBase is a subclass of SmartClientMedium)
	237	"""
	238	base_transport = get_transport(transport_base)
	239	client_medium = base_transport.get_smart_medium()
	240	cloned_transport = base_transport.clone(relpath)
	241	result = client_medium.remote_path_from_transport(cloned_transport)
	242	self.assertEqual(expected, result)
	243
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	244	def test_remote_path_from_transport_http(self):
	245	"""Remote paths for HTTP transports are calculated differently to other
	246	transports. They are just relative to the client base, not the root
	247	directory of the host.
	248	"""
	249	for scheme in ['http:', 'https:', 'bzr+http:', 'bzr+https:']:
3431.3.11 by Andrew Bennetts Push remote_path_from_transport logic into SmartClientMedium, removing special-casing of bzr+http from _SmartClient.	250	self.assertRemotePathHTTP(
	251	'../xyz/', scheme + '//host/path', '../xyz/')
	252	self.assertRemotePathHTTP(
	253	'xyz/', scheme + '//host/path', 'xyz/')
3313.3.3 by Andrew Bennetts Add tests for _SmartClient.remote_path_for_transport.	254
	255
3453.4.1 by Andrew Bennetts Better infrastructure on SmartClientMedium for tracking the remote version.	256	class Test_ClientMedium_remote_is_at_least(tests.TestCase):
	257	"""Tests for the behaviour of client_medium.remote_is_at_least."""
	258
	259	def test_initially_unlimited(self):
	260	"""A fresh medium assumes that the remote side supports all
	261	versions.
	262	"""
	263	client_medium = medium.SmartClientMedium('dummy base')
3453.4.10 by Andrew Bennetts Change _is_remote_at_least to _is_remote_before.	264	self.assertFalse(client_medium._is_remote_before((99, 99)))
3453.4.1 by Andrew Bennetts Better infrastructure on SmartClientMedium for tracking the remote version.	265
3453.4.9 by Andrew Bennetts Rename _remote_is_not to _remember_remote_is_before.	266	def test__remember_remote_is_before(self):
	267	"""Calling _remember_remote_is_before ratchets down the known remote
	268	version.
	269	"""
3453.4.1 by Andrew Bennetts Better infrastructure on SmartClientMedium for tracking the remote version.	270	client_medium = medium.SmartClientMedium('dummy base')
	271	# Mark the remote side as being less than 1.6. The remote side may
	272	# still be 1.5.
3453.4.9 by Andrew Bennetts Rename _remote_is_not to _remember_remote_is_before.	273	client_medium._remember_remote_is_before((1, 6))
3453.4.10 by Andrew Bennetts Change _is_remote_at_least to _is_remote_before.	274	self.assertTrue(client_medium._is_remote_before((1, 6)))
	275	self.assertFalse(client_medium._is_remote_before((1, 5)))
3453.4.9 by Andrew Bennetts Rename _remote_is_not to _remember_remote_is_before.	276	# Calling _remember_remote_is_before again with a lower value works.
	277	client_medium._remember_remote_is_before((1, 5))
3453.4.10 by Andrew Bennetts Change _is_remote_at_least to _is_remote_before.	278	self.assertTrue(client_medium._is_remote_before((1, 5)))
3453.4.9 by Andrew Bennetts Rename _remote_is_not to _remember_remote_is_before.	279	# You cannot call _remember_remote_is_before with a larger value.
3453.4.1 by Andrew Bennetts Better infrastructure on SmartClientMedium for tracking the remote version.	280	self.assertRaises(
3453.4.9 by Andrew Bennetts Rename _remote_is_not to _remember_remote_is_before.	281	AssertionError, client_medium._remember_remote_is_before, (1, 9))
3453.4.1 by Andrew Bennetts Better infrastructure on SmartClientMedium for tracking the remote version.	282
	283
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	284	class TestBzrDirOpenBranch(tests.TestCase):
	285
	286	def test_branch_present(self):
	287	transport = MemoryTransport()
	288	transport.mkdir('quack')
	289	transport = transport.clone('quack')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	290	client = FakeClient(transport.base)
	291	client.add_success_response('ok', '')
	292	client.add_success_response('ok', '', 'no', 'no', 'no')
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	293	bzrdir = RemoteBzrDir(transport, _client=client)
	294	result = bzrdir.open_branch()
	295	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	296	[('call', 'BzrDir.open_branch', ('quack/',)),
3221.3.3 by Robert Collins * Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so	297	('call', 'BzrDir.find_repositoryV2', ('quack/',))],
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	298	client._calls)
	299	self.assertIsInstance(result, RemoteBranch)
	300	self.assertEqual(bzrdir, result.bzrdir)
	301
	302	def test_branch_missing(self):
	303	transport = MemoryTransport()
	304	transport.mkdir('quack')
	305	transport = transport.clone('quack')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	306	client = FakeClient(transport.base)
	307	client.add_error_response('nobranch')
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	308	bzrdir = RemoteBzrDir(transport, _client=client)
	309	self.assertRaises(errors.NotBranchError, bzrdir.open_branch)
	310	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	311	[('call', 'BzrDir.open_branch', ('quack/',))],
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	312	client._calls)
	313
3211.4.1 by Robert Collins * ``RemoteBzrDir._get_tree_branch`` no longer triggers ``_ensure_real``,	314	def test__get_tree_branch(self):
	315	# _get_tree_branch is a form of open_branch, but it should only ask for
	316	# branch opening, not any other network requests.
	317	calls = []
	318	def open_branch():
	319	calls.append("Called")
	320	return "a-branch"
	321	transport = MemoryTransport()
	322	# no requests on the network - catches other api calls being made.
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	323	client = FakeClient(transport.base)
3211.4.1 by Robert Collins * ``RemoteBzrDir._get_tree_branch`` no longer triggers ``_ensure_real``,	324	bzrdir = RemoteBzrDir(transport, _client=client)
	325	# patch the open_branch call to record that it was called.
	326	bzrdir.open_branch = open_branch
	327	self.assertEqual((None, "a-branch"), bzrdir._get_tree_branch())
	328	self.assertEqual(["Called"], calls)
	329	self.assertEqual([], client._calls)
	330
3192.2.1 by Andrew Bennetts Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.	331	def test_url_quoting_of_path(self):
	332	# Relpaths on the wire should not be URL-escaped. So "~" should be
	333	# transmitted as "~", not "%7E".
3431.3.1 by Andrew Bennetts First rough cut of a fix for bug #230550, by adding .base to SmartClientMedia rather than relying on other objects to track this accurately while reusing client media.	334	transport = RemoteTCPTransport('bzr://localhost/~hello/')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	335	client = FakeClient(transport.base)
	336	client.add_success_response('ok', '')
	337	client.add_success_response('ok', '', 'no', 'no', 'no')
3192.2.1 by Andrew Bennetts Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.	338	bzrdir = RemoteBzrDir(transport, _client=client)
	339	result = bzrdir.open_branch()
	340	self.assertEqual(
	341	[('call', 'BzrDir.open_branch', ('~hello/',)),
3221.3.3 by Robert Collins * Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so	342	('call', 'BzrDir.find_repositoryV2', ('~hello/',))],
3192.2.1 by Andrew Bennetts Don't transmit URL-escaped relpaths in the smart protocol, which is back to how things worked in bzr 1.1 and earlier.	343	client._calls)
	344
3221.3.3 by Robert Collins * Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so	345	def check_open_repository(self, rich_root, subtrees, external_lookup='no'):
3104.4.2 by Andrew Bennetts All tests passing.	346	transport = MemoryTransport()
	347	transport.mkdir('quack')
	348	transport = transport.clone('quack')
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	349	if rich_root:
2018.5.166 by Andrew Bennetts Small changes in response to Aaron's review.	350	rich_response = 'yes'
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	351	else:
2018.5.166 by Andrew Bennetts Small changes in response to Aaron's review.	352	rich_response = 'no'
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	353	if subtrees:
2018.5.166 by Andrew Bennetts Small changes in response to Aaron's review.	354	subtree_response = 'yes'
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	355	else:
2018.5.166 by Andrew Bennetts Small changes in response to Aaron's review.	356	subtree_response = 'no'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	357	client = FakeClient(transport.base)
	358	client.add_success_response(
	359	'ok', '', rich_response, subtree_response, external_lookup)
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	360	bzrdir = RemoteBzrDir(transport, _client=client)
	361	result = bzrdir.open_repository()
	362	self.assertEqual(
3221.3.3 by Robert Collins * Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so	363	[('call', 'BzrDir.find_repositoryV2', ('quack/',))],
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	364	client._calls)
	365	self.assertIsInstance(result, RemoteRepository)
	366	self.assertEqual(bzrdir, result.bzrdir)
	367	self.assertEqual(rich_root, result._format.rich_root_data)
2018.5.138 by Robert Collins Merge bzr.dev.	368	self.assertEqual(subtrees, result._format.supports_tree_reference)
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	369
	370	def test_open_repository_sets_format_attributes(self):
	371	self.check_open_repository(True, True)
	372	self.check_open_repository(False, True)
	373	self.check_open_repository(True, False)
	374	self.check_open_repository(False, False)
3221.3.3 by Robert Collins * Hook up the new remote method ``RemoteBzrDir.find_repositoryV2`` so	375	self.check_open_repository(False, False, 'yes')
2018.5.118 by Robert Collins Fix RemoteRepositoryFormat to have appropriate rich_root_data and support_tree_reference.	376
2432.3.2 by Andrew Bennetts Add test, and tidy implementation.	377	def test_old_server(self):
	378	"""RemoteBzrDirFormat should fail to probe if the server version is too
	379	old.
	380	"""
	381	self.assertRaises(errors.NotBranchError,
	382	RemoteBzrDirFormat.probe_transport, OldServerTransport())
	383
	384
3297.3.3 by Andrew Bennetts SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.	385	class TestBzrDirOpenRepository(tests.TestCase):
	386
	387	def test_backwards_compat_1_2(self):
	388	transport = MemoryTransport()
	389	transport.mkdir('quack')
	390	transport = transport.clone('quack')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	391	client = FakeClient(transport.base)
	392	client.add_unknown_method_response('RemoteRepository.find_repositoryV2')
	393	client.add_success_response('ok', '', 'no', 'no')
3297.3.3 by Andrew Bennetts SmartClientRequestProtocol*.read_response_tuple can now raise UnknownSmartMethod. Callers no longer need to do their own ad hoc unknown smart method error detection.	394	bzrdir = RemoteBzrDir(transport, _client=client)
	395	repo = bzrdir.open_repository()
	396	self.assertEqual(
	397	[('call', 'BzrDir.find_repositoryV2', ('quack/',)),
	398	('call', 'BzrDir.find_repository', ('quack/',))],
	399	client._calls)
	400
	401
2432.3.2 by Andrew Bennetts Add test, and tidy implementation.	402	class OldSmartClient(object):
	403	"""A fake smart client for test_old_version that just returns a version one
	404	response to the 'hello' (query version) command.
	405	"""
	406
	407	def get_request(self):
	408	input_file = StringIO('ok\x011\n')
	409	output_file = StringIO()
	410	client_medium = medium.SmartSimplePipesClientMedium(
	411	input_file, output_file)
	412	return medium.SmartClientStreamMediumRequest(client_medium)
	413
3241.1.1 by Andrew Bennetts Shift protocol version querying from RemoteBzrDirFormat into SmartClientMedium.	414	def protocol_version(self):
	415	return 1
	416
2432.3.2 by Andrew Bennetts Add test, and tidy implementation.	417
	418	class OldServerTransport(object):
	419	"""A fake transport for test_old_server that reports it's smart server
	420	protocol version as version one.
	421	"""
	422
	423	def __init__(self):
	424	self.base = 'fake:'
	425
	426	def get_smart_client(self):
	427	return OldSmartClient()
	428
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	429
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	430	class RemoteBranchTestCase(tests.TestCase):
	431
	432	def make_remote_branch(self, transport, client):
	433	"""Make a RemoteBranch using 'client' as its _SmartClient.
	434
	435	A RemoteBzrDir and RemoteRepository will also be created to fill out
	436	the RemoteBranch, albeit with stub values for some of their attributes.
	437	"""
	438	# we do not want bzrdir to make any remote calls, so use False as its
	439	# _client. If it tries to make a remote call, this will fail
	440	# immediately.
	441	bzrdir = RemoteBzrDir(transport, _client=False)
	442	repo = RemoteRepository(bzrdir, None, _client=client)
	443	return RemoteBranch(bzrdir, repo, _client=client)
	444
	445
	446	class TestBranchLastRevisionInfo(RemoteBranchTestCase):
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	447
	448	def test_empty_branch(self):
	449	# in an empty branch we decode the response properly
	450	transport = MemoryTransport()
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	451	client = FakeClient(transport.base)
	452	client.add_success_response('ok', '0', 'null:')
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	453	transport.mkdir('quack')
	454	transport = transport.clone('quack')
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	455	branch = self.make_remote_branch(transport, client)
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	456	result = branch.last_revision_info()
	457
	458	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	459	[('call', 'Branch.last_revision_info', ('quack/',))],
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	460	client._calls)
	461	self.assertEqual((0, NULL_REVISION), result)
	462
	463	def test_non_empty_branch(self):
	464	# in a non-empty branch we also decode the response properly
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	465	revid = u'\xc8'.encode('utf8')
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	466	transport = MemoryTransport()
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	467	client = FakeClient(transport.base)
	468	client.add_success_response('ok', '2', revid)
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	469	transport.mkdir('kwaak')
	470	transport = transport.clone('kwaak')
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	471	branch = self.make_remote_branch(transport, client)
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	472	result = branch.last_revision_info()
	473
	474	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	475	[('call', 'Branch.last_revision_info', ('kwaak/',))],
2018.5.51 by Wouter van Heyst Test and implement RemoteBranch.last_revision_info()	476	client._calls)
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	477	self.assertEqual((2, revid), result)
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	478
	479
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	480	class TestBranchSetLastRevision(RemoteBranchTestCase):
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	481
	482	def test_set_empty(self):
	483	# set_revision_history([]) is translated to calling
	484	# Branch.set_last_revision(path, '') on the wire.
3104.4.2 by Andrew Bennetts All tests passing.	485	transport = MemoryTransport()
	486	transport.mkdir('branch')
	487	transport = transport.clone('branch')
	488
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	489	client = FakeClient(transport.base)
	490	# lock_write
	491	client.add_success_response('ok', 'branch token', 'repo token')
	492	# set_last_revision
	493	client.add_success_response('ok')
	494	# unlock
	495	client.add_success_response('ok')
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	496	branch = self.make_remote_branch(transport, client)
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	497	branch.lock_write()
	498	client._calls = []
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	499	result = branch.set_revision_history([])
	500	self.assertEqual(
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	501	[('call', 'Branch.set_last_revision',
3104.4.2 by Andrew Bennetts All tests passing.	502	('branch/', 'branch token', 'repo token', 'null:'))],
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	503	client._calls)
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	504	branch.unlock()
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	505	self.assertEqual(None, result)
	506
	507	def test_set_nonempty(self):
	508	# set_revision_history([rev-id1, ..., rev-idN]) is translated to calling
	509	# Branch.set_last_revision(path, rev-idN) on the wire.
3104.4.2 by Andrew Bennetts All tests passing.	510	transport = MemoryTransport()
	511	transport.mkdir('branch')
	512	transport = transport.clone('branch')
	513
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	514	client = FakeClient(transport.base)
	515	# lock_write
	516	client.add_success_response('ok', 'branch token', 'repo token')
	517	# set_last_revision
	518	client.add_success_response('ok')
	519	# unlock
	520	client.add_success_response('ok')
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	521	branch = self.make_remote_branch(transport, client)
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	522	# Lock the branch, reset the record of remote calls.
	523	branch.lock_write()
	524	client._calls = []
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	525
	526	result = branch.set_revision_history(['rev-id1', 'rev-id2'])
	527	self.assertEqual(
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	528	[('call', 'Branch.set_last_revision',
3104.4.2 by Andrew Bennetts All tests passing.	529	('branch/', 'branch token', 'repo token', 'rev-id2'))],
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	530	client._calls)
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	531	branch.unlock()
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	532	self.assertEqual(None, result)
	533
	534	def test_no_such_revision(self):
	535	transport = MemoryTransport()
	536	transport.mkdir('branch')
	537	transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	538	# A response of 'NoSuchRevision' is translated into an exception.
	539	client = FakeClient(transport.base)
	540	# lock_write
	541	client.add_success_response('ok', 'branch token', 'repo token')
	542	# set_last_revision
	543	client.add_error_response('NoSuchRevision', 'rev-id')
	544	# unlock
	545	client.add_success_response('ok')
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	546
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	547	branch = self.make_remote_branch(transport, client)
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	548	branch.lock_write()
	549	client._calls = []
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	550
	551	self.assertRaises(
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	552	errors.NoSuchRevision, branch.set_revision_history, ['rev-id'])
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	553	branch.unlock()
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	554
3577.1.1 by Andrew Bennetts Cherry-pick TipChangeRejected changes from pre-branch-tip-changed-hook loom.	555	def test_tip_change_rejected(self):
	556	"""TipChangeRejected responses cause a TipChangeRejected exception to
	557	be raised.
	558	"""
	559	transport = MemoryTransport()
	560	transport.mkdir('branch')
	561	transport = transport.clone('branch')
	562	client = FakeClient(transport.base)
	563	# lock_write
	564	client.add_success_response('ok', 'branch token', 'repo token')
	565	# set_last_revision
	566	rejection_msg_unicode = u'rejection message\N{INTERROBANG}'
	567	rejection_msg_utf8 = rejection_msg_unicode.encode('utf8')
	568	client.add_error_response('TipChangeRejected', rejection_msg_utf8)
	569	# unlock
	570	client.add_success_response('ok')
	571
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	572	branch = self.make_remote_branch(transport, client)
3577.1.1 by Andrew Bennetts Cherry-pick TipChangeRejected changes from pre-branch-tip-changed-hook loom.	573	branch.lock_write()
	574	self.addCleanup(branch.unlock)
	575	client._calls = []
	576
	577	# The 'TipChangeRejected' error response triggered by calling
	578	# set_revision_history causes a TipChangeRejected exception.
	579	err = self.assertRaises(
	580	errors.TipChangeRejected, branch.set_revision_history, ['rev-id'])
	581	# The UTF-8 message from the response has been decoded into a unicode
	582	# object.
	583	self.assertIsInstance(err.msg, unicode)
	584	self.assertEqual(rejection_msg_unicode, err.msg)
	585
2018.12.3 by Andrew Bennetts Add a Branch.set_last_revision smart method, and make RemoteBranch.set_revision_history use it.	586
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	587	class TestBranchSetLastRevisionInfo(RemoteBranchTestCase):
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	588
3297.4.2 by Andrew Bennetts Add backwards compatibility for servers older than 1.4.	589	def test_set_last_revision_info(self):
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	590	# set_last_revision_info(num, 'rev-id') is translated to calling
	591	# Branch.set_last_revision_info(num, 'rev-id') on the wire.
3297.4.1 by Andrew Bennetts Merge 'Add Branch.set_last_revision_info smart method'.	592	transport = MemoryTransport()
	593	transport.mkdir('branch')
	594	transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	595	client = FakeClient(transport.base)
	596	# lock_write
	597	client.add_success_response('ok', 'branch token', 'repo token')
	598	# set_last_revision
	599	client.add_success_response('ok')
	600	# unlock
	601	client.add_success_response('ok')
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	602
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	603	branch = self.make_remote_branch(transport, client)
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	604	# Lock the branch, reset the record of remote calls.
	605	branch.lock_write()
	606	client._calls = []
	607	result = branch.set_last_revision_info(1234, 'a-revision-id')
	608	self.assertEqual(
	609	[('call', 'Branch.set_last_revision_info',
3297.4.1 by Andrew Bennetts Merge 'Add Branch.set_last_revision_info smart method'.	610	('branch/', 'branch token', 'repo token',
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	611	'1234', 'a-revision-id'))],
	612	client._calls)
	613	self.assertEqual(None, result)
	614
	615	def test_no_such_revision(self):
	616	# A response of 'NoSuchRevision' is translated into an exception.
	617	transport = MemoryTransport()
	618	transport.mkdir('branch')
	619	transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	620	client = FakeClient(transport.base)
	621	# lock_write
	622	client.add_success_response('ok', 'branch token', 'repo token')
	623	# set_last_revision
	624	client.add_error_response('NoSuchRevision', 'revid')
	625	# unlock
	626	client.add_success_response('ok')
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	627
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	628	branch = self.make_remote_branch(transport, client)
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	629	# Lock the branch, reset the record of remote calls.
	630	branch.lock_write()
	631	client._calls = []
	632
	633	self.assertRaises(
	634	errors.NoSuchRevision, branch.set_last_revision_info, 123, 'revid')
	635	branch.unlock()
	636
3297.4.2 by Andrew Bennetts Add backwards compatibility for servers older than 1.4.	637	def lock_remote_branch(self, branch):
	638	"""Trick a RemoteBranch into thinking it is locked."""
	639	branch._lock_mode = 'w'
	640	branch._lock_count = 2
	641	branch._lock_token = 'branch token'
	642	branch._repo_lock_token = 'repo token'
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	643	branch.repository._lock_mode = 'w'
	644	branch.repository._lock_count = 2
	645	branch.repository._lock_token = 'repo token'
3297.4.2 by Andrew Bennetts Add backwards compatibility for servers older than 1.4.	646
	647	def test_backwards_compatibility(self):
	648	"""If the server does not support the Branch.set_last_revision_info
	649	verb (which is new in 1.4), then the client falls back to VFS methods.
	650	"""
	651	# This test is a little messy. Unlike most tests in this file, it
	652	# doesn't purely test what a Remote* object sends over the wire, and
	653	# how it reacts to responses from the wire. It instead relies partly
	654	# on asserting that the RemoteBranch will call
	655	# self._real_branch.set_last_revision_info(...).
	656
	657	# First, set up our RemoteBranch with a FakeClient that raises
	658	# UnknownSmartMethod, and a StubRealBranch that logs how it is called.
	659	transport = MemoryTransport()
	660	transport.mkdir('branch')
	661	transport = transport.clone('branch')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	662	client = FakeClient(transport.base)
	663	client.add_unknown_method_response('Branch.set_last_revision_info')
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	664	branch = self.make_remote_branch(transport, client)
3297.4.2 by Andrew Bennetts Add backwards compatibility for servers older than 1.4.	665	class StubRealBranch(object):
	666	def __init__(self):
	667	self.calls = []
	668	def set_last_revision_info(self, revno, revision_id):
	669	self.calls.append(
	670	('set_last_revision_info', revno, revision_id))
3441.5.5 by Andrew Bennetts Some small tweaks and comments.	671	def _clear_cached_state(self):
3441.5.5 by Andrew Bennetts Some small tweaks and comments.	672	pass
3297.4.2 by Andrew Bennetts Add backwards compatibility for servers older than 1.4.	673	real_branch = StubRealBranch()
	674	branch._real_branch = real_branch
	675	self.lock_remote_branch(branch)
	676
	677	# Call set_last_revision_info, and verify it behaved as expected.
	678	result = branch.set_last_revision_info(1234, 'a-revision-id')
	679	self.assertEqual(
	680	[('call', 'Branch.set_last_revision_info',
	681	('branch/', 'branch token', 'repo token',
	682	'1234', 'a-revision-id')),],
	683	client._calls)
	684	self.assertEqual(
	685	[('set_last_revision_info', 1234, 'a-revision-id')],
	686	real_branch.calls)
	687
3245.4.53 by Andrew Bennetts Add some missing 'raise' statements to test_remote.	688	def test_unexpected_error(self):
	689	# A response of 'NoSuchRevision' is translated into an exception.
	690	transport = MemoryTransport()
	691	transport.mkdir('branch')
	692	transport = transport.clone('branch')
	693	client = FakeClient(transport.base)
	694	# lock_write
	695	client.add_success_response('ok', 'branch token', 'repo token')
	696	# set_last_revision
	697	client.add_error_response('UnexpectedError')
	698	# unlock
	699	client.add_success_response('ok')
	700
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	701	branch = self.make_remote_branch(transport, client)
3245.4.53 by Andrew Bennetts Add some missing 'raise' statements to test_remote.	702	# Lock the branch, reset the record of remote calls.
	703	branch.lock_write()
	704	client._calls = []
	705
	706	err = self.assertRaises(
	707	errors.ErrorFromSmartServer,
	708	branch.set_last_revision_info, 123, 'revid')
	709	self.assertEqual(('UnexpectedError',), err.error_tuple)
	710	branch.unlock()
	711
3577.1.1 by Andrew Bennetts Cherry-pick TipChangeRejected changes from pre-branch-tip-changed-hook loom.	712	def test_tip_change_rejected(self):
	713	"""TipChangeRejected responses cause a TipChangeRejected exception to
	714	be raised.
	715	"""
	716	transport = MemoryTransport()
	717	transport.mkdir('branch')
	718	transport = transport.clone('branch')
	719	client = FakeClient(transport.base)
	720	# lock_write
	721	client.add_success_response('ok', 'branch token', 'repo token')
	722	# set_last_revision
	723	client.add_error_response('TipChangeRejected', 'rejection message')
	724	# unlock
	725	client.add_success_response('ok')
	726
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	727	branch = self.make_remote_branch(transport, client)
3577.1.1 by Andrew Bennetts Cherry-pick TipChangeRejected changes from pre-branch-tip-changed-hook loom.	728	# Lock the branch, reset the record of remote calls.
	729	branch.lock_write()
	730	self.addCleanup(branch.unlock)
	731	client._calls = []
	732
	733	# The 'TipChangeRejected' error response triggered by calling
	734	# set_last_revision_info causes a TipChangeRejected exception.
	735	err = self.assertRaises(
	736	errors.TipChangeRejected,
	737	branch.set_last_revision_info, 123, 'revid')
	738	self.assertEqual('rejection message', err.msg)
	739
2892.2.1 by Andrew Bennetts Add Branch.set_last_revision_info smart method, and make the RemoteBranch client use it.	740
2018.5.169 by Andrew Bennetts Add a _server_formats flag to BzrDir.open_from_transport and BzrDirFormat.find_format, make RemoteBranch.control_files into a property.	741	class TestBranchControlGetBranchConf(tests.TestCaseWithMemoryTransport):
3408.3.1 by Martin Pool Remove erroneous handling of branch.conf for RemoteBranch	742	"""Getting the branch configuration should use an abstract method not vfs.
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	743	"""
	744
	745	def test_get_branch_conf(self):
3408.3.1 by Martin Pool Remove erroneous handling of branch.conf for RemoteBranch	746	raise tests.KnownFailure('branch.conf is not retrieved by get_config_file')
3407.2.10 by Martin Pool Merge trunk	747	## # We should see that branch.get_config() does a single rpc to get the
	748	## # remote configuration file, abstracting away where that is stored on
	749	## # the server. However at the moment it always falls back to using the
	750	## # vfs, and this would need some changes in config.py.
3408.3.1 by Martin Pool Remove erroneous handling of branch.conf for RemoteBranch	751
3407.2.10 by Martin Pool Merge trunk	752	## # in an empty branch we decode the response properly
	753	## client = FakeClient([(('ok', ), '# config file body')], self.get_url())
	754	## # we need to make a real branch because the remote_branch.control_files
	755	## # will trigger _ensure_real.
	756	## branch = self.make_branch('quack')
	757	## transport = branch.bzrdir.root_transport
	758	## # we do not want bzrdir to make any remote calls
	759	## bzrdir = RemoteBzrDir(transport, _client=False)
	760	## branch = RemoteBranch(bzrdir, None, _client=client)
	761	## config = branch.get_config()
	762	## self.assertEqual(
	763	## [('call_expecting_body', 'Branch.get_config_file', ('quack/',))],
	764	## client._calls)
2018.5.59 by Robert Collins Get BranchConfig working somewhat on RemoteBranches (Robert Collins, Vincent Ladeuil).	765
	766
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	767	class TestBranchLockWrite(RemoteBranchTestCase):
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.	768
	769	def test_lock_write_unlockable(self):
	770	transport = MemoryTransport()
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	771	client = FakeClient(transport.base)
	772	client.add_error_response('UnlockableTransport')
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.	773	transport.mkdir('quack')
	774	transport = transport.clone('quack')
3692.1.1 by Andrew Bennetts Make RemoteBranch.lock_write lock the repository too.	775	branch = self.make_remote_branch(transport, client)
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.	776	self.assertRaises(errors.UnlockableTransport, branch.lock_write)
	777	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	778	[('call', 'Branch.lock_write', ('quack/', '', ''))],
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.	779	client._calls)
	780
	781
2466.2.2 by Andrew Bennetts Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.	782	class TestTransportIsReadonly(tests.TestCase):
	783
	784	def test_true(self):
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	785	client = FakeClient()
	786	client.add_success_response('yes')
2466.2.2 by Andrew Bennetts Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.	787	transport = RemoteTransport('bzr://example.com/', medium=False,
	788	_client=client)
	789	self.assertEqual(True, transport.is_readonly())
	790	self.assertEqual(
	791	[('call', 'Transport.is_readonly', ())],
	792	client._calls)
	793
	794	def test_false(self):
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	795	client = FakeClient()
	796	client.add_success_response('no')
2466.2.2 by Andrew Bennetts Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.	797	transport = RemoteTransport('bzr://example.com/', medium=False,
	798	_client=client)
	799	self.assertEqual(False, transport.is_readonly())
	800	self.assertEqual(
	801	[('call', 'Transport.is_readonly', ())],
	802	client._calls)
	803
	804	def test_error_from_old_server(self):
	805	"""bzr 0.15 and earlier servers don't recognise the is_readonly verb.
	806
	807	Clients should treat it as a "no" response, because is_readonly is only
	808	advisory anyway (a transport could be read-write, but then the
	809	underlying filesystem could be readonly anyway).
	810	"""
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	811	client = FakeClient()
	812	client.add_unknown_method_response('Transport.is_readonly')
2471.2.1 by Andrew Bennetts Fix trivial incompatibility with bzr 0.11 servers, which give a slightly different error to bzr 0.15 servers.	813	transport = RemoteTransport('bzr://example.com/', medium=False,
	814	_client=client)
	815	self.assertEqual(False, transport.is_readonly())
	816	self.assertEqual(
	817	[('call', 'Transport.is_readonly', ())],
	818	client._calls)
	819
2466.2.2 by Andrew Bennetts Add tests for RemoteTransport.is_readonly in the style of the other remote object tests.	820
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	821	class TestRemoteRepository(tests.TestCase):
2018.18.9 by Martin Pool remote Repository.tarball builds a temporary directory and tars that	822	"""Base for testing RemoteRepository protocol usage.
	823
	824	These tests contain frozen requests and responses. We want any changes to
	825	what is sent or expected to be require a thoughtful update to these tests
	826	because they might break compatibility with different-versioned servers.
	827	"""
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	828
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	829	def setup_fake_client_and_repository(self, transport_path):
2018.18.7 by Martin Pool (broken) Start addng client proxy test for Repository.tarball	830	"""Create the fake client and repository for testing with.
	831
	832	There's no real server here; we just have canned responses sent
	833	back one by one.
	834
	835	:param transport_path: Path below the root of the MemoryTransport
	836	where the repository will be created.
	837	"""
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	838	transport = MemoryTransport()
	839	transport.mkdir(transport_path)
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	840	client = FakeClient(transport.base)
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	841	transport = transport.clone(transport_path)
	842	# we do not want bzrdir to make any remote calls
	843	bzrdir = RemoteBzrDir(transport, _client=False)
	844	repo = RemoteRepository(bzrdir, None, _client=client)
	845	return repo, client
	846
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	847
2018.12.2 by Andrew Bennetts Remove some duplicate code in test_remote	848	class TestRepositoryGatherStats(TestRemoteRepository):
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	849
	850	def test_revid_none(self):
	851	# ('ok',), body with revisions and size
	852	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	853	repo, client = self.setup_fake_client_and_repository(transport_path)
	854	client.add_success_response_with_body(
	855	'revisions: 2\nsize: 18\n', 'ok')
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	856	result = repo.gather_stats(None)
	857	self.assertEqual(
2018.5.153 by Andrew Bennetts Rename call2 to call_expecting_body, and other small changes prompted by review.	858	[('call_expecting_body', 'Repository.gather_stats',
3104.4.2 by Andrew Bennetts All tests passing.	859	('quack/','','no'))],
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	860	client._calls)
	861	self.assertEqual({'revisions': 2, 'size': 18}, result)
	862
	863	def test_revid_no_committers(self):
	864	# ('ok',), body without committers
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	865	body = ('firstrev: 123456.300 3600\n'
	866	'latestrev: 654231.400 0\n'
	867	'revisions: 2\n'
	868	'size: 18\n')
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	869	transport_path = 'quick'
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	870	revid = u'\xc8'.encode('utf8')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	871	repo, client = self.setup_fake_client_and_repository(transport_path)
	872	client.add_success_response_with_body(body, 'ok')
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	873	result = repo.gather_stats(revid)
	874	self.assertEqual(
2018.5.153 by Andrew Bennetts Rename call2 to call_expecting_body, and other small changes prompted by review.	875	[('call_expecting_body', 'Repository.gather_stats',
3104.4.2 by Andrew Bennetts All tests passing.	876	('quick/', revid, 'no'))],
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	877	client._calls)
	878	self.assertEqual({'revisions': 2, 'size': 18,
	879	'firstrev': (123456.300, 3600),
	880	'latestrev': (654231.400, 0),},
	881	result)
	882
	883	def test_revid_with_committers(self):
	884	# ('ok',), body with committers
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	885	body = ('committers: 128\n'
	886	'firstrev: 123456.300 3600\n'
	887	'latestrev: 654231.400 0\n'
	888	'revisions: 2\n'
	889	'size: 18\n')
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	890	transport_path = 'buick'
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	891	revid = u'\xc8'.encode('utf8')
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	892	repo, client = self.setup_fake_client_and_repository(transport_path)
	893	client.add_success_response_with_body(body, 'ok')
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	894	result = repo.gather_stats(revid, True)
	895	self.assertEqual(
2018.5.153 by Andrew Bennetts Rename call2 to call_expecting_body, and other small changes prompted by review.	896	[('call_expecting_body', 'Repository.gather_stats',
3104.4.2 by Andrew Bennetts All tests passing.	897	('buick/', revid, 'yes'))],
2018.10.3 by v.ladeuil+lp at free more tests for gather_stats	898	client._calls)
	899	self.assertEqual({'revisions': 2, 'size': 18,
	900	'committers': 128,
	901	'firstrev': (123456.300, 3600),
	902	'latestrev': (654231.400, 0),},
	903	result)
	904
	905
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	906	class TestRepositoryGetGraph(TestRemoteRepository):
	907
	908	def test_get_graph(self):
3172.5.8 by Robert Collins Review feedback.	909	# get_graph returns a graph with the repository as the
3172.5.8 by Robert Collins Review feedback.	910	# parents_provider.
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	911	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	912	repo, client = self.setup_fake_client_and_repository(transport_path)
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	913	graph = repo.get_graph()
3441.5.24 by Andrew Bennetts Remove RemoteGraph experiment.	914	self.assertEqual(graph._parents_provider, repo)
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	915
	916
	917	class TestRepositoryGetParentMap(TestRemoteRepository):
	918
	919	def test_get_parent_map_caching(self):
	920	# get_parent_map returns from cache until unlock()
	921	# setup a reponse with two revisions
	922	r1 = u'\u0e33'.encode('utf8')
	923	r2 = u'\u0dab'.encode('utf8')
	924	lines = [' '.join([r2, r1]), r1]
3211.5.2 by Robert Collins Change RemoteRepository.get_parent_map to use bz2 not gzip for compression.	925	encoded_body = bz2.compress('\n'.join(lines))
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	926
	927	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	928	repo, client = self.setup_fake_client_and_repository(transport_path)
	929	client.add_success_response_with_body(encoded_body, 'ok')
	930	client.add_success_response_with_body(encoded_body, 'ok')
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	931	repo.lock_read()
	932	graph = repo.get_graph()
	933	parents = graph.get_parent_map([r2])
	934	self.assertEqual({r2: (r1,)}, parents)
	935	# locking and unlocking deeper should not reset
	936	repo.lock_read()
	937	repo.unlock()
	938	parents = graph.get_parent_map([r1])
3172.5.6 by Robert Collins Create new smart server verb Repository.get_parent_map.	939	self.assertEqual({r1: (NULL_REVISION,)}, parents)
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	940	self.assertEqual(
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.	941	[('call_with_body_bytes_expecting_body',
	942	'Repository.get_parent_map', ('quack/', r2), '\n\n0')],
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	943	client._calls)
	944	repo.unlock()
	945	# now we call again, and it should use the second response.
	946	repo.lock_read()
	947	graph = repo.get_graph()
3172.5.6 by Robert Collins Create new smart server verb Repository.get_parent_map.	948	parents = graph.get_parent_map([r1])
	949	self.assertEqual({r1: (NULL_REVISION,)}, parents)
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	950	self.assertEqual(
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.	951	[('call_with_body_bytes_expecting_body',
	952	'Repository.get_parent_map', ('quack/', r2), '\n\n0'),
	953	('call_with_body_bytes_expecting_body',
	954	'Repository.get_parent_map', ('quack/', r1), '\n\n0'),
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	955	],
	956	client._calls)
	957	repo.unlock()
	958
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	959	def test_get_parent_map_reconnects_if_unknown_method(self):
	960	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	961	repo, client = self.setup_fake_client_and_repository(transport_path)
	962	client.add_unknown_method_response('Repository,get_parent_map')
	963	client.add_success_response_with_body('', 'ok')
3453.4.10 by Andrew Bennetts Change _is_remote_at_least to _is_remote_before.	964	self.assertFalse(client._medium._is_remote_before((1, 2)))
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	965	rev_id = 'revision-id'
3297.3.5 by Andrew Bennetts Suppress a deprecation warning.	966	expected_deprecations = [
	967	'bzrlib.remote.RemoteRepository.get_revision_graph was deprecated '
	968	'in version 1.4.']
	969	parents = self.callDeprecated(
	970	expected_deprecations, repo.get_parent_map, [rev_id])
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	971	self.assertEqual(
3213.1.8 by Andrew Bennetts Merge from bzr.dev.	972	[('call_with_body_bytes_expecting_body',
3213.1.8 by Andrew Bennetts Merge from bzr.dev.	973	'Repository.get_parent_map', ('quack/', rev_id), '\n\n0'),
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	974	('disconnect medium',),
	975	('call_expecting_body', 'Repository.get_revision_graph',
	976	('quack/', ''))],
	977	client._calls)
3389.1.2 by Andrew Bennetts Add test for the bug John found.	978	# The medium is now marked as being connected to an older server
3453.4.10 by Andrew Bennetts Change _is_remote_at_least to _is_remote_before.	979	self.assertTrue(client._medium._is_remote_before((1, 2)))
3389.1.2 by Andrew Bennetts Add test for the bug John found.	980
	981	def test_get_parent_map_fallback_parentless_node(self):
	982	"""get_parent_map falls back to get_revision_graph on old servers. The
	983	results from get_revision_graph are tweaked to match the get_parent_map
	984	API.
	985
3389.1.3 by Andrew Bennetts Remove XXX from test description.	986	Specifically, a {key: ()} result from get_revision_graph means "no
3389.1.2 by Andrew Bennetts Add test for the bug John found.	987	parents" for that key, which in get_parent_map results should be
3389.1.3 by Andrew Bennetts Remove XXX from test description.	988	represented as {key: ('null:',)}.
3389.1.2 by Andrew Bennetts Add test for the bug John found.	989
	990	This is the test for https://bugs.launchpad.net/bzr/+bug/214894
	991	"""
	992	rev_id = 'revision-id'
	993	transport_path = 'quack'
3245.4.40 by Andrew Bennetts Merge from bzr.dev.	994	repo, client = self.setup_fake_client_and_repository(transport_path)
3245.4.40 by Andrew Bennetts Merge from bzr.dev.	995	client.add_success_response_with_body(rev_id, 'ok')
3453.4.9 by Andrew Bennetts Rename _remote_is_not to _remember_remote_is_before.	996	client._medium._remember_remote_is_before((1, 2))
3389.1.2 by Andrew Bennetts Add test for the bug John found.	997	expected_deprecations = [
	998	'bzrlib.remote.RemoteRepository.get_revision_graph was deprecated '
	999	'in version 1.4.']
	1000	parents = self.callDeprecated(
	1001	expected_deprecations, repo.get_parent_map, [rev_id])
	1002	self.assertEqual(
	1003	[('call_expecting_body', 'Repository.get_revision_graph',
	1004	('quack/', ''))],
	1005	client._calls)
	1006	self.assertEqual({rev_id: ('null:',)}, parents)
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	1007
3297.2.3 by Andrew Bennetts Test the code path that the typo is on.	1008	def test_get_parent_map_unexpected_response(self):
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1009	repo, client = self.setup_fake_client_and_repository('path')
	1010	client.add_success_response('something unexpected!')
3297.2.3 by Andrew Bennetts Test the code path that the typo is on.	1011	self.assertRaises(
	1012	errors.UnexpectedSmartServerResponse,
	1013	repo.get_parent_map, ['a-revision-id'])
3213.1.2 by Andrew Bennetts Add test for reconnection if get_parent_map is unknown by the server.	1014
3172.5.4 by Robert Collins Implement get_parent_map for RemoteRepository with caching, based on get_revision_graph.	1015
2018.5.68 by Wouter van Heyst Merge RemoteRepository.gather_stats.	1016	class TestRepositoryGetRevisionGraph(TestRemoteRepository):
	1017
	1018	def test_null_revision(self):
	1019	# a null revision has the predictable result {}, we should have no wire
	1020	# traffic when calling it with this argument
	1021	transport_path = 'empty'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1022	repo, client = self.setup_fake_client_and_repository(transport_path)
	1023	client.add_success_response('notused')
3287.6.4 by Robert Collins Fix up deprecation warnings for get_revision_graph.	1024	result = self.applyDeprecated(one_four, repo.get_revision_graph,
	1025	NULL_REVISION)
2018.5.68 by Wouter van Heyst Merge RemoteRepository.gather_stats.	1026	self.assertEqual([], client._calls)
	1027	self.assertEqual({}, result)
	1028
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1029	def test_none_revision(self):
	1030	# with none we want the entire graph
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	1031	r1 = u'\u0e33'.encode('utf8')
	1032	r2 = u'\u0dab'.encode('utf8')
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1033	lines = [' '.join([r2, r1]), r1]
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	1034	encoded_body = '\n'.join(lines)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1035
	1036	transport_path = 'sinhala'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1037	repo, client = self.setup_fake_client_and_repository(transport_path)
	1038	client.add_success_response_with_body(encoded_body, 'ok')
3287.6.4 by Robert Collins Fix up deprecation warnings for get_revision_graph.	1039	result = self.applyDeprecated(one_four, repo.get_revision_graph)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1040	self.assertEqual(
2018.5.153 by Andrew Bennetts Rename call2 to call_expecting_body, and other small changes prompted by review.	1041	[('call_expecting_body', 'Repository.get_revision_graph',
3104.4.2 by Andrew Bennetts All tests passing.	1042	('sinhala/', ''))],
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1043	client._calls)
2625.8.1 by Robert Collins LIBRARY API BREAKS:	1044	self.assertEqual({r1: (), r2: (r1, )}, result)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1045
	1046	def test_specific_revision(self):
	1047	# with a specific revision we want the graph for that
	1048	# with none we want the entire graph
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	1049	r11 = u'\u0e33'.encode('utf8')
	1050	r12 = u'\xc9'.encode('utf8')
	1051	r2 = u'\u0dab'.encode('utf8')
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1052	lines = [' '.join([r2, r11, r12]), r11, r12]
2018.5.106 by Andrew Bennetts Update tests in test_remote to use utf-8 byte strings for revision IDs, rather than unicode strings.	1053	encoded_body = '\n'.join(lines)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1054
	1055	transport_path = 'sinhala'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1056	repo, client = self.setup_fake_client_and_repository(transport_path)
	1057	client.add_success_response_with_body(encoded_body, 'ok')
3287.6.4 by Robert Collins Fix up deprecation warnings for get_revision_graph.	1058	result = self.applyDeprecated(one_four, repo.get_revision_graph, r2)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1059	self.assertEqual(
2018.5.153 by Andrew Bennetts Rename call2 to call_expecting_body, and other small changes prompted by review.	1060	[('call_expecting_body', 'Repository.get_revision_graph',
3104.4.2 by Andrew Bennetts All tests passing.	1061	('sinhala/', r2))],
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1062	client._calls)
2625.8.1 by Robert Collins LIBRARY API BREAKS:	1063	self.assertEqual({r11: (), r12: (), r2: (r11, r12), }, result)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1064
	1065	def test_no_such_revision(self):
	1066	revid = '123'
	1067	transport_path = 'sinhala'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1068	repo, client = self.setup_fake_client_and_repository(transport_path)
	1069	client.add_error_response('nosuchrevision', revid)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1070	# also check that the right revision is reported in the error
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1071	self.assertRaises(errors.NoSuchRevision,
3287.6.4 by Robert Collins Fix up deprecation warnings for get_revision_graph.	1072	self.applyDeprecated, one_four, repo.get_revision_graph, revid)
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1073	self.assertEqual(
2018.5.153 by Andrew Bennetts Rename call2 to call_expecting_body, and other small changes prompted by review.	1074	[('call_expecting_body', 'Repository.get_revision_graph',
3104.4.2 by Andrew Bennetts All tests passing.	1075	('sinhala/', revid))],
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1076	client._calls)
	1077
3245.4.53 by Andrew Bennetts Add some missing 'raise' statements to test_remote.	1078	def test_unexpected_error(self):
	1079	revid = '123'
	1080	transport_path = 'sinhala'
	1081	repo, client = self.setup_fake_client_and_repository(transport_path)
	1082	client.add_error_response('AnUnexpectedError')
	1083	e = self.assertRaises(errors.ErrorFromSmartServer,
	1084	self.applyDeprecated, one_four, repo.get_revision_graph, revid)
	1085	self.assertEqual(('AnUnexpectedError',), e.error_tuple)
	1086
2018.5.67 by Wouter van Heyst Implement RemoteRepository.get_revision_graph (Wouter van Heyst, Robert Collins)	1087
	1088	class TestRepositoryIsShared(TestRemoteRepository):
	1089
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1090	def test_is_shared(self):
	1091	# ('yes', ) for Repository.is_shared -> 'True'.
	1092	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1093	repo, client = self.setup_fake_client_and_repository(transport_path)
	1094	client.add_success_response('yes')
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1095	result = repo.is_shared()
	1096	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	1097	[('call', 'Repository.is_shared', ('quack/',))],
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1098	client._calls)
	1099	self.assertEqual(True, result)
	1100
	1101	def test_is_not_shared(self):
	1102	# ('no', ) for Repository.is_shared -> 'False'.
	1103	transport_path = 'qwack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1104	repo, client = self.setup_fake_client_and_repository(transport_path)
	1105	client.add_success_response('no')
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1106	result = repo.is_shared()
	1107	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	1108	[('call', 'Repository.is_shared', ('qwack/',))],
2018.5.57 by Robert Collins Implement RemoteRepository.is_shared (Robert Collins, Vincent Ladeuil).	1109	client._calls)
	1110	self.assertEqual(False, result)
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1111
	1112
	1113	class TestRepositoryLockWrite(TestRemoteRepository):
	1114
	1115	def test_lock_write(self):
	1116	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1117	repo, client = self.setup_fake_client_and_repository(transport_path)
	1118	client.add_success_response('ok', 'a token')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1119	result = repo.lock_write()
	1120	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	1121	[('call', 'Repository.lock_write', ('quack/', ''))],
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1122	client._calls)
	1123	self.assertEqual('a token', result)
	1124
	1125	def test_lock_write_already_locked(self):
	1126	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1127	repo, client = self.setup_fake_client_and_repository(transport_path)
	1128	client.add_error_response('LockContention')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1129	self.assertRaises(errors.LockContention, repo.lock_write)
	1130	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	1131	[('call', 'Repository.lock_write', ('quack/', ''))],
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.	1132	client._calls)
	1133
	1134	def test_lock_write_unlockable(self):
	1135	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1136	repo, client = self.setup_fake_client_and_repository(transport_path)
	1137	client.add_error_response('UnlockableTransport')
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.	1138	self.assertRaises(errors.UnlockableTransport, repo.lock_write)
	1139	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	1140	[('call', 'Repository.lock_write', ('quack/', ''))],
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1141	client._calls)
	1142
	1143
	1144	class TestRepositoryUnlock(TestRemoteRepository):
	1145
	1146	def test_unlock(self):
	1147	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1148	repo, client = self.setup_fake_client_and_repository(transport_path)
	1149	client.add_success_response('ok', 'a token')
	1150	client.add_success_response('ok')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1151	repo.lock_write()
	1152	repo.unlock()
	1153	self.assertEqual(
3104.4.2 by Andrew Bennetts All tests passing.	1154	[('call', 'Repository.lock_write', ('quack/', '')),
3104.4.2 by Andrew Bennetts All tests passing.	1155	('call', 'Repository.unlock', ('quack/', 'a token'))],
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1156	client._calls)
	1157
	1158	def test_unlock_wrong_token(self):
	1159	# If somehow the token is wrong, unlock will raise TokenMismatch.
	1160	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1161	repo, client = self.setup_fake_client_and_repository(transport_path)
	1162	client.add_success_response('ok', 'a token')
	1163	client.add_error_response('TokenMismatch')
2018.5.78 by Andrew Bennetts Implement RemoteRepository.lock_write/unlock to expect and send tokens over the	1164	repo.lock_write()
	1165	self.assertRaises(errors.TokenMismatch, repo.unlock)
	1166
	1167
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	1168	class TestRepositoryHasRevision(TestRemoteRepository):
	1169
	1170	def test_none(self):
	1171	# repo.has_revision(None) should not cause any traffic.
	1172	transport_path = 'quack'
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1173	repo, client = self.setup_fake_client_and_repository(transport_path)
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	1174
	1175	# The null revision is always there, so has_revision(None) == True.
3172.3.3 by Robert Collins Missed one occurence of None -> NULL_REVISION.	1176	self.assertEqual(True, repo.has_revision(NULL_REVISION))
2018.14.1 by Andrew Bennetts Update to current hpss branch? Fix lots of test failures.	1177
	1178	# The remote repo shouldn't be accessed.
	1179	self.assertEqual([], client._calls)
2018.18.7 by Martin Pool (broken) Start addng client proxy test for Repository.tarball	1180
	1181
	1182	class TestRepositoryTarball(TestRemoteRepository):
	1183
2018.18.9 by Martin Pool remote Repository.tarball builds a temporary directory and tars that	1184	# This is a canned tarball reponse we can validate against
2018.18.18 by Martin Pool reformat	1185	tarball_content = (
2018.18.23 by Martin Pool review cleanups	1186	'QlpoOTFBWSZTWdGkj3wAAWF/k8aQACBIB//A9+8cIX/v33AACEAYABAECEACNz'
	1187	'JqsgJJFPTSnk1A3qh6mTQAAAANPUHkagkSTEkaA09QaNAAAGgAAAcwCYCZGAEY'
	1188	'mJhMJghpiaYBUkKammSHqNMZQ0NABkNAeo0AGneAevnlwQoGzEzNVzaYxp/1Uk'
	1189	'xXzA1CQX0BJMZZLcPBrluJir5SQyijWHYZ6ZUtVqqlYDdB2QoCwa9GyWwGYDMA'
	1190	'OQYhkpLt/OKFnnlT8E0PmO8+ZNSo2WWqeCzGB5fBXZ3IvV7uNJVE7DYnWj6qwB'
	1191	'k5DJDIrQ5OQHHIjkS9KqwG3mc3t+F1+iujb89ufyBNIKCgeZBWrl5cXxbMGoMs'
	1192	'c9JuUkg5YsiVcaZJurc6KLi6yKOkgCUOlIlOpOoXyrTJjK8ZgbklReDdwGmFgt'
	1193	'dkVsAIslSVCd4AtACSLbyhLHryfb14PKegrVDba+U8OL6KQtzdM5HLjAc8/p6n'
	1194	'0lgaWU8skgO7xupPTkyuwheSckejFLK5T4ZOo0Gda9viaIhpD1Qn7JqqlKAJqC'
	1195	'QplPKp2nqBWAfwBGaOwVrz3y1T+UZZNismXHsb2Jq18T+VaD9k4P8DqE3g70qV'
	1196	'JLurpnDI6VS5oqDDPVbtVjMxMxMg4rzQVipn2Bv1fVNK0iq3Gl0hhnnHKm/egy'
	1197	'nWQ7QH/F3JFOFCQ0aSPfA='
	1198	).decode('base64')
2018.18.9 by Martin Pool remote Repository.tarball builds a temporary directory and tars that	1199
2018.18.7 by Martin Pool (broken) Start addng client proxy test for Repository.tarball	1200	def test_repository_tarball(self):
2018.18.9 by Martin Pool remote Repository.tarball builds a temporary directory and tars that	1201	# Test that Repository.tarball generates the right operations
2018.18.7 by Martin Pool (broken) Start addng client proxy test for Repository.tarball	1202	transport_path = 'repo'
2018.18.14 by Martin Pool merge hpss again; restore incorrectly removed RemoteRepository.break_lock	1203	expected_calls = [('call_expecting_body', 'Repository.tarball',
3104.4.2 by Andrew Bennetts All tests passing.	1204	('repo/', 'bz2',),),
2018.18.7 by Martin Pool (broken) Start addng client proxy test for Repository.tarball	1205	]
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1206	repo, client = self.setup_fake_client_and_repository(transport_path)
	1207	client.add_success_response_with_body(self.tarball_content, 'ok')
2018.18.7 by Martin Pool (broken) Start addng client proxy test for Repository.tarball	1208	# Now actually ask for the tarball
3245.4.24 by Andrew Bennetts Consistently raise errors from the server as ErrorFromSmartServer exceptions.	1209	tarball_file = repo._get_tarball('bz2')
2018.18.25 by Martin Pool Repository.tarball fixes for python2.4	1210	try:
	1211	self.assertEqual(expected_calls, client._calls)
	1212	self.assertEqual(self.tarball_content, tarball_file.read())
	1213	finally:
	1214	tarball_file.close()
2018.18.9 by Martin Pool remote Repository.tarball builds a temporary directory and tars that	1215
	1216
	1217	class TestRemoteRepositoryCopyContent(tests.TestCaseWithTransport):
	1218	"""RemoteRepository.copy_content_into optimizations"""
	1219
2018.18.10 by Martin Pool copy_content_into from Remote repositories by using temporary directories on both ends.	1220	def test_copy_content_remote_to_local(self):
	1221	self.transport_server = server.SmartTCPServer_for_testing
	1222	src_repo = self.make_repository('repo1')
	1223	src_repo = repository.Repository.open(self.get_url('repo1'))
	1224	# At the moment the tarball-based copy_content_into can't write back
	1225	# into a smart server. It would be good if it could upload the
	1226	# tarball; once that works we'd have to create repositories of
	1227	# different formats. -- mbp 20070410
	1228	dest_url = self.get_vfs_only_url('repo2')
	1229	dest_bzrdir = BzrDir.create(dest_url)
	1230	dest_repo = dest_bzrdir.create_repository()
	1231	self.assertFalse(isinstance(dest_repo, RemoteRepository))
	1232	self.assertTrue(isinstance(src_repo, RemoteRepository))
	1233	src_repo.copy_content_into(dest_repo)
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1234
	1235
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1236	class TestErrorTranslationBase(tests.TestCaseWithMemoryTransport):
	1237	"""Base class for unit tests for bzrlib.remote._translate_error."""
	1238
	1239	def translateTuple(self, error_tuple, **context):
	1240	"""Call _translate_error with an ErrorFromSmartServer built from the
	1241	given error_tuple.
	1242
	1243	:param error_tuple: A tuple of a smart server response, as would be
	1244	passed to an ErrorFromSmartServer.
	1245	:kwargs context: context items to call _translate_error with.
	1246
	1247	:returns: The error raised by _translate_error.
	1248	"""
	1249	# Raise the ErrorFromSmartServer before passing it as an argument,
	1250	# because _translate_error may need to re-raise it with a bare 'raise'
	1251	# statement.
	1252	server_error = errors.ErrorFromSmartServer(error_tuple)
	1253	translated_error = self.translateErrorFromSmartServer(
	1254	server_error, **context)
	1255	return translated_error
	1256
	1257	def translateErrorFromSmartServer(self, error_object, **context):
	1258	"""Like translateTuple, but takes an already constructed
	1259	ErrorFromSmartServer rather than a tuple.
	1260	"""
	1261	try:
	1262	raise error_object
	1263	except errors.ErrorFromSmartServer, server_error:
	1264	translated_error = self.assertRaises(
	1265	errors.BzrError, remote._translate_error, server_error,
	1266	**context)
	1267	return translated_error
	1268
	1269
	1270	class TestErrorTranslationSuccess(TestErrorTranslationBase):
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1271	"""Unit tests for bzrlib.remote._translate_error.
	1272
	1273	Given an ErrorFromSmartServer (which has an error tuple from a smart
	1274	server) and some context, _translate_error raises more specific errors from
	1275	bzrlib.errors.
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1276
	1277	This test case covers the cases where _translate_error succeeds in
	1278	translating an ErrorFromSmartServer to something better. See
	1279	TestErrorTranslationRobustness for other cases.
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1280	"""
	1281
	1282	def test_NoSuchRevision(self):
	1283	branch = self.make_branch('')
	1284	revid = 'revid'
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1285	translated_error = self.translateTuple(
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1286	('NoSuchRevision', revid), branch=branch)
	1287	expected_error = errors.NoSuchRevision(branch, revid)
	1288	self.assertEqual(expected_error, translated_error)
	1289
	1290	def test_nosuchrevision(self):
	1291	repository = self.make_repository('')
	1292	revid = 'revid'
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1293	translated_error = self.translateTuple(
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1294	('nosuchrevision', revid), repository=repository)
	1295	expected_error = errors.NoSuchRevision(repository, revid)
	1296	self.assertEqual(expected_error, translated_error)
	1297
	1298	def test_nobranch(self):
	1299	bzrdir = self.make_bzrdir('')
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1300	translated_error = self.translateTuple(('nobranch',), bzrdir=bzrdir)
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1301	expected_error = errors.NotBranchError(path=bzrdir.root_transport.base)
	1302	self.assertEqual(expected_error, translated_error)
	1303
	1304	def test_LockContention(self):
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1305	translated_error = self.translateTuple(('LockContention',))
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1306	expected_error = errors.LockContention('(remote lock)')
	1307	self.assertEqual(expected_error, translated_error)
	1308
	1309	def test_UnlockableTransport(self):
	1310	bzrdir = self.make_bzrdir('')
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1311	translated_error = self.translateTuple(
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1312	('UnlockableTransport',), bzrdir=bzrdir)
	1313	expected_error = errors.UnlockableTransport(bzrdir.root_transport)
	1314	self.assertEqual(expected_error, translated_error)
	1315
	1316	def test_LockFailed(self):
	1317	lock = 'str() of a server lock'
	1318	why = 'str() of why'
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1319	translated_error = self.translateTuple(('LockFailed', lock, why))
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1320	expected_error = errors.LockFailed(lock, why)
	1321	self.assertEqual(expected_error, translated_error)
	1322
	1323	def test_TokenMismatch(self):
	1324	token = 'a lock token'
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1325	translated_error = self.translateTuple(('TokenMismatch',), token=token)
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1326	expected_error = errors.TokenMismatch(token, '(remote token)')
	1327	self.assertEqual(expected_error, translated_error)
	1328
	1329	def test_Diverged(self):
	1330	branch = self.make_branch('a')
	1331	other_branch = self.make_branch('b')
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1332	translated_error = self.translateTuple(
3533.3.3 by Andrew Bennetts Add unit tests for bzrlib.remote._translate_error.	1333	('Diverged',), branch=branch, other_branch=other_branch)
	1334	expected_error = errors.DivergedBranches(branch, other_branch)
	1335	self.assertEqual(expected_error, translated_error)
	1336
	1337
3533.3.4 by Andrew Bennetts Add tests for _translate_error's robustness.	1338	class TestErrorTranslationRobustness(TestErrorTranslationBase):
	1339	"""Unit tests for bzrlib.remote._translate_error's robustness.
	1340
	1341	TestErrorTranslationSuccess is for cases where _translate_error can
	1342	translate successfully. This class about how _translate_err behaves when
	1343	it fails to translate: it re-raises the original error.
	1344	"""
	1345
	1346	def test_unrecognised_server_error(self):
	1347	"""If the error code from the server is not recognised, the original
	1348	ErrorFromSmartServer is propagated unmodified.
	1349	"""
	1350	error_tuple = ('An unknown error tuple',)
	1351	server_error = errors.ErrorFromSmartServer(error_tuple)
	1352	translated_error = self.translateErrorFromSmartServer(server_error)
	1353	self.assertEqual(server_error, translated_error)
	1354
	1355	def test_context_missing_a_key(self):
	1356	"""In case of a bug in the client, or perhaps an unexpected response
	1357	from a server, _translate_error returns the original error tuple from
	1358	the server and mutters a warning.
	1359	"""
	1360	# To translate a NoSuchRevision error _translate_error needs a 'branch'
	1361	# in the context dict. So let's give it an empty context dict instead
	1362	# to exercise its error recovery.
	1363	empty_context = {}
	1364	error_tuple = ('NoSuchRevision', 'revid')
	1365	server_error = errors.ErrorFromSmartServer(error_tuple)
	1366	translated_error = self.translateErrorFromSmartServer(server_error)
	1367	self.assertEqual(server_error, translated_error)
	1368	# In addition to re-raising ErrorFromSmartServer, some debug info has
	1369	# been muttered to the log file for developer to look at.
	1370	self.assertContainsRe(
	1371	self._get_log(keep_log_file=True),
	1372	"Missing key 'branch' in context")
	1373