~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_selftest.py

  • Committer: Robert Collins
  • Date: 2007-07-19 06:34:09 UTC
  • mto: (2592.3.46 repository)
  • mto: This revision was merged to the branch mainline in revision 2651.
  • Revision ID: robertc@robertcollins.net-20070719063409-stu9sckrxp8wp3mo
LIBRARY API BREAKS:

  * KnitIndex.get_parents now returns tuples. (Robert Collins)

INTERNALS:

  * Unused functions on the private interface KnitIndex have been removed.
    (Robert Collins)

  * New ``knit.KnitGraphIndex`` which provides a ``KnitIndex`` layered on top
    of a ``index.GraphIndex``. (Robert Collins)

  * New ``knit.KnitVersionedFile.iter_parents`` method that allows querying
    the parents of many knit nodes at once, reducing round trips to the 
    underlying index. (Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2008, 2009 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Tests for the test framework."""
18
18
 
26
26
 
27
27
import bzrlib
28
28
from bzrlib import (
29
 
    branchbuilder,
30
29
    bzrdir,
31
30
    errors,
32
31
    memorytree,
33
32
    osutils,
34
 
    remote,
35
33
    repository,
36
34
    symbol_versioning,
37
 
    tests,
38
 
    workingtree,
39
35
    )
40
36
from bzrlib.progress import _BaseProgressBar
41
 
from bzrlib.repofmt import (
42
 
    pack_repo,
43
 
    weaverepo,
44
 
    )
45
 
from bzrlib.symbol_versioning import (
46
 
    deprecated_function,
47
 
    deprecated_in,
48
 
    deprecated_method,
49
 
    )
 
37
from bzrlib.repofmt import weaverepo
 
38
from bzrlib.symbol_versioning import zero_ten, zero_eleven
50
39
from bzrlib.tests import (
51
40
                          ChrootedTestCase,
52
41
                          ExtendedTestResult,
56
45
                          TestCaseInTempDir,
57
46
                          TestCaseWithMemoryTransport,
58
47
                          TestCaseWithTransport,
59
 
                          TestNotApplicable,
60
48
                          TestSkipped,
61
49
                          TestSuite,
62
50
                          TestUtil,
63
51
                          TextTestRunner,
64
52
                          UnavailableFeature,
65
 
                          condition_id_re,
66
 
                          condition_isinstance,
67
 
                          exclude_tests_by_condition,
68
 
                          exclude_tests_by_re,
69
 
                          filter_suite_by_condition,
 
53
                          iter_suite_tests,
70
54
                          filter_suite_by_re,
71
 
                          iter_suite_tests,
72
 
                          preserve_input,
73
 
                          randomize_suite,
74
 
                          run_suite,
75
 
                          split_suite_by_condition,
76
 
                          split_suite_by_re,
 
55
                          sort_suite_by_re,
77
56
                          test_lsprof,
78
57
                          test_suite,
79
58
                          )
84
63
from bzrlib.version import _get_bzr_source_tree
85
64
 
86
65
 
87
 
def _test_ids(test_suite):
88
 
    """Get the ids for the tests in a test suite."""
89
 
    return [t.id() for t in iter_suite_tests(test_suite)]
90
 
 
91
 
 
92
66
class SelftestTests(TestCase):
93
67
 
94
68
    def test_import_tests(self):
110
84
                              'a test message\n')
111
85
 
112
86
 
113
 
class TestUnicodeFilename(TestCase):
114
 
 
115
 
    def test_probe_passes(self):
116
 
        """UnicodeFilename._probe passes."""
117
 
        # We can't test much more than that because the behaviour depends
118
 
        # on the platform.
119
 
        tests.UnicodeFilename._probe()
120
 
 
121
 
 
122
87
class TestTreeShape(TestCaseInTempDir):
123
88
 
124
89
    def test_unicode_paths(self):
125
 
        self.requireFeature(tests.UnicodeFilename)
126
 
 
127
90
        filename = u'hell\u00d8'
128
 
        self.build_tree_contents([(filename, 'contents of hello')])
 
91
        try:
 
92
            self.build_tree_contents([(filename, 'contents of hello')])
 
93
        except UnicodeEncodeError:
 
94
            raise TestSkipped("can't build unicode working tree in "
 
95
                "filesystem encoding %s" % sys.getfilesystemencoding())
129
96
        self.failUnlessExists(filename)
130
97
 
131
98
 
132
 
class TestTransportScenarios(TestCase):
 
99
class TestTransportProviderAdapter(TestCase):
133
100
    """A group of tests that test the transport implementation adaption core.
134
101
 
135
 
    This is a meta test that the tests are applied to all available
 
102
    This is a meta test that the tests are applied to all available 
136
103
    transports.
137
104
 
138
 
    This will be generalised in the future which is why it is in this
 
105
    This will be generalised in the future which is why it is in this 
139
106
    test file even though it is specific to transport tests at the moment.
140
107
    """
141
108
 
142
109
    def test_get_transport_permutations(self):
143
 
        # this checks that get_test_permutations defined by the module is
144
 
        # called by the get_transport_test_permutations function.
 
110
        # this checks that we the module get_test_permutations call
 
111
        # is made by the adapter get_transport_test_permitations method.
145
112
        class MockModule(object):
146
113
            def get_test_permutations(self):
147
114
                return sample_permutation
148
115
        sample_permutation = [(1,2), (3,4)]
149
116
        from bzrlib.tests.test_transport_implementations \
150
 
            import get_transport_test_permutations
 
117
            import TransportTestProviderAdapter
 
118
        adapter = TransportTestProviderAdapter()
151
119
        self.assertEqual(sample_permutation,
152
 
                         get_transport_test_permutations(MockModule()))
 
120
                         adapter.get_transport_test_permutations(MockModule()))
153
121
 
154
 
    def test_scenarios_invlude_all_modules(self):
155
 
        # this checks that the scenario generator returns as many permutations
156
 
        # as there are in all the registered transport modules - we assume if
157
 
        # this matches its probably doing the right thing especially in
158
 
        # combination with the tests for setting the right classes below.
 
122
    def test_adapter_checks_all_modules(self):
 
123
        # this checks that the adapter returns as many permurtations as
 
124
        # there are in all the registered# transport modules for there
 
125
        # - we assume if this matches its probably doing the right thing
 
126
        # especially in combination with the tests for setting the right
 
127
        # classes below.
159
128
        from bzrlib.tests.test_transport_implementations \
160
 
            import transport_test_permutations
 
129
            import TransportTestProviderAdapter
161
130
        from bzrlib.transport import _get_transport_modules
162
131
        modules = _get_transport_modules()
163
132
        permutation_count = 0
164
133
        for module in modules:
165
134
            try:
166
 
                permutation_count += len(reduce(getattr,
 
135
                permutation_count += len(reduce(getattr, 
167
136
                    (module + ".get_test_permutations").split('.')[1:],
168
137
                     __import__(module))())
169
138
            except errors.DependencyNotPresent:
170
139
                pass
171
 
        scenarios = transport_test_permutations()
172
 
        self.assertEqual(permutation_count, len(scenarios))
 
140
        input_test = TestTransportProviderAdapter(
 
141
            "test_adapter_sets_transport_class")
 
142
        adapter = TransportTestProviderAdapter()
 
143
        self.assertEqual(permutation_count,
 
144
                         len(list(iter(adapter.adapt(input_test)))))
173
145
 
174
 
    def test_scenarios_include_transport_class(self):
 
146
    def test_adapter_sets_transport_class(self):
 
147
        # Check that the test adapter inserts a transport and server into the
 
148
        # generated test.
 
149
        #
175
150
        # This test used to know about all the possible transports and the
176
151
        # order they were returned but that seems overly brittle (mbp
177
152
        # 20060307)
178
153
        from bzrlib.tests.test_transport_implementations \
179
 
            import transport_test_permutations
180
 
        scenarios = transport_test_permutations()
 
154
            import TransportTestProviderAdapter
 
155
        scenarios = TransportTestProviderAdapter().scenarios
181
156
        # there are at least that many builtin transports
182
157
        self.assertTrue(len(scenarios) > 6)
183
158
        one_scenario = scenarios[0]
188
163
                                   bzrlib.transport.Server))
189
164
 
190
165
 
191
 
class TestBranchScenarios(TestCase):
 
166
class TestBranchProviderAdapter(TestCase):
 
167
    """A group of tests that test the branch implementation test adapter."""
192
168
 
193
 
    def test_scenarios(self):
 
169
    def test_constructor(self):
194
170
        # check that constructor parameters are passed through to the adapted
195
171
        # test.
196
 
        from bzrlib.tests.branch_implementations import make_scenarios
 
172
        from bzrlib.tests.branch_implementations import BranchTestProviderAdapter
197
173
        server1 = "a"
198
174
        server2 = "b"
199
175
        formats = [("c", "C"), ("d", "D")]
200
 
        scenarios = make_scenarios(server1, server2, formats)
201
 
        self.assertEqual(2, len(scenarios))
 
176
        adapter = BranchTestProviderAdapter(server1, server2, formats)
 
177
        self.assertEqual(2, len(adapter.scenarios))
202
178
        self.assertEqual([
203
179
            ('str',
204
180
             {'branch_format': 'c',
210
186
              'bzrdir_format': 'D',
211
187
              'transport_readonly_server': 'b',
212
188
              'transport_server': 'a'})],
213
 
            scenarios)
214
 
 
215
 
 
216
 
class TestBzrDirScenarios(TestCase):
217
 
 
218
 
    def test_scenarios(self):
 
189
            adapter.scenarios)
 
190
 
 
191
 
 
192
class TestBzrDirProviderAdapter(TestCase):
 
193
    """A group of tests that test the bzr dir implementation test adapter."""
 
194
 
 
195
    def test_adapted_tests(self):
219
196
        # check that constructor parameters are passed through to the adapted
220
197
        # test.
221
 
        from bzrlib.tests.bzrdir_implementations import make_scenarios
 
198
        from bzrlib.tests.bzrdir_implementations import BzrDirTestProviderAdapter
222
199
        vfs_factory = "v"
223
200
        server1 = "a"
224
201
        server2 = "b"
225
202
        formats = ["c", "d"]
226
 
        scenarios = make_scenarios(vfs_factory, server1, server2, formats)
 
203
        adapter = BzrDirTestProviderAdapter(vfs_factory,
 
204
            server1, server2, formats)
227
205
        self.assertEqual([
228
206
            ('str',
229
207
             {'bzrdir_format': 'c',
235
213
              'transport_readonly_server': 'b',
236
214
              'transport_server': 'a',
237
215
              'vfs_transport_factory': 'v'})],
238
 
            scenarios)
239
 
 
240
 
 
241
 
class TestRepositoryScenarios(TestCase):
 
216
            adapter.scenarios)
 
217
 
 
218
 
 
219
class TestRepositoryProviderAdapter(TestCase):
 
220
    """A group of tests that test the repository implementation test adapter."""
 
221
 
 
222
    def test_constructor(self):
 
223
        # check that constructor parameters are passed through to the
 
224
        # scenarios.
 
225
        from bzrlib.tests.repository_implementations import RepositoryTestProviderAdapter
 
226
        server1 = "a"
 
227
        server2 = "b"
 
228
        formats = [("c", "C"), ("d", "D")]
 
229
        adapter = RepositoryTestProviderAdapter(server1, server2, formats)
 
230
        self.assertEqual([
 
231
            ('str',
 
232
             {'bzrdir_format': 'C',
 
233
              'repository_format': 'c',
 
234
              'transport_readonly_server': 'b',
 
235
              'transport_server': 'a'}),
 
236
            ('str',
 
237
             {'bzrdir_format': 'D',
 
238
              'repository_format': 'd',
 
239
              'transport_readonly_server': 'b',
 
240
              'transport_server': 'a'})],
 
241
            adapter.scenarios)
 
242
 
 
243
    def test_setting_vfs_transport(self):
 
244
        """The vfs_transport_factory can be set optionally."""
 
245
        from bzrlib.tests.repository_implementations import RepositoryTestProviderAdapter
 
246
        formats = [("a", "b"), ("c", "d")]
 
247
        adapter = RepositoryTestProviderAdapter(None, None, formats,
 
248
            vfs_transport_factory="vfs")
 
249
        self.assertEqual([
 
250
            ('str',
 
251
             {'bzrdir_format': 'b',
 
252
              'repository_format': 'a',
 
253
              'transport_readonly_server': None,
 
254
              'transport_server': None,
 
255
              'vfs_transport_factory': 'vfs'}),
 
256
            ('str',
 
257
             {'bzrdir_format': 'd',
 
258
              'repository_format': 'c',
 
259
              'transport_readonly_server': None,
 
260
              'transport_server': None,
 
261
              'vfs_transport_factory': 'vfs'})],
 
262
            adapter.scenarios)
242
263
 
243
264
    def test_formats_to_scenarios(self):
244
 
        from bzrlib.tests.per_repository import formats_to_scenarios
245
 
        formats = [("(c)", remote.RemoteRepositoryFormat()),
246
 
                   ("(d)", repository.format_registry.get(
247
 
                        'Bazaar pack repository format 1 (needs bzr 0.92)\n'))]
248
 
        no_vfs_scenarios = formats_to_scenarios(formats, "server", "readonly",
249
 
            None)
250
 
        vfs_scenarios = formats_to_scenarios(formats, "server", "readonly",
251
 
            vfs_transport_factory="vfs")
252
 
        # no_vfs generate scenarios without vfs_transport_factory
 
265
        """The adapter can generate all the scenarios needed."""
 
266
        from bzrlib.tests.repository_implementations import RepositoryTestProviderAdapter
 
267
        no_vfs_adapter = RepositoryTestProviderAdapter("server", "readonly",
 
268
            [], None)
 
269
        vfs_adapter = RepositoryTestProviderAdapter("server", "readonly",
 
270
            [], vfs_transport_factory="vfs")
 
271
        # no_vfs generate scenarios without vfs_transport_factor
 
272
        formats = [("c", "C"), (1, "D")]
253
273
        self.assertEqual([
254
 
            ('RemoteRepositoryFormat(c)',
255
 
             {'bzrdir_format': remote.RemoteBzrDirFormat(),
256
 
              'repository_format': remote.RemoteRepositoryFormat(),
 
274
            ('str',
 
275
             {'bzrdir_format': 'C',
 
276
              'repository_format': 'c',
257
277
              'transport_readonly_server': 'readonly',
258
278
              'transport_server': 'server'}),
259
 
            ('RepositoryFormatKnitPack1(d)',
260
 
             {'bzrdir_format': bzrdir.BzrDirMetaFormat1(),
261
 
              'repository_format': pack_repo.RepositoryFormatKnitPack1(),
 
279
            ('int',
 
280
             {'bzrdir_format': 'D',
 
281
              'repository_format': 1,
262
282
              'transport_readonly_server': 'readonly',
263
283
              'transport_server': 'server'})],
264
 
            no_vfs_scenarios)
 
284
            no_vfs_adapter.formats_to_scenarios(formats))
265
285
        self.assertEqual([
266
 
            ('RemoteRepositoryFormat(c)',
267
 
             {'bzrdir_format': remote.RemoteBzrDirFormat(),
268
 
              'repository_format': remote.RemoteRepositoryFormat(),
 
286
            ('str',
 
287
             {'bzrdir_format': 'C',
 
288
              'repository_format': 'c',
269
289
              'transport_readonly_server': 'readonly',
270
290
              'transport_server': 'server',
271
291
              'vfs_transport_factory': 'vfs'}),
272
 
            ('RepositoryFormatKnitPack1(d)',
273
 
             {'bzrdir_format': bzrdir.BzrDirMetaFormat1(),
274
 
              'repository_format': pack_repo.RepositoryFormatKnitPack1(),
 
292
            ('int',
 
293
             {'bzrdir_format': 'D',
 
294
              'repository_format': 1,
275
295
              'transport_readonly_server': 'readonly',
276
296
              'transport_server': 'server',
277
297
              'vfs_transport_factory': 'vfs'})],
278
 
            vfs_scenarios)
279
 
 
280
 
 
281
 
class TestTestScenarioApplication(TestCase):
 
298
            vfs_adapter.formats_to_scenarios(formats))
 
299
 
 
300
 
 
301
class TestTestScenarioApplier(TestCase):
282
302
    """Tests for the test adaption facilities."""
283
303
 
284
 
    def test_apply_scenario(self):
285
 
        from bzrlib.tests import apply_scenario
286
 
        input_test = TestTestScenarioApplication("test_apply_scenario")
 
304
    def test_adapt_applies_scenarios(self):
 
305
        from bzrlib.tests.repository_implementations import TestScenarioApplier
 
306
        input_test = TestTestScenarioApplier("test_adapt_test_to_scenario")
 
307
        adapter = TestScenarioApplier()
 
308
        adapter.scenarios = [("1", "dict"), ("2", "settings")]
 
309
        calls = []
 
310
        def capture_call(test, scenario):
 
311
            calls.append((test, scenario))
 
312
            return test
 
313
        adapter.adapt_test_to_scenario = capture_call
 
314
        adapter.adapt(input_test)
 
315
        self.assertEqual([(input_test, ("1", "dict")),
 
316
            (input_test, ("2", "settings"))], calls)
 
317
 
 
318
    def test_adapt_test_to_scenario(self):
 
319
        from bzrlib.tests.repository_implementations import TestScenarioApplier
 
320
        input_test = TestTestScenarioApplier("test_adapt_test_to_scenario")
 
321
        adapter = TestScenarioApplier()
287
322
        # setup two adapted tests
288
 
        adapted_test1 = apply_scenario(input_test,
 
323
        adapted_test1 = adapter.adapt_test_to_scenario(input_test,
289
324
            ("new id",
290
325
            {"bzrdir_format":"bzr_format",
291
326
             "repository_format":"repo_fmt",
292
327
             "transport_server":"transport_server",
293
328
             "transport_readonly_server":"readonly-server"}))
294
 
        adapted_test2 = apply_scenario(input_test,
 
329
        adapted_test2 = adapter.adapt_test_to_scenario(input_test,
295
330
            ("new id 2", {"bzrdir_format":None}))
296
331
        # input_test should have been altered.
297
332
        self.assertRaises(AttributeError, getattr, input_test, "bzrdir_format")
298
 
        # the new tests are mutually incompatible, ensuring it has
 
333
        # the new tests are mutually incompatible, ensuring it has 
299
334
        # made new ones, and unspecified elements in the scenario
300
335
        # should not have been altered.
301
336
        self.assertEqual("bzr_format", adapted_test1.bzrdir_format)
304
339
        self.assertEqual("readonly-server",
305
340
            adapted_test1.transport_readonly_server)
306
341
        self.assertEqual(
307
 
            "bzrlib.tests.test_selftest.TestTestScenarioApplication."
308
 
            "test_apply_scenario(new id)",
 
342
            "bzrlib.tests.test_selftest.TestTestScenarioApplier."
 
343
            "test_adapt_test_to_scenario(new id)",
309
344
            adapted_test1.id())
310
345
        self.assertEqual(None, adapted_test2.bzrdir_format)
311
346
        self.assertEqual(
312
 
            "bzrlib.tests.test_selftest.TestTestScenarioApplication."
313
 
            "test_apply_scenario(new id 2)",
 
347
            "bzrlib.tests.test_selftest.TestTestScenarioApplier."
 
348
            "test_adapt_test_to_scenario(new id 2)",
314
349
            adapted_test2.id())
315
350
 
316
351
 
317
 
class TestInterRepositoryScenarios(TestCase):
 
352
class TestInterRepositoryProviderAdapter(TestCase):
 
353
    """A group of tests that test the InterRepository test adapter."""
318
354
 
319
 
    def test_scenarios(self):
 
355
    def test_adapted_tests(self):
320
356
        # check that constructor parameters are passed through to the adapted
321
357
        # test.
322
358
        from bzrlib.tests.interrepository_implementations import \
323
 
            make_scenarios
 
359
            InterRepositoryTestProviderAdapter
324
360
        server1 = "a"
325
361
        server2 = "b"
326
362
        formats = [(str, "C1", "C2"), (int, "D1", "D2")]
327
 
        scenarios = make_scenarios(server1, server2, formats)
 
363
        adapter = InterRepositoryTestProviderAdapter(server1, server2, formats)
328
364
        self.assertEqual([
329
 
            ('str,str,str',
 
365
            ('str',
330
366
             {'interrepo_class': str,
331
367
              'repository_format': 'C1',
332
368
              'repository_format_to': 'C2',
333
369
              'transport_readonly_server': 'b',
334
370
              'transport_server': 'a'}),
335
 
            ('int,str,str',
 
371
            ('int',
336
372
             {'interrepo_class': int,
337
373
              'repository_format': 'D1',
338
374
              'repository_format_to': 'D2',
339
375
              'transport_readonly_server': 'b',
340
376
              'transport_server': 'a'})],
341
 
            scenarios)
342
 
 
343
 
 
344
 
class TestWorkingTreeScenarios(TestCase):
 
377
            adapter.formats_to_scenarios(formats))
 
378
 
 
379
 
 
380
class TestInterVersionedFileProviderAdapter(TestCase):
 
381
    """A group of tests that test the InterVersionedFile test adapter."""
 
382
 
 
383
    def test_scenarios(self):
 
384
        # check that constructor parameters are passed through to the adapted
 
385
        # test.
 
386
        from bzrlib.tests.interversionedfile_implementations \
 
387
            import InterVersionedFileTestProviderAdapter
 
388
        server1 = "a"
 
389
        server2 = "b"
 
390
        formats = [(str, "C1", "C2"), (int, "D1", "D2")]
 
391
        adapter = InterVersionedFileTestProviderAdapter(server1, server2, formats)
 
392
        self.assertEqual([
 
393
            ('str',
 
394
             {'interversionedfile_class':str,
 
395
              'transport_readonly_server': 'b',
 
396
              'transport_server': 'a',
 
397
              'versionedfile_factory': 'C1',
 
398
              'versionedfile_factory_to': 'C2'}),
 
399
            ('int',
 
400
             {'interversionedfile_class': int,
 
401
              'transport_readonly_server': 'b',
 
402
              'transport_server': 'a',
 
403
              'versionedfile_factory': 'D1',
 
404
              'versionedfile_factory_to': 'D2'})],
 
405
            adapter.scenarios)
 
406
 
 
407
 
 
408
class TestRevisionStoreProviderAdapter(TestCase):
 
409
    """A group of tests that test the RevisionStore test adapter."""
 
410
 
 
411
    def test_scenarios(self):
 
412
        # check that constructor parameters are passed through to the adapted
 
413
        # test.
 
414
        from bzrlib.tests.revisionstore_implementations \
 
415
            import RevisionStoreTestProviderAdapter
 
416
        # revision stores need a store factory - i.e. RevisionKnit
 
417
        #, a readonly and rw transport 
 
418
        # transport servers:
 
419
        server1 = "a"
 
420
        server2 = "b"
 
421
        store_factories = ["c", "d"]
 
422
        adapter = RevisionStoreTestProviderAdapter(server1, server2, store_factories)
 
423
        self.assertEqual([
 
424
            ('c',
 
425
             {'store_factory': 'c',
 
426
              'transport_readonly_server': 'b',
 
427
              'transport_server': 'a'}),
 
428
            ('d',
 
429
             {'store_factory': 'd',
 
430
              'transport_readonly_server': 'b',
 
431
              'transport_server': 'a'})],
 
432
            adapter.scenarios)
 
433
 
 
434
 
 
435
class TestWorkingTreeProviderAdapter(TestCase):
 
436
    """A group of tests that test the workingtree implementation test adapter."""
345
437
 
346
438
    def test_scenarios(self):
347
439
        # check that constructor parameters are passed through to the adapted
348
440
        # test.
349
441
        from bzrlib.tests.workingtree_implementations \
350
 
            import make_scenarios
 
442
            import WorkingTreeTestProviderAdapter
351
443
        server1 = "a"
352
444
        server2 = "b"
353
 
        formats = [workingtree.WorkingTreeFormat2(),
354
 
                   workingtree.WorkingTreeFormat3(),]
355
 
        scenarios = make_scenarios(server1, server2, formats)
 
445
        formats = [("c", "C"), ("d", "D")]
 
446
        adapter = WorkingTreeTestProviderAdapter(server1, server2, formats)
356
447
        self.assertEqual([
357
 
            ('WorkingTreeFormat2',
358
 
             {'bzrdir_format': formats[0]._matchingbzrdir,
359
 
              'transport_readonly_server': 'b',
360
 
              'transport_server': 'a',
361
 
              'workingtree_format': formats[0]}),
362
 
            ('WorkingTreeFormat3',
363
 
             {'bzrdir_format': formats[1]._matchingbzrdir,
364
 
              'transport_readonly_server': 'b',
365
 
              'transport_server': 'a',
366
 
              'workingtree_format': formats[1]})],
367
 
            scenarios)
368
 
 
369
 
 
370
 
class TestTreeScenarios(TestCase):
371
 
 
372
 
    def test_scenarios(self):
373
 
        # the tree implementation scenario generator is meant to setup one
374
 
        # instance for each working tree format, and one additional instance
375
 
        # that will use the default wt format, but create a revision tree for
376
 
        # the tests.  this means that the wt ones should have the
377
 
        # workingtree_to_test_tree attribute set to 'return_parameter' and the
378
 
        # revision one set to revision_tree_from_workingtree.
 
448
            ('str',
 
449
             {'bzrdir_format': 'C',
 
450
              'transport_readonly_server': 'b',
 
451
              'transport_server': 'a',
 
452
              'workingtree_format': 'c'}),
 
453
            ('str',
 
454
             {'bzrdir_format': 'D',
 
455
              'transport_readonly_server': 'b',
 
456
              'transport_server': 'a',
 
457
              'workingtree_format': 'd'})],
 
458
            adapter.scenarios)
 
459
 
 
460
 
 
461
class TestTreeProviderAdapter(TestCase):
 
462
    """Test the setup of tree_implementation tests."""
 
463
 
 
464
    def test_adapted_tests(self):
 
465
        # the tree implementation adapter is meant to setup one instance for
 
466
        # each working tree format, and one additional instance that will
 
467
        # use the default wt format, but create a revision tree for the tests.
 
468
        # this means that the wt ones should have the workingtree_to_test_tree
 
469
        # attribute set to 'return_parameter' and the revision one set to
 
470
        # revision_tree_from_workingtree.
379
471
 
380
472
        from bzrlib.tests.tree_implementations import (
381
 
            _dirstate_tree_from_workingtree,
382
 
            make_scenarios,
383
 
            preview_tree_pre,
384
 
            preview_tree_post,
 
473
            TreeTestProviderAdapter,
385
474
            return_parameter,
386
475
            revision_tree_from_workingtree
387
476
            )
 
477
        from bzrlib.workingtree import WorkingTreeFormat, WorkingTreeFormat3
 
478
        input_test = TestTreeProviderAdapter(
 
479
            "test_adapted_tests")
388
480
        server1 = "a"
389
481
        server2 = "b"
390
 
        formats = [workingtree.WorkingTreeFormat2(),
391
 
                   workingtree.WorkingTreeFormat3(),]
392
 
        scenarios = make_scenarios(server1, server2, formats)
393
 
        self.assertEqual(7, len(scenarios))
394
 
        default_wt_format = workingtree.WorkingTreeFormat4._default_format
395
 
        wt4_format = workingtree.WorkingTreeFormat4()
396
 
        wt5_format = workingtree.WorkingTreeFormat5()
397
 
        expected_scenarios = [
398
 
            ('WorkingTreeFormat2',
399
 
             {'bzrdir_format': formats[0]._matchingbzrdir,
400
 
              'transport_readonly_server': 'b',
401
 
              'transport_server': 'a',
402
 
              'workingtree_format': formats[0],
403
 
              '_workingtree_to_test_tree': return_parameter,
404
 
              }),
405
 
            ('WorkingTreeFormat3',
406
 
             {'bzrdir_format': formats[1]._matchingbzrdir,
407
 
              'transport_readonly_server': 'b',
408
 
              'transport_server': 'a',
409
 
              'workingtree_format': formats[1],
410
 
              '_workingtree_to_test_tree': return_parameter,
411
 
             }),
412
 
            ('RevisionTree',
413
 
             {'_workingtree_to_test_tree': revision_tree_from_workingtree,
414
 
              'bzrdir_format': default_wt_format._matchingbzrdir,
415
 
              'transport_readonly_server': 'b',
416
 
              'transport_server': 'a',
417
 
              'workingtree_format': default_wt_format,
418
 
             }),
419
 
            ('DirStateRevisionTree,WT4',
420
 
             {'_workingtree_to_test_tree': _dirstate_tree_from_workingtree,
421
 
              'bzrdir_format': wt4_format._matchingbzrdir,
422
 
              'transport_readonly_server': 'b',
423
 
              'transport_server': 'a',
424
 
              'workingtree_format': wt4_format,
425
 
             }),
426
 
            ('DirStateRevisionTree,WT5',
427
 
             {'_workingtree_to_test_tree': _dirstate_tree_from_workingtree,
428
 
              'bzrdir_format': wt5_format._matchingbzrdir,
429
 
              'transport_readonly_server': 'b',
430
 
              'transport_server': 'a',
431
 
              'workingtree_format': wt5_format,
432
 
             }),
433
 
            ('PreviewTree',
434
 
             {'_workingtree_to_test_tree': preview_tree_pre,
435
 
              'bzrdir_format': default_wt_format._matchingbzrdir,
436
 
              'transport_readonly_server': 'b',
437
 
              'transport_server': 'a',
438
 
              'workingtree_format': default_wt_format}),
439
 
            ('PreviewTreePost',
440
 
             {'_workingtree_to_test_tree': preview_tree_post,
441
 
              'bzrdir_format': default_wt_format._matchingbzrdir,
442
 
              'transport_readonly_server': 'b',
443
 
              'transport_server': 'a',
444
 
              'workingtree_format': default_wt_format}),
445
 
             ]
446
 
        self.assertEqual(expected_scenarios, scenarios)
447
 
 
448
 
 
449
 
class TestInterTreeScenarios(TestCase):
 
482
        formats = [("c", "C"), ("d", "D")]
 
483
        adapter = TreeTestProviderAdapter(server1, server2, formats)
 
484
        suite = adapter.adapt(input_test)
 
485
        tests = list(iter(suite))
 
486
        self.assertEqual(4, len(tests))
 
487
        # this must match the default format setp up in
 
488
        # TreeTestProviderAdapter.adapt
 
489
        default_format = WorkingTreeFormat3
 
490
        self.assertEqual(tests[0].workingtree_format, formats[0][0])
 
491
        self.assertEqual(tests[0].bzrdir_format, formats[0][1])
 
492
        self.assertEqual(tests[0].transport_server, server1)
 
493
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
494
        self.assertEqual(tests[0].workingtree_to_test_tree, return_parameter)
 
495
        self.assertEqual(tests[1].workingtree_format, formats[1][0])
 
496
        self.assertEqual(tests[1].bzrdir_format, formats[1][1])
 
497
        self.assertEqual(tests[1].transport_server, server1)
 
498
        self.assertEqual(tests[1].transport_readonly_server, server2)
 
499
        self.assertEqual(tests[1].workingtree_to_test_tree, return_parameter)
 
500
        self.assertIsInstance(tests[2].workingtree_format, default_format)
 
501
        #self.assertEqual(tests[2].bzrdir_format,
 
502
        #                 default_format._matchingbzrdir)
 
503
        self.assertEqual(tests[2].transport_server, server1)
 
504
        self.assertEqual(tests[2].transport_readonly_server, server2)
 
505
        self.assertEqual(tests[2].workingtree_to_test_tree,
 
506
            revision_tree_from_workingtree)
 
507
 
 
508
 
 
509
class TestInterTreeProviderAdapter(TestCase):
450
510
    """A group of tests that test the InterTreeTestAdapter."""
451
511
 
452
 
    def test_scenarios(self):
 
512
    def test_adapted_tests(self):
453
513
        # check that constructor parameters are passed through to the adapted
454
514
        # test.
455
515
        # for InterTree tests we want the machinery to bring up two trees in
457
517
        # because each optimiser can be direction specific, we need to test
458
518
        # each optimiser in its chosen direction.
459
519
        # unlike the TestProviderAdapter we dont want to automatically add a
460
 
        # parameterized one for WorkingTree - the optimisers will tell us what
 
520
        # parameterised one for WorkingTree - the optimisers will tell us what
461
521
        # ones to add.
462
522
        from bzrlib.tests.tree_implementations import (
463
523
            return_parameter,
464
524
            revision_tree_from_workingtree
465
525
            )
466
526
        from bzrlib.tests.intertree_implementations import (
467
 
            make_scenarios,
 
527
            InterTreeTestProviderAdapter,
468
528
            )
469
529
        from bzrlib.workingtree import WorkingTreeFormat2, WorkingTreeFormat3
470
 
        input_test = TestInterTreeScenarios(
471
 
            "test_scenarios")
 
530
        input_test = TestInterTreeProviderAdapter(
 
531
            "test_adapted_tests")
472
532
        server1 = "a"
473
533
        server2 = "b"
474
534
        format1 = WorkingTreeFormat2()
475
535
        format2 = WorkingTreeFormat3()
476
 
        formats = [("1", str, format1, format2, "converter1"),
477
 
            ("2", int, format2, format1, "converter2")]
478
 
        scenarios = make_scenarios(server1, server2, formats)
479
 
        self.assertEqual(2, len(scenarios))
480
 
        expected_scenarios = [
481
 
            ("1", {
482
 
                "bzrdir_format": format1._matchingbzrdir,
483
 
                "intertree_class": formats[0][1],
484
 
                "workingtree_format": formats[0][2],
485
 
                "workingtree_format_to": formats[0][3],
486
 
                "mutable_trees_to_test_trees": formats[0][4],
487
 
                "_workingtree_to_test_tree": return_parameter,
488
 
                "transport_server": server1,
489
 
                "transport_readonly_server": server2,
490
 
                }),
491
 
            ("2", {
492
 
                "bzrdir_format": format2._matchingbzrdir,
493
 
                "intertree_class": formats[1][1],
494
 
                "workingtree_format": formats[1][2],
495
 
                "workingtree_format_to": formats[1][3],
496
 
                "mutable_trees_to_test_trees": formats[1][4],
497
 
                "_workingtree_to_test_tree": return_parameter,
498
 
                "transport_server": server1,
499
 
                "transport_readonly_server": server2,
500
 
                }),
501
 
            ]
502
 
        self.assertEqual(scenarios, expected_scenarios)
 
536
        formats = [(str, format1, format2, "converter1"),
 
537
            (int, format2, format1, "converter2")]
 
538
        adapter = InterTreeTestProviderAdapter(server1, server2, formats)
 
539
        suite = adapter.adapt(input_test)
 
540
        tests = list(iter(suite))
 
541
        self.assertEqual(2, len(tests))
 
542
        self.assertEqual(tests[0].intertree_class, formats[0][0])
 
543
        self.assertEqual(tests[0].workingtree_format, formats[0][1])
 
544
        self.assertEqual(tests[0].workingtree_format_to, formats[0][2])
 
545
        self.assertEqual(tests[0].mutable_trees_to_test_trees, formats[0][3])
 
546
        self.assertEqual(tests[0].workingtree_to_test_tree, return_parameter)
 
547
        self.assertEqual(tests[0].transport_server, server1)
 
548
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
549
        self.assertEqual(tests[1].intertree_class, formats[1][0])
 
550
        self.assertEqual(tests[1].workingtree_format, formats[1][1])
 
551
        self.assertEqual(tests[1].workingtree_format_to, formats[1][2])
 
552
        self.assertEqual(tests[1].mutable_trees_to_test_trees, formats[1][3])
 
553
        self.assertEqual(tests[1].workingtree_to_test_tree, return_parameter)
 
554
        self.assertEqual(tests[1].transport_server, server1)
 
555
        self.assertEqual(tests[1].transport_readonly_server, server2)
503
556
 
504
557
 
505
558
class TestTestCaseInTempDir(TestCaseInTempDir):
507
560
    def test_home_is_not_working(self):
508
561
        self.assertNotEqual(self.test_dir, self.test_home_dir)
509
562
        cwd = osutils.getcwd()
510
 
        self.assertIsSameRealPath(self.test_dir, cwd)
511
 
        self.assertIsSameRealPath(self.test_home_dir, os.environ['HOME'])
512
 
 
513
 
    def test_assertEqualStat_equal(self):
514
 
        from bzrlib.tests.test_dirstate import _FakeStat
515
 
        self.build_tree(["foo"])
516
 
        real = os.lstat("foo")
517
 
        fake = _FakeStat(real.st_size, real.st_mtime, real.st_ctime,
518
 
            real.st_dev, real.st_ino, real.st_mode)
519
 
        self.assertEqualStat(real, fake)
520
 
 
521
 
    def test_assertEqualStat_notequal(self):
522
 
        self.build_tree(["foo", "bar"])
523
 
        self.assertRaises(AssertionError, self.assertEqualStat,
524
 
            os.lstat("foo"), os.lstat("bar"))
 
563
        self.assertEqual(self.test_dir, cwd)
 
564
        self.assertEqual(self.test_home_dir, os.environ['HOME'])
525
565
 
526
566
 
527
567
class TestTestCaseWithMemoryTransport(TestCaseWithMemoryTransport):
530
570
        """The test_home_dir for TestCaseWithMemoryTransport is missing.
531
571
 
532
572
        This is because TestCaseWithMemoryTransport is for tests that do not
533
 
        need any disk resources: they should be hooked into bzrlib in such a
534
 
        way that no global settings are being changed by the test (only a
 
573
        need any disk resources: they should be hooked into bzrlib in such a 
 
574
        way that no global settings are being changed by the test (only a 
535
575
        few tests should need to do that), and having a missing dir as home is
536
576
        an effective way to ensure that this is the case.
537
577
        """
538
 
        self.assertIsSameRealPath(
539
 
            self.TEST_ROOT + "/MemoryTransportMissingHomeDir",
 
578
        self.assertEqual(self.TEST_ROOT + "/MemoryTransportMissingHomeDir",
540
579
            self.test_home_dir)
541
 
        self.assertIsSameRealPath(self.test_home_dir, os.environ['HOME'])
542
 
 
 
580
        self.assertEqual(self.test_home_dir, os.environ['HOME'])
 
581
        
543
582
    def test_cwd_is_TEST_ROOT(self):
544
 
        self.assertIsSameRealPath(self.test_dir, self.TEST_ROOT)
 
583
        self.assertEqual(self.test_dir, self.TEST_ROOT)
545
584
        cwd = osutils.getcwd()
546
 
        self.assertIsSameRealPath(self.test_dir, cwd)
 
585
        self.assertEqual(self.test_dir, cwd)
547
586
 
548
587
    def test_make_branch_and_memory_tree(self):
549
588
        """In TestCaseWithMemoryTransport we should not make the branch on disk.
569
608
        self.assertEqual(format.repository_format.__class__,
570
609
            tree.branch.repository._format.__class__)
571
610
 
572
 
    def test_make_branch_builder(self):
573
 
        builder = self.make_branch_builder('dir')
574
 
        self.assertIsInstance(builder, branchbuilder.BranchBuilder)
575
 
        # Guard against regression into MemoryTransport leaking
576
 
        # files to disk instead of keeping them in memory.
577
 
        self.failIf(osutils.lexists('dir'))
578
 
 
579
 
    def test_make_branch_builder_with_format(self):
580
 
        # Use a repo layout that doesn't conform to a 'named' layout, to ensure
581
 
        # that the format objects are used.
582
 
        format = bzrdir.BzrDirMetaFormat1()
583
 
        repo_format = weaverepo.RepositoryFormat7()
584
 
        format.repository_format = repo_format
585
 
        builder = self.make_branch_builder('dir', format=format)
586
 
        the_branch = builder.get_branch()
587
 
        # Guard against regression into MemoryTransport leaking
588
 
        # files to disk instead of keeping them in memory.
589
 
        self.failIf(osutils.lexists('dir'))
590
 
        self.assertEqual(format.repository_format.__class__,
591
 
                         the_branch.repository._format.__class__)
592
 
        self.assertEqual(repo_format.get_format_string(),
593
 
                         self.get_transport().get_bytes(
594
 
                            'dir/.bzr/repository/format'))
595
 
 
596
 
    def test_make_branch_builder_with_format_name(self):
597
 
        builder = self.make_branch_builder('dir', format='knit')
598
 
        the_branch = builder.get_branch()
599
 
        # Guard against regression into MemoryTransport leaking
600
 
        # files to disk instead of keeping them in memory.
601
 
        self.failIf(osutils.lexists('dir'))
602
 
        dir_format = bzrdir.format_registry.make_bzrdir('knit')
603
 
        self.assertEqual(dir_format.repository_format.__class__,
604
 
                         the_branch.repository._format.__class__)
605
 
        self.assertEqual('Bazaar-NG Knit Repository Format 1',
606
 
                         self.get_transport().get_bytes(
607
 
                            'dir/.bzr/repository/format'))
608
 
 
609
 
    def test_safety_net(self):
610
 
        """No test should modify the safety .bzr directory.
611
 
 
612
 
        We just test that the _check_safety_net private method raises
613
 
        AssertionError, it's easier than building a test suite with the same
614
 
        test.
615
 
        """
616
 
        # Oops, a commit in the current directory (i.e. without local .bzr
617
 
        # directory) will crawl up the hierarchy to find a .bzr directory.
618
 
        self.run_bzr(['commit', '-mfoo', '--unchanged'])
619
 
        # But we have a safety net in place.
620
 
        self.assertRaises(AssertionError, self._check_safety_net)
621
 
 
622
611
 
623
612
class TestTestCaseWithTransport(TestCaseWithTransport):
624
613
    """Tests for the convenience functions TestCaseWithTransport introduces."""
640
629
        self.assertEqual(t2.base[:-1], t.abspath('foo/bar'))
641
630
 
642
631
    def test_get_readonly_url_http(self):
643
 
        from bzrlib.tests.http_server import HttpServer
 
632
        from bzrlib.tests.HttpServer import HttpServer
644
633
        from bzrlib.transport import get_transport
645
634
        from bzrlib.transport.local import LocalURLServer
646
635
        from bzrlib.transport.http import HttpTransportBase
664
653
        self.assertRaises(AssertionError, self.assertIsDirectory, 'a_file', t)
665
654
        self.assertRaises(AssertionError, self.assertIsDirectory, 'not_here', t)
666
655
 
667
 
    def test_make_branch_builder(self):
668
 
        builder = self.make_branch_builder('dir')
669
 
        rev_id = builder.build_commit()
670
 
        self.failUnlessExists('dir')
671
 
        a_dir = bzrdir.BzrDir.open('dir')
672
 
        self.assertRaises(errors.NoWorkingTree, a_dir.open_workingtree)
673
 
        a_branch = a_dir.open_branch()
674
 
        builder_branch = builder.get_branch()
675
 
        self.assertEqual(a_branch.base, builder_branch.base)
676
 
        self.assertEqual((1, rev_id), builder_branch.last_revision_info())
677
 
        self.assertEqual((1, rev_id), a_branch.last_revision_info())
678
 
 
679
656
 
680
657
class TestTestCaseTransports(TestCaseWithTransport):
681
658
 
686
663
    def test_make_bzrdir_preserves_transport(self):
687
664
        t = self.get_transport()
688
665
        result_bzrdir = self.make_bzrdir('subdir')
689
 
        self.assertIsInstance(result_bzrdir.transport,
 
666
        self.assertIsInstance(result_bzrdir.transport, 
690
667
                              MemoryTransport)
691
668
        # should not be on disk, should only be in memory
692
669
        self.failIfExists('subdir')
726
703
 
727
704
class TestTestResult(TestCase):
728
705
 
729
 
    def check_timing(self, test_case, expected_re):
 
706
    def test_elapsed_time_with_benchmarking(self):
730
707
        result = bzrlib.tests.TextTestResult(self._log_file,
731
 
                descriptions=0,
732
 
                verbosity=1,
733
 
                )
734
 
        test_case.run(result)
735
 
        timed_string = result._testTimeString(test_case)
736
 
        self.assertContainsRe(timed_string, expected_re)
737
 
 
738
 
    def test_test_reporting(self):
739
 
        class ShortDelayTestCase(TestCase):
740
 
            def test_short_delay(self):
741
 
                time.sleep(0.003)
742
 
            def test_short_benchmark(self):
743
 
                self.time(time.sleep, 0.003)
744
 
        self.check_timing(ShortDelayTestCase('test_short_delay'),
745
 
                          r"^ +[0-9]+ms$")
 
708
                                        descriptions=0,
 
709
                                        verbosity=1,
 
710
                                        )
 
711
        result._recordTestStartTime()
 
712
        time.sleep(0.003)
 
713
        result.extractBenchmarkTime(self)
 
714
        timed_string = result._testTimeString()
 
715
        # without explicit benchmarking, we should get a simple time.
 
716
        self.assertContainsRe(timed_string, "^ +[0-9]+ms$")
746
717
        # if a benchmark time is given, we want a x of y style result.
747
 
        self.check_timing(ShortDelayTestCase('test_short_benchmark'),
748
 
                          r"^ +[0-9]+ms/ +[0-9]+ms$")
749
 
 
750
 
    def test_unittest_reporting_unittest_class(self):
751
 
        # getting the time from a non-bzrlib test works ok
752
 
        class ShortDelayTestCase(unittest.TestCase):
753
 
            def test_short_delay(self):
754
 
                time.sleep(0.003)
755
 
        self.check_timing(ShortDelayTestCase('test_short_delay'),
756
 
                          r"^ +[0-9]+ms$")
 
718
        self.time(time.sleep, 0.001)
 
719
        result.extractBenchmarkTime(self)
 
720
        timed_string = result._testTimeString()
 
721
        self.assertContainsRe(
 
722
            timed_string, "^ +[0-9]+ms/ +[0-9]+ms$")
 
723
        # extracting the time from a non-bzrlib testcase sets to None
 
724
        result._recordTestStartTime()
 
725
        result.extractBenchmarkTime(
 
726
            unittest.FunctionTestCase(self.test_elapsed_time_with_benchmarking))
 
727
        timed_string = result._testTimeString()
 
728
        self.assertContainsRe(timed_string, "^ +[0-9]+ms$")
 
729
        # cheat. Yes, wash thy mouth out with soap.
 
730
        self._benchtime = None
757
731
 
758
732
    def test_assigned_benchmark_file_stores_date(self):
759
733
        output = StringIO()
763
737
                                        bench_history=output
764
738
                                        )
765
739
        output_string = output.getvalue()
 
740
        
766
741
        # if you are wondering about the regexp please read the comment in
767
742
        # test_bench_history (bzrlib.tests.test_selftest.TestRunner)
768
743
        # XXX: what comment?  -- Andrew Bennetts
787
762
        self.assertContainsRe(lines[1],
788
763
            " *[0-9]+ms bzrlib.tests.test_selftest.TestTestResult"
789
764
            "._time_hello_world_encoding")
790
 
 
 
765
 
791
766
    def _time_hello_world_encoding(self):
792
767
        """Profile two sleep calls
793
 
 
 
768
        
794
769
        This is used to exercise the test framework.
795
770
        """
796
771
        self.time(unicode, 'hello', errors='replace')
814
789
        # execute the test, which should succeed and record profiles
815
790
        example_test_case.run(result)
816
791
        # lsprofile_something()
817
 
        # if this worked we want
 
792
        # if this worked we want 
818
793
        # LSProf output for <built in function unicode> (['hello'], {'errors': 'replace'})
819
794
        #    CallCount    Recursive    Total(ms)   Inline(ms) module:lineno(function)
820
795
        # (the lsprof header)
821
796
        # ... an arbitrary number of lines
822
797
        # and the function call which is time.sleep.
823
 
        #           1        0            ???         ???       ???(sleep)
 
798
        #           1        0            ???         ???       ???(sleep) 
824
799
        # and then repeated but with 'world', rather than 'hello'.
825
800
        # this should appear in the output stream of our test result.
826
801
        output = result_stream.getvalue()
867
842
            )
