1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
# Copyright (C) 2006 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""Test read_bundle works properly across various transports."""
import cStringIO
import os
import bzrlib.bundle
from bzrlib.bundle.serializer import write_bundle
import bzrlib.bzrdir
import bzrlib.errors as errors
from bzrlib.symbol_versioning import deprecated_in
from bzrlib import tests
from bzrlib.tests.test_transport import TestTransportImplementation
from bzrlib.tests.test_transport_implementations import transport_test_permutations
import bzrlib.transport
from bzrlib.transport.memory import MemoryTransport
import bzrlib.urlutils
def load_tests(standard_tests, module, loader):
"""Multiply tests for tranport implementations."""
transport_tests, remaining_tests = tests.split_suite_by_condition(
standard_tests, tests.condition_isinstance(TestReadBundleFromURL))
return tests.multiply_tests(transport_tests, transport_test_permutations(),
remaining_tests)
def create_bundle_file(test_case):
test_case.build_tree(['tree/', 'tree/a', 'tree/subdir/'])
format = bzrlib.bzrdir.BzrDirFormat.get_default_format()
bzrdir = format.initialize('tree')
repo = bzrdir.create_repository()
branch = repo.bzrdir.create_branch()
wt = branch.bzrdir.create_workingtree()
wt.add(['a', 'subdir/'])
wt.commit('new project', rev_id='commit-1')
out = cStringIO.StringIO()
rev_ids = write_bundle(wt.branch.repository,
wt.get_parent_ids()[0], 'null:', out)
out.seek(0)
return out, wt
class TestDeprecations(tests.TestCaseInTempDir):
def create_test_bundle(self):
out, wt = create_bundle_file(self)
f = open('test_bundle', 'wb')
try:
f.write(out.getvalue())
finally:
f.close()
return wt
def test_read_bundle_from_url_deprecated(self):
wt = self.create_test_bundle()
t = bzrlib.transport.get_transport(self.test_dir)
url = t.abspath('test_bundle')
self.callDeprecated([deprecated_in((1, 12, 0))
% 'bzrlib.bundle.read_bundle_from_url'],
bzrlib.bundle.read_bundle_from_url,
url)
class TestReadBundleFromURL(TestTransportImplementation):
"""Test that read_bundle works properly across multiple transports"""
def get_url(self, relpath=''):
return bzrlib.urlutils.join(self._server.get_url(), relpath)
def create_test_bundle(self):
out, wt = create_bundle_file(self)
if self.get_transport().is_readonly():
f = open('test_bundle', 'wb')
try:
f.write(out.getvalue())
finally:
f.close()
else:
self.get_transport().put_file('test_bundle', out)
self.log('Put to: %s', self.get_url('test_bundle'))
return wt
def test_read_mergeable_from_url(self):
self._captureVar('BZR_NO_SMART_VFS', None)
wt = self.create_test_bundle()
if wt is None:
return
# read_mergeable_from_url will invoke get_transport which may *not*
# respect self._transport (i.e. returns a transport that is different
# from the one we want to test, so we must inject a correct transport
# into possible_transports first.
t = self.get_transport('test_bundle')
possible_transports = [t]
info = bzrlib.bundle.read_mergeable_from_url(
unicode(self.get_url('test_bundle')),
possible_transports=possible_transports)
revision = info.real_revisions[-1]
self.assertEqual('commit-1', revision.revision_id)
def test_read_fail(self):
# Trying to read from a directory, or non-bundle file
# should fail with NotABundle
self._captureVar('BZR_NO_SMART_VFS', None)
wt = self.create_test_bundle()
if wt is None:
return
self.assertRaises(errors.NotABundle,
bzrlib.bundle.read_mergeable_from_url,
self.get_url('tree'))
self.assertRaises(errors.NotABundle,
bzrlib.bundle.read_mergeable_from_url,
self.get_url('tree/a'))
def test_read_mergeable_respects_possible_transports(self):
t = self.get_transport('test_bundle')
if not isinstance(t, bzrlib.transport.ConnectedTransport):
# There is no point testing transport reuse for not connected
# transports (the test will fail even).
return
self._captureVar('BZR_NO_SMART_VFS', None)
wt = self.create_test_bundle()
if wt is None:
return
# read_mergeable_from_url will invoke get_transport which may *not*
# respect self._transport (i.e. returns a transport that is different
# from the one we want to test, so we must inject a correct transport
# into possible_transports first.
possible_transports = [t]
url = unicode(self.get_url('test_bundle'))
info = bzrlib.bundle.read_mergeable_from_url(url,
possible_transports=possible_transports)
self.assertEqual(1, len(possible_transports))
|