/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_read_bundle.py

merge bzr.dev r4054

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_read_bundle.py
23
23
from bzrlib.bundle.serializer import write_bundle
24
24
import bzrlib.bzrdir
25
25
import bzrlib.errors as errors
 
26
from bzrlib.symbol_versioning import deprecated_in
26
27
from bzrlib import tests
27
28
from bzrlib.tests.test_transport import TestTransportImplementation
28
29
from bzrlib.tests.test_transport_implementations import TransportTestProviderAdapter
34
35
def load_tests(standard_tests, module, loader):
35
36
    """Multiply tests for tranport implementations."""
36
37
    result = loader.suiteClass()
 
38
    transport_tests, remaining_tests = tests.split_suite_by_condition(
 
39
        standard_tests, tests.condition_isinstance((TestReadBundleFromURL)))
 
40
 
37
41
    adapter = TransportTestProviderAdapter()
38
 
    for test in tests.iter_suite_tests(standard_tests):
39
 
        result.addTests(adapter.adapt(test))
 
42
    tests.adapt_tests(transport_tests, adapter, result)
 
43
 
 
44
    # No parametrization for the remaining tests
 
45
    result.addTests(remaining_tests)
 
46
 
40
47
    return result
41
48
 
42
49
 
60
67
    return out, wt
61
68
 
62
69
 
 
70
class TestDeprecations(tests.TestCaseInTempDir):
 
71
 
 
72
    def create_test_bundle(self):
 
73
        out, wt = create_bundle_file(self)
 
74
        f = open('test_bundle', 'wb')
 
75
        try:
 
76
            f.write(out.getvalue())
 
77
        finally:
 
78
            f.close()
 
79
        return wt
 
80
 
 
81
    def test_read_bundle_from_url_deprecated(self):
 
82
        wt = self.create_test_bundle()
 
83
        t = bzrlib.transport.get_transport(self.test_dir)
 
84
        url = t.abspath('test_bundle')
 
85
        self.callDeprecated([deprecated_in((1, 12, 0))
 
86
                             % 'bzrlib.bundle.read_bundle_from_url'],
 
87
                            bzrlib.bundle.read_bundle_from_url,
 
88
                            url)
 
89
 
 
90
 
63
91
class TestReadBundleFromURL(TestTransportImplementation):
64
92
    """Test that read_bundle works properly across multiple transports"""
65
93
 
70
98
        out, wt = create_bundle_file(self)
71
99
        if self.get_transport().is_readonly():
72
100
            f = open('test_bundle', 'wb')
73
 
            f.write(out.getvalue())
74
 
            f.close()
 
101
            try:
 
102
                f.write(out.getvalue())
 
103
            finally:
 
104
                f.close()
75
105
        else:
76
106
            self.get_transport().put_file('test_bundle', out)
77
107
            self.log('Put to: %s', self.get_url('test_bundle'))
78
108
        return wt
79
109
 
80
 
    def test_read_bundle_from_url(self):
 
110
    def test_read_mergeable_from_url(self):
81
111
        self._captureVar('BZR_NO_SMART_VFS', None)
82
112
        wt = self.create_test_bundle()
83
113
        if wt is None:
84
114
            return
85
 
        info = bzrlib.bundle.read_bundle_from_url(
86
 
                    unicode(self.get_url('test_bundle')))
 
115
        # read_mergeable_from_url will invoke get_transport which may *not*
 
116
        # respect self._transport (i.e. returns a transport that is different
 
117
        # from the one we want to test, so we must inject a correct transport
 
118
        # into possible_transports first.
 
119
        t = self.get_transport('test_bundle')
 
120
        possible_transports = [t]
 
121
        info = bzrlib.bundle.read_mergeable_from_url(
 
122
                    unicode(self.get_url('test_bundle')),
 
123
                    possible_transports=possible_transports)
87
124
        revision = info.real_revisions[-1]
88
125
        self.assertEqual('commit-1', revision.revision_id)
89
126
 
95
132
        if wt is None:
96
133
            return
97
134
 
98
 
        self.assertRaises(errors.NotABundle, 
99
 
            bzrlib.bundle.read_bundle_from_url, 
 
135
        self.assertRaises(errors.NotABundle,
 
136
            bzrlib.bundle.read_mergeable_from_url,
100
137
            self.get_url('tree'))
101
 
        self.assertRaises(errors.NotABundle, 
102
 
            bzrlib.bundle.read_bundle_from_url, 
 
138
        self.assertRaises(errors.NotABundle,
 
139
            bzrlib.bundle.read_mergeable_from_url,
103
140
            self.get_url('tree/a'))
104
141
 
105
 
    def test_read_mergeable_populates_possible_transports(self):
 
142
    def test_read_mergeable_respects_possible_transports(self):
 
143
        t = self.get_transport('test_bundle')
 
144
        if not isinstance(t, bzrlib.transport.ConnectedTransport):
 
145
            # There is no point testing transport reuse for not connected
 
146
            # transports (the test will fail even).
 
147
            return
106
148
        self._captureVar('BZR_NO_SMART_VFS', None)
107
149
        wt = self.create_test_bundle()
108
150
        if wt is None:
109
151
            return
110
 
        possible_transports = []
 
152
        # read_mergeable_from_url will invoke get_transport which may *not*
 
153
        # respect self._transport (i.e. returns a transport that is different
 
154
        # from the one we want to test, so we must inject a correct transport
 
155
        # into possible_transports first.
 
156
        possible_transports = [t]
111
157
        url = unicode(self.get_url('test_bundle'))
112
158
        info = bzrlib.bundle.read_mergeable_from_url(url,
113
159
            possible_transports=possible_transports)