~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_selftest.py

  • Committer: Wouter van Heyst
  • Date: 2006-06-07 16:05:27 UTC
  • mto: This revision was merged to the branch mainline in revision 1752.
  • Revision ID: larstiq@larstiq.dyndns.org-20060607160527-2b3649154d0e2e84
more code cleanup

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2005, 2006 by Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
 
4
# it under the terms of the GNU General Public License version 2 as published by
 
5
# the Free Software Foundation.
7
6
#
8
7
# This program is distributed in the hope that it will be useful,
9
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
16
15
 
17
16
"""Tests for the test framework."""
18
17
 
19
 
import cStringIO
20
18
import os
21
19
from StringIO import StringIO
22
20
import sys
25
23
import warnings
26
24
 
27
25
import bzrlib
28
 
from bzrlib import (
29
 
    bzrdir,
30
 
    errors,
31
 
    memorytree,
32
 
    osutils,
33
 
    repository,
34
 
    symbol_versioning,
35
 
    )
36
26
from bzrlib.progress import _BaseProgressBar
37
 
from bzrlib.repofmt import weaverepo
38
 
from bzrlib.symbol_versioning import zero_ten, zero_eleven
39
27
from bzrlib.tests import (
40
28
                          ChrootedTestCase,
41
 
                          ExtendedTestResult,
42
 
                          Feature,
43
 
                          KnownFailure,
44
29
                          TestCase,
45
30
                          TestCaseInTempDir,
46
 
                          TestCaseWithMemoryTransport,
47
31
                          TestCaseWithTransport,
48
32
                          TestSkipped,
49
33
                          TestSuite,
50
 
                          TestUtil,
51
34
                          TextTestRunner,
52
 
                          UnavailableFeature,
53
 
                          clean_selftest_output,
54
 
                          iter_suite_tests,
55
 
                          filter_suite_by_re,
56
 
                          sort_suite_by_re,
57
 
                          test_lsprof,
58
 
                          test_suite,
59
35
                          )
60
 
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
61
36
from bzrlib.tests.TestUtil import _load_module_by_name
62
 
from bzrlib.trace import note
63
 
from bzrlib.transport.memory import MemoryServer, MemoryTransport
64
 
from bzrlib.version import _get_bzr_source_tree
 
37
import bzrlib.errors as errors
65
38
 
66
39
 
67
40
class SelftestTests(TestCase):
75
48
                          _load_module_by_name,
76
49
                          'bzrlib.no-name-yet')
77
50
 
 
51
 
78
52
class MetaTestLog(TestCase):
79
53
 
80
54
    def test_logging(self):
81
55
        """Test logs are captured when a test fails."""
82
56
        self.log('a test message')
83
57
        self._log_file.flush()
84
 
        self.assertContainsRe(self._get_log(keep_log_file=True),
85
 
                              'a test message\n')
 
58
        self.assertContainsRe(self._get_log(), 'a test message\n')
86
59
 
87
60
 
88
61
class TestTreeShape(TestCaseInTempDir):
114
87
            def get_test_permutations(self):
115
88
                return sample_permutation
116
89
        sample_permutation = [(1,2), (3,4)]
117
 
        from bzrlib.tests.test_transport_implementations \
118
 
            import TransportTestProviderAdapter
 
90
        from bzrlib.transport import TransportTestProviderAdapter
119
91
        adapter = TransportTestProviderAdapter()
120
92
        self.assertEqual(sample_permutation,
121
93
                         adapter.get_transport_test_permutations(MockModule()))
126
98
        # - we assume if this matches its probably doing the right thing
127
99
        # especially in combination with the tests for setting the right
128
100
        # classes below.
129
 
        from bzrlib.tests.test_transport_implementations \
130
 
            import TransportTestProviderAdapter
131
 
        from bzrlib.transport import _get_transport_modules
 
101
        from bzrlib.transport import (TransportTestProviderAdapter,
 
102
                                      _get_transport_modules
 
103
                                      )
132
104
        modules = _get_transport_modules()
133
105
        permutation_count = 0
134
106
        for module in modules:
151
123
        # This test used to know about all the possible transports and the
152
124
        # order they were returned but that seems overly brittle (mbp
153
125
        # 20060307)
154
 
        from bzrlib.tests.test_transport_implementations \
155
 
            import TransportTestProviderAdapter
156
 
        scenarios = TransportTestProviderAdapter().scenarios
 
126
        input_test = TestTransportProviderAdapter(
 
127
            "test_adapter_sets_transport_class")
 
128
        from bzrlib.transport import TransportTestProviderAdapter
 
129
        suite = TransportTestProviderAdapter().adapt(input_test)
 
130
        tests = list(iter(suite))
 
131
        self.assertTrue(len(tests) > 6)
157
132
        # there are at least that many builtin transports
158
 
        self.assertTrue(len(scenarios) > 6)
159
 
        one_scenario = scenarios[0]
160
 
        self.assertIsInstance(one_scenario[0], str)