868
843
        test = self.get_passing_test()
869
844
        result.startTest(test)
 
845
        result.extractBenchmarkTime(test)
870
846
        prefix = len(result_stream.getvalue())
871
847
        # the err parameter has the shape:
872
848
        # (class, exception object, traceback)
892
868
        test = self.get_passing_test()
893
869
        # this seeds the state to handle reporting the test.
894
870
        result.startTest(test)
 
871
        result.extractBenchmarkTime(test)
895
872
        # the err parameter has the shape:
896
873
        # (class, exception object, traceback)
897
874
        # KnownFailures dont get their tracebacks shown though, so we
911
888
        test.run(result)
912
889
        self.assertEqual(
913
890
            [
914
 
            ('update', '[2 in 0s] passing_test', None, None),
 
891
            ('update', '[2 in 0s, 3 known failures] passing_test', None, None),
915
892
            ],
916
893
            pb.calls[2:])
917
894
 
956
933
        test = self.get_passing_test()
957
934
        feature = Feature()
958
935
        result.startTest(test)
 
936
        result.extractBenchmarkTime(test)
959
937
        prefix = len(result_stream.getvalue())
960
938
        result.report_unsupported(test, feature)
961
939
        output = result_stream.getvalue()[prefix:]
962
940
        lines = output.splitlines()
963
941
        self.assertEqual(lines, ['NODEP                   0ms', "    The feature 'Feature' is not available."])
964
 
 
 
942
    
965
943
    def test_text_report_unsupported(self):
966
944
        # text test output formatting
967
945
        pb = MockProgress()
975
953
        feature = Feature()
976
954
        # this seeds the state to handle reporting the test.
977
955
        result.startTest(test)
 
956
        result.extractBenchmarkTime(test)
978
957
        result.report_unsupported(test, feature)
979
958
        # no output on unsupported features
980
959
        self.assertEqual(
987
966
        test.run(result)
988
967
        self.assertEqual(
989
968
            [
990
 
            ('update', '[2 in 0s, 2 missing] passing_test', None, None),
 
969
            ('update', '[2 in 0s, 2 missing features] passing_test', None, None),
991
970
            ],
992
971
            pb.calls[1:])
993
 
 
 
972
    
994
973
    def test_unavailable_exception(self):
995
974
        """An UnavailableFeature being raised should invoke addNotSupported."""
996
975
        class InstrumentedTestResult(ExtendedTestResult):
1011
990
        # and not count as an error
1012
991
        self.assertEqual(0, result.error_count)
1013
992
 
1014
 
    def test_strict_with_unsupported_feature(self):
1015
 
        result = bzrlib.tests.TextTestResult(self._log_file, descriptions=0,
1016
 
                                             verbosity=1)
1017
 
        test = self.get_passing_test()
1018
 
        feature = "Unsupported Feature"
1019
 
        result.addNotSupported(test, feature)
1020
 
        self.assertFalse(result.wasStrictlySuccessful())
1021
 
        self.assertEqual(None, result._extractBenchmarkTime(test))
1022
 
 
1023
 
    def test_strict_with_known_failure(self):
1024
 
        result = bzrlib.tests.TextTestResult(self._log_file, descriptions=0,
1025
 
                                             verbosity=1)
1026
 
        test = self.get_passing_test()
1027
 
        err = (KnownFailure, KnownFailure('foo'), None)
1028
 
        result._addKnownFailure(test, err)
1029
 
        self.assertFalse(result.wasStrictlySuccessful())
1030
 
        self.assertEqual(None, result._extractBenchmarkTime(test))
1031
 
 
1032
 
    def test_strict_with_success(self):
1033
 
        result = bzrlib.tests.TextTestResult(self._log_file, descriptions=0,
1034
 
                                             verbosity=1)
1035
 
        test = self.get_passing_test()
1036
 
        result.addSuccess(test)
1037
 
        self.assertTrue(result.wasStrictlySuccessful())
1038
 
        self.assertEqual(None, result._extractBenchmarkTime(test))
1039
 
 
1040
 
 
1041
 
class TestUnicodeFilenameFeature(TestCase):
1042
 
 
1043
 
    def test_probe_passes(self):
1044
 
        """UnicodeFilenameFeature._probe passes."""
1045
 
        # We can't test much more than that because the behaviour depends
1046
 
        # on the platform.
1047
 
        tests.UnicodeFilenameFeature._probe()
1048
 
 
1049
993
 
1050
994
class TestRunner(TestCase):
1051
995
 
1057
1001
 
1058
1002
        This current saves and restores:
1059
1003
        TestCaseInTempDir.TEST_ROOT
1060
 
 
 
1004
        
1061
1005
        There should be no tests in this file that use bzrlib.tests.TextTestRunner
1062
1006
        without using this convenience method, because of our use of global state.
1063
1007
        """
1115
1059
        # run a test that is skipped, and check the suite as a whole still
1116
1060
        # succeeds.
1117
1061
        # skipping_test must be hidden in here so it's not run as a real test
1118
 
        class SkippingTest(TestCase):
1119
 
            def skipping_test(self):
1120
 
                raise TestSkipped('test intentionally skipped')
 
1062
        def skipping_test():
 
1063
            raise TestSkipped('test intentionally skipped')
 
1064
 
1121
1065
        runner = TextTestRunner(stream=self._log_file)
1122
 
        test = SkippingTest("skipping_test")
 
1066
        test = unittest.FunctionTestCase(skipping_test)
1123
1067
        result = self.run_test_runner(runner, test)
1124
1068
        self.assertTrue(result.wasSuccessful())
1125
1069
 
1126
1070
    def test_skipped_from_setup(self):
1127
 
        calls = []
1128
1071
        class SkippedSetupTest(TestCase):
1129
1072
 
1130
1073
            def setUp(self):
1131
 
                calls.append('setUp')
 
1074
                self.counter = 1
1132
1075
                self.addCleanup(self.cleanup)
1133
1076
                raise TestSkipped('skipped setup')
1134
1077
 
1136
1079
                self.fail('test reached')
1137
1080
 
1138
1081
            def cleanup(self):
1139
 
                calls.append('cleanup')
 
1082
                self.counter -= 1
1140
1083
 
1141
1084
        runner = TextTestRunner(stream=self._log_file)
1142
1085
        test = SkippedSetupTest('test_skip')
1143
1086
        result = self.run_test_runner(runner, test)
1144
1087
        self.assertTrue(result.wasSuccessful())
1145
1088
        # Check if cleanup was called the right number of times.
1146
 
        self.assertEqual(['setUp', 'cleanup'], calls)
 
1089
        self.assertEqual(0, test.counter)
1147
1090
 
1148
1091
    def test_skipped_from_test(self):
1149
 
        calls = []
1150
1092
        class SkippedTest(TestCase):
1151
1093
 
1152
1094
            def setUp(self):
1153
 
                TestCase.setUp(self)
1154
 
                calls.append('setUp')
 
1095
                self.counter = 1
1155
1096
                self.addCleanup(self.cleanup)
1156
1097
 
1157
1098
            def test_skip(self):
1158
1099
                raise TestSkipped('skipped test')
1159
1100
 
1160
1101
            def cleanup(self):
1161
 
                calls.append('cleanup')
 
1102
                self.counter -= 1
1162
1103
 
1163
1104
        runner = TextTestRunner(stream=self._log_file)
1164
1105
        test = SkippedTest('test_skip')
1165
1106
        result = self.run_test_runner(runner, test)
1166
1107
        self.assertTrue(result.wasSuccessful())
1167
1108
        # Check if cleanup was called the right number of times.
1168
 
        self.assertEqual(['setUp', 'cleanup'], calls)
1169
 
 
1170
 
    def test_not_applicable(self):
1171
 
        # run a test that is skipped because it's not applicable
1172
 
        def not_applicable_test():
1173
 
            from bzrlib.tests import TestNotApplicable
1174
 
            raise TestNotApplicable('this test never runs')
1175
 
        out = StringIO()
1176
 
        runner = TextTestRunner(stream=out, verbosity=2)
1177
 
        test = unittest.FunctionTestCase(not_applicable_test)
1178
 
        result = self.run_test_runner(runner, test)
1179
 
        self._log_file.write(out.getvalue())
1180
 
        self.assertTrue(result.wasSuccessful())
1181
 
        self.assertTrue(result.wasStrictlySuccessful())
1182
 
        self.assertContainsRe(out.getvalue(),
1183
 
                r'(?m)not_applicable_test   * N/A')
1184
 
        self.assertContainsRe(out.getvalue(),
1185
 
                r'(?m)^    this test never runs')
1186
 
 
1187
 
    def test_not_applicable_demo(self):
1188
 
        # just so you can see it in the test output
1189
 
        raise TestNotApplicable('this test is just a demonstation')
 
1109
        self.assertEqual(0, test.counter)
1190
1110
 
1191
1111
    def test_unsupported_features_listed(self):
1192
1112
        """When unsupported features are encountered they are detailed."""
1228
1148
            revision_id = workingtree.get_parent_ids()[0]
1229
1149
            self.assertEndsWith(output_string.rstrip(), revision_id)
1230
1150
 
1231
 
    def assertLogDeleted(self, test):
1232
 
        log = test._get_log()
1233
 
        self.assertEqual("DELETED log file to reduce memory footprint", log)
1234
 
        self.assertEqual('', test._log_contents)
1235
 
        self.assertIs(None, test._log_file_name)
1236
 
 
1237
1151
    def test_success_log_deleted(self):
1238
1152
        """Successful tests have their log deleted"""
1239
1153
 
1247
1161
        test = LogTester('test_success')
1248
1162
        result = self.run_test_runner(runner, test)
1249
1163
 
1250
 
        self.assertLogDeleted(test)
1251
 
 
1252
 
    def test_skipped_log_deleted(self):
1253
 
        """Skipped tests have their log deleted"""
1254
 
 
1255
 
        class LogTester(TestCase):
1256
 
 
1257
 
            def test_skipped(self):
1258
 
                self.log('this will be removed\n')
1259
 
                raise tests.TestSkipped()
1260
 
 
1261
 
        sio = cStringIO.StringIO()
1262
 
        runner = TextTestRunner(stream=sio)
1263
 
        test = LogTester('test_skipped')
1264
 
        result = self.run_test_runner(runner, test)
1265
 
 
1266
 
        self.assertLogDeleted(test)
1267
 
 
1268
 
    def test_not_aplicable_log_deleted(self):
1269
 
        """Not applicable tests have their log deleted"""
1270
 
 
1271
 
        class LogTester(TestCase):
1272
 
 
1273
 
            def test_not_applicable(self):
1274
 
                self.log('this will be removed\n')
1275
 
                raise tests.TestNotApplicable()
1276
 
 
1277
 
        sio = cStringIO.StringIO()
1278
 
        runner = TextTestRunner(stream=sio)
1279
 
        test = LogTester('test_not_applicable')
1280
 
        result = self.run_test_runner(runner, test)
1281
 
 
1282
 
        self.assertLogDeleted(test)
1283
 
 
1284
 
    def test_known_failure_log_deleted(self):
1285
 
        """Know failure tests have their log deleted"""
1286
 
 
1287
 
        class LogTester(TestCase):
1288
 
 
1289
 
            def test_known_failure(self):
1290
 
                self.log('this will be removed\n')
1291
 
                raise tests.KnownFailure()
1292
 
 
1293
 
        sio = cStringIO.StringIO()
1294
 
        runner = TextTestRunner(stream=sio)
1295
 
        test = LogTester('test_known_failure')
1296
 
        result = self.run_test_runner(runner, test)
1297
 
 
1298
 
        self.assertLogDeleted(test)
 
1164
        log = test._get_log()
 
1165
        self.assertEqual("DELETED log file to reduce memory footprint", log)
 
1166
        self.assertEqual('', test._log_contents)
 
1167
        self.assertIs(None, test._log_file_name)
1299
1168
 
1300
1169
    def test_fail_log_kept(self):
1301
1170
        """Failed tests have their log kept"""
1347
1216
    def _test_pass(self):
1348
1217
        pass
1349
1218
 
1350
 
class _TestException(Exception):
1351
 
    pass
1352
1219
 
1353
1220
class TestTestCase(TestCase):
1354
1221
    """Tests that test the core bzrlib TestCase."""
1355
1222
 
1356
 
    def test_assertLength_matches_empty(self):
1357
 
        a_list = []
1358
 
        self.assertLength(0, a_list)
1359
 
 
1360
 
    def test_assertLength_matches_nonempty(self):
1361
 
        a_list = [1, 2, 3]
1362
 
        self.assertLength(3, a_list)
1363
 
 
1364
 
    def test_assertLength_fails_different(self):
1365
 
        a_list = []
1366
 
        self.assertRaises(AssertionError, self.assertLength, 1, a_list)
1367
 
 
1368
 
    def test_assertLength_shows_sequence_in_failure(self):
1369
 
        a_list = [1, 2, 3]
1370
 
        exception = self.assertRaises(AssertionError, self.assertLength, 2,
1371
 
            a_list)
1372
 
        self.assertEqual('Incorrect length: wanted 2, got 3 for [1, 2, 3]',
1373
 
            exception.args[0])
1374
 
 
1375
 
    def test_base_setUp_not_called_causes_failure(self):
1376
 
        class TestCaseWithBrokenSetUp(TestCase):
1377
 
            def setUp(self):
1378
 
                pass # does not call TestCase.setUp
1379
 
            def test_foo(self):
1380
 
                pass
1381
 
        test = TestCaseWithBrokenSetUp('test_foo')
1382
 
        result = unittest.TestResult()
1383
 
        test.run(result)
1384
 
        self.assertFalse(result.wasSuccessful())
1385
 
        self.assertEqual(1, result.testsRun)
1386
 
 
1387
 
    def test_base_tearDown_not_called_causes_failure(self):
1388
 
        class TestCaseWithBrokenTearDown(TestCase):
1389
 
            def tearDown(self):
1390
 
                pass # does not call TestCase.tearDown
1391
 
            def test_foo(self):
1392
 
                pass
1393
 
        test = TestCaseWithBrokenTearDown('test_foo')
1394
 
        result = unittest.TestResult()
1395
 
        test.run(result)
1396
 
        self.assertFalse(result.wasSuccessful())
1397
 
        self.assertEqual(1, result.testsRun)
1398
 
 
1399
1223
    def test_debug_flags_sanitised(self):
1400
1224
        """The bzrlib debug flags should be sanitised by setUp."""
1401
 
        if 'allow_debug' in tests.selftest_debug_flags:
1402
 
            raise TestNotApplicable(
1403
 
                '-Eallow_debug option prevents debug flag sanitisation')
1404
1225
        # we could set something and run a test that will check
1405
1226
        # it gets santised, but this is probably sufficient for now:
1406
1227
        # if someone runs the test with -Dsomething it will error.
1407
1228
        self.assertEqual(set(), bzrlib.debug.debug_flags)
1408
1229
 
1409
 
    def change_selftest_debug_flags(self, new_flags):
1410
 
        orig_selftest_flags = tests.selftest_debug_flags
1411
 
        self.addCleanup(self._restore_selftest_debug_flags, orig_selftest_flags)
1412
 
        tests.selftest_debug_flags = set(new_flags)
1413
 
 
1414
 
    def _restore_selftest_debug_flags(self, flags):
1415
 
        tests.selftest_debug_flags = flags
1416
 
 
1417
 
    def test_allow_debug_flag(self):
1418
 
        """The -Eallow_debug flag prevents bzrlib.debug.debug_flags from being
1419
 
        sanitised (i.e. cleared) before running a test.
1420
 
        """
1421
 
        self.change_selftest_debug_flags(set(['allow_debug']))
1422
 
        bzrlib.debug.debug_flags = set(['a-flag'])
1423
 
        class TestThatRecordsFlags(TestCase):
1424
 
            def test_foo(nested_self):
1425
 
                self.flags = set(bzrlib.debug.debug_flags)
1426
 
        test = TestThatRecordsFlags('test_foo')
1427
 
        test.run(self.make_test_result())
1428
 
        self.assertEqual(set(['a-flag']), self.flags)
1429
 
 
1430
 
    def test_debug_flags_restored(self):
1431
 
        """The bzrlib debug flags should be restored to their original state
1432
 
        after the test was run, even if allow_debug is set.
1433
 
        """
1434
 
        self.change_selftest_debug_flags(set(['allow_debug']))
1435
 
        # Now run a test that modifies debug.debug_flags.
1436
 
        bzrlib.debug.debug_flags = set(['original-state'])
1437
 
        class TestThatModifiesFlags(TestCase):
1438
 
            def test_foo(self):
1439
 
                bzrlib.debug.debug_flags = set(['modified'])
1440
 
        test = TestThatModifiesFlags('test_foo')
1441
 
        test.run(self.make_test_result())
1442
 
        self.assertEqual(set(['original-state']), bzrlib.debug.debug_flags)
1443
 
 
1444
 
    def make_test_result(self):
1445
 
        return bzrlib.tests.TextTestResult(
1446
 
            self._log_file, descriptions=0, verbosity=1)
1447
 
 
1448
1230
    def inner_test(self):
1449
1231
        # the inner child test
1450
1232
        note("inner_test")
1453
1235
        # the outer child test
1454
1236
        note("outer_start")
1455
1237
        self.inner_test = TestTestCase("inner_child")
1456
 
        result = self.make_test_result()
 
1238
        result = bzrlib.tests.TextTestResult(self._log_file,
 
1239
                                        descriptions=0,
 
1240
                                        verbosity=1)
1457
1241
        self.inner_test.run(result)
1458
1242
        note("outer finish")
1459
1243
 
1463
1247
        # should setup a new log, log content to it, setup a child case (B),
1464
1248
        # which should log independently, then case (A) should log a trailer
1465
1249
        # and return.
1466
 
        # we do two nested children so that we can verify the state of the
 
1250
        # we do two nested children so that we can verify the state of the 
1467
1251
        # logs after the outer child finishes is correct, which a bad clean
1468
1252
        # up routine in tearDown might trigger a fault in our test with only
1469
1253
        # one child, we should instead see the bad result inside our test with
1471
1255
        # the outer child test
1472
1256
        original_trace = bzrlib.trace._trace_file
1473
1257
        outer_test = TestTestCase("outer_child")
1474
 
        result = self.make_test_result()
 
1258
        result = bzrlib.tests.TextTestResult(self._log_file,
 
1259
                                        descriptions=0,
 
1260
                                        verbosity=1)
1475
1261
        outer_test.run(result)
1476
1262
        self.assertEqual(original_trace, bzrlib.trace._trace_file)
1477
1263
 
1496
1282
 
1497
1283
    def test_hooks_sanitised(self):
1498
1284
        """The bzrlib hooks should be sanitised by setUp."""
1499
 
        # Note this test won't fail with hooks that the core library doesn't
1500
 
        # use - but it trigger with a plugin that adds hooks, so its still a
1501
 
        # useful warning in that case.
1502
1285
        self.assertEqual(bzrlib.branch.BranchHooks(),
1503
1286
            bzrlib.branch.Branch.hooks)
1504
1287
        self.assertEqual(bzrlib.smart.server.SmartServerHooks(),
1505
1288
            bzrlib.smart.server.SmartTCPServer.hooks)
1506
 
        self.assertEqual(bzrlib.commands.CommandHooks(),
1507
 
            bzrlib.commands.Command.hooks)
1508
1289
 
1509
1290
    def test__gather_lsprof_in_benchmarks(self):
1510
1291
        """When _gather_lsprof_in_benchmarks is on, accumulate profile data.
1511
 
 
 
1292
        
1512
1293
        Each self.time() call is individually and separately profiled.
1513
1294
        """
1514
1295
        self.requireFeature(test_lsprof.LSProfFeature)
1515
 
        # overrides the class member with an instance member so no cleanup
 
1296
        # overrides the class member with an instance member so no cleanup 
1516
1297
        # needed.
1517
1298
        self._gather_lsprof_in_benchmarks = True
1518
1299
        self.time(time.sleep, 0.000)
1544
1325
    def test_run_no_parameters(self):
1545
1326
        test = SampleTestCase('_test_pass')
1546
1327
        test.run()
1547
 
 
 
1328
    
1548
1329
    def test_run_enabled_unittest_result(self):
1549
1330
        """Test we revert to regular behaviour when the test is enabled."""
1550
1331
        test = SampleTestCase('_test_pass')
1598
1379
            ],
1599
1380
            result.calls)
1600
1381
 
1601
 
    def test_assert_list_raises_on_generator(self):
1602
 
        def generator_which_will_raise():
1603
 
            # This will not raise until after the first yield
1604
 
            yield 1
1605
 
            raise _TestException()
1606
 
 
1607
 
        e = self.assertListRaises(_TestException, generator_which_will_raise)
1608
 
        self.assertIsInstance(e, _TestException)
1609
 
 
1610
 
        e = self.assertListRaises(Exception, generator_which_will_raise)
1611
 
        self.assertIsInstance(e, _TestException)
1612
 
 
1613
 
    def test_assert_list_raises_on_plain(self):
1614
 
        def plain_exception():
1615
 
            raise _TestException()
1616
 
            return []
1617
 
 
1618
 
        e = self.assertListRaises(_TestException, plain_exception)
1619
 
        self.assertIsInstance(e, _TestException)
1620
 
 
1621
 
        e = self.assertListRaises(Exception, plain_exception)
1622
 
        self.assertIsInstance(e, _TestException)
1623
 
 
1624
 
    def test_assert_list_raises_assert_wrong_exception(self):
1625
 
        class _NotTestException(Exception):
1626
 
            pass
1627
 
 
1628
 
        def wrong_exception():
1629
 
            raise _NotTestException()
1630
 
 
1631
 
        def wrong_exception_generator():
1632
 
            yield 1
1633
 
            yield 2
1634
 
            raise _NotTestException()
1635
 
 
1636
 
        # Wrong exceptions are not intercepted
1637
 
        self.assertRaises(_NotTestException,
1638
 
            self.assertListRaises, _TestException, wrong_exception)
1639
 
        self.assertRaises(_NotTestException,
1640
 
            self.assertListRaises, _TestException, wrong_exception_generator)
1641
 
 
1642
 
    def test_assert_list_raises_no_exception(self):
1643
 
        def success():
1644
 
            return []
1645
 
 
1646
 
        def success_generator():
1647
 
            yield 1
1648
 
            yield 2
1649
 
 
1650
 
        self.assertRaises(AssertionError,
1651
 
            self.assertListRaises, _TestException, success)
1652
 
 
1653
 
        self.assertRaises(AssertionError,
1654
 
            self.assertListRaises, _TestException, success_generator)
1655
 
 
1656
 
 
1657
 
# NB: Don't delete this; it's not actually from 0.11!
1658
 
@deprecated_function(deprecated_in((0, 11, 0)))
 
1382
 
 
1383
@symbol_versioning.deprecated_function(zero_eleven)
1659
1384
def sample_deprecated_function():
1660
1385
    """A deprecated function to test applyDeprecated with."""
1661
1386
    return 2
1668
1393
class ApplyDeprecatedHelper(object):
1669
1394
    """A helper class for ApplyDeprecated tests."""
1670
1395
 
1671
 
    @deprecated_method(deprecated_in((0, 11, 0)))
 
1396
    @symbol_versioning.deprecated_method(zero_eleven)
1672
1397
    def sample_deprecated_method(self, param_one):
1673
1398
        """A deprecated method for testing with."""
1674
1399
        return param_one
1676
1401
    def sample_normal_method(self):
1677
1402
        """A undeprecated method."""
1678
1403
 
1679
 
    @deprecated_method(deprecated_in((0, 10, 0)))
 
1404
    @symbol_versioning.deprecated_method(zero_ten)
1680
1405
    def sample_nested_deprecation(self):
1681
1406
        return sample_deprecated_function()
1682
1407
 
1697
1422
    def test_applyDeprecated_not_deprecated(self):
1698
1423
        sample_object = ApplyDeprecatedHelper()
1699
1424
        # calling an undeprecated callable raises an assertion
1700
 
        self.assertRaises(AssertionError, self.applyDeprecated,
1701
 
            deprecated_in((0, 11, 0)),
 
1425
        self.assertRaises(AssertionError, self.applyDeprecated, zero_eleven,
1702
1426
            sample_object.sample_normal_method)
1703
 
        self.assertRaises(AssertionError, self.applyDeprecated,
1704
 
            deprecated_in((0, 11, 0)),
 
1427
        self.assertRaises(AssertionError, self.applyDeprecated, zero_eleven,
1705
1428
            sample_undeprecated_function, "a param value")
1706
1429
        # calling a deprecated callable (function or method) with the wrong
1707
1430
        # expected deprecation fails.
1708
 
        self.assertRaises(AssertionError, self.applyDeprecated,
1709
 
            deprecated_in((0, 10, 0)),
 
1431
        self.assertRaises(AssertionError, self.applyDeprecated, zero_ten,
1710
1432
            sample_object.sample_deprecated_method, "a param value")
1711
 
        self.assertRaises(AssertionError, self.applyDeprecated,
1712
 
            deprecated_in((0, 10, 0)),
 
1433
        self.assertRaises(AssertionError, self.applyDeprecated, zero_ten,
1713
1434
            sample_deprecated_function)
1714
1435
        # calling a deprecated callable (function or method) with the right
1715
1436
        # expected deprecation returns the functions result.
1716
 
        self.assertEqual("a param value",
1717
 
            self.applyDeprecated(deprecated_in((0, 11, 0)),
 
1437
        self.assertEqual("a param value", self.applyDeprecated(zero_eleven,
1718
1438
            sample_object.sample_deprecated_method, "a param value"))
1719
 
        self.assertEqual(2, self.applyDeprecated(deprecated_in((0, 11, 0)),
 
1439
        self.assertEqual(2, self.applyDeprecated(zero_eleven,
1720
1440
            sample_deprecated_function))
1721
1441
        # calling a nested deprecation with the wrong deprecation version
1722
 
        # fails even if a deeper nested function was deprecated with the
 
1442
        # fails even if a deeper nested function was deprecated with the 
1723
1443
        # supplied version.
1724
1444
        self.assertRaises(AssertionError, self.applyDeprecated,
1725
 
            deprecated_in((0, 11, 0)), sample_object.sample_nested_deprecation)
 
1445
            zero_eleven, sample_object.sample_nested_deprecation)
1726
1446
        # calling a nested deprecation with the right deprecation value
1727
1447
        # returns the calls result.
1728
 
        self.assertEqual(2, self.applyDeprecated(deprecated_in((0, 10, 0)),
 
1448
        self.assertEqual(2, self.applyDeprecated(zero_ten,
1729
1449
            sample_object.sample_nested_deprecation))
1730
1450
 
1731
1451
    def test_callDeprecated(self):
1732
1452
        def testfunc(be_deprecated, result=None):
1733
1453
            if be_deprecated is True:
1734
 
                symbol_versioning.warn('i am deprecated', DeprecationWarning,
 
1454
                symbol_versioning.warn('i am deprecated', DeprecationWarning, 
1735
1455
                                       stacklevel=1)
1736
1456
            return result
1737
1457
        result = self.callDeprecated(['i am deprecated'], testfunc, True)
1742
1462
        self.callDeprecated([], testfunc, be_deprecated=False)
1743
1463
 
1744
1464
 
1745
 
class TestWarningTests(TestCase):
1746
 
    """Tests for calling methods that raise warnings."""
1747
 
 
1748
 
    def test_callCatchWarnings(self):
1749
 
        def meth(a, b):
1750
 
            warnings.warn("this is your last warning")
1751
 
            return a + b
1752
 
        wlist, result = self.callCatchWarnings(meth, 1, 2)
1753
 
        self.assertEquals(3, result)
1754
 
        # would like just to compare them, but UserWarning doesn't implement
1755
 
        # eq well
1756
 
        w0, = wlist
1757
 
        self.assertIsInstance(w0, UserWarning)
1758
 
        self.assertEquals("this is your last warning", str(w0))
1759
 
 
1760
 
 
1761
1465
class TestConvenienceMakers(TestCaseWithTransport):
1762
1466
    """Test for the make_* convenience functions."""
1763
1467
 
1802
1506
            return TestSuite()
1803
1507
        out = StringIO()
1804
1508
        err = StringIO()
1805
 
        self.apply_redirected(out, err, None, bzrlib.tests.selftest,
 
1509
        self.apply_redirected(out, err, None, bzrlib.tests.selftest, 
1806
1510
            test_suite_factory=factory)
1807
1511
        self.assertEqual([True], factory_called)
1808
1512
 
1873
1577
class TestSelftestFiltering(TestCase):
1874
1578
 
1875
1579
    def setUp(self):
1876
 
        TestCase.setUp(self)
1877
1580
        self.suite = TestUtil.TestSuite()
1878
1581
        self.loader = TestUtil.TestLoader()
1879
1582
        self.suite.addTest(self.loader.loadTestsFromModuleNames([
1880
1583
            'bzrlib.tests.test_selftest']))
1881
 
        self.all_names = _test_ids(self.suite)
1882
 
 
1883
 
    def test_condition_id_re(self):
1884
 
        test_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
1885
 
            'test_condition_id_re')
1886
 
        filtered_suite = filter_suite_by_condition(self.suite,
1887
 
            condition_id_re('test_condition_id_re'))
1888
 
        self.assertEqual([test_name], _test_ids(filtered_suite))
1889
 
 
1890
 
    def test_condition_id_in_list(self):
1891
 
        test_names = ['bzrlib.tests.test_selftest.TestSelftestFiltering.'
1892
 
                      'test_condition_id_in_list']
1893
 
        id_list = tests.TestIdList(test_names)
1894
 
        filtered_suite = filter_suite_by_condition(
1895
 
            self.suite, tests.condition_id_in_list(id_list))
1896
 
        my_pattern = 'TestSelftestFiltering.*test_condition_id_in_list'
1897
 
        re_filtered = filter_suite_by_re(self.suite, my_pattern)
1898
 
        self.assertEqual(_test_ids(re_filtered), _test_ids(filtered_suite))
1899
 
 
1900
 
    def test_condition_id_startswith(self):
1901
 
        klass = 'bzrlib.tests.test_selftest.TestSelftestFiltering.'
1902
 
        start1 = klass + 'test_condition_id_starts'
1903
 
        start2 = klass + 'test_condition_id_in'
1904
 
        test_names = [ klass + 'test_condition_id_in_list',
1905
 
                      klass + 'test_condition_id_startswith',
1906
 
                     ]
1907
 
        filtered_suite = filter_suite_by_condition(
1908
 
            self.suite, tests.condition_id_startswith([start1, start2]))
1909
 
        self.assertEqual(test_names, _test_ids(filtered_suite))
1910
 
 
1911
 
    def test_condition_isinstance(self):
1912
 
        filtered_suite = filter_suite_by_condition(self.suite,
1913
 
            condition_isinstance(self.__class__))
1914
 
        class_pattern = 'bzrlib.tests.test_selftest.TestSelftestFiltering.'
1915
 
        re_filtered = filter_suite_by_re(self.suite, class_pattern)
1916
 
        self.assertEqual(_test_ids(re_filtered), _test_ids(filtered_suite))
1917
 
 
1918
 
    def test_exclude_tests_by_condition(self):
1919
 
        excluded_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
1920
 
            'test_exclude_tests_by_condition')
1921
 
        filtered_suite = exclude_tests_by_condition(self.suite,
1922
 
            lambda x:x.id() == excluded_name)
1923
 
        self.assertEqual(len(self.all_names) - 1,
1924
 
            filtered_suite.countTestCases())
1925
 
        self.assertFalse(excluded_name in _test_ids(filtered_suite))
1926
 
        remaining_names = list(self.all_names)
1927
 
        remaining_names.remove(excluded_name)
1928
 
        self.assertEqual(remaining_names, _test_ids(filtered_suite))
1929
 
 
1930
 
    def test_exclude_tests_by_re(self):
1931
 
        self.all_names = _test_ids(self.suite)
1932
 
        filtered_suite = exclude_tests_by_re(self.suite, 'exclude_tests_by_re')
1933
 
        excluded_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
1934
 
            'test_exclude_tests_by_re')
1935
 
        self.assertEqual(len(self.all_names) - 1,
1936
 
            filtered_suite.countTestCases())
1937
 
        self.assertFalse(excluded_name in _test_ids(filtered_suite))
1938
 
        remaining_names = list(self.all_names)
1939
 
        remaining_names.remove(excluded_name)
1940
 
        self.assertEqual(remaining_names, _test_ids(filtered_suite))
1941
 
 
1942
 
    def test_filter_suite_by_condition(self):
1943
 
        test_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
1944
 
            'test_filter_suite_by_condition')
1945
 
        filtered_suite = filter_suite_by_condition(self.suite,
1946
 
            lambda x:x.id() == test_name)
1947
 
        self.assertEqual([test_name], _test_ids(filtered_suite))
 
1584
        self.all_names = [t.id() for t in iter_suite_tests(self.suite)]
1948
1585
 
1949
1586
    def test_filter_suite_by_re(self):
1950
 
        filtered_suite = filter_suite_by_re(self.suite, 'test_filter_suite_by_r')
1951
 
        filtered_names = _test_ids(filtered_suite)
 
1587
        filtered_suite = filter_suite_by_re(self.suite, 'test_filter')
 
1588
        filtered_names = [t.id() for t in iter_suite_tests(filtered_suite)]
1952
1589
        self.assertEqual(filtered_names, ['bzrlib.tests.test_selftest.'
1953
1590
            'TestSelftestFiltering.test_filter_suite_by_re'])
1954
 
 
1955
 
    def test_filter_suite_by_id_list(self):
1956
 
        test_list = ['bzrlib.tests.test_selftest.'
1957
 
                     'TestSelftestFiltering.test_filter_suite_by_id_list']
1958
 
        filtered_suite = tests.filter_suite_by_id_list(
1959
 
            self.suite, tests.TestIdList(test_list))
1960
 
        filtered_names = _test_ids(filtered_suite)
1961
 
        self.assertEqual(
1962
 
            filtered_names,
1963
 
            ['bzrlib.tests.test_selftest.'
1964
 
             'TestSelftestFiltering.test_filter_suite_by_id_list'])
1965
 
 
1966
 
    def test_filter_suite_by_id_startswith(self):
1967
 
        # By design this test may fail if another test is added whose name also
1968
 
        # begins with one of the start value used.
1969
 
        klass = 'bzrlib.tests.test_selftest.TestSelftestFiltering.'
1970
 
        start1 = klass + 'test_filter_suite_by_id_starts'
1971
 
        start2 = klass + 'test_filter_suite_by_id_li'
1972
 
        test_list = [klass + 'test_filter_suite_by_id_list',
1973
 
                     klass + 'test_filter_suite_by_id_startswith',
1974
 
                     ]
1975
 
        filtered_suite = tests.filter_suite_by_id_startswith(
1976
 
            self.suite, [start1, start2])
1977
 
        self.assertEqual(
1978
 
            test_list,
1979
 
            _test_ids(filtered_suite),
1980
 
            )
1981
 
 
1982
 
    def test_preserve_input(self):
1983
 
        # NB: Surely this is something in the stdlib to do this?
1984
 
        self.assertTrue(self.suite is preserve_input(self.suite))
1985
 
        self.assertTrue("@#$" is preserve_input("@#$"))
1986
 
 
1987
 
    def test_randomize_suite(self):
1988
 
        randomized_suite = randomize_suite(self.suite)
1989
 
        # randomizing should not add or remove test names.
1990
 
        self.assertEqual(set(_test_ids(self.suite)),
1991
 
                         set(_test_ids(randomized_suite)))
1992
 
        # Technically, this *can* fail, because random.shuffle(list) can be
1993
 
        # equal to list. Trying multiple times just pushes the frequency back.
1994
 
        # As its len(self.all_names)!:1, the failure frequency should be low
1995
 
        # enough to ignore. RBC 20071021.
1996
 
        # It should change the order.
1997
 
        self.assertNotEqual(self.all_names, _test_ids(randomized_suite))
1998
 
        # But not the length. (Possibly redundant with the set test, but not
1999
 
        # necessarily.)
2000
 
        self.assertEqual(len(self.all_names), len(_test_ids(randomized_suite)))
2001
 
 
2002
 
    def test_split_suit_by_condition(self):
2003
 
        self.all_names = _test_ids(self.suite)
2004
 
        condition = condition_id_re('test_filter_suite_by_r')
2005
 
        split_suite = split_suite_by_condition(self.suite, condition)
2006
 
        filtered_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
2007
 
            'test_filter_suite_by_re')
2008
 
        self.assertEqual([filtered_name], _test_ids(split_suite[0]))
2009
 
        self.assertFalse(filtered_name in _test_ids(split_suite[1]))
2010
 
        remaining_names = list(self.all_names)
2011
 
        remaining_names.remove(filtered_name)
2012
 
        self.assertEqual(remaining_names, _test_ids(split_suite[1]))
2013
 
 
2014
 
    def test_split_suit_by_re(self):
2015
 
        self.all_names = _test_ids(self.suite)
2016
 
        split_suite = split_suite_by_re(self.suite, 'test_filter_suite_by_r')
2017
 
        filtered_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
2018
 
            'test_filter_suite_by_re')
2019
 
        self.assertEqual([filtered_name], _test_ids(split_suite[0]))
2020
 
        self.assertFalse(filtered_name in _test_ids(split_suite[1]))
2021
 
        remaining_names = list(self.all_names)
2022
 
        remaining_names.remove(filtered_name)
2023
 
        self.assertEqual(remaining_names, _test_ids(split_suite[1]))
 
1591
            
 
1592
    def test_sort_suite_by_re(self):
 
1593
        sorted_suite = sort_suite_by_re(self.suite, 'test_filter')
 
1594
        sorted_names = [t.id() for t in iter_suite_tests(sorted_suite)]
 
1595
        self.assertEqual(sorted_names[0], 'bzrlib.tests.test_selftest.'
 
1596
            'TestSelftestFiltering.test_filter_suite_by_re')
 
1597
        self.assertEquals(sorted(self.all_names), sorted(sorted_names))
2024
1598
 
2025
1599
 
2026
1600
class TestCheckInventoryShape(TestCaseWithTransport):
2035
1609
            self.check_inventory_shape(tree.inventory, files)
2036
1610
        finally:
2037
1611
            tree.unlock()
2038
 
 
2039
 
 
2040
 
class TestBlackboxSupport(TestCase):
2041
 
    """Tests for testsuite blackbox features."""
2042
 
 
2043
 
    def test_run_bzr_failure_not_caught(self):
2044
 
        # When we run bzr in blackbox mode, we want any unexpected errors to
2045
 
        # propagate up to the test suite so that it can show the error in the
2046
 
        # usual way, and we won't get a double traceback.
2047
 
        e = self.assertRaises(
2048
 
            AssertionError,
2049
 
            self.run_bzr, ['assert-fail'])
2050
 
        # make sure we got the real thing, not an error from somewhere else in
2051
 
        # the test framework
2052
 
        self.assertEquals('always fails', str(e))
2053
 
        # check that there's no traceback in the test log
2054
 
        self.assertNotContainsRe(self._get_log(keep_log_file=True),
2055
 
            r'Traceback')
2056
 
 
2057
 
    def test_run_bzr_user_error_caught(self):
2058
 
        # Running bzr in blackbox mode, normal/expected/user errors should be
2059
 
        # caught in the regular way and turned into an error message plus exit
2060
 
        # code.
2061
 
        out, err = self.run_bzr(["log", "/nonexistantpath"], retcode=3)
2062
 
        self.assertEqual(out, '')
2063
 
        self.assertContainsRe(err,
2064
 
            'bzr: ERROR: Not a branch: ".*nonexistantpath/".\n')
2065
 
 
2066
 
 
2067
 
class TestTestLoader(TestCase):
2068
 
    """Tests for the test loader."""
2069
 
 
2070
 
    def _get_loader_and_module(self):
2071
 
        """Gets a TestLoader and a module with one test in it."""
2072
 
        loader = TestUtil.TestLoader()
2073
 
        module = {}
2074
 
        class Stub(TestCase):
2075
 
            def test_foo(self):
2076
 
                pass
2077
 
        class MyModule(object):
2078
 
            pass
2079
 
        MyModule.a_class = Stub
2080
 
        module = MyModule()
2081
 
        return loader, module
2082
 
 
2083
 
    def test_module_no_load_tests_attribute_loads_classes(self):
2084
 
        loader, module = self._get_loader_and_module()
2085
 
        self.assertEqual(1, loader.loadTestsFromModule(module).countTestCases())
2086
 
 
2087
 
    def test_module_load_tests_attribute_gets_called(self):
2088
 
        loader, module = self._get_loader_and_module()
2089
 
        # 'self' is here because we're faking the module with a class. Regular
2090
 
        # load_tests do not need that :)
2091
 
        def load_tests(self, standard_tests, module, loader):
2092
 
            result = loader.suiteClass()
2093
 
            for test in iter_suite_tests(standard_tests):
2094
 
                result.addTests([test, test])
2095
 
            return result
2096
 
        # add a load_tests() method which multiplies the tests from the module.
2097
 
        module.__class__.load_tests = load_tests
2098
 
        self.assertEqual(2, loader.loadTestsFromModule(module).countTestCases())
2099
 
 
2100
 
    def test_load_tests_from_module_name_smoke_test(self):
2101
 
        loader = TestUtil.TestLoader()
2102
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2103
 
        self.assertEquals(['bzrlib.tests.test_sampler.DemoTest.test_nothing'],
2104
 
                          _test_ids(suite))
2105
 
 
2106
 
    def test_load_tests_from_module_name_with_bogus_module_name(self):
2107
 
        loader = TestUtil.TestLoader()
2108
 
        self.assertRaises(ImportError, loader.loadTestsFromModuleName, 'bogus')
2109
 
 
2110
 
 
2111
 
class TestTestIdList(tests.TestCase):
2112
 
 
2113
 
    def _create_id_list(self, test_list):
2114
 
        return tests.TestIdList(test_list)
2115
 
 
2116
 
    def _create_suite(self, test_id_list):
2117
 
 
2118
 
        class Stub(TestCase):
2119
 
            def test_foo(self):
2120
 
                pass
2121
 
 
2122
 
        def _create_test_id(id):
2123
 
            return lambda: id
2124
 
 
2125
 
        suite = TestUtil.TestSuite()
2126
 
        for id in test_id_list:
2127
 
            t  = Stub('test_foo')
2128
 
            t.id = _create_test_id(id)
2129
 
            suite.addTest(t)
2130
 
        return suite
2131
 
 
2132
 
    def _test_ids(self, test_suite):
2133
 
        """Get the ids for the tests in a test suite."""
2134
 
        return [t.id() for t in iter_suite_tests(test_suite)]
2135
 
 
2136
 
    def test_empty_list(self):
2137
 
        id_list = self._create_id_list([])
2138
 
        self.assertEquals({}, id_list.tests)
2139
 
        self.assertEquals({}, id_list.modules)
2140
 
 
2141
 
    def test_valid_list(self):
2142
 
        id_list = self._create_id_list(
2143
 
            ['mod1.cl1.meth1', 'mod1.cl1.meth2',
2144
 
             'mod1.func1', 'mod1.cl2.meth2',
2145
 
             'mod1.submod1',
2146
 
             'mod1.submod2.cl1.meth1', 'mod1.submod2.cl2.meth2',
2147
 
             ])
2148
 
        self.assertTrue(id_list.refers_to('mod1'))
2149
 
        self.assertTrue(id_list.refers_to('mod1.submod1'))
2150
 
        self.assertTrue(id_list.refers_to('mod1.submod2'))
2151
 
        self.assertTrue(id_list.includes('mod1.cl1.meth1'))
2152
 
        self.assertTrue(id_list.includes('mod1.submod1'))
2153
 
        self.assertTrue(id_list.includes('mod1.func1'))
2154
 
 
2155
 
    def test_bad_chars_in_params(self):
2156
 
        id_list = self._create_id_list(['mod1.cl1.meth1(xx.yy)'])
2157
 
        self.assertTrue(id_list.refers_to('mod1'))
2158
 
        self.assertTrue(id_list.includes('mod1.cl1.meth1(xx.yy)'))
2159
 
 
2160
 
    def test_module_used(self):
2161
 
        id_list = self._create_id_list(['mod.class.meth'])
2162
 
        self.assertTrue(id_list.refers_to('mod'))
2163
 
        self.assertTrue(id_list.refers_to('mod.class'))
2164
 
        self.assertTrue(id_list.refers_to('mod.class.meth'))
2165
 
 
2166
 
    def test_test_suite(self):
2167
 
        # This test is slow, so we do a single test with one test in each
2168
 
        # category
2169
 
        test_list = [
2170
 
            # testmod_names
2171
 
            'bzrlib.tests.blackbox.test_branch.TestBranch.test_branch',
2172
 
            'bzrlib.tests.test_selftest.TestTestIdList.test_test_suite',
2173
 
            # transport implementations
2174
 
            'bzrlib.tests.test_transport_implementations.TransportTests'
2175
 
            '.test_abspath(LocalURLServer)',
2176
 
            # modules_to_doctest
2177
 
            'bzrlib.timestamp.format_highres_date',
2178
 
            # plugins can't be tested that way since selftest may be run with
2179
 
            # --no-plugins
2180
 
            ]
2181
 
        suite = tests.test_suite(test_list)
2182
 
        self.assertEquals(test_list, _test_ids(suite))
2183
 
 
2184
 
    def test_test_suite_matches_id_list_with_unknown(self):
2185
 
        loader = TestUtil.TestLoader()
2186
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2187
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing',
2188
 
                     'bogus']
2189
 
        not_found, duplicates = tests.suite_matches_id_list(suite, test_list)
2190
 
        self.assertEquals(['bogus'], not_found)
2191
 
        self.assertEquals([], duplicates)
2192
 
 
2193
 
    def test_suite_matches_id_list_with_duplicates(self):
2194
 
        loader = TestUtil.TestLoader()
2195
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2196
 
        dupes = loader.suiteClass()
2197
 
        for test in iter_suite_tests(suite):
2198
 
            dupes.addTest(test)
2199
 
            dupes.addTest(test) # Add it again
2200
 
 
2201
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing',]
2202
 
        not_found, duplicates = tests.suite_matches_id_list(
2203
 
            dupes, test_list)
2204
 
        self.assertEquals([], not_found)
2205
 
        self.assertEquals(['bzrlib.tests.test_sampler.DemoTest.test_nothing'],
2206
 
                          duplicates)
2207
 
 
2208
 
 
2209
 
class TestLoadTestIdList(tests.TestCaseInTempDir):
2210
 
 
2211
 
    def _create_test_list_file(self, file_name, content):
2212
 
        fl = open(file_name, 'wt')
2213
 
        fl.write(content)
2214
 
        fl.close()
2215
 
 
2216
 
    def test_load_unknown(self):
2217
 
        self.assertRaises(errors.NoSuchFile,
2218
 
                          tests.load_test_id_list, 'i_do_not_exist')
2219
 
 
2220
 
    def test_load_test_list(self):
2221
 
        test_list_fname = 'test.list'
2222
 
        self._create_test_list_file(test_list_fname,
2223
 
                                    'mod1.cl1.meth1\nmod2.cl2.meth2\n')
2224
 
        tlist = tests.load_test_id_list(test_list_fname)
2225
 
        self.assertEquals(2, len(tlist))
2226
 
        self.assertEquals('mod1.cl1.meth1', tlist[0])
2227
 
        self.assertEquals('mod2.cl2.meth2', tlist[1])
2228
 
 
2229
 
    def test_load_dirty_file(self):
2230
 
        test_list_fname = 'test.list'
2231
 
        self._create_test_list_file(test_list_fname,
2232
 
                                    '  mod1.cl1.meth1\n\nmod2.cl2.meth2  \n'
2233
 
                                    'bar baz\n')
2234
 
        tlist = tests.load_test_id_list(test_list_fname)
2235
 
        self.assertEquals(4, len(tlist))
2236
 
        self.assertEquals('mod1.cl1.meth1', tlist[0])
2237
 
        self.assertEquals('', tlist[1])
2238
 
        self.assertEquals('mod2.cl2.meth2', tlist[2])
2239
 
        self.assertEquals('bar baz', tlist[3])
2240
 
 
2241
 
 
2242
 
class TestFilteredByModuleTestLoader(tests.TestCase):
2243
 
 
2244
 
    def _create_loader(self, test_list):
2245
 
        id_filter = tests.TestIdList(test_list)
2246
 
        loader = TestUtil.FilteredByModuleTestLoader(id_filter.refers_to)
2247
 
        return loader
2248
 
 
2249
 
    def test_load_tests(self):
2250
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing']
2251
 
        loader = self._create_loader(test_list)
2252
 
 
2253
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2254
 
        self.assertEquals(test_list, _test_ids(suite))
2255
 
 
2256
 
    def test_exclude_tests(self):
2257
 
        test_list = ['bogus']
2258
 
        loader = self._create_loader(test_list)
2259
 
 
2260
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2261
 
        self.assertEquals([], _test_ids(suite))
2262
 
 
2263
 
 
2264
 
class TestFilteredByNameStartTestLoader(tests.TestCase):
2265
 
 
2266
 
    def _create_loader(self, name_start):
2267
 
        def needs_module(name):
2268
 
            return name.startswith(name_start) or name_start.startswith(name)
2269
 
        loader = TestUtil.FilteredByModuleTestLoader(needs_module)
2270
 
        return loader
2271
 
 
2272
 
    def test_load_tests(self):
2273
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing']
2274
 
        loader = self._create_loader('bzrlib.tests.test_samp')
2275
 
 
2276
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2277
 
        self.assertEquals(test_list, _test_ids(suite))
2278
 
 
2279
 
    def test_load_tests_inside_module(self):
2280
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing']
2281
 
        loader = self._create_loader('bzrlib.tests.test_sampler.Demo')
2282
 
 
2283
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2284
 
        self.assertEquals(test_list, _test_ids(suite))
2285
 
 
2286
 
    def test_exclude_tests(self):
2287
 
        test_list = ['bogus']
2288
 
        loader = self._create_loader('bogus')
2289
 
 
2290
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2291
 
        self.assertEquals([], _test_ids(suite))
2292
 
 
2293
 
 
2294
 
class TestTestPrefixRegistry(tests.TestCase):
2295
 
 
2296
 
    def _get_registry(self):
2297
 
        tp_registry = tests.TestPrefixAliasRegistry()
2298
 
        return tp_registry
2299
 
 
2300
 
    def test_register_new_prefix(self):
2301
 
        tpr = self._get_registry()
2302
 
        tpr.register('foo', 'fff.ooo.ooo')
2303
 
        self.assertEquals('fff.ooo.ooo', tpr.get('foo'))
2304
 
 
2305
 
    def test_register_existing_prefix(self):
2306
 
        tpr = self._get_registry()
2307
 
        tpr.register('bar', 'bbb.aaa.rrr')
2308
 
        tpr.register('bar', 'bBB.aAA.rRR')
2309
 
        self.assertEquals('bbb.aaa.rrr', tpr.get('bar'))
2310
 
        self.assertContainsRe(self._get_log(keep_log_file=True),
2311
 
                              r'.*bar.*bbb.aaa.rrr.*bBB.aAA.rRR')
2312
 
 
2313
 
    def test_get_unknown_prefix(self):
2314
 
        tpr = self._get_registry()
2315
 
        self.assertRaises(KeyError, tpr.get, 'I am not a prefix')
2316
 
 
2317
 
    def test_resolve_prefix(self):
2318
 
        tpr = self._get_registry()
2319
 
        tpr.register('bar', 'bb.aa.rr')
2320
 
        self.assertEquals('bb.aa.rr', tpr.resolve_alias('bar'))
2321
 
 
2322
 
    def test_resolve_unknown_alias(self):
2323
 
        tpr = self._get_registry()
2324
 
        self.assertRaises(errors.BzrCommandError,
2325
 
                          tpr.resolve_alias, 'I am not a prefix')
2326
 
 
2327
 
    def test_predefined_prefixes(self):
2328
 
        tpr = tests.test_prefix_alias_registry
2329
 
        self.assertEquals('bzrlib', tpr.resolve_alias('bzrlib'))
2330
 
        self.assertEquals('bzrlib.doc', tpr.resolve_alias('bd'))
2331
 
        self.assertEquals('bzrlib.utils', tpr.resolve_alias('bu'))
2332
 
        self.assertEquals('bzrlib.tests', tpr.resolve_alias('bt'))
2333
 
        self.assertEquals('bzrlib.tests.blackbox', tpr.resolve_alias('bb'))
2334
 
        self.assertEquals('bzrlib.plugins', tpr.resolve_alias('bp'))
2335
 
 
2336
 
 
2337
 
class TestRunSuite(TestCase):
2338
 
 
2339
 
    def test_runner_class(self):
2340
 
        """run_suite accepts and uses a runner_class keyword argument."""
2341
 
        class Stub(TestCase):
2342
 
            def test_foo(self):
2343
 
                pass
2344
 
        suite = Stub("test_foo")
2345
 
        calls = []
2346
 
        class MyRunner(TextTestRunner):
2347
 
            def run(self, test):
2348
 
                calls.append(test)
2349
 
                return ExtendedTestResult(self.stream, self.descriptions,
2350
 
                    self.verbosity)
2351
 
        run_suite(suite, runner_class=MyRunner)
2352
 
        self.assertEqual(calls, [suite])