161
 
        self.assertTrue(issubclass(one_scenario[1]["transport_class"],
 
133
        one_test = tests[0]
 
134
        self.assertTrue(issubclass(one_test.transport_class, 
162
135
                                   bzrlib.transport.Transport))
163
 
        self.assertTrue(issubclass(one_scenario[1]["transport_server"],
 
136
        self.assertTrue(issubclass(one_test.transport_server, 
164
137
                                   bzrlib.transport.Server))
165
138
 
166
139
 
167
140
class TestBranchProviderAdapter(TestCase):
168
141
    """A group of tests that test the branch implementation test adapter."""
169
142
 
170
 
    def test_constructor(self):
 
143
    def test_adapted_tests(self):
171
144
        # check that constructor parameters are passed through to the adapted
172
145
        # test.
173
 
        from bzrlib.tests.branch_implementations import BranchTestProviderAdapter
 
146
        from bzrlib.branch import BranchTestProviderAdapter
 
147
        input_test = TestBranchProviderAdapter(
 
148
            "test_adapted_tests")
174
149
        server1 = "a"
175
150
        server2 = "b"
176
151
        formats = [("c", "C"), ("d", "D")]
177
152
        adapter = BranchTestProviderAdapter(server1, server2, formats)
178
 
        self.assertEqual(2, len(adapter.scenarios))
179
 
        self.assertEqual([
180
 
            ('str',
181
 
             {'branch_format': 'c',
182
 
              'bzrdir_format': 'C',
183
 
              'transport_readonly_server': 'b',
184
 
              'transport_server': 'a'}),
185
 
            ('str',
186
 
             {'branch_format': 'd',
187
 
              'bzrdir_format': 'D',
188
 
              'transport_readonly_server': 'b',
189
 
              'transport_server': 'a'})],
190
 
            adapter.scenarios)
 
153
        suite = adapter.adapt(input_test)
 
154
        tests = list(iter(suite))
 
155
        self.assertEqual(2, len(tests))
 
156
        self.assertEqual(tests[0].branch_format, formats[0][0])
 
157
        self.assertEqual(tests[0].bzrdir_format, formats[0][1])
 
158
        self.assertEqual(tests[0].transport_server, server1)
 
159
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
160
        self.assertEqual(tests[1].branch_format, formats[1][0])
 
161
        self.assertEqual(tests[1].bzrdir_format, formats[1][1])
 
162
        self.assertEqual(tests[1].transport_server, server1)
 
163
        self.assertEqual(tests[1].transport_readonly_server, server2)
191
164
 
192
165
 
193
166
class TestBzrDirProviderAdapter(TestCase):
196
169
    def test_adapted_tests(self):
197
170
        # check that constructor parameters are passed through to the adapted
198
171
        # test.
199
 
        from bzrlib.tests.bzrdir_implementations import BzrDirTestProviderAdapter
200
 
        vfs_factory = "v"
 
172
        from bzrlib.bzrdir import BzrDirTestProviderAdapter
 
173
        input_test = TestBzrDirProviderAdapter(
 
174
            "test_adapted_tests")
201
175
        server1 = "a"
202
176
        server2 = "b"
203
177
        formats = ["c", "d"]
204
 
        adapter = BzrDirTestProviderAdapter(vfs_factory,
205
 
            server1, server2, formats)
206
 
        self.assertEqual([
207
 
            ('str',
208
 
             {'bzrdir_format': 'c',
209
 
              'transport_readonly_server': 'b',
210
 
              'transport_server': 'a',
211
 
              'vfs_transport_factory': 'v'}),
212
 
            ('str',
213
 
             {'bzrdir_format': 'd',
214
 
              'transport_readonly_server': 'b',
215
 
              'transport_server': 'a',
216
 
              'vfs_transport_factory': 'v'})],
217
 
            adapter.scenarios)
 
178
        adapter = BzrDirTestProviderAdapter(server1, server2, formats)
 
179
        suite = adapter.adapt(input_test)
 
180
        tests = list(iter(suite))
 
181
        self.assertEqual(2, len(tests))
 
182
        self.assertEqual(tests[0].bzrdir_format, formats[0])
 
183
        self.assertEqual(tests[0].transport_server, server1)
 
184
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
185
        self.assertEqual(tests[1].bzrdir_format, formats[1])
 
186
        self.assertEqual(tests[1].transport_server, server1)
 
187
        self.assertEqual(tests[1].transport_readonly_server, server2)
218
188
 
219
189
 
220
190
class TestRepositoryProviderAdapter(TestCase):
221
191
    """A group of tests that test the repository implementation test adapter."""
222
192
 
223
 
    def test_constructor(self):
224
 
        # check that constructor parameters are passed through to the
225
 
        # scenarios.
226
 
        from bzrlib.tests.repository_implementations import RepositoryTestProviderAdapter
 
193
    def test_adapted_tests(self):
 
194
        # check that constructor parameters are passed through to the adapted
 
195
        # test.
 
196
        from bzrlib.repository import RepositoryTestProviderAdapter
 
197
        input_test = TestRepositoryProviderAdapter(
 
198
            "test_adapted_tests")
227
199
        server1 = "a"
228
200
        server2 = "b"
229
201
        formats = [("c", "C"), ("d", "D")]
230
202
        adapter = RepositoryTestProviderAdapter(server1, server2, formats)
231
 
        self.assertEqual([
232
 
            ('str',
233
 
             {'bzrdir_format': 'C',
234
 
              'repository_format': 'c',
235
 
              'transport_readonly_server': 'b',
236
 
              'transport_server': 'a'}),
237
 
            ('str',
238
 
             {'bzrdir_format': 'D',
239
 
              'repository_format': 'd',
240
 
              'transport_readonly_server': 'b',
241
 
              'transport_server': 'a'})],
242
 
            adapter.scenarios)
243
 
 
244
 
    def test_setting_vfs_transport(self):
245
 
        """The vfs_transport_factory can be set optionally."""
246
 
        from bzrlib.tests.repository_implementations import RepositoryTestProviderAdapter
247
 
        formats = [("a", "b"), ("c", "d")]
248
 
        adapter = RepositoryTestProviderAdapter(None, None, formats,
249
 
            vfs_transport_factory="vfs")
250
 
        self.assertEqual([
251
 
            ('str',
252
 
             {'bzrdir_format': 'b',
253
 
              'repository_format': 'a',
254
 
              'transport_readonly_server': None,
255
 
              'transport_server': None,
256
 
              'vfs_transport_factory': 'vfs'}),
257
 
            ('str',
258
 
             {'bzrdir_format': 'd',
259
 
              'repository_format': 'c',
260
 
              'transport_readonly_server': None,
261
 
              'transport_server': None,
262
 
              'vfs_transport_factory': 'vfs'})],
263
 
            adapter.scenarios)
264
 
 
265
 
    def test_formats_to_scenarios(self):
266
 
        """The adapter can generate all the scenarios needed."""
267
 
        from bzrlib.tests.repository_implementations import RepositoryTestProviderAdapter
268
 
        no_vfs_adapter = RepositoryTestProviderAdapter("server", "readonly",
269
 
            [], None)
270
 
        vfs_adapter = RepositoryTestProviderAdapter("server", "readonly",
271
 
            [], vfs_transport_factory="vfs")
272
 
        # no_vfs generate scenarios without vfs_transport_factor
273
 
        formats = [("c", "C"), (1, "D")]
274
 
        self.assertEqual([
275
 
            ('str',
276
 
             {'bzrdir_format': 'C',
277
 
              'repository_format': 'c',
278
 
              'transport_readonly_server': 'readonly',
279
 
              'transport_server': 'server'}),
280
 
            ('int',
281
 
             {'bzrdir_format': 'D',
282
 
              'repository_format': 1,
283
 
              'transport_readonly_server': 'readonly',
284
 
              'transport_server': 'server'})],
285
 
            no_vfs_adapter.formats_to_scenarios(formats))
286
 
        self.assertEqual([
287
 
            ('str',
288
 
             {'bzrdir_format': 'C',
289
 
              'repository_format': 'c',
290
 
              'transport_readonly_server': 'readonly',
291
 
              'transport_server': 'server',
292
 
              'vfs_transport_factory': 'vfs'}),
293
 
            ('int',
294
 
             {'bzrdir_format': 'D',
295
 
              'repository_format': 1,
296
 
              'transport_readonly_server': 'readonly',
297
 
              'transport_server': 'server',
298
 
              'vfs_transport_factory': 'vfs'})],
299
 
            vfs_adapter.formats_to_scenarios(formats))
300
 
 
301
 
 
302
 
class TestTestScenarioApplier(TestCase):
303
 
    """Tests for the test adaption facilities."""
304
 
 
305
 
    def test_adapt_applies_scenarios(self):
306
 
        from bzrlib.tests.repository_implementations import TestScenarioApplier
307
 
        input_test = TestTestScenarioApplier("test_adapt_test_to_scenario")
308
 
        adapter = TestScenarioApplier()
309
 
        adapter.scenarios = [("1", "dict"), ("2", "settings")]
310
 
        calls = []
311
 
        def capture_call(test, scenario):
312
 
            calls.append((test, scenario))
313
 
            return test
314
 
        adapter.adapt_test_to_scenario = capture_call
315
 
        adapter.adapt(input_test)
316
 
        self.assertEqual([(input_test, ("1", "dict")),
317
 
            (input_test, ("2", "settings"))], calls)
318
 
 
319
 
    def test_adapt_test_to_scenario(self):
320
 
        from bzrlib.tests.repository_implementations import TestScenarioApplier
321
 
        input_test = TestTestScenarioApplier("test_adapt_test_to_scenario")
322
 
        adapter = TestScenarioApplier()
323
 
        # setup two adapted tests
324
 
        adapted_test1 = adapter.adapt_test_to_scenario(input_test,
325
 
            ("new id",
326
 
            {"bzrdir_format":"bzr_format",
327
 
             "repository_format":"repo_fmt",
328
 
             "transport_server":"transport_server",
329
 
             "transport_readonly_server":"readonly-server"}))
330
 
        adapted_test2 = adapter.adapt_test_to_scenario(input_test,
331
 
            ("new id 2", {"bzrdir_format":None}))
332
 
        # input_test should have been altered.
333
 
        self.assertRaises(AttributeError, getattr, input_test, "bzrdir_format")
334
 
        # the new tests are mutually incompatible, ensuring it has 
335
 
        # made new ones, and unspecified elements in the scenario
336
 
        # should not have been altered.
337
 
        self.assertEqual("bzr_format", adapted_test1.bzrdir_format)
338
 
        self.assertEqual("repo_fmt", adapted_test1.repository_format)
339
 
        self.assertEqual("transport_server", adapted_test1.transport_server)
340
 
        self.assertEqual("readonly-server",
341
 
            adapted_test1.transport_readonly_server)
342
 
        self.assertEqual(
343
 
            "bzrlib.tests.test_selftest.TestTestScenarioApplier."
344
 
            "test_adapt_test_to_scenario(new id)",
345
 
            adapted_test1.id())
346
 
        self.assertEqual(None, adapted_test2.bzrdir_format)
347
 
        self.assertEqual(
348
 
            "bzrlib.tests.test_selftest.TestTestScenarioApplier."
349
 
            "test_adapt_test_to_scenario(new id 2)",
350
 
            adapted_test2.id())
 
203
        suite = adapter.adapt(input_test)
 
204
        tests = list(iter(suite))
 
205
        self.assertEqual(2, len(tests))
 
206
        self.assertEqual(tests[0].bzrdir_format, formats[0][1])
 
207
        self.assertEqual(tests[0].repository_format, formats[0][0])
 
208
        self.assertEqual(tests[0].transport_server, server1)
 
209
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
210
        self.assertEqual(tests[1].bzrdir_format, formats[1][1])
 
211
        self.assertEqual(tests[1].repository_format, formats[1][0])
 
212
        self.assertEqual(tests[1].transport_server, server1)
 
213
        self.assertEqual(tests[1].transport_readonly_server, server2)
351
214
 
352
215
 
353
216
class TestInterRepositoryProviderAdapter(TestCase):
356
219
    def test_adapted_tests(self):
357
220
        # check that constructor parameters are passed through to the adapted
358
221
        # test.
359
 
        from bzrlib.tests.interrepository_implementations import \
360
 
            InterRepositoryTestProviderAdapter
 
222
        from bzrlib.repository import InterRepositoryTestProviderAdapter
 
223
        input_test = TestInterRepositoryProviderAdapter(
 
224
            "test_adapted_tests")
361
225
        server1 = "a"
362
226
        server2 = "b"
363
227
        formats = [(str, "C1", "C2"), (int, "D1", "D2")]
364
228
        adapter = InterRepositoryTestProviderAdapter(server1, server2, formats)
365
 
        self.assertEqual([
366
 
            ('str',
367
 
             {'interrepo_class': str,
368
 
              'repository_format': 'C1',
369
 
              'repository_format_to': 'C2',
370
 
              'transport_readonly_server': 'b',
371
 
              'transport_server': 'a'}),
372
 
            ('int',
373
 
             {'interrepo_class': int,
374
 
              'repository_format': 'D1',
375
 
              'repository_format_to': 'D2',
376
 
              'transport_readonly_server': 'b',
377
 
              'transport_server': 'a'})],
378
 
            adapter.formats_to_scenarios(formats))
 
229
        suite = adapter.adapt(input_test)
 
230
        tests = list(iter(suite))
 
231
        self.assertEqual(2, len(tests))
 
232
        self.assertEqual(tests[0].interrepo_class, formats[0][0])
 
233
        self.assertEqual(tests[0].repository_format, formats[0][1])
 
234
        self.assertEqual(tests[0].repository_format_to, formats[0][2])
 
235
        self.assertEqual(tests[0].transport_server, server1)
 
236
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
237
        self.assertEqual(tests[1].interrepo_class, formats[1][0])
 
238
        self.assertEqual(tests[1].repository_format, formats[1][1])
 
239
        self.assertEqual(tests[1].repository_format_to, formats[1][2])
 
240
        self.assertEqual(tests[1].transport_server, server1)
 
241
        self.assertEqual(tests[1].transport_readonly_server, server2)
379
242
 
380
243
 
381
244
class TestInterVersionedFileProviderAdapter(TestCase):
382
245
    """A group of tests that test the InterVersionedFile test adapter."""
383
246
 
384
 
    def test_scenarios(self):
 
247
    def test_adapted_tests(self):
385
248
        # check that constructor parameters are passed through to the adapted
386
249
        # test.
387
 
        from bzrlib.tests.interversionedfile_implementations \
388
 
            import InterVersionedFileTestProviderAdapter
 
250
        from bzrlib.versionedfile import InterVersionedFileTestProviderAdapter
 
251
        input_test = TestInterRepositoryProviderAdapter(
 
252
            "test_adapted_tests")
389
253
        server1 = "a"
390
254
        server2 = "b"
391
255
        formats = [(str, "C1", "C2"), (int, "D1", "D2")]
392
256
        adapter = InterVersionedFileTestProviderAdapter(server1, server2, formats)
393
 
        self.assertEqual([
394
 
            ('str',
395
 
             {'interversionedfile_class':str,
396
 
              'transport_readonly_server': 'b',
397
 
              'transport_server': 'a',
398
 
              'versionedfile_factory': 'C1',
399
 
              'versionedfile_factory_to': 'C2'}),
400
 
            ('int',
401
 
             {'interversionedfile_class': int,
402
 
              'transport_readonly_server': 'b',
403
 
              'transport_server': 'a',
404
 
              'versionedfile_factory': 'D1',
405
 
              'versionedfile_factory_to': 'D2'})],
406
 
            adapter.scenarios)
 
257
        suite = adapter.adapt(input_test)
 
258
        tests = list(iter(suite))
 
259
        self.assertEqual(2, len(tests))
 
260
        self.assertEqual(tests[0].interversionedfile_class, formats[0][0])
 
261
        self.assertEqual(tests[0].versionedfile_factory, formats[0][1])
 
262
        self.assertEqual(tests[0].versionedfile_factory_to, formats[0][2])
 
263
        self.assertEqual(tests[0].transport_server, server1)
 
264
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
265
        self.assertEqual(tests[1].interversionedfile_class, formats[1][0])
 
266
        self.assertEqual(tests[1].versionedfile_factory, formats[1][1])
 
267
        self.assertEqual(tests[1].versionedfile_factory_to, formats[1][2])
 
268
        self.assertEqual(tests[1].transport_server, server1)
 
269
        self.assertEqual(tests[1].transport_readonly_server, server2)
407
270
 
408
271
 
409
272
class TestRevisionStoreProviderAdapter(TestCase):
410
273
    """A group of tests that test the RevisionStore test adapter."""
411
274
 
412
 
    def test_scenarios(self):
 
275
    def test_adapted_tests(self):
413
276
        # check that constructor parameters are passed through to the adapted
414
277
        # test.
415
 
        from bzrlib.tests.revisionstore_implementations \
416
 
            import RevisionStoreTestProviderAdapter
 
278
        from bzrlib.store.revision import RevisionStoreTestProviderAdapter
 
279
        input_test = TestRevisionStoreProviderAdapter(
 
280
            "test_adapted_tests")
417
281
        # revision stores need a store factory - i.e. RevisionKnit
418
282
        #, a readonly and rw transport 
419
283
        # transport servers:
421
285
        server2 = "b"
422
286
        store_factories = ["c", "d"]
423
287
        adapter = RevisionStoreTestProviderAdapter(server1, server2, store_factories)
424
 
        self.assertEqual([
425
 
            ('c',
426
 
             {'store_factory': 'c',
427
 
              'transport_readonly_server': 'b',
428
 
              'transport_server': 'a'}),
429
 
            ('d',
430
 
             {'store_factory': 'd',
431
 
              'transport_readonly_server': 'b',
432
 
              'transport_server': 'a'})],
433
 
            adapter.scenarios)
 
288
        suite = adapter.adapt(input_test)
 
289
        tests = list(iter(suite))
 
290
        self.assertEqual(2, len(tests))
 
291
        self.assertEqual(tests[0].store_factory, store_factories[0][0])
 
292
        self.assertEqual(tests[0].transport_server, server1)
 
293
        self.assertEqual(tests[0].transport_readonly_server, server2)
 
294
        self.assertEqual(tests[1].store_factory, store_factories[1][0])
 
295
        self.assertEqual(tests[1].transport_server, server1)
 
296
        self.assertEqual(tests[1].transport_readonly_server, server2)
434
297
 
435
298
 
436
299
class TestWorkingTreeProviderAdapter(TestCase):
437
300
    """A group of tests that test the workingtree implementation test adapter."""
438
301
 
439
 
    def test_scenarios(self):
 
302
    def test_adapted_tests(self):
440
303
        # check that constructor parameters are passed through to the adapted
441
304
        # test.
442
 
        from bzrlib.tests.workingtree_implementations \
443
 
            import WorkingTreeTestProviderAdapter
 
305
        from bzrlib.workingtree import WorkingTreeTestProviderAdapter
 
306
        input_test = TestWorkingTreeProviderAdapter(
 
307
            "test_adapted_tests")
444
308
        server1 = "a"
445
309
        server2 = "b"
446
310
        formats = [("c", "C"), ("d", "D")]
447
311
        adapter = WorkingTreeTestProviderAdapter(server1, server2, formats)
448
 
        self.assertEqual([
449
 
            ('str',
450
 
             {'bzrdir_format': 'C',
451
 
              'transport_readonly_server': 'b',
452
 
              'transport_server': 'a',
453
 
              'workingtree_format': 'c'}),
454
 
            ('str',
455
 
             {'bzrdir_format': 'D',
456
 
              'transport_readonly_server': 'b',
457
 
              'transport_server': 'a',
458
 
              'workingtree_format': 'd'})],
459
 
            adapter.scenarios)
460
 
 
461
 
 
462
 
class TestTreeProviderAdapter(TestCase):
463
 
    """Test the setup of tree_implementation tests."""
464
 
 
465
 
    def test_adapted_tests(self):
466
 
        # the tree implementation adapter is meant to setup one instance for
467
 
        # each working tree format, and one additional instance that will
468
 
        # use the default wt format, but create a revision tree for the tests.
469
 
        # this means that the wt ones should have the workingtree_to_test_tree
470
 
        # attribute set to 'return_parameter' and the revision one set to
471
 
        # revision_tree_from_workingtree.
472
 
 
473
 
        from bzrlib.tests.tree_implementations import (
474
 
            TreeTestProviderAdapter,
475
 
            return_parameter,
476
 
            revision_tree_from_workingtree
477
 
            )
478
 
        from bzrlib.workingtree import WorkingTreeFormat, WorkingTreeFormat3
479
 
        input_test = TestTreeProviderAdapter(
480
 
            "test_adapted_tests")
481
 
        server1 = "a"
482
 
        server2 = "b"
483
 
        formats = [("c", "C"), ("d", "D")]
484
 
        adapter = TreeTestProviderAdapter(server1, server2, formats)
485
312
        suite = adapter.adapt(input_test)
486
313
        tests = list(iter(suite))
487
 
        self.assertEqual(4, len(tests))
488
 
        # this must match the default format setp up in
489
 
        # TreeTestProviderAdapter.adapt
490
 
        default_format = WorkingTreeFormat3
 
314
        self.assertEqual(2, len(tests))
491
315
        self.assertEqual(tests[0].workingtree_format, formats[0][0])
492
316
        self.assertEqual(tests[0].bzrdir_format, formats[0][1])
493
317
        self.assertEqual(tests[0].transport_server, server1)
494
318
        self.assertEqual(tests[0].transport_readonly_server, server2)
495
 
        self.assertEqual(tests[0].workingtree_to_test_tree, return_parameter)
496
319
        self.assertEqual(tests[1].workingtree_format, formats[1][0])
497
320
        self.assertEqual(tests[1].bzrdir_format, formats[1][1])
498
321
        self.assertEqual(tests[1].transport_server, server1)
499
322
        self.assertEqual(tests[1].transport_readonly_server, server2)
500
 
        self.assertEqual(tests[1].workingtree_to_test_tree, return_parameter)
501
 
        self.assertIsInstance(tests[2].workingtree_format, default_format)
502
 
        #self.assertEqual(tests[2].bzrdir_format,
503
 
        #                 default_format._matchingbzrdir)
504
 
        self.assertEqual(tests[2].transport_server, server1)
505
 
        self.assertEqual(tests[2].transport_readonly_server, server2)
506
 
        self.assertEqual(tests[2].workingtree_to_test_tree,
507
 
            revision_tree_from_workingtree)
508
 
 
509
 
 
510
 
class TestInterTreeProviderAdapter(TestCase):
511
 
    """A group of tests that test the InterTreeTestAdapter."""
512
 
 
513
 
    def test_adapted_tests(self):
514
 
        # check that constructor parameters are passed through to the adapted
515
 
        # test.
516
 
        # for InterTree tests we want the machinery to bring up two trees in
517
 
        # each instance: the base one, and the one we are interacting with.
518
 
        # because each optimiser can be direction specific, we need to test
519
 
        # each optimiser in its chosen direction.
520
 
        # unlike the TestProviderAdapter we dont want to automatically add a
521
 
        # parameterised one for WorkingTree - the optimisers will tell us what
522
 
        # ones to add.
523
 
        from bzrlib.tests.tree_implementations import (
524
 
            return_parameter,
525
 
            revision_tree_from_workingtree
526
 
            )
527
 
        from bzrlib.tests.intertree_implementations import (
528
 
            InterTreeTestProviderAdapter,
529
 
            )
530
 
        from bzrlib.workingtree import WorkingTreeFormat2, WorkingTreeFormat3
531
 
        input_test = TestInterTreeProviderAdapter(
532
 
            "test_adapted_tests")
533
 
        server1 = "a"
534
 
        server2 = "b"
535
 
        format1 = WorkingTreeFormat2()
536
 
        format2 = WorkingTreeFormat3()
537
 
        formats = [(str, format1, format2, "converter1"),
538
 
            (int, format2, format1, "converter2")]
539
 
        adapter = InterTreeTestProviderAdapter(server1, server2, formats)
540
 
        suite = adapter.adapt(input_test)
541
 
        tests = list(iter(suite))
542
 
        self.assertEqual(2, len(tests))
543
 
        self.assertEqual(tests[0].intertree_class, formats[0][0])
544
 
        self.assertEqual(tests[0].workingtree_format, formats[0][1])
545
 
        self.assertEqual(tests[0].workingtree_format_to, formats[0][2])
546
 
        self.assertEqual(tests[0].mutable_trees_to_test_trees, formats[0][3])
547
 
        self.assertEqual(tests[0].workingtree_to_test_tree, return_parameter)
548
 
        self.assertEqual(tests[0].transport_server, server1)
549
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
550
 
        self.assertEqual(tests[1].intertree_class, formats[1][0])
551
 
        self.assertEqual(tests[1].workingtree_format, formats[1][1])
552
 
        self.assertEqual(tests[1].workingtree_format_to, formats[1][2])
553
 
        self.assertEqual(tests[1].mutable_trees_to_test_trees, formats[1][3])
554
 
        self.assertEqual(tests[1].workingtree_to_test_tree, return_parameter)
555
 
        self.assertEqual(tests[1].transport_server, server1)
556
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
557
 
 
558
 
 
559
 
class TestTestCaseInTempDir(TestCaseInTempDir):
560
 
 
561
 
    def test_home_is_not_working(self):
562
 
        self.assertNotEqual(self.test_dir, self.test_home_dir)
563
 
        cwd = osutils.getcwd()
564
 
        self.assertEqual(self.test_dir, cwd)
565
 
        self.assertEqual(self.test_home_dir, os.environ['HOME'])
566
 
 
567
 
 
568
 
class TestTestCaseWithMemoryTransport(TestCaseWithMemoryTransport):
569
 
 
570
 
    def test_home_is_non_existant_dir_under_root(self):
571
 
        """The test_home_dir for TestCaseWithMemoryTransport is missing.
572
 
 
573
 
        This is because TestCaseWithMemoryTransport is for tests that do not
574
 
        need any disk resources: they should be hooked into bzrlib in such a 
575
 
        way that no global settings are being changed by the test (only a 
576
 
        few tests should need to do that), and having a missing dir as home is
577
 
        an effective way to ensure that this is the case.
578
 
        """
579
 
        self.assertEqual(self.TEST_ROOT + "/MemoryTransportMissingHomeDir",
580
 
            self.test_home_dir)
581
 
        self.assertEqual(self.test_home_dir, os.environ['HOME'])
582
 
        
583
 
    def test_cwd_is_TEST_ROOT(self):
584
 
        self.assertEqual(self.test_dir, self.TEST_ROOT)
585
 
        cwd = osutils.getcwd()
586
 
        self.assertEqual(self.test_dir, cwd)
587
 
 
588
 
    def test_make_branch_and_memory_tree(self):
589
 
        """In TestCaseWithMemoryTransport we should not make the branch on disk.
590
 
 
591
 
        This is hard to comprehensively robustly test, so we settle for making
592
 
        a branch and checking no directory was created at its relpath.
593
 
        """
594
 
        tree = self.make_branch_and_memory_tree('dir')
595
 
        # Guard against regression into MemoryTransport leaking
596
 
        # files to disk instead of keeping them in memory.
597
 
        self.failIf(osutils.lexists('dir'))
598
 
        self.assertIsInstance(tree, memorytree.MemoryTree)
599
 
 
600
 
    def test_make_branch_and_memory_tree_with_format(self):
601
 
        """make_branch_and_memory_tree should accept a format option."""
602
 
        format = bzrdir.BzrDirMetaFormat1()
603
 
        format.repository_format = weaverepo.RepositoryFormat7()
604
 
        tree = self.make_branch_and_memory_tree('dir', format=format)
605
 
        # Guard against regression into MemoryTransport leaking
606
 
        # files to disk instead of keeping them in memory.
607
 
        self.failIf(osutils.lexists('dir'))
608
 
        self.assertIsInstance(tree, memorytree.MemoryTree)
609
 
        self.assertEqual(format.repository_format.__class__,
610
 
            tree.branch.repository._format.__class__)
611
323
 
612
324
 
613
325
class TestTestCaseWithTransport(TestCaseWithTransport):
617
329
        from bzrlib.transport import get_transport
618
330
        from bzrlib.transport.memory import MemoryServer
619
331
        from bzrlib.transport.readonly import ReadonlyTransportDecorator
620
 
        self.vfs_transport_factory = MemoryServer
 
332
        self.transport_server = MemoryServer
621
333
        self.transport_readonly_server = None
622
334
        # calling get_readonly_transport() constructs a decorator on the url
623
335
        # for the server
630
342
        self.assertEqual(t2.base[:-1], t.abspath('foo/bar'))
631
343
 
632
344
    def test_get_readonly_url_http(self):
633
 
        from bzrlib.tests.HttpServer import HttpServer
634
345
        from bzrlib.transport import get_transport
635
 
        from bzrlib.transport.local import LocalURLServer
636
 
        from bzrlib.transport.http import HttpTransportBase
637
 
        self.transport_server = LocalURLServer
 
346
        from bzrlib.transport.local import LocalRelpathServer
 
347
        from bzrlib.transport.http import HttpServer, HttpTransportBase
 
348
        self.transport_server = LocalRelpathServer
638
349
        self.transport_readonly_server = HttpServer
639
350
        # calling get_readonly_transport() gives us a HTTP server instance.
640
351
        url = self.get_readonly_url()
655
366
        self.assertRaises(AssertionError, self.assertIsDirectory, 'not_here', t)
656
367
 
657
368
 
658
 
class TestTestCaseTransports(TestCaseWithTransport):
659
 
 
660
 
    def setUp(self):
661
 
        super(TestTestCaseTransports, self).setUp()
662
 
        self.vfs_transport_factory = MemoryServer
663
 
 
664
 
    def test_make_bzrdir_preserves_transport(self):
665
 
        t = self.get_transport()
666
 
        result_bzrdir = self.make_bzrdir('subdir')
667
 
        self.assertIsInstance(result_bzrdir.transport, 
668
 
                              MemoryTransport)
669
 
        # should not be on disk, should only be in memory
670
 
        self.failIfExists('subdir')
671
 
 
672
 
 
673
369
class TestChrootedTest(ChrootedTestCase):
674
370
 
675
371
    def test_root_is_root(self):
698
394
    def clear(self):
699
395
        self.calls.append(('clear',))
700
396
 
701
 
    def note(self, msg, *args):
702
 
        self.calls.append(('note', msg, args))
703
 
 
704
397
 
705
398
class TestTestResult(TestCase):
706
399
 
 
400
    def test_progress_bar_style_quiet(self):
 
401
        # test using a progress bar.
 
402
        dummy_test = TestTestResult('test_progress_bar_style_quiet')
 
403
        dummy_error = (Exception, None, [])
 
404
        mypb = MockProgress()
 
405
        mypb.update('Running tests', 0, 4)
 
406
        last_calls = mypb.calls[:]
 
407
        result = bzrlib.tests._MyResult(self._log_file,
 
408
                                        descriptions=0,
 
409
                                        verbosity=1,
 
410
                                        pb=mypb)
 
411
        self.assertEqual(last_calls, mypb.calls)
 
412
 
 
413
        # an error 
 
414
        result.startTest(dummy_test)
 
415
        # starting a test prints the test name
 
416
        self.assertEqual(last_calls + [('update', '...tyle_quiet', 0, None)], mypb.calls)
 
417
        last_calls = mypb.calls[:]
 
418
        result.addError(dummy_test, dummy_error)
 
419
        self.assertEqual(last_calls + [('update', 'ERROR        ', 1, None)], mypb.calls)
 
420
        last_calls = mypb.calls[:]
 
421
 
 
422
        # a failure
 
423
        result.startTest(dummy_test)
 
424
        self.assertEqual(last_calls + [('update', '...tyle_quiet', 1, None)], mypb.calls)
 
425
        last_calls = mypb.calls[:]
 
426
        result.addFailure(dummy_test, dummy_error)
 
427
        self.assertEqual(last_calls + [('update', 'FAIL         ', 2, None)], mypb.calls)
 
428
        last_calls = mypb.calls[:]
 
429
 
 
430
        # a success
 
431
        result.startTest(dummy_test)
 
432
        self.assertEqual(last_calls + [('update', '...tyle_quiet', 2, None)], mypb.calls)
 
433
        last_calls = mypb.calls[:]
 
434
        result.addSuccess(dummy_test)
 
435
        self.assertEqual(last_calls + [('update', 'OK           ', 3, None)], mypb.calls)
 
436
        last_calls = mypb.calls[:]
 
437
 
 
438
        # a skip
 
439
        result.startTest(dummy_test)
 
440
        self.assertEqual(last_calls + [('update', '...tyle_quiet', 3, None)], mypb.calls)
 
441
        last_calls = mypb.calls[:]
 
442
        result.addSkipped(dummy_test, dummy_error)
 
443
        self.assertEqual(last_calls + [('update', 'SKIP         ', 4, None)], mypb.calls)
 
444
        last_calls = mypb.calls[:]
 
445
 
707
446
    def test_elapsed_time_with_benchmarking(self):
708
 
        result = bzrlib.tests.TextTestResult(self._log_file,
 
447
        result = bzrlib.tests._MyResult(self._log_file,
709
448
                                        descriptions=0,
710
449
                                        verbosity=1,
711
450
                                        )
714
453
        result.extractBenchmarkTime(self)
715
454
        timed_string = result._testTimeString()
716
455
        # without explicit benchmarking, we should get a simple time.
717
 
        self.assertContainsRe(timed_string, "^ +[0-9]+ms$")
 
456
        self.assertContainsRe(timed_string, "^         [ 1-9][0-9]ms$")
718
457
        # if a benchmark time is given, we want a x of y style result.
719
458
        self.time(time.sleep, 0.001)
720
459
        result.extractBenchmarkTime(self)
721
460
        timed_string = result._testTimeString()
722
 
        self.assertContainsRe(
723
 
            timed_string, "^ +[0-9]+ms/ +[0-9]+ms$")
 
461
        self.assertContainsRe(timed_string, "^   [ 1-9][0-9]ms/   [ 1-9][0-9]ms$")
724
462
        # extracting the time from a non-bzrlib testcase sets to None
725
463
        result._recordTestStartTime()
726
464
        result.extractBenchmarkTime(
727
465
            unittest.FunctionTestCase(self.test_elapsed_time_with_benchmarking))
728
466
        timed_string = result._testTimeString()
729
 
        self.assertContainsRe(timed_string, "^ +[0-9]+ms$")
 
467
        self.assertContainsRe(timed_string, "^         [ 1-9][0-9]ms$")
730
468
        # cheat. Yes, wash thy mouth out with soap.
731
469
        self._benchtime = None
732
470
 
733
 
    def test_assigned_benchmark_file_stores_date(self):
734
 
        output = StringIO()
735
 
        result = bzrlib.tests.TextTestResult(self._log_file,
736
 
                                        descriptions=0,
737
 
                                        verbosity=1,
738
 
                                        bench_history=output
739
 
                                        )
740
 
        output_string = output.getvalue()
741
 
        
742
 
        # if you are wondering about the regexp please read the comment in
743
 
        # test_bench_history (bzrlib.tests.test_selftest.TestRunner)
744
 
        # XXX: what comment?  -- Andrew Bennetts
745
 
        self.assertContainsRe(output_string, "--date [0-9.]+")
746
 
 
747
 
    def test_benchhistory_records_test_times(self):
748
 
        result_stream = StringIO()
749
 
        result = bzrlib.tests.TextTestResult(
750
 
            self._log_file,
751
 
            descriptions=0,
752
 
            verbosity=1,
753
 
            bench_history=result_stream
754
 
            )
755
 
 
756
 
        # we want profile a call and check that its test duration is recorded
757
 
        # make a new test instance that when run will generate a benchmark
758
 
        example_test_case = TestTestResult("_time_hello_world_encoding")
759
 
        # execute the test, which should succeed and record times
760
 
        example_test_case.run(result)
761
 
        lines = result_stream.getvalue().splitlines()
762
 
        self.assertEqual(2, len(lines))
763
 
        self.assertContainsRe(lines[1],
764
 
            " *[0-9]+ms bzrlib.tests.test_selftest.TestTestResult"
765
 
            "._time_hello_world_encoding")
766
 
 
767
471
    def _time_hello_world_encoding(self):
768
472
        """Profile two sleep calls
769
473
        
774
478
 
775
479
    def test_lsprofiling(self):
776
480
        """Verbose test result prints lsprof statistics from test cases."""
777
 
        self.requireFeature(test_lsprof.LSProfFeature)
 
481
        try:
 
482
            import bzrlib.lsprof
 
483
        except ImportError:
 
484
            raise TestSkipped("lsprof not installed.")
778
485
        result_stream = StringIO()
779
 
        result = bzrlib.tests.VerboseTestResult(
 
486
        result = bzrlib.tests._MyResult(
780
487
            unittest._WritelnDecorator(result_stream),
781
488
            descriptions=0,
782
489
            verbosity=2,
799
506
        #           1        0            ???         ???       ???(sleep) 
800
507
        # and then repeated but with 'world', rather than 'hello'.
801
508
        # this should appear in the output stream of our test result.
802
 
        output = result_stream.getvalue()
803
 
        self.assertContainsRe(output,
804
 
            r"LSProf output for <type 'unicode'>\(\('hello',\), {'errors': 'replace'}\)")
805
 
        self.assertContainsRe(output,
806
 
            r" *CallCount *Recursive *Total\(ms\) *Inline\(ms\) *module:lineno\(function\)\n")
807
 
        self.assertContainsRe(output,
808
 
            r"( +1 +0 +0\.\d+ +0\.\d+ +<method 'disable' of '_lsprof\.Profiler' objects>\n)?")
809
 
        self.assertContainsRe(output,
810
 
            r"LSProf output for <type 'unicode'>\(\('world',\), {'errors': 'replace'}\)\n")
811
 
 
812
 
    def test_known_failure(self):
813
 
        """A KnownFailure being raised should trigger several result actions."""
814
 
        class InstrumentedTestResult(ExtendedTestResult):
815
 
 
816
 
            def report_test_start(self, test): pass
817
 
            def report_known_failure(self, test, err):
818
 
                self._call = test, err
819
 
        result = InstrumentedTestResult(None, None, None, None)
820
 
        def test_function():
821
 
            raise KnownFailure('failed!')
822
 
        test = unittest.FunctionTestCase(test_function)
823
 
        test.run(result)
824
 
        # it should invoke 'report_known_failure'.
825
 
        self.assertEqual(2, len(result._call))
826
 
        self.assertEqual(test, result._call[0])
827
 
        self.assertEqual(KnownFailure, result._call[1][0])
828
 
        self.assertIsInstance(result._call[1][1], KnownFailure)
829
 
        # we dont introspec the traceback, if the rest is ok, it would be
830
 
        # exceptional for it not to be.
831
 
        # it should update the known_failure_count on the object.
832
 
        self.assertEqual(1, result.known_failure_count)
833
 
        # the result should be successful.
834
 
        self.assertTrue(result.wasSuccessful())
835
 
 
836
 
    def test_verbose_report_known_failure(self):
837
 
        # verbose test output formatting
838
 
        result_stream = StringIO()
839
 
        result = bzrlib.tests.VerboseTestResult(
840
 
            unittest._WritelnDecorator(result_stream),
841
 
            descriptions=0,
842
 
            verbosity=2,
843
 
            )
844
 
        test = self.get_passing_test()
845
 
        result.startTest(test)
846
 
        result.extractBenchmarkTime(test)
847
 
        prefix = len(result_stream.getvalue())
848
 
        # the err parameter has the shape:
849
 
        # (class, exception object, traceback)
850
 
        # KnownFailures dont get their tracebacks shown though, so we
851
 
        # can skip that.
852
 
        err = (KnownFailure, KnownFailure('foo'), None)
853
 
        result.report_known_failure(test, err)
854
 
        output = result_stream.getvalue()[prefix:]
855
 
        lines = output.splitlines()
856
 
        self.assertContainsRe(lines[0], r'XFAIL *\d+ms$')
857
 
        self.assertEqual(lines[1], '    foo')
858
 
        self.assertEqual(2, len(lines))
859
 
 
860
 
    def test_text_report_known_failure(self):
861
 
        # text test output formatting
862
 
        pb = MockProgress()
863
 
        result = bzrlib.tests.TextTestResult(
864
 
            None,
865
 
            descriptions=0,
866
 
            verbosity=1,
867
 
            pb=pb,
868
 
            )
869
 
        test = self.get_passing_test()
870
 
        # this seeds the state to handle reporting the test.
871
 
        result.startTest(test)
872
 
        result.extractBenchmarkTime(test)
873
 
        # the err parameter has the shape:
874
 
        # (class, exception object, traceback)
875
 
        # KnownFailures dont get their tracebacks shown though, so we
876
 
        # can skip that.
877
 
        err = (KnownFailure, KnownFailure('foo'), None)
878
 
        result.report_known_failure(test, err)
879
 
        self.assertEqual(
880
 
            [
881
 
            ('update', '[1 in 0s] passing_test', None, None),
882
 
            ('note', 'XFAIL: %s\n%s\n', ('passing_test', err[1]))
883
 
            ],
884
 
            pb.calls)
885
 
        # known_failures should be printed in the summary, so if we run a test
886
 
        # after there are some known failures, the update prefix should match
887
 
        # this.
888
 
        result.known_failure_count = 3
889
 
        test.run(result)
890
 
        self.assertEqual(
891
 
            [
892
 
            ('update', '[2 in 0s, 3 known failures] passing_test', None, None),
893
 
            ],
894
 
            pb.calls[2:])
895
 
 
896
 
    def get_passing_test(self):
897
 
        """Return a test object that can't be run usefully."""
898
 
        def passing_test():
899
 
            pass
900
 
        return unittest.FunctionTestCase(passing_test)
901
 
 
902
 
    def test_add_not_supported(self):
903
 
        """Test the behaviour of invoking addNotSupported."""
904
 
        class InstrumentedTestResult(ExtendedTestResult):
905
 
            def report_test_start(self, test): pass
906
 
            def report_unsupported(self, test, feature):
907
 
                self._call = test, feature
908
 
        result = InstrumentedTestResult(None, None, None, None)
909
 
        test = SampleTestCase('_test_pass')
910
 
        feature = Feature()
911
 
        result.startTest(test)
912
 
        result.addNotSupported(test, feature)
913
 
        # it should invoke 'report_unsupported'.
914
 
        self.assertEqual(2, len(result._call))
915
 
        self.assertEqual(test, result._call[0])
916
 
        self.assertEqual(feature, result._call[1])
917
 
        # the result should be successful.
918
 
        self.assertTrue(result.wasSuccessful())
919
 
        # it should record the test against a count of tests not run due to
920
 
        # this feature.
921
 
        self.assertEqual(1, result.unsupported['Feature'])
922
 
        # and invoking it again should increment that counter
923
 
        result.addNotSupported(test, feature)
924
 
        self.assertEqual(2, result.unsupported['Feature'])
925
 
 
926
 
    def test_verbose_report_unsupported(self):
927
 
        # verbose test output formatting
928
 
        result_stream = StringIO()
929
 
        result = bzrlib.tests.VerboseTestResult(
930
 
            unittest._WritelnDecorator(result_stream),
931
 
            descriptions=0,
932
 
            verbosity=2,
933
 
            )
934
 
        test = self.get_passing_test()
935
 
        feature = Feature()
936
 
        result.startTest(test)
937
 
        result.extractBenchmarkTime(test)
938
 
        prefix = len(result_stream.getvalue())
939
 
        result.report_unsupported(test, feature)
940
 
        output = result_stream.getvalue()[prefix:]
941
 
        lines = output.splitlines()
942
 
        self.assertEqual(lines, ['NODEP                   0ms', "    The feature 'Feature' is not available."])
943
 
    
944
 
    def test_text_report_unsupported(self):
945
 
        # text test output formatting
946
 
        pb = MockProgress()
947
 
        result = bzrlib.tests.TextTestResult(
948
 
            None,
949
 
            descriptions=0,
950
 
            verbosity=1,
951
 
            pb=pb,
952
 
            )
953
 
        test = self.get_passing_test()
954
 
        feature = Feature()
955
 
        # this seeds the state to handle reporting the test.
956
 
        result.startTest(test)
957
 
        result.extractBenchmarkTime(test)
958
 
        result.report_unsupported(test, feature)
959
 
        # no output on unsupported features
960
 
        self.assertEqual(
961
 
            [('update', '[1 in 0s] passing_test', None, None)
962
 
            ],
963
 
            pb.calls)
964
 
        # the number of missing features should be printed in the progress
965
 
        # summary, so check for that.
966
 
        result.unsupported = {'foo':0, 'bar':0}
967
 
        test.run(result)
968
 
        self.assertEqual(
969
 
            [
970
 
            ('update', '[2 in 0s, 2 missing features] passing_test', None, None),
971
 
            ],
972
 
            pb.calls[1:])
973
 
    
974
 
    def test_unavailable_exception(self):
975
 
        """An UnavailableFeature being raised should invoke addNotSupported."""
976
 
        class InstrumentedTestResult(ExtendedTestResult):
977
 
 
978
 
            def report_test_start(self, test): pass
979
 
            def addNotSupported(self, test, feature):
980
 
                self._call = test, feature
981
 
        result = InstrumentedTestResult(None, None, None, None)
982
 
        feature = Feature()
983
 
        def test_function():
984
 
            raise UnavailableFeature(feature)
985
 
        test = unittest.FunctionTestCase(test_function)
986
 
        test.run(result)
987
 
        # it should invoke 'addNotSupported'.
988
 
        self.assertEqual(2, len(result._call))
989
 
        self.assertEqual(test, result._call[0])
990
 
        self.assertEqual(feature, result._call[1])
991
 
        # and not count as an error
992
 
        self.assertEqual(0, result.error_count)
 
509
        self.assertContainsRe(result_stream.getvalue(), 
 
510
            r"LSProf output for <type 'unicode'>\(\('hello',\), {'errors': 'replace'}\)\n"
 
511
            r" *CallCount *Recursive *Total\(ms\) *Inline\(ms\) *module:lineno\(function\)\n"
 
512
            r"( +1 +0 +0\.\d+ +0\.\d+ +<method 'disable' of '_lsprof\.Profiler' objects>\n)?"
 
513
            r"LSProf output for <type 'unicode'>\(\('world',\), {'errors': 'replace'}\)\n"
 
514
            r" *CallCount *Recursive *Total\(ms\) *Inline\(ms\) *module:lineno\(function\)\n"
 
515
            r"( +1 +0 +0\.\d+ +0\.\d+ +<method 'disable' of '_lsprof\.Profiler' objects>\n)?"
 
516
            )
993
517
 
994
518
 
995
519
class TestRunner(TestCase):
1013
537
        finally:
1014
538
            TestCaseInTempDir.TEST_ROOT = old_root
1015
539
 
1016
 
    def test_known_failure_failed_run(self):
1017
 
        # run a test that generates a known failure which should be printed in
1018
 
        # the final output when real failures occur.
1019
 
        def known_failure_test():
1020
 
            raise KnownFailure('failed')
1021
 
        test = unittest.TestSuite()
1022
 
        test.addTest(unittest.FunctionTestCase(known_failure_test))
1023
 
        def failing_test():
1024
 
            raise AssertionError('foo')
1025
 
        test.addTest(unittest.FunctionTestCase(failing_test))
1026
 
        stream = StringIO()
1027
 
        runner = TextTestRunner(stream=stream)
1028
 
        result = self.run_test_runner(runner, test)
1029
 
        lines = stream.getvalue().splitlines()
1030
 
        self.assertEqual([
1031
 
            '',
1032
 
            '======================================================================',
1033
 
            'FAIL: unittest.FunctionTestCase (failing_test)',
1034
 
            '----------------------------------------------------------------------',
1035
 
            'Traceback (most recent call last):',
1036
 
            '    raise AssertionError(\'foo\')',
1037
 
            'AssertionError: foo',
1038
 
            '',
1039
 
            '----------------------------------------------------------------------',
1040
 
            '',
1041
 
            'FAILED (failures=1, known_failure_count=1)'],
1042
 
            lines[0:5] + lines[6:10] + lines[11:])
1043
 
 
1044
 
    def test_known_failure_ok_run(self):
1045
 
        # run a test that generates a known failure which should be printed in the final output.
1046
 
        def known_failure_test():
1047
 
            raise KnownFailure('failed')
1048
 
        test = unittest.FunctionTestCase(known_failure_test)
1049
 
        stream = StringIO()
1050
 
        runner = TextTestRunner(stream=stream)
1051
 
        result = self.run_test_runner(runner, test)
1052
 
        self.assertContainsRe(stream.getvalue(),
1053
 
            '\n'
1054
 
            '-*\n'
1055
 
            'Ran 1 test in .*\n'
1056
 
            '\n'
1057
 
            'OK \\(known_failures=1\\)\n')
 
540
    def test_accepts_and_uses_pb_parameter(self):
 
541
        test = TestRunner('dummy_test')
 
542
        mypb = MockProgress()
 
543
        self.assertEqual([], mypb.calls)
 
544
        runner = TextTestRunner(stream=self._log_file, pb=mypb)
 
545
        result = self.run_test_runner(runner, test)
 
546
        self.assertEqual(1, result.testsRun)
 
547
        self.assertEqual(('update', 'Running tests', 0, 1), mypb.calls[0])
 
548
        self.assertEqual(('update', '...dummy_test', 0, None), mypb.calls[1])
 
549
        self.assertEqual(('update', 'OK           ', 1, None), mypb.calls[2])
 
550
        self.assertEqual(('update', 'Cleaning up', 0, 1), mypb.calls[3])
 
551
        self.assertEqual(('clear',), mypb.calls[4])
 
552
        self.assertEqual(5, len(mypb.calls))
1058
553
 
1059
554
    def test_skipped_test(self):
1060
555
        # run a test that is skipped, and check the suite as a whole still
1062
557
        # skipping_test must be hidden in here so it's not run as a real test
1063
558
        def skipping_test():
1064
559
            raise TestSkipped('test intentionally skipped')
1065
 
 
1066
 
        runner = TextTestRunner(stream=self._log_file)
 
560
        runner = TextTestRunner(stream=self._log_file, keep_output=True)
1067
561
        test = unittest.FunctionTestCase(skipping_test)
1068
562
        result = self.run_test_runner(runner, test)
1069
563
        self.assertTrue(result.wasSuccessful())
1070
564
 
1071
 
    def test_skipped_from_setup(self):
1072
 
        class SkippedSetupTest(TestCase):
1073
 
 
1074
 
            def setUp(self):
1075
 
                self.counter = 1
1076
 
                self.addCleanup(self.cleanup)
1077
 
                raise TestSkipped('skipped setup')
1078
 
 
1079
 
            def test_skip(self):
1080
 
                self.fail('test reached')
1081
 
 
1082
 
            def cleanup(self):
1083
 
                self.counter -= 1
1084
 
 
1085
 
        runner = TextTestRunner(stream=self._log_file)
1086
 
        test = SkippedSetupTest('test_skip')
1087
 
        result = self.run_test_runner(runner, test)
1088
 
        self.assertTrue(result.wasSuccessful())
1089
 
        # Check if cleanup was called the right number of times.
1090
 
        self.assertEqual(0, test.counter)
1091
 
 
1092
 
    def test_skipped_from_test(self):
1093
 
        class SkippedTest(TestCase):
1094
 
 
1095
 
            def setUp(self):
1096
 
                self.counter = 1
1097
 
                self.addCleanup(self.cleanup)
1098
 
 
1099
 
            def test_skip(self):
1100
 
                raise TestSkipped('skipped test')
1101
 
 
1102
 
            def cleanup(self):
1103
 
                self.counter -= 1
1104
 
 
1105
 
        runner = TextTestRunner(stream=self._log_file)
1106
 
        test = SkippedTest('test_skip')
1107
 
        result = self.run_test_runner(runner, test)
1108
 
        self.assertTrue(result.wasSuccessful())
1109
 
        # Check if cleanup was called the right number of times.
1110
 
        self.assertEqual(0, test.counter)
1111
 
 
1112
 
    def test_unsupported_features_listed(self):
1113
 
        """When unsupported features are encountered they are detailed."""
1114
 
        class Feature1(Feature):
1115
 
            def _probe(self): return False
1116
 
        class Feature2(Feature):
1117
 
            def _probe(self): return False
1118
 
        # create sample tests
1119
 
        test1 = SampleTestCase('_test_pass')
1120
 
        test1._test_needs_features = [Feature1()]
1121
 
        test2 = SampleTestCase('_test_pass')
1122
 
        test2._test_needs_features = [Feature2()]
1123
 
        test = unittest.TestSuite()
1124
 
        test.addTest(test1)
1125
 
        test.addTest(test2)
1126
 
        stream = StringIO()
1127
 
        runner = TextTestRunner(stream=stream)
1128
 
        result = self.run_test_runner(runner, test)
1129
 
        lines = stream.getvalue().splitlines()
1130
 
        self.assertEqual([
1131
 
            'OK',
1132
 
            "Missing feature 'Feature1' skipped 1 tests.",
1133
 
            "Missing feature 'Feature2' skipped 1 tests.",
1134
 
            ],
1135
 
            lines[-3:])
1136
 
 
1137
 
    def test_bench_history(self):
1138
 
        # tests that the running the benchmark produces a history file
1139
 
        # containing a timestamp and the revision id of the bzrlib source which
1140
 
        # was tested.
1141
 
        workingtree = _get_bzr_source_tree()
1142
 
        test = TestRunner('dummy_test')
1143
 
        output = StringIO()
1144
 
        runner = TextTestRunner(stream=self._log_file, bench_history=output)
1145
 
        result = self.run_test_runner(runner, test)
1146
 
        output_string = output.getvalue()
1147
 
        self.assertContainsRe(output_string, "--date [0-9.]+")
1148
 
        if workingtree is not None:
1149
 
            revision_id = workingtree.get_parent_ids()[0]
1150
 
            self.assertEndsWith(output_string.rstrip(), revision_id)
1151
 
 
1152
 
    def test_success_log_deleted(self):
1153
 
        """Successful tests have their log deleted"""
1154
 
 
1155
 
        class LogTester(TestCase):
1156
 
 
1157
 
            def test_success(self):
1158
 
                self.log('this will be removed\n')
1159
 
 
1160
 
        sio = cStringIO.StringIO()
1161
 
        runner = TextTestRunner(stream=sio)
1162
 
        test = LogTester('test_success')
1163
 
        result = self.run_test_runner(runner, test)
1164
 
 
1165
 
        log = test._get_log()
1166
 
        self.assertEqual("DELETED log file to reduce memory footprint", log)
1167
 
        self.assertEqual('', test._log_contents)
1168
 
        self.assertIs(None, test._log_file_name)
1169
 
 
1170
 
    def test_fail_log_kept(self):
1171
 
        """Failed tests have their log kept"""
1172
 
 
1173
 
        class LogTester(TestCase):
1174
 
 
1175
 
            def test_fail(self):
1176
 
                self.log('this will be kept\n')
1177
 
                self.fail('this test fails')
1178
 
 
1179
 
        sio = cStringIO.StringIO()
1180
 
        runner = TextTestRunner(stream=sio)
1181
 
        test = LogTester('test_fail')
1182
 
        result = self.run_test_runner(runner, test)
1183
 
 
1184
 
        text = sio.getvalue()
1185
 
        self.assertContainsRe(text, 'this will be kept')
1186
 
        self.assertContainsRe(text, 'this test fails')
1187
 
 
1188
 
        log = test._get_log()
1189
 
        self.assertContainsRe(log, 'this will be kept')
1190
 
        self.assertEqual(log, test._log_contents)
1191
 
 
1192
 
    def test_error_log_kept(self):
1193
 
        """Tests with errors have their log kept"""
1194
 
 
1195
 
        class LogTester(TestCase):
1196
 
 
1197
 
            def test_error(self):
1198
 
                self.log('this will be kept\n')
1199
 
                raise ValueError('random exception raised')
1200
 
 
1201
 
        sio = cStringIO.StringIO()
1202
 
        runner = TextTestRunner(stream=sio)
1203
 
        test = LogTester('test_error')
1204
 
        result = self.run_test_runner(runner, test)
1205
 
 
1206
 
        text = sio.getvalue()
1207
 
        self.assertContainsRe(text, 'this will be kept')
1208
 
        self.assertContainsRe(text, 'random exception raised')
1209
 
 
1210
 
        log = test._get_log()
1211
 
        self.assertContainsRe(log, 'this will be kept')
1212
 
        self.assertEqual(log, test._log_contents)
1213
 
 
1214
 
 
1215
 
class SampleTestCase(TestCase):
1216
 
 
1217
 
    def _test_pass(self):
1218
 
        pass
1219
 
 
1220
565
 
1221
566
class TestTestCase(TestCase):
1222
567
    """Tests that test the core bzrlib TestCase."""
1223
568
 
1224
 
    def test_debug_flags_sanitised(self):
1225
 
        """The bzrlib debug flags should be sanitised by setUp."""
1226
 
        # we could set something and run a test that will check
1227
 
        # it gets santised, but this is probably sufficient for now:
1228
 
        # if someone runs the test with -Dsomething it will error.
1229
 
        self.assertEqual(set(), bzrlib.debug.debug_flags)
1230
 
 
1231
569
    def inner_test(self):
1232
570
        # the inner child test
1233
571
        note("inner_test")
1236
574
        # the outer child test
1237
575
        note("outer_start")
1238
576
        self.inner_test = TestTestCase("inner_child")
1239
 
        result = bzrlib.tests.TextTestResult(self._log_file,
 
577
        result = bzrlib.tests._MyResult(self._log_file,
1240
578
                                        descriptions=0,
1241
579
                                        verbosity=1)
1242
580
        self.inner_test.run(result)
1256
594
        # the outer child test
1257
595
        original_trace = bzrlib.trace._trace_file
1258
596
        outer_test = TestTestCase("outer_child")
1259
 
        result = bzrlib.tests.TextTestResult(self._log_file,
 
597
        result = bzrlib.tests._MyResult(self._log_file,
1260
598
                                        descriptions=0,
1261
599
                                        verbosity=1)
1262
600
        outer_test.run(result)
1271
609
        """Test that the TestCase.time() method accumulates a benchmark time."""
1272
610
        sample_test = TestTestCase("method_that_times_a_bit_twice")
1273
611
        output_stream = StringIO()
1274
 
        result = bzrlib.tests.VerboseTestResult(
 
612
        result = bzrlib.tests._MyResult(
1275
613
            unittest._WritelnDecorator(output_stream),
1276
614
            descriptions=0,
1277
 
            verbosity=2,
1278
 
            num_tests=sample_test.countTestCases())
 
615
            verbosity=2)
1279
616
        sample_test.run(result)
1280
617
        self.assertContainsRe(
1281
618
            output_stream.getvalue(),
1282
 
            r"\d+ms/ +\d+ms\n$")
1283
 
 
1284
 
    def test_hooks_sanitised(self):
1285
 
        """The bzrlib hooks should be sanitised by setUp."""
1286
 
        self.assertEqual(bzrlib.branch.BranchHooks(),
1287
 
            bzrlib.branch.Branch.hooks)
1288
 
        self.assertEqual(bzrlib.smart.server.SmartServerHooks(),
1289
 
            bzrlib.smart.server.SmartTCPServer.hooks)
1290
 
 
 
619
            "[1-9][0-9]ms/   [1-9][0-9]ms\n$")
 
620
        
1291
621
    def test__gather_lsprof_in_benchmarks(self):
1292
622
        """When _gather_lsprof_in_benchmarks is on, accumulate profile data.
1293
623
        
1294
624
        Each self.time() call is individually and separately profiled.
1295
625
        """
1296
 
        self.requireFeature(test_lsprof.LSProfFeature)
 
626
        try:
 
627
            import bzrlib.lsprof
 
628
        except ImportError:
 
629
            raise TestSkipped("lsprof not installed.")
1297
630
        # overrides the class member with an instance member so no cleanup 
1298
631
        # needed.
1299
632
        self._gather_lsprof_in_benchmarks = True
1305
638
        self.assertIsInstance(self._benchcalls[0][1], bzrlib.lsprof.Stats)
1306
639
        self.assertIsInstance(self._benchcalls[1][1], bzrlib.lsprof.Stats)
1307
640
 
1308
 
    def test_knownFailure(self):
1309
 
        """Self.knownFailure() should raise a KnownFailure exception."""
1310
 
        self.assertRaises(KnownFailure, self.knownFailure, "A Failure")
1311
 
 
1312
 
    def test_requireFeature_available(self):
1313
 
        """self.requireFeature(available) is a no-op."""
1314
 
        class Available(Feature):
1315
 
            def _probe(self):return True
1316
 
        feature = Available()
1317
 
        self.requireFeature(feature)
1318
 
 
1319
 
    def test_requireFeature_unavailable(self):
1320
 
        """self.requireFeature(unavailable) raises UnavailableFeature."""
1321
 
        class Unavailable(Feature):
1322
 
            def _probe(self):return False
1323
 
        feature = Unavailable()
1324
 
        self.assertRaises(UnavailableFeature, self.requireFeature, feature)
1325
 
 
1326
 
    def test_run_no_parameters(self):
1327
 
        test = SampleTestCase('_test_pass')
1328
 
        test.run()
1329
 
    
1330
 
    def test_run_enabled_unittest_result(self):
1331
 
        """Test we revert to regular behaviour when the test is enabled."""
1332
 
        test = SampleTestCase('_test_pass')
1333
 
        class EnabledFeature(object):
1334
 
            def available(self):
1335
 
                return True
1336
 
        test._test_needs_features = [EnabledFeature()]
1337
 
        result = unittest.TestResult()
1338
 
        test.run(result)
1339
 
        self.assertEqual(1, result.testsRun)
1340
 
        self.assertEqual([], result.errors)
1341
 
        self.assertEqual([], result.failures)
1342
 
 
1343
 
    def test_run_disabled_unittest_result(self):
1344
 
        """Test our compatability for disabled tests with unittest results."""
1345
 
        test = SampleTestCase('_test_pass')
1346
 
        class DisabledFeature(object):
1347
 
            def available(self):
1348
 
                return False
1349
 
        test._test_needs_features = [DisabledFeature()]
1350
 
        result = unittest.TestResult()
1351
 
        test.run(result)
1352
 
        self.assertEqual(1, result.testsRun)
1353
 
        self.assertEqual([], result.errors)
1354
 
        self.assertEqual([], result.failures)
1355
 
 
1356
 
    def test_run_disabled_supporting_result(self):
1357
 
        """Test disabled tests behaviour with support aware results."""
1358
 
        test = SampleTestCase('_test_pass')
1359
 
        class DisabledFeature(object):
1360
 
            def available(self):
1361
 
                return False
1362
 
        the_feature = DisabledFeature()
1363
 
        test._test_needs_features = [the_feature]
1364
 
        class InstrumentedTestResult(unittest.TestResult):
1365
 
            def __init__(self):
1366
 
                unittest.TestResult.__init__(self)
1367
 
                self.calls = []
1368
 
            def startTest(self, test):
1369
 
                self.calls.append(('startTest', test))
1370
 
            def stopTest(self, test):
1371
 
                self.calls.append(('stopTest', test))
1372
 
            def addNotSupported(self, test, feature):
1373
 
                self.calls.append(('addNotSupported', test, feature))
1374
 
        result = InstrumentedTestResult()
1375
 
        test.run(result)
1376
 
        self.assertEqual([
1377
 
            ('startTest', test),
1378
 
            ('addNotSupported', test, the_feature),
1379
 
            ('stopTest', test),
1380
 
            ],
1381
 
            result.calls)
1382
 
 
1383
 
 
1384
 
@symbol_versioning.deprecated_function(zero_eleven)
1385
 
def sample_deprecated_function():
1386
 
    """A deprecated function to test applyDeprecated with."""
1387
 
    return 2
1388
 
 
1389
 
 
1390
 
def sample_undeprecated_function(a_param):
1391
 
    """A undeprecated function to test applyDeprecated with."""
1392
 
 
1393
 
 
1394
 
class ApplyDeprecatedHelper(object):
1395
 
    """A helper class for ApplyDeprecated tests."""
1396
 
 
1397
 
    @symbol_versioning.deprecated_method(zero_eleven)
1398
 
    def sample_deprecated_method(self, param_one):
1399
 
        """A deprecated method for testing with."""
1400
 
        return param_one
1401
 
 
1402
 
    def sample_normal_method(self):
1403
 
        """A undeprecated method."""
1404
 
 
1405
 
    @symbol_versioning.deprecated_method(zero_ten)
1406
 
    def sample_nested_deprecation(self):
1407
 
        return sample_deprecated_function()
1408
 
 
1409
641
 
1410
642
class TestExtraAssertions(TestCase):
1411
643
    """Tests for new test assertions in bzrlib test suite"""
1420
652
        self.assertEndsWith('foo', 'oo')
1421
653
        self.assertRaises(AssertionError, self.assertEndsWith, 'o', 'oo')
1422
654
 
1423
 
    def test_applyDeprecated_not_deprecated(self):
1424
 
        sample_object = ApplyDeprecatedHelper()
1425
 
        # calling an undeprecated callable raises an assertion
1426
 
        self.assertRaises(AssertionError, self.applyDeprecated, zero_eleven,
1427
 
            sample_object.sample_normal_method)
1428
 
        self.assertRaises(AssertionError, self.applyDeprecated, zero_eleven,
1429
 
            sample_undeprecated_function, "a param value")
1430
 
        # calling a deprecated callable (function or method) with the wrong
1431
 
        # expected deprecation fails.
1432
 
        self.assertRaises(AssertionError, self.applyDeprecated, zero_ten,
1433
 
            sample_object.sample_deprecated_method, "a param value")
1434
 
        self.assertRaises(AssertionError, self.applyDeprecated, zero_ten,
1435
 
            sample_deprecated_function)
1436
 
        # calling a deprecated callable (function or method) with the right
1437
 
        # expected deprecation returns the functions result.
1438
 
        self.assertEqual("a param value", self.applyDeprecated(zero_eleven,
1439
 
            sample_object.sample_deprecated_method, "a param value"))
1440
 
        self.assertEqual(2, self.applyDeprecated(zero_eleven,
1441
 
            sample_deprecated_function))
1442
 
        # calling a nested deprecation with the wrong deprecation version
1443
 
        # fails even if a deeper nested function was deprecated with the 
1444
 
        # supplied version.
1445
 
        self.assertRaises(AssertionError, self.applyDeprecated,
1446
 
            zero_eleven, sample_object.sample_nested_deprecation)
1447
 
        # calling a nested deprecation with the right deprecation value
1448
 
        # returns the calls result.
1449
 
        self.assertEqual(2, self.applyDeprecated(zero_ten,
1450
 
            sample_object.sample_nested_deprecation))
1451
 
 
1452
 
    def test_callDeprecated(self):
1453
 
        def testfunc(be_deprecated, result=None):
1454
 
            if be_deprecated is True:
1455
 
                symbol_versioning.warn('i am deprecated', DeprecationWarning, 
1456
 
                                       stacklevel=1)
1457
 
            return result
1458
 
        result = self.callDeprecated(['i am deprecated'], testfunc, True)
1459
 
        self.assertIs(None, result)
1460
 
        result = self.callDeprecated([], testfunc, False, 'result')
1461
 
        self.assertEqual('result', result)
1462
 
        self.callDeprecated(['i am deprecated'], testfunc, be_deprecated=True)
1463
 
        self.callDeprecated([], testfunc, be_deprecated=False)
1464
 
 
1465
655
 
1466
656
class TestConvenienceMakers(TestCaseWithTransport):
1467
657
    """Test for the make_* convenience functions."""
1475
665
        self.assertIsInstance(bzrlib.bzrdir.BzrDir.open('b')._format,
1476
666
                              bzrlib.bzrdir.BzrDirFormat6)
1477
667
 
1478
 
    def test_make_branch_and_memory_tree(self):
1479
 
        # we should be able to get a new branch and a mutable tree from
1480
 
        # TestCaseWithTransport
1481
 
        tree = self.make_branch_and_memory_tree('a')
1482
 
        self.assertIsInstance(tree, bzrlib.memorytree.MemoryTree)
1483
 
 
1484
 
 
1485
 
class TestSFTPMakeBranchAndTree(TestCaseWithSFTPServer):
1486
 
 
1487
 
    def test_make_tree_for_sftp_branch(self):
1488
 
        """Transports backed by local directories create local trees."""
1489
 
 
1490
 
        tree = self.make_branch_and_tree('t1')
1491
 
        base = tree.bzrdir.root_transport.base
1492
 
        self.failIf(base.startswith('sftp'),
1493
 
                'base %r is on sftp but should be local' % base)
1494
 
        self.assertEquals(tree.bzrdir.root_transport,
1495
 
                tree.branch.bzrdir.root_transport)
1496
 
        self.assertEquals(tree.bzrdir.root_transport,
1497
 
                tree.branch.repository.bzrdir.root_transport)
1498
 
 
1499
668
 
1500
669
class TestSelftest(TestCase):
1501
670
    """Tests of bzrlib.tests.selftest."""
1510
679
        self.apply_redirected(out, err, None, bzrlib.tests.selftest, 
1511
680
            test_suite_factory=factory)
1512
681
        self.assertEqual([True], factory_called)
1513
 
 
1514
 
 
1515
 
class TestSelftestCleanOutput(TestCaseInTempDir):
1516
 
 
1517
 
    def test_clean_output(self):
1518
 
        # test functionality of clean_selftest_output()
1519
 
        self.build_tree(['test0000.tmp/', 'test0001.tmp/',
1520
 
                         'bzrlib/', 'tests/',
1521
 
                         'bzr', 'setup.py', 'test9999.tmp'])
1522
 
 
1523
 
        root = os.getcwdu()
1524
 
        before = os.listdir(root)
1525
 
        before.sort()
1526
 
        self.assertEquals(['bzr','bzrlib','setup.py',
1527
 
                           'test0000.tmp','test0001.tmp',
1528
 
                           'test9999.tmp','tests'],
1529
 
                           before)
1530
 
        clean_selftest_output(root, quiet=True)
1531
 
        after = os.listdir(root)
1532
 
        after.sort()
1533
 
        self.assertEquals(['bzr','bzrlib','setup.py',
1534
 
                           'test9999.tmp','tests'],
1535
 
                           after)
1536
 
 
1537
 
    def test_clean_readonly(self):
1538
 
        # test for delete read-only files
1539
 
        self.build_tree(['test0000.tmp/', 'test0000.tmp/foo'])
1540
 
        osutils.make_readonly('test0000.tmp/foo')
1541
 
        root = os.getcwdu()
1542
 
        before = os.listdir(root);  before.sort()
1543
 
        self.assertEquals(['test0000.tmp'], before)
1544
 
        clean_selftest_output(root, quiet=True)
1545
 
        after = os.listdir(root);   after.sort()
1546
 
        self.assertEquals([], after)
1547
 
 
1548
 
 
1549
 
class TestKnownFailure(TestCase):
1550
 
 
1551
 
    def test_known_failure(self):
1552
 
        """Check that KnownFailure is defined appropriately."""
1553
 
        # a KnownFailure is an assertion error for compatability with unaware
1554
 
        # runners.
1555
 
        self.assertIsInstance(KnownFailure(""), AssertionError)
1556
 
 
1557
 
    def test_expect_failure(self):
1558
 
        try:
1559
 
            self.expectFailure("Doomed to failure", self.assertTrue, False)
1560
 
        except KnownFailure, e:
1561
 
            self.assertEqual('Doomed to failure', e.args[0])
1562
 
        try:
1563
 
            self.expectFailure("Doomed to failure", self.assertTrue, True)
1564
 
        except AssertionError, e:
1565
 
            self.assertEqual('Unexpected success.  Should have failed:'
1566
 
                             ' Doomed to failure', e.args[0])
1567
 
        else:
1568
 
            self.fail('Assertion not raised')
1569
 
 
1570
 
 
1571
 
class TestFeature(TestCase):
1572
 
 
1573
 
    def test_caching(self):
1574
 
        """Feature._probe is called by the feature at most once."""
1575
 
        class InstrumentedFeature(Feature):
1576
 
            def __init__(self):
1577
 
                Feature.__init__(self)
1578
 
                self.calls = []
1579
 
            def _probe(self):
1580
 
                self.calls.append('_probe')
1581
 
                return False
1582
 
        feature = InstrumentedFeature()
1583
 
        feature.available()
1584
 
        self.assertEqual(['_probe'], feature.calls)
1585
 
        feature.available()
1586
 
        self.assertEqual(['_probe'], feature.calls)
1587
 
 
1588
 
    def test_named_str(self):
1589
 
        """Feature.__str__ should thunk to feature_name()."""
1590
 
        class NamedFeature(Feature):
1591
 
            def feature_name(self):
1592
 
                return 'symlinks'
1593
 
        feature = NamedFeature()
1594
 
        self.assertEqual('symlinks', str(feature))
1595
 
 
1596
 
    def test_default_str(self):
1597
 
        """Feature.__str__ should default to __class__.__name__."""
1598
 
        class NamedFeature(Feature):
1599
 
            pass
1600
 
        feature = NamedFeature()
1601
 
        self.assertEqual('NamedFeature', str(feature))
1602
 
 
1603
 
 
1604
 
class TestUnavailableFeature(TestCase):
1605
 
 
1606
 
    def test_access_feature(self):
1607
 
        feature = Feature()
1608
 
        exception = UnavailableFeature(feature)
1609
 
        self.assertIs(feature, exception.args[0])
1610
 
 
1611
 
 
1612
 
class TestSelftestFiltering(TestCase):
1613
 
 
1614
 
    def setUp(self):
1615
 
        self.suite = TestUtil.TestSuite()
1616
 
        self.loader = TestUtil.TestLoader()
1617
 
        self.suite.addTest(self.loader.loadTestsFromModuleNames([
1618
 
            'bzrlib.tests.test_selftest']))
1619
 
        self.all_names = [t.id() for t in iter_suite_tests(self.suite)]
1620
 
 
1621
 
    def test_filter_suite_by_re(self):
1622
 
        filtered_suite = filter_suite_by_re(self.suite, 'test_filter')
1623
 
        filtered_names = [t.id() for t in iter_suite_tests(filtered_suite)]
1624
 
        self.assertEqual(filtered_names, ['bzrlib.tests.test_selftest.'
1625
 
            'TestSelftestFiltering.test_filter_suite_by_re'])
1626
 
            
1627
 
    def test_sort_suite_by_re(self):
1628
 
        sorted_suite = sort_suite_by_re(self.suite, 'test_filter')
1629
 
        sorted_names = [t.id() for t in iter_suite_tests(sorted_suite)]
1630
 
        self.assertEqual(sorted_names[0], 'bzrlib.tests.test_selftest.'
1631
 
            'TestSelftestFiltering.test_filter_suite_by_re')
1632
 
        self.assertEquals(sorted(self.all_names), sorted(sorted_names))
1633
 
 
1634
 
 
1635
 
class TestCheckInventoryShape(TestCaseWithTransport):
1636
 
 
1637
 
    def test_check_inventory_shape(self):
1638
 
        files = ['a', 'b/', 'b/c']
1639
 
        tree = self.make_branch_and_tree('.')
1640
 
        self.build_tree(files)
1641
 
        tree.add(files)
1642
 
        tree.lock_read()
1643
 
        try:
1644
 
            self.check_inventory_shape(tree.inventory, files)
1645
 
        finally:
1646
 
            tree.unlock()