/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to breezy/tests/test_selftest.py

  • Committer: Jelmer Vernooij
  • Date: 2017-06-10 21:59:15 UTC
  • mto: This revision was merged to the branch mainline in revision 6690.
  • Revision ID: jelmer@jelmer.uk-20170610215915-zcpu0in3r1irx3ml
Move serializer to bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005-2013, 2016 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
16
16
 
17
17
"""Tests for the test framework."""
18
18
 
19
 
from cStringIO import StringIO
20
 
from doctest import ELLIPSIS
 
19
import gc
 
20
import doctest
21
21
import os
22
22
import signal
23
23
import sys
 
24
import threading
24
25
import time
25
26
import unittest
26
27
import warnings
27
28
 
28
 
from testtools import MultiTestResult
 
29
from testtools import (
 
30
    ExtendedToOriginalDecorator,
 
31
    MultiTestResult,
 
32
    )
 
33
from testtools.content import Content
29
34
from testtools.content_type import ContentType
30
35
from testtools.matchers import (
31
36
    DocTestMatches,
32
37
    Equals,
33
38
    )
34
 
import testtools.tests.helpers
 
39
import testtools.testresult.doubles
35
40
 
36
 
import bzrlib
37
 
from bzrlib import (
 
41
import breezy
 
42
from .. import (
38
43
    branchbuilder,
39
 
    bzrdir,
40
 
    debug,
 
44
    controldir,
41
45
    errors,
 
46
    hooks,
42
47
    lockdir,
43
48
    memorytree,
44
49
    osutils,
45
 
    progress,
46
50
    remote,
47
51
    repository,
48
52
    symbol_versioning,
50
54
    transport,
51
55
    workingtree,
52
56
    )
53
 
from bzrlib.repofmt import (
 
57
from ..bzr import (
 
58
    bzrdir,
 
59
    workingtree_3,
 
60
    workingtree_4,
 
61
    )
 
62
from ..bzr import (
54
63
    groupcompress_repo,
55
 
    pack_repo,
56
 
    weaverepo,
57
 
    )
58
 
from bzrlib.symbol_versioning import (
 
64
    )
 
65
from ..sixish import (
 
66
    StringIO,
 
67
    )
 
68
from ..symbol_versioning import (
59
69
    deprecated_function,
60
70
    deprecated_in,
61
71
    deprecated_method,
62
72
    )
63
 
from bzrlib.tests import (
 
73
from . import (
64
74
    features,
65
75
    test_lsprof,
66
76
    test_server,
67
 
    test_sftp_transport,
68
77
    TestUtil,
69
78
    )
70
 
from bzrlib.trace import note
71
 
from bzrlib.transport import memory
72
 
from bzrlib.version import _get_bzr_source_tree
 
79
from ..trace import note, mutter
 
80
from ..transport import memory
73
81
 
74
82
 
75
83
def _test_ids(test_suite):
77
85
    return [t.id() for t in tests.iter_suite_tests(test_suite)]
78
86
 
79
87
 
80
 
class SelftestTests(tests.TestCase):
81
 
 
82
 
    def test_import_tests(self):
83
 
        mod = TestUtil._load_module_by_name('bzrlib.tests.test_selftest')
84
 
        self.assertEqual(mod.SelftestTests, SelftestTests)
85
 
 
86
 
    def test_import_test_failure(self):
87
 
        self.assertRaises(ImportError,
88
 
                          TestUtil._load_module_by_name,
89
 
                          'bzrlib.no-name-yet')
90
 
 
91
 
 
92
88
class MetaTestLog(tests.TestCase):
93
89
 
94
90
    def test_logging(self):
100
96
            "text", "plain", {"charset": "utf8"})))
101
97
        self.assertThat(u"".join(log.iter_text()), Equals(self.get_log()))
102
98
        self.assertThat(self.get_log(),
103
 
            DocTestMatches(u"...a test message\n", ELLIPSIS))
104
 
 
105
 
 
106
 
class TestUnicodeFilename(tests.TestCase):
107
 
 
108
 
    def test_probe_passes(self):
109
 
        """UnicodeFilename._probe passes."""
110
 
        # We can't test much more than that because the behaviour depends
111
 
        # on the platform.
112
 
        tests.UnicodeFilename._probe()
 
99
            DocTestMatches(u"...a test message\n", doctest.ELLIPSIS))
113
100
 
114
101
 
115
102
class TestTreeShape(tests.TestCaseInTempDir):
116
103
 
117
104
    def test_unicode_paths(self):
118
 
        self.requireFeature(tests.UnicodeFilename)
 
105
        self.requireFeature(features.UnicodeFilenameFeature)
119
106
 
120
107
        filename = u'hell\u00d8'
121
108
        self.build_tree_contents([(filename, 'contents of hello')])
122
 
        self.failUnlessExists(filename)
 
109
        self.assertPathExists(filename)
 
110
 
 
111
 
 
112
class TestClassesAvailable(tests.TestCase):
 
113
    """As a convenience we expose Test* classes from breezy.tests"""
 
114
 
 
115
    def test_test_case(self):
 
116
        from . import TestCase
 
117
 
 
118
    def test_test_loader(self):
 
119
        from . import TestLoader
 
120
 
 
121
    def test_test_suite(self):
 
122
        from . import TestSuite
123
123
 
124
124
 
125
125
class TestTransportScenarios(tests.TestCase):
139
139
            def get_test_permutations(self):
140
140
                return sample_permutation
141
141
        sample_permutation = [(1,2), (3,4)]
142
 
        from bzrlib.tests.per_transport import get_transport_test_permutations
 
142
        from .per_transport import get_transport_test_permutations
143
143
        self.assertEqual(sample_permutation,
144
144
                         get_transport_test_permutations(MockModule()))
145
145
 
148
148
        # as there are in all the registered transport modules - we assume if
149
149
        # this matches its probably doing the right thing especially in
150
150
        # combination with the tests for setting the right classes below.
151
 
        from bzrlib.tests.per_transport import transport_test_permutations
152
 
        from bzrlib.transport import _get_transport_modules
 
151
        from .per_transport import transport_test_permutations
 
152
        from ..transport import _get_transport_modules
153
153
        modules = _get_transport_modules()
154
154
        permutation_count = 0
155
155
        for module in modules:
166
166
        # This test used to know about all the possible transports and the
167
167
        # order they were returned but that seems overly brittle (mbp
168
168
        # 20060307)
169
 
        from bzrlib.tests.per_transport import transport_test_permutations
 
169
        from .per_transport import transport_test_permutations
170
170
        scenarios = transport_test_permutations()
171
171
        # there are at least that many builtin transports
172
172
        self.assertTrue(len(scenarios) > 6)
173
173
        one_scenario = scenarios[0]
174
174
        self.assertIsInstance(one_scenario[0], str)
175
175
        self.assertTrue(issubclass(one_scenario[1]["transport_class"],
176
 
                                   bzrlib.transport.Transport))
 
176
                                   breezy.transport.Transport))
177
177
        self.assertTrue(issubclass(one_scenario[1]["transport_server"],
178
 
                                   bzrlib.transport.Server))
 
178
                                   breezy.transport.Server))
179
179
 
180
180
 
181
181
class TestBranchScenarios(tests.TestCase):
183
183
    def test_scenarios(self):
184
184
        # check that constructor parameters are passed through to the adapted
185
185
        # test.
186
 
        from bzrlib.tests.per_branch import make_scenarios
 
186
        from .per_branch import make_scenarios
187
187
        server1 = "a"
188
188
        server2 = "b"
189
189
        formats = [("c", "C"), ("d", "D")]
208
208
    def test_scenarios(self):
209
209
        # check that constructor parameters are passed through to the adapted
210
210
        # test.
211
 
        from bzrlib.tests.per_bzrdir import make_scenarios
 
211
        from .per_controldir import make_scenarios
212
212
        vfs_factory = "v"
213
213
        server1 = "a"
214
214
        server2 = "b"
231
231
class TestRepositoryScenarios(tests.TestCase):
232
232
 
233
233
    def test_formats_to_scenarios(self):
234
 
        from bzrlib.tests.per_repository import formats_to_scenarios
 
234
        from .per_repository import formats_to_scenarios
235
235
        formats = [("(c)", remote.RemoteRepositoryFormat()),
236
236
                   ("(d)", repository.format_registry.get(
237
237
                    'Bazaar repository format 2a (needs bzr 1.16 or later)\n'))]
272
272
    """Tests for the test adaption facilities."""
273
273
 
274
274
    def test_apply_scenario(self):
275
 
        from bzrlib.tests import apply_scenario
 
275
        from breezy.tests import apply_scenario
276
276
        input_test = TestTestScenarioApplication("test_apply_scenario")
277
277
        # setup two adapted tests
278
278
        adapted_test1 = apply_scenario(input_test,
294
294
        self.assertEqual("readonly-server",
295
295
            adapted_test1.transport_readonly_server)
296
296
        self.assertEqual(
297
 
            "bzrlib.tests.test_selftest.TestTestScenarioApplication."
 
297
            "breezy.tests.test_selftest.TestTestScenarioApplication."
298
298
            "test_apply_scenario(new id)",
299
299
            adapted_test1.id())
300
300
        self.assertEqual(None, adapted_test2.bzrdir_format)
301
301
        self.assertEqual(
302
 
            "bzrlib.tests.test_selftest.TestTestScenarioApplication."
 
302
            "breezy.tests.test_selftest.TestTestScenarioApplication."
303
303
            "test_apply_scenario(new id 2)",
304
304
            adapted_test2.id())
305
305
 
309
309
    def test_scenarios(self):
310
310
        # check that constructor parameters are passed through to the adapted
311
311
        # test.
312
 
        from bzrlib.tests.per_interrepository import make_scenarios
 
312
        from .per_interrepository import make_scenarios
313
313
        server1 = "a"
314
314
        server2 = "b"
315
 
        formats = [("C0", "C1", "C2"), ("D0", "D1", "D2")]
 
315
        formats = [("C0", "C1", "C2", "C3"), ("D0", "D1", "D2", "D3")]
316
316
        scenarios = make_scenarios(server1, server2, formats)
317
317
        self.assertEqual([
318
318
            ('C0,str,str',
319
319
             {'repository_format': 'C1',
320
320
              'repository_format_to': 'C2',
321
321
              'transport_readonly_server': 'b',
322
 
              'transport_server': 'a'}),
 
322
              'transport_server': 'a',
 
323
              'extra_setup': 'C3'}),
323
324
            ('D0,str,str',
324
325
             {'repository_format': 'D1',
325
326
              'repository_format_to': 'D2',
326
327
              'transport_readonly_server': 'b',
327
 
              'transport_server': 'a'})],
 
328
              'transport_server': 'a',
 
329
              'extra_setup': 'D3'})],
328
330
            scenarios)
329
331
 
330
332
 
333
335
    def test_scenarios(self):
334
336
        # check that constructor parameters are passed through to the adapted
335
337
        # test.
336
 
        from bzrlib.tests.per_workingtree import make_scenarios
 
338
        from .per_workingtree import make_scenarios
337
339
        server1 = "a"
338
340
        server2 = "b"
339
 
        formats = [workingtree.WorkingTreeFormat2(),
340
 
                   workingtree.WorkingTreeFormat3(),]
341
 
        scenarios = make_scenarios(server1, server2, formats)
 
341
        formats = [workingtree_4.WorkingTreeFormat4(),
 
342
                   workingtree_3.WorkingTreeFormat3(),
 
343
                   workingtree_4.WorkingTreeFormat6()]
 
344
        scenarios = make_scenarios(server1, server2, formats,
 
345
            remote_server='c', remote_readonly_server='d',
 
346
            remote_backing_server='e')
342
347
        self.assertEqual([
343
 
            ('WorkingTreeFormat2',
 
348
            ('WorkingTreeFormat4',
344
349
             {'bzrdir_format': formats[0]._matchingbzrdir,
345
350
              'transport_readonly_server': 'b',
346
351
              'transport_server': 'a',
349
354
             {'bzrdir_format': formats[1]._matchingbzrdir,
350
355
              'transport_readonly_server': 'b',
351
356
              'transport_server': 'a',
352
 
              'workingtree_format': formats[1]})],
353
 
            scenarios)
 
357
              'workingtree_format': formats[1]}),
 
358
            ('WorkingTreeFormat6',
 
359
             {'bzrdir_format': formats[2]._matchingbzrdir,
 
360
              'transport_readonly_server': 'b',
 
361
              'transport_server': 'a',
 
362
              'workingtree_format': formats[2]}),
 
363
            ('WorkingTreeFormat6,remote',
 
364
             {'bzrdir_format': formats[2]._matchingbzrdir,
 
365
              'repo_is_remote': True,
 
366
              'transport_readonly_server': 'd',
 
367
              'transport_server': 'c',
 
368
              'vfs_transport_factory': 'e',
 
369
              'workingtree_format': formats[2]}),
 
370
            ], scenarios)
354
371
 
355
372
 
356
373
class TestTreeScenarios(tests.TestCase):
357
374
 
358
375
    def test_scenarios(self):
359
376
        # the tree implementation scenario generator is meant to setup one
360
 
        # instance for each working tree format, and one additional instance
 
377
        # instance for each working tree format, one additional instance
361
378
        # that will use the default wt format, but create a revision tree for
362
 
        # the tests.  this means that the wt ones should have the
363
 
        # workingtree_to_test_tree attribute set to 'return_parameter' and the
364
 
        # revision one set to revision_tree_from_workingtree.
 
379
        # the tests, and one more that uses the default wt format as a
 
380
        # lightweight checkout of a remote repository.  This means that the wt
 
381
        # ones should have the workingtree_to_test_tree attribute set to
 
382
        # 'return_parameter' and the revision one set to
 
383
        # revision_tree_from_workingtree.
365
384
 
366
 
        from bzrlib.tests.per_tree import (
 
385
        from .per_tree import (
367
386
            _dirstate_tree_from_workingtree,
368
387
            make_scenarios,
369
388
            preview_tree_pre,
373
392
            )
374
393
        server1 = "a"
375
394
        server2 = "b"
376
 
        formats = [workingtree.WorkingTreeFormat2(),
377
 
                   workingtree.WorkingTreeFormat3(),]
 
395
        smart_server = test_server.SmartTCPServer_for_testing
 
396
        smart_readonly_server = test_server.ReadonlySmartTCPServer_for_testing
 
397
        mem_server = memory.MemoryServer
 
398
        formats = [workingtree_4.WorkingTreeFormat4(),
 
399
                   workingtree_3.WorkingTreeFormat3(),]
378
400
        scenarios = make_scenarios(server1, server2, formats)
379
 
        self.assertEqual(7, len(scenarios))
380
 
        default_wt_format = workingtree.WorkingTreeFormat4._default_format
381
 
        wt4_format = workingtree.WorkingTreeFormat4()
382
 
        wt5_format = workingtree.WorkingTreeFormat5()
 
401
        self.assertEqual(8, len(scenarios))
 
402
        default_wt_format = workingtree.format_registry.get_default()
 
403
        wt4_format = workingtree_4.WorkingTreeFormat4()
 
404
        wt5_format = workingtree_4.WorkingTreeFormat5()
 
405
        wt6_format = workingtree_4.WorkingTreeFormat6()
383
406
        expected_scenarios = [
384
 
            ('WorkingTreeFormat2',
 
407
            ('WorkingTreeFormat4',
385
408
             {'bzrdir_format': formats[0]._matchingbzrdir,
386
409
              'transport_readonly_server': 'b',
387
410
              'transport_server': 'a',
395
418
              'workingtree_format': formats[1],
396
419
              '_workingtree_to_test_tree': return_parameter,
397
420
             }),
 
421
            ('WorkingTreeFormat6,remote',
 
422
             {'bzrdir_format': wt6_format._matchingbzrdir,
 
423
              'repo_is_remote': True,
 
424
              'transport_readonly_server': smart_readonly_server,
 
425
              'transport_server': smart_server,
 
426
              'vfs_transport_factory': mem_server,
 
427
              'workingtree_format': wt6_format,
 
428
              '_workingtree_to_test_tree': return_parameter,
 
429
             }),
398
430
            ('RevisionTree',
399
431
             {'_workingtree_to_test_tree': revision_tree_from_workingtree,
400
432
              'bzrdir_format': default_wt_format._matchingbzrdir,
445
477
        # unlike the TestProviderAdapter we dont want to automatically add a
446
478
        # parameterized one for WorkingTree - the optimisers will tell us what
447
479
        # ones to add.
448
 
        from bzrlib.tests.per_tree import (
 
480
        from .per_tree import (
449
481
            return_parameter,
450
 
            revision_tree_from_workingtree
451
482
            )
452
 
        from bzrlib.tests.per_intertree import (
 
483
        from .per_intertree import (
453
484
            make_scenarios,
454
485
            )
455
 
        from bzrlib.workingtree import WorkingTreeFormat2, WorkingTreeFormat3
 
486
        from ..bzr.workingtree_3 import WorkingTreeFormat3
 
487
        from ..bzr.workingtree_4 import WorkingTreeFormat4
456
488
        input_test = TestInterTreeScenarios(
457
489
            "test_scenarios")
458
490
        server1 = "a"
459
491
        server2 = "b"
460
 
        format1 = WorkingTreeFormat2()
 
492
        format1 = WorkingTreeFormat4()
461
493
        format2 = WorkingTreeFormat3()
462
494
        formats = [("1", str, format1, format2, "converter1"),
463
495
            ("2", int, format2, format1, "converter2")]
497
529
        self.assertIsSameRealPath(self.test_home_dir, os.environ['HOME'])
498
530
 
499
531
    def test_assertEqualStat_equal(self):
500
 
        from bzrlib.tests.test_dirstate import _FakeStat
 
532
        from .test_dirstate import _FakeStat
501
533
        self.build_tree(["foo"])
502
534
        real = os.lstat("foo")
503
535
        fake = _FakeStat(real.st_size, real.st_mtime, real.st_ctime,
509
541
        self.assertRaises(AssertionError, self.assertEqualStat,
510
542
            os.lstat("foo"), os.lstat("longname"))
511
543
 
 
544
    def test_assertPathExists(self):
 
545
        self.assertPathExists('.')
 
546
        self.build_tree(['foo/', 'foo/bar'])
 
547
        self.assertPathExists('foo/bar')
 
548
        self.assertPathDoesNotExist('foo/foo')
 
549
 
512
550
 
513
551
class TestTestCaseWithMemoryTransport(tests.TestCaseWithMemoryTransport):
514
552
 
516
554
        """The test_home_dir for TestCaseWithMemoryTransport is missing.
517
555
 
518
556
        This is because TestCaseWithMemoryTransport is for tests that do not
519
 
        need any disk resources: they should be hooked into bzrlib in such a
 
557
        need any disk resources: they should be hooked into breezy in such a
520
558
        way that no global settings are being changed by the test (only a
521
559
        few tests should need to do that), and having a missing dir as home is
522
560
        an effective way to ensure that this is the case.
531
569
        cwd = osutils.getcwd()
532
570
        self.assertIsSameRealPath(self.test_dir, cwd)
533
571
 
534
 
    def test_BZR_HOME_and_HOME_are_bytestrings(self):
535
 
        """The $BZR_HOME and $HOME environment variables should not be unicode.
 
572
    def test_BRZ_HOME_and_HOME_are_bytestrings(self):
 
573
        """The $BRZ_HOME and $HOME environment variables should not be unicode.
536
574
 
537
575
        See https://bugs.launchpad.net/bzr/+bug/464174
538
576
        """
539
 
        self.assertIsInstance(os.environ['BZR_HOME'], str)
 
577
        self.assertIsInstance(os.environ['BRZ_HOME'], str)
540
578
        self.assertIsInstance(os.environ['HOME'], str)
541
579
 
542
580
    def test_make_branch_and_memory_tree(self):
548
586
        tree = self.make_branch_and_memory_tree('dir')
549
587
        # Guard against regression into MemoryTransport leaking
550
588
        # files to disk instead of keeping them in memory.
551
 
        self.failIf(osutils.lexists('dir'))
 
589
        self.assertFalse(osutils.lexists('dir'))
552
590
        self.assertIsInstance(tree, memorytree.MemoryTree)
553
591
 
554
592
    def test_make_branch_and_memory_tree_with_format(self):
555
593
        """make_branch_and_memory_tree should accept a format option."""
556
594
        format = bzrdir.BzrDirMetaFormat1()
557
 
        format.repository_format = weaverepo.RepositoryFormat7()
 
595
        format.repository_format = repository.format_registry.get_default()
558
596
        tree = self.make_branch_and_memory_tree('dir', format=format)
559
597
        # Guard against regression into MemoryTransport leaking
560
598
        # files to disk instead of keeping them in memory.
561
 
        self.failIf(osutils.lexists('dir'))
 
599
        self.assertFalse(osutils.lexists('dir'))
562
600
        self.assertIsInstance(tree, memorytree.MemoryTree)
563
601
        self.assertEqual(format.repository_format.__class__,
564
602
            tree.branch.repository._format.__class__)
568
606
        self.assertIsInstance(builder, branchbuilder.BranchBuilder)
569
607
        # Guard against regression into MemoryTransport leaking
570
608
        # files to disk instead of keeping them in memory.
571
 
        self.failIf(osutils.lexists('dir'))
 
609
        self.assertFalse(osutils.lexists('dir'))
572
610
 
573
611
    def test_make_branch_builder_with_format(self):
574
612
        # Use a repo layout that doesn't conform to a 'named' layout, to ensure
575
613
        # that the format objects are used.
576
614
        format = bzrdir.BzrDirMetaFormat1()
577
 
        repo_format = weaverepo.RepositoryFormat7()
 
615
        repo_format = repository.format_registry.get_default()
578
616
        format.repository_format = repo_format
579
617
        builder = self.make_branch_builder('dir', format=format)
580
618
        the_branch = builder.get_branch()
581
619
        # Guard against regression into MemoryTransport leaking
582
620
        # files to disk instead of keeping them in memory.
583
 
        self.failIf(osutils.lexists('dir'))
 
621
        self.assertFalse(osutils.lexists('dir'))
584
622
        self.assertEqual(format.repository_format.__class__,
585
623
                         the_branch.repository._format.__class__)
586
624
        self.assertEqual(repo_format.get_format_string(),
592
630
        the_branch = builder.get_branch()
593
631
        # Guard against regression into MemoryTransport leaking
594
632
        # files to disk instead of keeping them in memory.
595
 
        self.failIf(osutils.lexists('dir'))
596
 
        dir_format = bzrdir.format_registry.make_bzrdir('knit')
 
633
        self.assertFalse(osutils.lexists('dir'))
 
634
        dir_format = controldir.format_registry.make_controldir('knit')
597
635
        self.assertEqual(dir_format.repository_format.__class__,
598
636
                         the_branch.repository._format.__class__)
599
637
        self.assertEqual('Bazaar-NG Knit Repository Format 1',
603
641
    def test_dangling_locks_cause_failures(self):
604
642
        class TestDanglingLock(tests.TestCaseWithMemoryTransport):
605
643
            def test_function(self):
606
 
                t = self.get_transport('.')
 
644
                t = self.get_transport_from_path('.')
607
645
                l = lockdir.LockDir(t, 'lock')
608
646
                l.create()
609
647
                l.attempt_lock()
610
648
        test = TestDanglingLock('test_function')
611
649
        result = test.run()
 
650
        total_failures = result.errors + result.failures
612
651
        if self._lock_check_thorough:
613
 
            self.assertEqual(1, len(result.errors))
 
652
            self.assertEqual(1, len(total_failures))
614
653
        else:
615
654
            # When _lock_check_thorough is disabled, then we don't trigger a
616
655
            # failure
617
 
            self.assertEqual(0, len(result.errors))
 
656
            self.assertEqual(0, len(total_failures))
618
657
 
619
658
 
620
659
class TestTestCaseWithTransport(tests.TestCaseWithTransport):
621
660
    """Tests for the convenience functions TestCaseWithTransport introduces."""
622
661
 
623
662
    def test_get_readonly_url_none(self):
624
 
        from bzrlib.transport import get_transport
625
 
        from bzrlib.transport.readonly import ReadonlyTransportDecorator
 
663
        from ..transport.readonly import ReadonlyTransportDecorator
626
664
        self.vfs_transport_factory = memory.MemoryServer
627
665
        self.transport_readonly_server = None
628
666
        # calling get_readonly_transport() constructs a decorator on the url
629
667
        # for the server
630
668
        url = self.get_readonly_url()
631
669
        url2 = self.get_readonly_url('foo/bar')
632
 
        t = get_transport(url)
633
 
        t2 = get_transport(url2)
634
 
        self.failUnless(isinstance(t, ReadonlyTransportDecorator))
635
 
        self.failUnless(isinstance(t2, ReadonlyTransportDecorator))
 
670
        t = transport.get_transport_from_url(url)
 
671
        t2 = transport.get_transport_from_url(url2)
 
672
        self.assertIsInstance(t, ReadonlyTransportDecorator)
 
673
        self.assertIsInstance(t2, ReadonlyTransportDecorator)
636
674
        self.assertEqual(t2.base[:-1], t.abspath('foo/bar'))
637
675
 
638
676
    def test_get_readonly_url_http(self):
639
 
        from bzrlib.tests.http_server import HttpServer
640
 
        from bzrlib.transport import get_transport
641
 
        from bzrlib.transport.http import HttpTransportBase
 
677
        from .http_server import HttpServer
 
678
        from ..transport.http import HttpTransportBase
642
679
        self.transport_server = test_server.LocalURLServer
643
680
        self.transport_readonly_server = HttpServer
644
681
        # calling get_readonly_transport() gives us a HTTP server instance.
645
682
        url = self.get_readonly_url()
646
683
        url2 = self.get_readonly_url('foo/bar')
647
684
        # the transport returned may be any HttpTransportBase subclass
648
 
        t = get_transport(url)
649
 
        t2 = get_transport(url2)
650
 
        self.failUnless(isinstance(t, HttpTransportBase))
651
 
        self.failUnless(isinstance(t2, HttpTransportBase))
 
685
        t = transport.get_transport_from_url(url)
 
686
        t2 = transport.get_transport_from_url(url2)
 
687
        self.assertIsInstance(t, HttpTransportBase)
 
688
        self.assertIsInstance(t2, HttpTransportBase)
652
689
        self.assertEqual(t2.base[:-1], t.abspath('foo/bar'))
653
690
 
654
691
    def test_is_directory(self):
662
699
    def test_make_branch_builder(self):
663
700
        builder = self.make_branch_builder('dir')
664
701
        rev_id = builder.build_commit()
665
 
        self.failUnlessExists('dir')
666
 
        a_dir = bzrdir.BzrDir.open('dir')
 
702
        self.assertPathExists('dir')
 
703
        a_dir = controldir.ControlDir.open('dir')
667
704
        self.assertRaises(errors.NoWorkingTree, a_dir.open_workingtree)
668
705
        a_branch = a_dir.open_branch()
669
706
        builder_branch = builder.get_branch()
678
715
        super(TestTestCaseTransports, self).setUp()
679
716
        self.vfs_transport_factory = memory.MemoryServer
680
717
 
681
 
    def test_make_bzrdir_preserves_transport(self):
 
718
    def test_make_controldir_preserves_transport(self):
682
719
        t = self.get_transport()
683
 
        result_bzrdir = self.make_bzrdir('subdir')
 
720
        result_bzrdir = self.make_controldir('subdir')
684
721
        self.assertIsInstance(result_bzrdir.transport,
685
722
                              memory.MemoryTransport)
686
723
        # should not be on disk, should only be in memory
687
 
        self.failIfExists('subdir')
 
724
        self.assertPathDoesNotExist('subdir')
688
725
 
689
726
 
690
727
class TestChrootedTest(tests.ChrootedTestCase):
691
728
 
692
729
    def test_root_is_root(self):
693
 
        from bzrlib.transport import get_transport
694
 
        t = get_transport(self.get_readonly_url())
 
730
        t = transport.get_transport_from_url(self.get_readonly_url())
695
731
        url = t.base
696
732
        self.assertEqual(url, t.clone('..').base)
697
733
 
699
735
class TestProfileResult(tests.TestCase):
700
736
 
701
737
    def test_profiles_tests(self):
702
 
        self.requireFeature(test_lsprof.LSProfFeature)
703
 
        terminal = testtools.tests.helpers.ExtendedTestResult()
 
738
        self.requireFeature(features.lsprof_feature)
 
739
        terminal = testtools.testresult.doubles.ExtendedTestResult()
704
740
        result = tests.ProfileResult(terminal)
705
741
        class Sample(tests.TestCase):
706
742
            def a(self):
719
755
class TestTestResult(tests.TestCase):
720
756
 
721
757
    def check_timing(self, test_case, expected_re):
722
 
        result = bzrlib.tests.TextTestResult(self._log_file,
723
 
                descriptions=0,
724
 
                verbosity=1,
725
 
                )
726
 
        capture = testtools.tests.helpers.ExtendedTestResult()
 
758
        result = tests.TextTestResult(StringIO(), descriptions=0, verbosity=1)
 
759
        capture = testtools.testresult.doubles.ExtendedTestResult()
727
760
        test_case.run(MultiTestResult(result, capture))
728
761
        run_case = capture._events[0][1]
729
762
        timed_string = result._testTimeString(run_case)
743
776
                          r"^ +[0-9]+ms\*$")
744
777
 
745
778
    def test_unittest_reporting_unittest_class(self):
746
 
        # getting the time from a non-bzrlib test works ok
 
779
        # getting the time from a non-breezy test works ok
747
780
        class ShortDelayTestCase(unittest.TestCase):
748
781
            def test_short_delay(self):
749
782
                time.sleep(0.003)
750
783
        self.check_timing(ShortDelayTestCase('test_short_delay'),
751
784
                          r"^ +[0-9]+ms$")
752
785
 
753
 
    def _patch_get_bzr_source_tree(self):
754
 
        # Reading from the actual source tree breaks isolation, but we don't
755
 
        # want to assume that thats *all* that would happen.
756
 
        self.overrideAttr(bzrlib.version, '_get_bzr_source_tree', lambda: None)
757
 
 
758
 
    def test_assigned_benchmark_file_stores_date(self):
759
 
        self._patch_get_bzr_source_tree()
760
 
        output = StringIO()
761
 
        result = bzrlib.tests.TextTestResult(self._log_file,
762
 
                                        descriptions=0,
763
 
                                        verbosity=1,
764
 
                                        bench_history=output
765
 
                                        )
766
 
        output_string = output.getvalue()
767
 
        # if you are wondering about the regexp please read the comment in
768
 
        # test_bench_history (bzrlib.tests.test_selftest.TestRunner)
769
 
        # XXX: what comment?  -- Andrew Bennetts
770
 
        self.assertContainsRe(output_string, "--date [0-9.]+")
771
 
 
772
 
    def test_benchhistory_records_test_times(self):
773
 
        self._patch_get_bzr_source_tree()
774
 
        result_stream = StringIO()
775
 
        result = bzrlib.tests.TextTestResult(
776
 
            self._log_file,
777
 
            descriptions=0,
778
 
            verbosity=1,
779
 
            bench_history=result_stream
780
 
            )
781
 
 
782
 
        # we want profile a call and check that its test duration is recorded
783
 
        # make a new test instance that when run will generate a benchmark
784
 
        example_test_case = TestTestResult("_time_hello_world_encoding")
785
 
        # execute the test, which should succeed and record times
786
 
        example_test_case.run(result)
787
 
        lines = result_stream.getvalue().splitlines()
788
 
        self.assertEqual(2, len(lines))
789
 
        self.assertContainsRe(lines[1],
790
 
            " *[0-9]+ms bzrlib.tests.test_selftest.TestTestResult"
791
 
            "._time_hello_world_encoding")
792
 
 
793
786
    def _time_hello_world_encoding(self):
794
787
        """Profile two sleep calls
795
788
 
800
793
 
801
794
    def test_lsprofiling(self):
802
795
        """Verbose test result prints lsprof statistics from test cases."""
803
 
        self.requireFeature(test_lsprof.LSProfFeature)
 
796
        self.requireFeature(features.lsprof_feature)
804
797
        result_stream = StringIO()
805
 
        result = bzrlib.tests.VerboseTestResult(
806
 
            unittest._WritelnDecorator(result_stream),
 
798
        result = breezy.tests.VerboseTestResult(
 
799
            result_stream,
807
800
            descriptions=0,
808
801
            verbosity=2,
809
802
            )
835
828
        self.assertContainsRe(output,
836
829
            r"LSProf output for <type 'unicode'>\(\('world',\), {'errors': 'replace'}\)\n")
837
830
 
 
831
    def test_uses_time_from_testtools(self):
 
832
        """Test case timings in verbose results should use testtools times"""
 
833
        import datetime
 
834
        class TimeAddedVerboseTestResult(tests.VerboseTestResult):
 
835
            def startTest(self, test):
 
836
                self.time(datetime.datetime.utcfromtimestamp(1.145))
 
837
                super(TimeAddedVerboseTestResult, self).startTest(test)
 
838
            def addSuccess(self, test):
 
839
                self.time(datetime.datetime.utcfromtimestamp(51.147))
 
840
                super(TimeAddedVerboseTestResult, self).addSuccess(test)
 
841
            def report_tests_starting(self): pass
 
842
        sio = StringIO()
 
843
        self.get_passing_test().run(TimeAddedVerboseTestResult(sio, 0, 2))
 
844
        self.assertEndsWith(sio.getvalue(), "OK    50002ms\n")
 
845
 
838
846
    def test_known_failure(self):
839
 
        """A KnownFailure being raised should trigger several result actions."""
 
847
        """Using knownFailure should trigger several result actions."""
840
848
        class InstrumentedTestResult(tests.ExtendedTestResult):
841
849
            def stopTestRun(self): pass
842
 
            def startTests(self): pass
843
 
            def report_test_start(self, test): pass
 
850
            def report_tests_starting(self): pass
844
851
            def report_known_failure(self, test, err=None, details=None):
845
852
                self._call = test, 'known failure'
846
853
        result = InstrumentedTestResult(None, None, None, None)
847
854
        class Test(tests.TestCase):
848
855
            def test_function(self):
849
 
                raise tests.KnownFailure('failed!')
 
856
                self.knownFailure('failed!')
850
857
        test = Test("test_function")
851
858
        test.run(result)
852
859
        # it should invoke 'report_known_failure'.
863
870
    def test_verbose_report_known_failure(self):
864
871
        # verbose test output formatting
865
872
        result_stream = StringIO()
866
 
        result = bzrlib.tests.VerboseTestResult(
867
 
            unittest._WritelnDecorator(result_stream),
 
873
        result = breezy.tests.VerboseTestResult(
 
874
            result_stream,
868
875
            descriptions=0,
869
876
            verbosity=2,
870
877
            )
871
 
        test = self.get_passing_test()
872
 
        result.startTest(test)
873
 
        prefix = len(result_stream.getvalue())
874
 
        # the err parameter has the shape:
875
 
        # (class, exception object, traceback)
876
 
        # KnownFailures dont get their tracebacks shown though, so we
877
 
        # can skip that.
878
 
        err = (tests.KnownFailure, tests.KnownFailure('foo'), None)
879
 
        result.report_known_failure(test, err)
880
 
        output = result_stream.getvalue()[prefix:]
881
 
        lines = output.splitlines()
882
 
        self.assertContainsRe(lines[0], r'XFAIL *\d+ms$')
883
 
        self.assertEqual(lines[1], '    foo')
884
 
        self.assertEqual(2, len(lines))
 
878
        _get_test("test_xfail").run(result)
 
879
        self.assertContainsRe(result_stream.getvalue(),
 
880
            "\n\\S+\\.test_xfail\\s+XFAIL\\s+\\d+ms\n"
 
881
            "\\s*(?:Text attachment: )?reason"
 
882
            "(?:\n-+\n|: {{{)"
 
883
            "this_fails"
 
884
            "(?:\n-+\n|}}}\n)")
885
885
 
886
886
    def get_passing_test(self):
887
887
        """Return a test object that can't be run usefully."""
893
893
        """Test the behaviour of invoking addNotSupported."""
894
894
        class InstrumentedTestResult(tests.ExtendedTestResult):
895
895
            def stopTestRun(self): pass
896
 
            def startTests(self): pass
897
 
            def report_test_start(self, test): pass
 
896
            def report_tests_starting(self): pass
898
897
            def report_unsupported(self, test, feature):
899
898
                self._call = test, feature
900
899
        result = InstrumentedTestResult(None, None, None, None)
901
900
        test = SampleTestCase('_test_pass')
902
 
        feature = tests.Feature()
 
901
        feature = features.Feature()
903
902
        result.startTest(test)
904
903
        result.addNotSupported(test, feature)
905
904
        # it should invoke 'report_unsupported'.
918
917
    def test_verbose_report_unsupported(self):
919
918
        # verbose test output formatting
920
919
        result_stream = StringIO()
921
 
        result = bzrlib.tests.VerboseTestResult(
922
 
            unittest._WritelnDecorator(result_stream),
 
920
        result = breezy.tests.VerboseTestResult(
 
921
            result_stream,
923
922
            descriptions=0,
924
923
            verbosity=2,
925
924
            )
926
925
        test = self.get_passing_test()
927
 
        feature = tests.Feature()
 
926
        feature = features.Feature()
928
927
        result.startTest(test)
929
928
        prefix = len(result_stream.getvalue())
930
929
        result.report_unsupported(test, feature)
939
938
        """An UnavailableFeature being raised should invoke addNotSupported."""
940
939
        class InstrumentedTestResult(tests.ExtendedTestResult):
941
940
            def stopTestRun(self): pass
942
 
            def startTests(self): pass
943
 
            def report_test_start(self, test): pass
 
941
            def report_tests_starting(self): pass
944
942
            def addNotSupported(self, test, feature):
945
943
                self._call = test, feature
946
944
        result = InstrumentedTestResult(None, None, None, None)
947
 
        feature = tests.Feature()
 
945
        feature = features.Feature()
948
946
        class Test(tests.TestCase):
949
947
            def test_function(self):
950
948
                raise tests.UnavailableFeature(feature)
958
956
        self.assertEqual(0, result.error_count)
959
957
 
960
958
    def test_strict_with_unsupported_feature(self):
961
 
        result = bzrlib.tests.TextTestResult(self._log_file, descriptions=0,
962
 
                                             verbosity=1)
 
959
        result = tests.TextTestResult(StringIO(), descriptions=0, verbosity=1)
963
960
        test = self.get_passing_test()
964
961
        feature = "Unsupported Feature"
965
962
        result.addNotSupported(test, feature)
967
964
        self.assertEqual(None, result._extractBenchmarkTime(test))
968
965
 
969
966
    def test_strict_with_known_failure(self):
970
 
        result = bzrlib.tests.TextTestResult(self._log_file, descriptions=0,
971
 
                                             verbosity=1)
972
 
        test = self.get_passing_test()
973
 
        err = (tests.KnownFailure, tests.KnownFailure('foo'), None)
974
 
        result.addExpectedFailure(test, err)
 
967
        result = tests.TextTestResult(StringIO(), descriptions=0, verbosity=1)
 
968
        test = _get_test("test_xfail")
 
969
        test.run(result)
975
970
        self.assertFalse(result.wasStrictlySuccessful())
976
971
        self.assertEqual(None, result._extractBenchmarkTime(test))
977
972
 
978
973
    def test_strict_with_success(self):
979
 
        result = bzrlib.tests.TextTestResult(self._log_file, descriptions=0,
980
 
                                             verbosity=1)
 
974
        result = tests.TextTestResult(StringIO(), descriptions=0, verbosity=1)
981
975
        test = self.get_passing_test()
982
976
        result.addSuccess(test)
983
977
        self.assertTrue(result.wasStrictlySuccessful())
988
982
        class InstrumentedTestResult(tests.ExtendedTestResult):
989
983
            calls = 0
990
984
            def startTests(self): self.calls += 1
991
 
            def report_test_start(self, test): pass
992
985
        result = InstrumentedTestResult(None, None, None, None)
993
986
        def test_function():
994
987
            pass
995
988
        test = unittest.FunctionTestCase(test_function)
996
989
        test.run(result)
997
 
        self.assertEquals(1, result.calls)
998
 
 
999
 
 
1000
 
class TestUnicodeFilenameFeature(tests.TestCase):
1001
 
 
1002
 
    def test_probe_passes(self):
1003
 
        """UnicodeFilenameFeature._probe passes."""
1004
 
        # We can't test much more than that because the behaviour depends
1005
 
        # on the platform.
1006
 
        tests.UnicodeFilenameFeature._probe()
 
990
        self.assertEqual(1, result.calls)
 
991
 
 
992
    def test_startTests_only_once(self):
 
993
        """With multiple tests startTests should still only be called once"""
 
994
        class InstrumentedTestResult(tests.ExtendedTestResult):
 
995
            calls = 0
 
996
            def startTests(self): self.calls += 1
 
997
        result = InstrumentedTestResult(None, None, None, None)
 
998
        suite = unittest.TestSuite([
 
999
            unittest.FunctionTestCase(lambda: None),
 
1000
            unittest.FunctionTestCase(lambda: None)])
 
1001
        suite.run(result)
 
1002
        self.assertEqual(1, result.calls)
 
1003
        self.assertEqual(2, result.count)
1007
1004
 
1008
1005
 
1009
1006
class TestRunner(tests.TestCase):
1018
1015
        TestCaseInTempDir.TEST_ROOT
1019
1016
 
1020
1017
        There should be no tests in this file that use
1021
 
        bzrlib.tests.TextTestRunner without using this convenience method,
 
1018
        breezy.tests.TextTestRunner without using this convenience method,
1022
1019
        because of our use of global state.
1023
1020
        """
1024
1021
        old_root = tests.TestCaseInTempDir.TEST_ROOT
1025
 
        old_leak = tests.TestCase._first_thread_leaker_id
1026
1022
        try:
1027
1023
            tests.TestCaseInTempDir.TEST_ROOT = None
1028
 
            tests.TestCase._first_thread_leaker_id = None
1029
1024
            return testrunner.run(test)
1030
1025
        finally:
1031
1026
            tests.TestCaseInTempDir.TEST_ROOT = old_root
1032
 
            tests.TestCase._first_thread_leaker_id = old_leak
1033
1027
 
1034
1028
    def test_known_failure_failed_run(self):
1035
1029
        # run a test that generates a known failure which should be printed in
1040
1034
        test = unittest.TestSuite()
1041
1035
        test.addTest(Test("known_failure_test"))
1042
1036
        def failing_test():
1043
 
            self.fail('foo')
 
1037
            raise AssertionError('foo')
1044
1038
        test.addTest(unittest.FunctionTestCase(failing_test))
1045
1039
        stream = StringIO()
1046
1040
        runner = tests.TextTestRunner(stream=stream)
1047
1041
        result = self.run_test_runner(runner, test)
1048
1042
        lines = stream.getvalue().splitlines()
1049
1043
        self.assertContainsRe(stream.getvalue(),
1050
 
            '(?sm)^bzr selftest.*$'
 
1044
            '(?sm)^brz selftest.*$'
1051
1045
            '.*'
1052
1046
            '^======================================================================\n'
1053
1047
            '^FAIL: failing_test\n'
1054
1048
            '^----------------------------------------------------------------------\n'
1055
1049
            'Traceback \\(most recent call last\\):\n'
1056
1050
            '  .*' # File .*, line .*, in failing_test' - but maybe not from .pyc
1057
 
            '    self.fail\\(\'foo\'\\)\n'
 
1051
            '    raise AssertionError\\(\'foo\'\\)\n'
1058
1052
            '.*'
1059
1053
            '^----------------------------------------------------------------------\n'
1060
1054
            '.*'
1066
1060
        # the final output.
1067
1061
        class Test(tests.TestCase):
1068
1062
            def known_failure_test(self):
1069
 
                self.expectFailure('failed', self.assertTrue, False)
 
1063
                self.knownFailure("Never works...")
1070
1064
        test = Test("known_failure_test")
1071
1065
        stream = StringIO()
1072
1066
        runner = tests.TextTestRunner(stream=stream)
1078
1072
            '\n'
1079
1073
            'OK \\(known_failures=1\\)\n')
1080
1074
 
 
1075
    def test_unexpected_success_bad(self):
 
1076
        class Test(tests.TestCase):
 
1077
            def test_truth(self):
 
1078
                self.expectFailure("No absolute truth", self.assertTrue, True)
 
1079
        runner = tests.TextTestRunner(stream=StringIO())
 
1080
        result = self.run_test_runner(runner, Test("test_truth"))
 
1081
        self.assertContainsRe(runner.stream.getvalue(),
 
1082
            "=+\n"
 
1083
            "FAIL: \\S+\.test_truth\n"
 
1084
            "-+\n"
 
1085
            "(?:.*\n)*"
 
1086
            "\\s*(?:Text attachment: )?reason"
 
1087
            "(?:\n-+\n|: {{{)"
 
1088
            "No absolute truth"
 
1089
            "(?:\n-+\n|}}}\n)"
 
1090
            "(?:.*\n)*"
 
1091
            "-+\n"
 
1092
            "Ran 1 test in .*\n"
 
1093
            "\n"
 
1094
            "FAILED \\(failures=1\\)\n\\Z")
 
1095
 
1081
1096
    def test_result_decorator(self):
1082
1097
        # decorate results
1083
1098
        calls = []
1084
 
        class LoggingDecorator(tests.ForwardingResult):
 
1099
        class LoggingDecorator(ExtendedToOriginalDecorator):
1085
1100
            def startTest(self, test):
1086
 
                tests.ForwardingResult.startTest(self, test)
 
1101
                ExtendedToOriginalDecorator.startTest(self, test)
1087
1102
                calls.append('start')
1088
1103
        test = unittest.FunctionTestCase(lambda:None)
1089
1104
        stream = StringIO()
1099
1114
        class SkippingTest(tests.TestCase):
1100
1115
            def skipping_test(self):
1101
1116
                raise tests.TestSkipped('test intentionally skipped')
1102
 
        runner = tests.TextTestRunner(stream=self._log_file)
 
1117
        runner = tests.TextTestRunner(stream=StringIO())
1103
1118
        test = SkippingTest("skipping_test")
1104
1119
        result = self.run_test_runner(runner, test)
1105
1120
        self.assertTrue(result.wasSuccessful())
1119
1134
            def cleanup(self):
1120
1135
                calls.append('cleanup')
1121
1136
 
1122
 
        runner = tests.TextTestRunner(stream=self._log_file)
 
1137
        runner = tests.TextTestRunner(stream=StringIO())
1123
1138
        test = SkippedSetupTest('test_skip')
1124
1139
        result = self.run_test_runner(runner, test)
1125
1140
        self.assertTrue(result.wasSuccessful())
1131
1146
        class SkippedTest(tests.TestCase):
1132
1147
 
1133
1148
            def setUp(self):
1134
 
                tests.TestCase.setUp(self)
 
1149
                super(SkippedTest, self).setUp()
1135
1150
                calls.append('setUp')
1136
1151
                self.addCleanup(self.cleanup)
1137
1152
 
1141
1156
            def cleanup(self):
1142
1157
                calls.append('cleanup')
1143
1158
 
1144
 
        runner = tests.TextTestRunner(stream=self._log_file)
 
1159
        runner = tests.TextTestRunner(stream=StringIO())
1145
1160
        test = SkippedTest('test_skip')
1146
1161
        result = self.run_test_runner(runner, test)
1147
1162
        self.assertTrue(result.wasSuccessful())
1157
1172
        runner = tests.TextTestRunner(stream=out, verbosity=2)
1158
1173
        test = Test("not_applicable_test")
1159
1174
        result = self.run_test_runner(runner, test)
1160
 
        self._log_file.write(out.getvalue())
 
1175
        self.log(out.getvalue())
1161
1176
        self.assertTrue(result.wasSuccessful())
1162
1177
        self.assertTrue(result.wasStrictlySuccessful())
1163
1178
        self.assertContainsRe(out.getvalue(),
1164
 
                r'(?m)not_applicable_test   * N/A')
 
1179
                r'(?m)not_applicable_test  * N/A')
1165
1180
        self.assertContainsRe(out.getvalue(),
1166
1181
                r'(?m)^    this test never runs')
1167
1182
 
1168
1183
    def test_unsupported_features_listed(self):
1169
1184
        """When unsupported features are encountered they are detailed."""
1170
 
        class Feature1(tests.Feature):
 
1185
        class Feature1(features.Feature):
1171
1186
            def _probe(self): return False
1172
 
        class Feature2(tests.Feature):
 
1187
        class Feature2(features.Feature):
1173
1188
            def _probe(self): return False
1174
1189
        # create sample tests
1175
1190
        test1 = SampleTestCase('_test_pass')
1190
1205
            ],
1191
1206
            lines[-3:])
1192
1207
 
1193
 
    def _patch_get_bzr_source_tree(self):
1194
 
        # Reading from the actual source tree breaks isolation, but we don't
1195
 
        # want to assume that thats *all* that would happen.
1196
 
        self._get_source_tree_calls = []
1197
 
        def new_get():
1198
 
            self._get_source_tree_calls.append("called")
1199
 
            return None
1200
 
        self.overrideAttr(bzrlib.version, '_get_bzr_source_tree',  new_get)
1201
 
 
1202
 
    def test_bench_history(self):
1203
 
        # tests that the running the benchmark passes bench_history into
1204
 
        # the test result object. We can tell that happens if
1205
 
        # _get_bzr_source_tree is called.
1206
 
        self._patch_get_bzr_source_tree()
1207
 
        test = TestRunner('dummy_test')
1208
 
        output = StringIO()
1209
 
        runner = tests.TextTestRunner(stream=self._log_file,
1210
 
                                      bench_history=output)
1211
 
        result = self.run_test_runner(runner, test)
1212
 
        output_string = output.getvalue()
1213
 
        self.assertContainsRe(output_string, "--date [0-9.]+")
1214
 
        self.assertLength(1, self._get_source_tree_calls)
 
1208
    def test_verbose_test_count(self):
 
1209
        """A verbose test run reports the right test count at the start"""
 
1210
        suite = TestUtil.TestSuite([
 
1211
            unittest.FunctionTestCase(lambda:None),
 
1212
            unittest.FunctionTestCase(lambda:None)])
 
1213
        self.assertEqual(suite.countTestCases(), 2)
 
1214
        stream = StringIO()
 
1215
        runner = tests.TextTestRunner(stream=stream, verbosity=2)
 
1216
        # Need to use the CountingDecorator as that's what sets num_tests
 
1217
        result = self.run_test_runner(runner, tests.CountingDecorator(suite))
 
1218
        self.assertStartsWith(stream.getvalue(), "running 2 tests")
1215
1219
 
1216
1220
    def test_startTestRun(self):
1217
1221
        """run should call result.startTestRun()"""
1218
1222
        calls = []
1219
 
        class LoggingDecorator(tests.ForwardingResult):
 
1223
        class LoggingDecorator(ExtendedToOriginalDecorator):
1220
1224
            def startTestRun(self):
1221
 
                tests.ForwardingResult.startTestRun(self)
 
1225
                ExtendedToOriginalDecorator.startTestRun(self)
1222
1226
                calls.append('startTestRun')
1223
1227
        test = unittest.FunctionTestCase(lambda:None)
1224
1228
        stream = StringIO()
1230
1234
    def test_stopTestRun(self):
1231
1235
        """run should call result.stopTestRun()"""
1232
1236
        calls = []
1233
 
        class LoggingDecorator(tests.ForwardingResult):
 
1237
        class LoggingDecorator(ExtendedToOriginalDecorator):
1234
1238
            def stopTestRun(self):
1235
 
                tests.ForwardingResult.stopTestRun(self)
 
1239
                ExtendedToOriginalDecorator.stopTestRun(self)
1236
1240
                calls.append('stopTestRun')
1237
1241
        test = unittest.FunctionTestCase(lambda:None)
1238
1242
        stream = StringIO()
1241
1245
        result = self.run_test_runner(runner, test)
1242
1246
        self.assertLength(1, calls)
1243
1247
 
 
1248
    def test_unicode_test_output_on_ascii_stream(self):
 
1249
        """Showing results should always succeed even on an ascii console"""
 
1250
        class FailureWithUnicode(tests.TestCase):
 
1251
            def test_log_unicode(self):
 
1252
                self.log(u"\u2606")
 
1253
                self.fail("Now print that log!")
 
1254
        out = StringIO()
 
1255
        self.overrideAttr(osutils, "get_terminal_encoding",
 
1256
            lambda trace=False: "ascii")
 
1257
        result = self.run_test_runner(tests.TextTestRunner(stream=out),
 
1258
            FailureWithUnicode("test_log_unicode"))
 
1259
        self.assertContainsRe(out.getvalue(),
 
1260
            "(?:Text attachment: )?log"
 
1261
            "(?:\n-+\n|: {{{)"
 
1262
            "\d+\.\d+  \\\\u2606"
 
1263
            "(?:\n-+\n|}}}\n)")
 
1264
 
1244
1265
 
1245
1266
class SampleTestCase(tests.TestCase):
1246
1267
 
1252
1273
 
1253
1274
 
1254
1275
class TestTestCase(tests.TestCase):
1255
 
    """Tests that test the core bzrlib TestCase."""
 
1276
    """Tests that test the core breezy TestCase."""
1256
1277
 
1257
1278
    def test_assertLength_matches_empty(self):
1258
1279
        a_list = []
1298
1319
        self.assertEqual(1, result.testsRun)
1299
1320
 
1300
1321
    def test_debug_flags_sanitised(self):
1301
 
        """The bzrlib debug flags should be sanitised by setUp."""
 
1322
        """The breezy debug flags should be sanitised by setUp."""
1302
1323
        if 'allow_debug' in tests.selftest_debug_flags:
1303
1324
            raise tests.TestNotApplicable(
1304
1325
                '-Eallow_debug option prevents debug flag sanitisation')
1308
1329
        flags = set()
1309
1330
        if self._lock_check_thorough:
1310
1331
            flags.add('strict_locks')
1311
 
        self.assertEqual(flags, bzrlib.debug.debug_flags)
 
1332
        self.assertEqual(flags, breezy.debug.debug_flags)
1312
1333
 
1313
1334
    def change_selftest_debug_flags(self, new_flags):
1314
1335
        self.overrideAttr(tests, 'selftest_debug_flags', set(new_flags))
1315
1336
 
1316
1337
    def test_allow_debug_flag(self):
1317
 
        """The -Eallow_debug flag prevents bzrlib.debug.debug_flags from being
 
1338
        """The -Eallow_debug flag prevents breezy.debug.debug_flags from being
1318
1339
        sanitised (i.e. cleared) before running a test.
1319
1340
        """
1320
 
        self.change_selftest_debug_flags(set(['allow_debug']))
1321
 
        bzrlib.debug.debug_flags = set(['a-flag'])
 
1341
        self.change_selftest_debug_flags({'allow_debug'})
 
1342
        breezy.debug.debug_flags = {'a-flag'}
1322
1343
        class TestThatRecordsFlags(tests.TestCase):
1323
1344
            def test_foo(nested_self):
1324
 
                self.flags = set(bzrlib.debug.debug_flags)
 
1345
                self.flags = set(breezy.debug.debug_flags)
1325
1346
        test = TestThatRecordsFlags('test_foo')
1326
1347
        test.run(self.make_test_result())
1327
 
        flags = set(['a-flag'])
 
1348
        flags = {'a-flag'}
1328
1349
        if 'disable_lock_checks' not in tests.selftest_debug_flags:
1329
1350
            flags.add('strict_locks')
1330
1351
        self.assertEqual(flags, self.flags)
1333
1354
        """The -Edisable_lock_checks flag disables thorough checks."""
1334
1355
        class TestThatRecordsFlags(tests.TestCase):
1335
1356
            def test_foo(nested_self):
1336
 
                self.flags = set(bzrlib.debug.debug_flags)
 
1357
                self.flags = set(breezy.debug.debug_flags)
1337
1358
                self.test_lock_check_thorough = nested_self._lock_check_thorough
1338
1359
        self.change_selftest_debug_flags(set())
1339
1360
        test = TestThatRecordsFlags('test_foo')
1341
1362
        # By default we do strict lock checking and thorough lock/unlock
1342
1363
        # tracking.
1343
1364
        self.assertTrue(self.test_lock_check_thorough)
1344
 
        self.assertEqual(set(['strict_locks']), self.flags)
 
1365
        self.assertEqual({'strict_locks'}, self.flags)
1345
1366
        # Now set the disable_lock_checks flag, and show that this changed.
1346
 
        self.change_selftest_debug_flags(set(['disable_lock_checks']))
 
1367
        self.change_selftest_debug_flags({'disable_lock_checks'})
1347
1368
        test = TestThatRecordsFlags('test_foo')
1348
1369
        test.run(self.make_test_result())
1349
1370
        self.assertFalse(self.test_lock_check_thorough)
1352
1373
    def test_this_fails_strict_lock_check(self):
1353
1374
        class TestThatRecordsFlags(tests.TestCase):
1354
1375
            def test_foo(nested_self):
1355
 
                self.flags1 = set(bzrlib.debug.debug_flags)
 
1376
                self.flags1 = set(breezy.debug.debug_flags)
1356
1377
                self.thisFailsStrictLockCheck()
1357
 
                self.flags2 = set(bzrlib.debug.debug_flags)
 
1378
                self.flags2 = set(breezy.debug.debug_flags)
1358
1379
        # Make sure lock checking is active
1359
1380
        self.change_selftest_debug_flags(set())
1360
1381
        test = TestThatRecordsFlags('test_foo')
1361
1382
        test.run(self.make_test_result())
1362
 
        self.assertEqual(set(['strict_locks']), self.flags1)
 
1383
        self.assertEqual({'strict_locks'}, self.flags1)
1363
1384
        self.assertEqual(set(), self.flags2)
1364
1385
 
1365
1386
    def test_debug_flags_restored(self):
1366
 
        """The bzrlib debug flags should be restored to their original state
 
1387
        """The breezy debug flags should be restored to their original state
1367
1388
        after the test was run, even if allow_debug is set.
1368
1389
        """
1369
 
        self.change_selftest_debug_flags(set(['allow_debug']))
 
1390
        self.change_selftest_debug_flags({'allow_debug'})
1370
1391
        # Now run a test that modifies debug.debug_flags.
1371
 
        bzrlib.debug.debug_flags = set(['original-state'])
 
1392
        breezy.debug.debug_flags = {'original-state'}
1372
1393
        class TestThatModifiesFlags(tests.TestCase):
1373
1394
            def test_foo(self):
1374
 
                bzrlib.debug.debug_flags = set(['modified'])
 
1395
                breezy.debug.debug_flags = {'modified'}
1375
1396
        test = TestThatModifiesFlags('test_foo')
1376
1397
        test.run(self.make_test_result())
1377
 
        self.assertEqual(set(['original-state']), bzrlib.debug.debug_flags)
 
1398
        self.assertEqual({'original-state'}, breezy.debug.debug_flags)
1378
1399
 
1379
1400
    def make_test_result(self):
1380
 
        """Get a test result that writes to the test log file."""
1381
 
        return tests.TextTestResult(self._log_file, descriptions=0, verbosity=1)
 
1401
        """Get a test result that writes to a StringIO."""
 
1402
        return tests.TextTestResult(StringIO(), descriptions=0, verbosity=1)
1382
1403
 
1383
1404
    def inner_test(self):
1384
1405
        # the inner child test
1405
1426
        # one child, we should instead see the bad result inside our test with
1406
1427
        # the two children.
1407
1428
        # the outer child test
1408
 
        original_trace = bzrlib.trace._trace_file
 
1429
        original_trace = breezy.trace._trace_file
1409
1430
        outer_test = TestTestCase("outer_child")
1410
1431
        result = self.make_test_result()
1411
1432
        outer_test.run(result)
1412
 
        self.assertEqual(original_trace, bzrlib.trace._trace_file)
 
1433
        self.assertEqual(original_trace, breezy.trace._trace_file)
1413
1434
 
1414
1435
    def method_that_times_a_bit_twice(self):
1415
1436
        # call self.time twice to ensure it aggregates
1420
1441
        """Test that the TestCase.time() method accumulates a benchmark time."""
1421
1442
        sample_test = TestTestCase("method_that_times_a_bit_twice")
1422
1443
        output_stream = StringIO()
1423
 
        result = bzrlib.tests.VerboseTestResult(
1424
 
            unittest._WritelnDecorator(output_stream),
 
1444
        result = breezy.tests.VerboseTestResult(
 
1445
            output_stream,
1425
1446
            descriptions=0,
1426
1447
            verbosity=2)
1427
1448
        sample_test.run(result)
1430
1451
            r"\d+ms\*\n$")
1431
1452
 
1432
1453
    def test_hooks_sanitised(self):
1433
 
        """The bzrlib hooks should be sanitised by setUp."""
 
1454
        """The breezy hooks should be sanitised by setUp."""
1434
1455
        # Note this test won't fail with hooks that the core library doesn't
1435
1456
        # use - but it trigger with a plugin that adds hooks, so its still a
1436
1457
        # useful warning in that case.
1437
 
        self.assertEqual(bzrlib.branch.BranchHooks(),
1438
 
            bzrlib.branch.Branch.hooks)
1439
 
        self.assertEqual(bzrlib.smart.server.SmartServerHooks(),
1440
 
            bzrlib.smart.server.SmartTCPServer.hooks)
1441
 
        self.assertEqual(bzrlib.commands.CommandHooks(),
1442
 
            bzrlib.commands.Command.hooks)
 
1458
        self.assertEqual(breezy.branch.BranchHooks(), breezy.branch.Branch.hooks)
 
1459
        self.assertEqual(
 
1460
            breezy.smart.server.SmartServerHooks(),
 
1461
            breezy.smart.server.SmartTCPServer.hooks)
 
1462
        self.assertEqual(
 
1463
            breezy.commands.CommandHooks(), breezy.commands.Command.hooks)
1443
1464
 
1444
1465
    def test__gather_lsprof_in_benchmarks(self):
1445
1466
        """When _gather_lsprof_in_benchmarks is on, accumulate profile data.
1446
1467
 
1447
1468
        Each self.time() call is individually and separately profiled.
1448
1469
        """
1449
 
        self.requireFeature(test_lsprof.LSProfFeature)
 
1470
        self.requireFeature(features.lsprof_feature)
1450
1471
        # overrides the class member with an instance member so no cleanup
1451
1472
        # needed.
1452
1473
        self._gather_lsprof_in_benchmarks = True
1455
1476
        self.assertEqual(2, len(self._benchcalls))
1456
1477
        self.assertEqual((time.sleep, (0.000,), {}), self._benchcalls[0][0])
1457
1478
        self.assertEqual((time.sleep, (0.003,), {}), self._benchcalls[1][0])
1458
 
        self.assertIsInstance(self._benchcalls[0][1], bzrlib.lsprof.Stats)
1459
 
        self.assertIsInstance(self._benchcalls[1][1], bzrlib.lsprof.Stats)
 
1479
        self.assertIsInstance(self._benchcalls[0][1], breezy.lsprof.Stats)
 
1480
        self.assertIsInstance(self._benchcalls[1][1], breezy.lsprof.Stats)
1460
1481
        del self._benchcalls[:]
1461
1482
 
1462
1483
    def test_knownFailure(self):
1471
1492
        transport_server = memory.MemoryServer()
1472
1493
        transport_server.start_server()
1473
1494
        self.addCleanup(transport_server.stop_server)
1474
 
        t = transport.get_transport(transport_server.get_url())
1475
 
        bzrdir.BzrDir.create(t.base)
 
1495
        t = transport.get_transport_from_url(transport_server.get_url())
 
1496
        controldir.ControlDir.create(t.base)
1476
1497
        self.assertRaises(errors.BzrError,
1477
 
            bzrdir.BzrDir.open_from_transport, t)
 
1498
            controldir.ControlDir.open_from_transport, t)
1478
1499
        # But if we declare this as safe, we can open the bzrdir.
1479
1500
        self.permit_url(t.base)
1480
1501
        self._bzr_selftest_roots.append(t.base)
1481
 
        bzrdir.BzrDir.open_from_transport(t)
 
1502
        controldir.ControlDir.open_from_transport(t)
1482
1503
 
1483
1504
    def test_requireFeature_available(self):
1484
1505
        """self.requireFeature(available) is a no-op."""
1485
 
        class Available(tests.Feature):
 
1506
        class Available(features.Feature):
1486
1507
            def _probe(self):return True
1487
1508
        feature = Available()
1488
1509
        self.requireFeature(feature)
1489
1510
 
1490
1511
    def test_requireFeature_unavailable(self):
1491
1512
        """self.requireFeature(unavailable) raises UnavailableFeature."""
1492
 
        class Unavailable(tests.Feature):
 
1513
        class Unavailable(features.Feature):
1493
1514
            def _probe(self):return False
1494
1515
        feature = Unavailable()
1495
1516
        self.assertRaises(tests.UnavailableFeature,
1618
1639
        self.assertRaises(AssertionError,
1619
1640
            self.assertListRaises, _TestException, success_generator)
1620
1641
 
 
1642
    def _run_successful_test(self, test):
 
1643
        result = testtools.TestResult()
 
1644
        test.run(result)
 
1645
        self.assertTrue(result.wasSuccessful())
 
1646
        return result
 
1647
 
1621
1648
    def test_overrideAttr_without_value(self):
1622
1649
        self.test_attr = 'original' # Define a test attribute
1623
1650
        obj = self # Make 'obj' visible to the embedded test
1624
1651
        class Test(tests.TestCase):
1625
1652
 
1626
1653
            def setUp(self):
1627
 
                tests.TestCase.setUp(self)
 
1654
                super(Test, self).setUp()
1628
1655
                self.orig = self.overrideAttr(obj, 'test_attr')
1629
1656
 
1630
1657
            def test_value(self):
1633
1660
                obj.test_attr = 'modified'
1634
1661
                self.assertEqual('modified', obj.test_attr)
1635
1662
 
1636
 
        test = Test('test_value')
1637
 
        test.run(unittest.TestResult())
 
1663
        self._run_successful_test(Test('test_value'))
1638
1664
        self.assertEqual('original', obj.test_attr)
1639
1665
 
1640
1666
    def test_overrideAttr_with_value(self):
1643
1669
        class Test(tests.TestCase):
1644
1670
 
1645
1671
            def setUp(self):
1646
 
                tests.TestCase.setUp(self)
 
1672
                super(Test, self).setUp()
1647
1673
                self.orig = self.overrideAttr(obj, 'test_attr', new='modified')
1648
1674
 
1649
1675
            def test_value(self):
1650
1676
                self.assertEqual('original', self.orig)
1651
1677
                self.assertEqual('modified', obj.test_attr)
1652
1678
 
1653
 
        test = Test('test_value')
1654
 
        test.run(unittest.TestResult())
 
1679
        self._run_successful_test(Test('test_value'))
1655
1680
        self.assertEqual('original', obj.test_attr)
1656
1681
 
 
1682
    def test_overrideAttr_with_no_existing_value_and_value(self):
 
1683
        # Do not define the test_attribute
 
1684
        obj = self # Make 'obj' visible to the embedded test
 
1685
        class Test(tests.TestCase):
 
1686
 
 
1687
            def setUp(self):
 
1688
                tests.TestCase.setUp(self)
 
1689
                self.orig = self.overrideAttr(obj, 'test_attr', new='modified')
 
1690
 
 
1691
            def test_value(self):
 
1692
                self.assertEqual(tests._unitialized_attr, self.orig)
 
1693
                self.assertEqual('modified', obj.test_attr)
 
1694
 
 
1695
        self._run_successful_test(Test('test_value'))
 
1696
        self.assertRaises(AttributeError, getattr, obj, 'test_attr')
 
1697
 
 
1698
    def test_overrideAttr_with_no_existing_value_and_no_value(self):
 
1699
        # Do not define the test_attribute
 
1700
        obj = self # Make 'obj' visible to the embedded test
 
1701
        class Test(tests.TestCase):
 
1702
 
 
1703
            def setUp(self):
 
1704
                tests.TestCase.setUp(self)
 
1705
                self.orig = self.overrideAttr(obj, 'test_attr')
 
1706
 
 
1707
            def test_value(self):
 
1708
                self.assertEqual(tests._unitialized_attr, self.orig)
 
1709
                self.assertRaises(AttributeError, getattr, obj, 'test_attr')
 
1710
 
 
1711
        self._run_successful_test(Test('test_value'))
 
1712
        self.assertRaises(AttributeError, getattr, obj, 'test_attr')
 
1713
 
 
1714
    def test_recordCalls(self):
 
1715
        from breezy.tests import test_selftest
 
1716
        calls = self.recordCalls(
 
1717
            test_selftest, '_add_numbers')
 
1718
        self.assertEqual(test_selftest._add_numbers(2, 10),
 
1719
            12)
 
1720
        self.assertEqual(calls, [((2, 10), {})])
 
1721
 
 
1722
 
 
1723
def _add_numbers(a, b):
 
1724
    return a + b
 
1725
 
 
1726
 
 
1727
class _MissingFeature(features.Feature):
 
1728
    def _probe(self):
 
1729
        return False
 
1730
missing_feature = _MissingFeature()
 
1731
 
 
1732
 
 
1733
def _get_test(name):
 
1734
    """Get an instance of a specific example test.
 
1735
 
 
1736
    We protect this in a function so that they don't auto-run in the test
 
1737
    suite.
 
1738
    """
 
1739
 
 
1740
    class ExampleTests(tests.TestCase):
 
1741
 
 
1742
        def test_fail(self):
 
1743
            mutter('this was a failing test')
 
1744
            self.fail('this test will fail')
 
1745
 
 
1746
        def test_error(self):
 
1747
            mutter('this test errored')
 
1748
            raise RuntimeError('gotcha')
 
1749
 
 
1750
        def test_missing_feature(self):
 
1751
            mutter('missing the feature')
 
1752
            self.requireFeature(missing_feature)
 
1753
 
 
1754
        def test_skip(self):
 
1755
            mutter('this test will be skipped')
 
1756
            raise tests.TestSkipped('reason')
 
1757
 
 
1758
        def test_success(self):
 
1759
            mutter('this test succeeds')
 
1760
 
 
1761
        def test_xfail(self):
 
1762
            mutter('test with expected failure')
 
1763
            self.knownFailure('this_fails')
 
1764
 
 
1765
        def test_unexpected_success(self):
 
1766
            mutter('test with unexpected success')
 
1767
            self.expectFailure('should_fail', lambda: None)
 
1768
 
 
1769
    return ExampleTests(name)
 
1770
 
 
1771
 
 
1772
def _get_skip_reasons(result):
 
1773
    # GZ 2017-06-06: Newer testtools doesn't have this, uses detail instead
 
1774
    return result.skip_reasons
 
1775
 
 
1776
 
 
1777
class TestTestCaseLogDetails(tests.TestCase):
 
1778
 
 
1779
    def _run_test(self, test_name):
 
1780
        test = _get_test(test_name)
 
1781
        result = testtools.TestResult()
 
1782
        test.run(result)
 
1783
        return result
 
1784
 
 
1785
    def test_fail_has_log(self):
 
1786
        result = self._run_test('test_fail')
 
1787
        self.assertEqual(1, len(result.failures))
 
1788
        result_content = result.failures[0][1]
 
1789
        self.assertContainsRe(result_content,
 
1790
            '(?m)^(?:Text attachment: )?log(?:$|: )')
 
1791
        self.assertContainsRe(result_content, 'this was a failing test')
 
1792
 
 
1793
    def test_error_has_log(self):
 
1794
        result = self._run_test('test_error')
 
1795
        self.assertEqual(1, len(result.errors))
 
1796
        result_content = result.errors[0][1]
 
1797
        self.assertContainsRe(result_content,
 
1798
            '(?m)^(?:Text attachment: )?log(?:$|: )')
 
1799
        self.assertContainsRe(result_content, 'this test errored')
 
1800
 
 
1801
    def test_skip_has_no_log(self):
 
1802
        result = self._run_test('test_skip')
 
1803
        reasons = _get_skip_reasons(result)
 
1804
        self.assertEqual({'reason'}, set(reasons))
 
1805
        skips = reasons['reason']
 
1806
        self.assertEqual(1, len(skips))
 
1807
        test = skips[0]
 
1808
        self.assertFalse('log' in test.getDetails())
 
1809
 
 
1810
    def test_missing_feature_has_no_log(self):
 
1811
        # testtools doesn't know about addNotSupported, so it just gets
 
1812
        # considered as a skip
 
1813
        result = self._run_test('test_missing_feature')
 
1814
        reasons = _get_skip_reasons(result)
 
1815
        self.assertEqual({missing_feature}, set(reasons))
 
1816
        skips = reasons[missing_feature]
 
1817
        self.assertEqual(1, len(skips))
 
1818
        test = skips[0]
 
1819
        self.assertFalse('log' in test.getDetails())
 
1820
 
 
1821
    def test_xfail_has_no_log(self):
 
1822
        result = self._run_test('test_xfail')
 
1823
        self.assertEqual(1, len(result.expectedFailures))
 
1824
        result_content = result.expectedFailures[0][1]
 
1825
        self.assertNotContainsRe(result_content,
 
1826
            '(?m)^(?:Text attachment: )?log(?:$|: )')
 
1827
        self.assertNotContainsRe(result_content, 'test with expected failure')
 
1828
 
 
1829
    def test_unexpected_success_has_log(self):
 
1830
        result = self._run_test('test_unexpected_success')
 
1831
        self.assertEqual(1, len(result.unexpectedSuccesses))
 
1832
        # Inconsistency, unexpectedSuccesses is a list of tests,
 
1833
        # expectedFailures is a list of reasons?
 
1834
        test = result.unexpectedSuccesses[0]
 
1835
        details = test.getDetails()
 
1836
        self.assertTrue('log' in details)
 
1837
 
 
1838
 
 
1839
class TestTestCloning(tests.TestCase):
 
1840
    """Tests that test cloning of TestCases (as used by multiply_tests)."""
 
1841
 
 
1842
    def test_cloned_testcase_does_not_share_details(self):
 
1843
        """A TestCase cloned with clone_test does not share mutable attributes
 
1844
        such as details or cleanups.
 
1845
        """
 
1846
        class Test(tests.TestCase):
 
1847
            def test_foo(self):
 
1848
                self.addDetail('foo', Content('text/plain', lambda: 'foo'))
 
1849
        orig_test = Test('test_foo')
 
1850
        cloned_test = tests.clone_test(orig_test, orig_test.id() + '(cloned)')
 
1851
        orig_test.run(unittest.TestResult())
 
1852
        self.assertEqual('foo', orig_test.getDetails()['foo'].iter_bytes())
 
1853
        self.assertEqual(None, cloned_test.getDetails().get('foo'))
 
1854
 
 
1855
    def test_double_apply_scenario_preserves_first_scenario(self):
 
1856
        """Applying two levels of scenarios to a test preserves the attributes
 
1857
        added by both scenarios.
 
1858
        """
 
1859
        class Test(tests.TestCase):
 
1860
            def test_foo(self):
 
1861
                pass
 
1862
        test = Test('test_foo')
 
1863
        scenarios_x = [('x=1', {'x': 1}), ('x=2', {'x': 2})]
 
1864
        scenarios_y = [('y=1', {'y': 1}), ('y=2', {'y': 2})]
 
1865
        suite = tests.multiply_tests(test, scenarios_x, unittest.TestSuite())
 
1866
        suite = tests.multiply_tests(suite, scenarios_y, unittest.TestSuite())
 
1867
        all_tests = list(tests.iter_suite_tests(suite))
 
1868
        self.assertLength(4, all_tests)
 
1869
        all_xys = sorted((t.x, t.y) for t in all_tests)
 
1870
        self.assertEqual([(1, 1), (1, 2), (2, 1), (2, 2)], all_xys)
 
1871
 
1657
1872
 
1658
1873
# NB: Don't delete this; it's not actually from 0.11!
1659
1874
@deprecated_function(deprecated_in((0, 11, 0)))
1683
1898
 
1684
1899
 
1685
1900
class TestExtraAssertions(tests.TestCase):
1686
 
    """Tests for new test assertions in bzrlib test suite"""
 
1901
    """Tests for new test assertions in breezy test suite"""
1687
1902
 
1688
1903
    def test_assert_isinstance(self):
1689
1904
        self.assertIsInstance(2, int)
1690
1905
        self.assertIsInstance(u'', basestring)
1691
1906
        e = self.assertRaises(AssertionError, self.assertIsInstance, None, int)
1692
 
        self.assertEquals(str(e),
 
1907
        self.assertEqual(str(e),
1693
1908
            "None is an instance of <type 'NoneType'> rather than <type 'int'>")
1694
1909
        self.assertRaises(AssertionError, self.assertIsInstance, 23.3, int)
1695
1910
        e = self.assertRaises(AssertionError,
1696
1911
            self.assertIsInstance, None, int, "it's just not")
1697
 
        self.assertEquals(str(e),
 
1912
        self.assertEqual(str(e),
1698
1913
            "None is an instance of <type 'NoneType'> rather than <type 'int'>"
1699
1914
            ": it's just not")
1700
1915
 
1705
1920
    def test_assertEqualDiff(self):
1706
1921
        e = self.assertRaises(AssertionError,
1707
1922
                              self.assertEqualDiff, '', '\n')
1708
 
        self.assertEquals(str(e),
 
1923
        self.assertEqual(str(e),
1709
1924
                          # Don't blink ! The '+' applies to the second string
1710
1925
                          'first string is missing a final newline.\n+ \n')
1711
1926
        e = self.assertRaises(AssertionError,
1712
1927
                              self.assertEqualDiff, '\n', '')
1713
 
        self.assertEquals(str(e),
 
1928
        self.assertEqual(str(e),
1714
1929
                          # Don't blink ! The '-' applies to the second string
1715
1930
                          'second string is missing a final newline.\n- \n')
1716
1931
 
1773
1988
            warnings.warn("this is your last warning")
1774
1989
            return a + b
1775
1990
        wlist, result = self.callCatchWarnings(meth, 1, 2)
1776
 
        self.assertEquals(3, result)
 
1991
        self.assertEqual(3, result)
1777
1992
        # would like just to compare them, but UserWarning doesn't implement
1778
1993
        # eq well
1779
1994
        w0, = wlist
1780
1995
        self.assertIsInstance(w0, UserWarning)
1781
 
        self.assertEquals("this is your last warning", str(w0))
 
1996
        self.assertEqual("this is your last warning", str(w0))
1782
1997
 
1783
1998
 
1784
1999
class TestConvenienceMakers(tests.TestCaseWithTransport):
1786
2001
 
1787
2002
    def test_make_branch_and_tree_with_format(self):
1788
2003
        # we should be able to supply a format to make_branch_and_tree
1789
 
        self.make_branch_and_tree('a', format=bzrlib.bzrdir.BzrDirMetaFormat1())
1790
 
        self.make_branch_and_tree('b', format=bzrlib.bzrdir.BzrDirFormat6())
1791
 
        self.assertIsInstance(bzrlib.bzrdir.BzrDir.open('a')._format,
1792
 
                              bzrlib.bzrdir.BzrDirMetaFormat1)
1793
 
        self.assertIsInstance(bzrlib.bzrdir.BzrDir.open('b')._format,
1794
 
                              bzrlib.bzrdir.BzrDirFormat6)
 
2004
        self.make_branch_and_tree('a', format=breezy.bzr.bzrdir.BzrDirMetaFormat1())
 
2005
        self.assertIsInstance(breezy.controldir.ControlDir.open('a')._format,
 
2006
                              breezy.bzr.bzrdir.BzrDirMetaFormat1)
1795
2007
 
1796
2008
    def test_make_branch_and_memory_tree(self):
1797
2009
        # we should be able to get a new branch and a mutable tree from
1798
2010
        # TestCaseWithTransport
1799
2011
        tree = self.make_branch_and_memory_tree('a')
1800
 
        self.assertIsInstance(tree, bzrlib.memorytree.MemoryTree)
 
2012
        self.assertIsInstance(tree, breezy.memorytree.MemoryTree)
1801
2013
 
1802
2014
    def test_make_tree_for_local_vfs_backed_transport(self):
1803
2015
        # make_branch_and_tree has to use local branch and repositories
1806
2018
        self.transport_server = test_server.FakeVFATServer
1807
2019
        self.assertFalse(self.get_url('t1').startswith('file://'))
1808
2020
        tree = self.make_branch_and_tree('t1')
1809
 
        base = tree.bzrdir.root_transport.base
 
2021
        base = tree.controldir.root_transport.base
1810
2022
        self.assertStartsWith(base, 'file://')
1811
 
        self.assertEquals(tree.bzrdir.root_transport,
1812
 
                tree.branch.bzrdir.root_transport)
1813
 
        self.assertEquals(tree.bzrdir.root_transport,
1814
 
                tree.branch.repository.bzrdir.root_transport)
1815
 
 
1816
 
 
1817
 
class SelfTestHelper:
 
2023
        self.assertEqual(tree.controldir.root_transport,
 
2024
                tree.branch.controldir.root_transport)
 
2025
        self.assertEqual(tree.controldir.root_transport,
 
2026
                tree.branch.repository.controldir.root_transport)
 
2027
 
 
2028
 
 
2029
class SelfTestHelper(object):
1818
2030
 
1819
2031
    def run_selftest(self, **kwargs):
1820
2032
        """Run selftest returning its output."""
1821
2033
        output = StringIO()
1822
 
        old_transport = bzrlib.tests.default_transport
 
2034
        old_transport = breezy.tests.default_transport
1823
2035
        old_root = tests.TestCaseWithMemoryTransport.TEST_ROOT
1824
2036
        tests.TestCaseWithMemoryTransport.TEST_ROOT = None
1825
2037
        try:
1826
2038
            self.assertEqual(True, tests.selftest(stream=output, **kwargs))
1827
2039
        finally:
1828
 
            bzrlib.tests.default_transport = old_transport
 
2040
            breezy.tests.default_transport = old_transport
1829
2041
            tests.TestCaseWithMemoryTransport.TEST_ROOT = old_root
1830
2042
        output.seek(0)
1831
2043
        return output
1832
2044
 
1833
2045
 
1834
2046
class TestSelftest(tests.TestCase, SelfTestHelper):
1835
 
    """Tests of bzrlib.tests.selftest."""
 
2047
    """Tests of breezy.tests.selftest."""
1836
2048
 
1837
2049
    def test_selftest_benchmark_parameter_invokes_test_suite__benchmark__(self):
1838
2050
        factory_called = []
1841
2053
            return TestUtil.TestSuite()
1842
2054
        out = StringIO()
1843
2055
        err = StringIO()
1844
 
        self.apply_redirected(out, err, None, bzrlib.tests.selftest,
 
2056
        self.apply_redirected(out, err, None, breezy.tests.selftest,
1845
2057
            test_suite_factory=factory)
1846
2058
        self.assertEqual([True], factory_called)
1847
2059
 
1874
2086
        self.assertLength(2, output.readlines())
1875
2087
 
1876
2088
    def test_lsprof_tests(self):
1877
 
        self.requireFeature(test_lsprof.LSProfFeature)
1878
 
        calls = []
 
2089
        self.requireFeature(features.lsprof_feature)
 
2090
        results = []
1879
2091
        class Test(object):
1880
2092
            def __call__(test, result):
1881
2093
                test.run(result)
1882
2094
            def run(test, result):
1883
 
                self.assertIsInstance(result, tests.ForwardingResult)
1884
 
                calls.append("called")
 
2095
                results.append(result)
1885
2096
            def countTestCases(self):
1886
2097
                return 1
1887
2098
        self.run_selftest(test_suite_factory=Test, lsprof_tests=True)
1888
 
        self.assertLength(1, calls)
 
2099
        self.assertLength(1, results)
 
2100
        self.assertIsInstance(results.pop(), ExtendedToOriginalDecorator)
1889
2101
 
1890
2102
    def test_random(self):
1891
2103
        # test randomising by listing a number of tests.
1918
2130
 
1919
2131
    def test_starting_with_single_argument(self):
1920
2132
        output = self.run_selftest(test_suite_factory=self.factory,
1921
 
            starting_with=['bzrlib.tests.test_selftest.Test.a'],
 
2133
            starting_with=['breezy.tests.test_selftest.Test.a'],
1922
2134
            list_only=True)
1923
 
        self.assertEqual('bzrlib.tests.test_selftest.Test.a\n',
 
2135
        self.assertEqual('breezy.tests.test_selftest.Test.a\n',
1924
2136
            output.getvalue())
1925
2137
 
1926
2138
    def test_starting_with_multiple_argument(self):
1927
2139
        output = self.run_selftest(test_suite_factory=self.factory,
1928
 
            starting_with=['bzrlib.tests.test_selftest.Test.a',
1929
 
                'bzrlib.tests.test_selftest.Test.b'],
 
2140
            starting_with=['breezy.tests.test_selftest.Test.a',
 
2141
                'breezy.tests.test_selftest.Test.b'],
1930
2142
            list_only=True)
1931
 
        self.assertEqual('bzrlib.tests.test_selftest.Test.a\n'
1932
 
            'bzrlib.tests.test_selftest.Test.b\n',
 
2143
        self.assertEqual('breezy.tests.test_selftest.Test.a\n'
 
2144
            'breezy.tests.test_selftest.Test.b\n',
1933
2145
            output.getvalue())
1934
2146
 
1935
2147
    def check_transport_set(self, transport_server):
1938
2150
            captured_transport.append(a_transport)
1939
2151
        class Capture(tests.TestCase):
1940
2152
            def a(self):
1941
 
                seen_transport(bzrlib.tests.default_transport)
 
2153
                seen_transport(breezy.tests.default_transport)
1942
2154
        def factory():
1943
2155
            return TestUtil.TestSuite([Capture("a")])
1944
2156
        self.run_selftest(transport=transport_server, test_suite_factory=factory)
1946
2158
 
1947
2159
    def test_transport_sftp(self):
1948
2160
        self.requireFeature(features.paramiko)
1949
 
        from bzrlib.tests import stub_sftp
 
2161
        from breezy.tests import stub_sftp
1950
2162
        self.check_transport_set(stub_sftp.SFTPAbsoluteServer)
1951
2163
 
1952
2164
    def test_transport_memory(self):
1971
2183
            load_list='missing file name', list_only=True)
1972
2184
 
1973
2185
 
 
2186
class TestSubunitLogDetails(tests.TestCase, SelfTestHelper):
 
2187
 
 
2188
    _test_needs_features = [features.subunit]
 
2189
 
 
2190
    def run_subunit_stream(self, test_name):
 
2191
        from subunit import ProtocolTestCase
 
2192
        def factory():
 
2193
            return TestUtil.TestSuite([_get_test(test_name)])
 
2194
        stream = self.run_selftest(runner_class=tests.SubUnitBzrRunner,
 
2195
            test_suite_factory=factory)
 
2196
        test = ProtocolTestCase(stream)
 
2197
        result = testtools.TestResult()
 
2198
        test.run(result)
 
2199
        content = stream.getvalue()
 
2200
        return content, result
 
2201
 
 
2202
    def test_fail_has_log(self):
 
2203
        content, result = self.run_subunit_stream('test_fail')
 
2204
        self.assertEqual(1, len(result.failures))
 
2205
        self.assertContainsRe(content, '(?m)^log$')
 
2206
        self.assertContainsRe(content, 'this test will fail')
 
2207
 
 
2208
    def test_error_has_log(self):
 
2209
        content, result = self.run_subunit_stream('test_error')
 
2210
        self.assertContainsRe(content, '(?m)^log$')
 
2211
        self.assertContainsRe(content, 'this test errored')
 
2212
 
 
2213
    def test_skip_has_no_log(self):
 
2214
        content, result = self.run_subunit_stream('test_skip')
 
2215
        self.assertNotContainsRe(content, '(?m)^log$')
 
2216
        self.assertNotContainsRe(content, 'this test will be skipped')
 
2217
        reasons = _get_skip_reasons(result)
 
2218
        self.assertEqual({'reason'}, set(reasons))
 
2219
        skips = reasons['reason']
 
2220
        self.assertEqual(1, len(skips))
 
2221
        test = skips[0]
 
2222
        # RemotedTestCase doesn't preserve the "details"
 
2223
        ## self.assertFalse('log' in test.getDetails())
 
2224
 
 
2225
    def test_missing_feature_has_no_log(self):
 
2226
        content, result = self.run_subunit_stream('test_missing_feature')
 
2227
        self.assertNotContainsRe(content, '(?m)^log$')
 
2228
        self.assertNotContainsRe(content, 'missing the feature')
 
2229
        reasons = _get_skip_reasons(result)
 
2230
        self.assertEqual({'_MissingFeature\n'}, set(reasons))
 
2231
        skips = reasons['_MissingFeature\n']
 
2232
        self.assertEqual(1, len(skips))
 
2233
        test = skips[0]
 
2234
        # RemotedTestCase doesn't preserve the "details"
 
2235
        ## self.assertFalse('log' in test.getDetails())
 
2236
 
 
2237
    def test_xfail_has_no_log(self):
 
2238
        content, result = self.run_subunit_stream('test_xfail')
 
2239
        self.assertNotContainsRe(content, '(?m)^log$')
 
2240
        self.assertNotContainsRe(content, 'test with expected failure')
 
2241
        self.assertEqual(1, len(result.expectedFailures))
 
2242
        result_content = result.expectedFailures[0][1]
 
2243
        self.assertNotContainsRe(result_content,
 
2244
            '(?m)^(?:Text attachment: )?log(?:$|: )')
 
2245
        self.assertNotContainsRe(result_content, 'test with expected failure')
 
2246
 
 
2247
    def test_unexpected_success_has_log(self):
 
2248
        content, result = self.run_subunit_stream('test_unexpected_success')
 
2249
        self.assertContainsRe(content, '(?m)^log$')
 
2250
        self.assertContainsRe(content, 'test with unexpected success')
 
2251
        # GZ 2011-05-18: Old versions of subunit treat unexpected success as a
 
2252
        #                success, if a min version check is added remove this
 
2253
        from subunit import TestProtocolClient as _Client
 
2254
        if _Client.addUnexpectedSuccess.__func__ is _Client.addSuccess.__func__:
 
2255
            self.expectFailure('subunit treats "unexpectedSuccess"'
 
2256
                               ' as a plain success',
 
2257
                self.assertEqual, 1, len(result.unexpectedSuccesses))
 
2258
        self.assertEqual(1, len(result.unexpectedSuccesses))
 
2259
        test = result.unexpectedSuccesses[0]
 
2260
        # RemotedTestCase doesn't preserve the "details"
 
2261
        ## self.assertTrue('log' in test.getDetails())
 
2262
 
 
2263
    def test_success_has_no_log(self):
 
2264
        content, result = self.run_subunit_stream('test_success')
 
2265
        self.assertEqual(1, result.testsRun)
 
2266
        self.assertNotContainsRe(content, '(?m)^log$')
 
2267
        self.assertNotContainsRe(content, 'this test succeeds')
 
2268
 
 
2269
 
1974
2270
class TestRunBzr(tests.TestCase):
1975
2271
 
1976
2272
    out = ''
1998
2294
        self.assertEqual(['rocks'], self.argv)
1999
2295
        self.assertEqual(34, self.retcode)
2000
2296
        self.assertEqual('It sure does!\n', out)
2001
 
        self.assertEquals(out, self.out)
 
2297
        self.assertEqual(out, self.out)
2002
2298
        self.assertEqual('', err)
2003
 
        self.assertEquals(err, self.err)
 
2299
        self.assertEqual(err, self.err)
2004
2300
 
2005
2301
    def test_run_bzr_error_regexes(self):
2006
2302
        self.out = ''
2073
2369
    def apply_redirected(self, stdin=None, stdout=None, stderr=None,
2074
2370
                         a_callable=None, *args, **kwargs):
2075
2371
        self.stdin = stdin
2076
 
        self.factory_stdin = getattr(bzrlib.ui.ui_factory, "stdin", None)
2077
 
        self.factory = bzrlib.ui.ui_factory
 
2372
        self.factory_stdin = getattr(breezy.ui.ui_factory, "stdin", None)
 
2373
        self.factory = breezy.ui.ui_factory
2078
2374
        self.working_dir = osutils.getcwd()
2079
2375
        stdout.write('foo\n')
2080
2376
        stderr.write('bar\n')
2097
2393
        # own UI factory, which is an instance of TestUIFactory,
2098
2394
        # with stdin, stdout and stderr attached to the stdin,
2099
2395
        # stdout and stderr of the invoked run_bzr
2100
 
        current_factory = bzrlib.ui.ui_factory
 
2396
        current_factory = breezy.ui.ui_factory
2101
2397
        self.run_bzr(['foo'])
2102
 
        self.failIf(current_factory is self.factory)
 
2398
        self.assertFalse(current_factory is self.factory)
2103
2399
        self.assertNotEqual(sys.stdout, self.factory.stdout)
2104
2400
        self.assertNotEqual(sys.stderr, self.factory.stderr)
2105
2401
        self.assertEqual('foo\n', self.factory.stdout.getvalue())
2146
2442
    """Base class for tests testing how we might run bzr."""
2147
2443
 
2148
2444
    def setUp(self):
2149
 
        tests.TestCaseWithTransport.setUp(self)
 
2445
        super(TestWithFakedStartBzrSubprocess, self).setUp()
2150
2446
        self.subprocess_calls = []
2151
2447
 
2152
2448
    def start_bzr_subprocess(self, process_args, env_changes=None,
2175
2471
            result = self.run_bzr_subprocess(*args, **kwargs)
2176
2472
        except:
2177
2473
            self.next_subprocess = None
2178
 
            for key, expected in expected_args.iteritems():
 
2474
            for key, expected in expected_args.items():
2179
2475
                self.assertEqual(expected, self.subprocess_calls[-1][key])
2180
2476
            raise
2181
2477
        else:
2182
2478
            self.next_subprocess = None
2183
 
            for key, expected in expected_args.iteritems():
 
2479
            for key, expected in expected_args.items():
2184
2480
                self.assertEqual(expected, self.subprocess_calls[-1][key])
2185
2481
            return result
2186
2482
 
2262
2558
 
2263
2559
 
2264
2560
class TestStartBzrSubProcess(tests.TestCase):
 
2561
    """Stub test start_bzr_subprocess."""
2265
2562
 
2266
 
    def check_popen_state(self):
2267
 
        """Replace to make assertions when popen is called."""
 
2563
    def _subprocess_log_cleanup(self):
 
2564
        """Inhibits the base version as we don't produce a log file."""
2268
2565
 
2269
2566
    def _popen(self, *args, **kwargs):
2270
 
        """Record the command that is run, so that we can ensure it is correct"""
 
2567
        """Override the base version to record the command that is run.
 
2568
 
 
2569
        From there we can ensure it is correct without spawning a real process.
 
2570
        """
2271
2571
        self.check_popen_state()
2272
2572
        self._popen_args = args
2273
2573
        self._popen_kwargs = kwargs
2274
2574
        raise _DontSpawnProcess()
2275
2575
 
 
2576
    def check_popen_state(self):
 
2577
        """Replace to make assertions when popen is called."""
 
2578
 
2276
2579
    def test_run_bzr_subprocess_no_plugins(self):
2277
2580
        self.assertRaises(_DontSpawnProcess, self.start_bzr_subprocess, [])
2278
2581
        command = self._popen_args[0]
2279
2582
        self.assertEqual(sys.executable, command[0])
2280
 
        self.assertEqual(self.get_bzr_path(), command[1])
 
2583
        self.assertEqual(self.get_brz_path(), command[1])
2281
2584
        self.assertEqual(['--no-plugins'], command[2:])
2282
2585
 
2283
2586
    def test_allow_plugins(self):
2284
2587
        self.assertRaises(_DontSpawnProcess, self.start_bzr_subprocess, [],
2285
 
            allow_plugins=True)
 
2588
                          allow_plugins=True)
2286
2589
        command = self._popen_args[0]
2287
2590
        self.assertEqual([], command[2:])
2288
2591
 
2289
2592
    def test_set_env(self):
2290
 
        self.failIf('EXISTANT_ENV_VAR' in os.environ)
 
2593
        self.assertFalse('EXISTANT_ENV_VAR' in os.environ)
2291
2594
        # set in the child
2292
2595
        def check_environment():
2293
2596
            self.assertEqual('set variable', os.environ['EXISTANT_ENV_VAR'])
2294
2597
        self.check_popen_state = check_environment
2295
2598
        self.assertRaises(_DontSpawnProcess, self.start_bzr_subprocess, [],
2296
 
            env_changes={'EXISTANT_ENV_VAR':'set variable'})
 
2599
                          env_changes={'EXISTANT_ENV_VAR':'set variable'})
2297
2600
        # not set in theparent
2298
2601
        self.assertFalse('EXISTANT_ENV_VAR' in os.environ)
2299
2602
 
2300
2603
    def test_run_bzr_subprocess_env_del(self):
2301
2604
        """run_bzr_subprocess can remove environment variables too."""
2302
 
        self.failIf('EXISTANT_ENV_VAR' in os.environ)
 
2605
        self.assertFalse('EXISTANT_ENV_VAR' in os.environ)
2303
2606
        def check_environment():
2304
2607
            self.assertFalse('EXISTANT_ENV_VAR' in os.environ)
2305
2608
        os.environ['EXISTANT_ENV_VAR'] = 'set variable'
2306
2609
        self.check_popen_state = check_environment
2307
2610
        self.assertRaises(_DontSpawnProcess, self.start_bzr_subprocess, [],
2308
 
            env_changes={'EXISTANT_ENV_VAR':None})
 
2611
                          env_changes={'EXISTANT_ENV_VAR':None})
2309
2612
        # Still set in parent
2310
2613
        self.assertEqual('set variable', os.environ['EXISTANT_ENV_VAR'])
2311
2614
        del os.environ['EXISTANT_ENV_VAR']
2312
2615
 
2313
2616
    def test_env_del_missing(self):
2314
 
        self.failIf('NON_EXISTANT_ENV_VAR' in os.environ)
 
2617
        self.assertFalse('NON_EXISTANT_ENV_VAR' in os.environ)
2315
2618
        def check_environment():
2316
2619
            self.assertFalse('NON_EXISTANT_ENV_VAR' in os.environ)
2317
2620
        self.check_popen_state = check_environment
2318
2621
        self.assertRaises(_DontSpawnProcess, self.start_bzr_subprocess, [],
2319
 
            env_changes={'NON_EXISTANT_ENV_VAR':None})
 
2622
                          env_changes={'NON_EXISTANT_ENV_VAR':None})
2320
2623
 
2321
2624
    def test_working_dir(self):
2322
2625
        """Test that we can specify the working dir for the child"""
2325
2628
        chdirs = []
2326
2629
        def chdir(path):
2327
2630
            chdirs.append(path)
2328
 
        os.chdir = chdir
2329
 
        try:
2330
 
            def getcwd():
2331
 
                return 'current'
2332
 
            osutils.getcwd = getcwd
2333
 
            try:
2334
 
                self.assertRaises(_DontSpawnProcess, self.start_bzr_subprocess, [],
2335
 
                    working_dir='foo')
2336
 
            finally:
2337
 
                osutils.getcwd = orig_getcwd
2338
 
        finally:
2339
 
            os.chdir = orig_chdir
 
2631
        self.overrideAttr(os, 'chdir', chdir)
 
2632
        def getcwd():
 
2633
            return 'current'
 
2634
        self.overrideAttr(osutils, 'getcwd', getcwd)
 
2635
        self.assertRaises(_DontSpawnProcess, self.start_bzr_subprocess, [],
 
2636
                          working_dir='foo')
2340
2637
        self.assertEqual(['foo', 'current'], chdirs)
2341
2638
 
 
2639
    def test_get_brz_path_with_cwd_breezy(self):
 
2640
        self.get_source_path = lambda: ""
 
2641
        self.overrideAttr(os.path, "isfile", lambda path: True)
 
2642
        self.assertEqual(self.get_brz_path(), "brz")
 
2643
 
2342
2644
 
2343
2645
class TestActuallyStartBzrSubprocess(tests.TestCaseWithTransport):
2344
2646
    """Tests that really need to do things with an external bzr."""
2354
2656
        result = self.finish_bzr_subprocess(process, send_signal=signal.SIGINT,
2355
2657
                                            retcode=3)
2356
2658
        self.assertEqual('', result[0])
2357
 
        self.assertEqual('bzr: interrupted\n', result[1])
2358
 
 
2359
 
 
2360
 
class TestFeature(tests.TestCase):
2361
 
 
2362
 
    def test_caching(self):
2363
 
        """Feature._probe is called by the feature at most once."""
2364
 
        class InstrumentedFeature(tests.Feature):
2365
 
            def __init__(self):
2366
 
                super(InstrumentedFeature, self).__init__()
2367
 
                self.calls = []
2368
 
            def _probe(self):
2369
 
                self.calls.append('_probe')
2370
 
                return False
2371
 
        feature = InstrumentedFeature()
2372
 
        feature.available()
2373
 
        self.assertEqual(['_probe'], feature.calls)
2374
 
        feature.available()
2375
 
        self.assertEqual(['_probe'], feature.calls)
2376
 
 
2377
 
    def test_named_str(self):
2378
 
        """Feature.__str__ should thunk to feature_name()."""
2379
 
        class NamedFeature(tests.Feature):
2380
 
            def feature_name(self):
2381
 
                return 'symlinks'
2382
 
        feature = NamedFeature()
2383
 
        self.assertEqual('symlinks', str(feature))
2384
 
 
2385
 
    def test_default_str(self):
2386
 
        """Feature.__str__ should default to __class__.__name__."""
2387
 
        class NamedFeature(tests.Feature):
2388
 
            pass
2389
 
        feature = NamedFeature()
2390
 
        self.assertEqual('NamedFeature', str(feature))
2391
 
 
2392
 
 
2393
 
class TestUnavailableFeature(tests.TestCase):
2394
 
 
2395
 
    def test_access_feature(self):
2396
 
        feature = tests.Feature()
2397
 
        exception = tests.UnavailableFeature(feature)
2398
 
        self.assertIs(feature, exception.args[0])
2399
 
 
2400
 
 
2401
 
simple_thunk_feature = tests._CompatabilityThunkFeature(
2402
 
    deprecated_in((2, 1, 0)),
2403
 
    'bzrlib.tests.test_selftest',
2404
 
    'simple_thunk_feature','UnicodeFilename',
2405
 
    replacement_module='bzrlib.tests'
2406
 
    )
2407
 
 
2408
 
class Test_CompatibilityFeature(tests.TestCase):
2409
 
 
2410
 
    def test_does_thunk(self):
2411
 
        res = self.callDeprecated(
2412
 
            ['bzrlib.tests.test_selftest.simple_thunk_feature was deprecated'
2413
 
             ' in version 2.1.0. Use bzrlib.tests.UnicodeFilename instead.'],
2414
 
            simple_thunk_feature.available)
2415
 
        self.assertEqual(tests.UnicodeFilename.available(), res)
2416
 
 
2417
 
 
2418
 
class TestModuleAvailableFeature(tests.TestCase):
2419
 
 
2420
 
    def test_available_module(self):
2421
 
        feature = tests.ModuleAvailableFeature('bzrlib.tests')
2422
 
        self.assertEqual('bzrlib.tests', feature.module_name)
2423
 
        self.assertEqual('bzrlib.tests', str(feature))
2424
 
        self.assertTrue(feature.available())
2425
 
        self.assertIs(tests, feature.module)
2426
 
 
2427
 
    def test_unavailable_module(self):
2428
 
        feature = tests.ModuleAvailableFeature('bzrlib.no_such_module_exists')
2429
 
        self.assertEqual('bzrlib.no_such_module_exists', str(feature))
2430
 
        self.assertFalse(feature.available())
2431
 
        self.assertIs(None, feature.module)
 
2659
        self.assertEqual('brz: interrupted\n', result[1])
2432
2660
 
2433
2661
 
2434
2662
class TestSelftestFiltering(tests.TestCase):
2435
2663
 
2436
2664
    def setUp(self):
2437
 
        tests.TestCase.setUp(self)
 
2665
        super(TestSelftestFiltering, self).setUp()
2438
2666
        self.suite = TestUtil.TestSuite()
2439
2667
        self.loader = TestUtil.TestLoader()
2440
2668
        self.suite.addTest(self.loader.loadTestsFromModule(
2441
 
            sys.modules['bzrlib.tests.test_selftest']))
 
2669
            sys.modules['breezy.tests.test_selftest']))
2442
2670
        self.all_names = _test_ids(self.suite)
2443
2671
 
2444
2672
    def test_condition_id_re(self):
2445
 
        test_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2673
        test_name = ('breezy.tests.test_selftest.TestSelftestFiltering.'
2446
2674
            'test_condition_id_re')
2447
2675
        filtered_suite = tests.filter_suite_by_condition(
2448
2676
            self.suite, tests.condition_id_re('test_condition_id_re'))
2449
2677
        self.assertEqual([test_name], _test_ids(filtered_suite))
2450
2678
 
2451
2679
    def test_condition_id_in_list(self):
2452
 
        test_names = ['bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2680
        test_names = ['breezy.tests.test_selftest.TestSelftestFiltering.'
2453
2681
                      'test_condition_id_in_list']
2454
2682
        id_list = tests.TestIdList(test_names)
2455
2683
        filtered_suite = tests.filter_suite_by_condition(
2459
2687
        self.assertEqual(_test_ids(re_filtered), _test_ids(filtered_suite))
2460
2688
 
2461
2689
    def test_condition_id_startswith(self):
2462
 
        klass = 'bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2690
        klass = 'breezy.tests.test_selftest.TestSelftestFiltering.'
2463
2691
        start1 = klass + 'test_condition_id_starts'
2464
2692
        start2 = klass + 'test_condition_id_in'
2465
2693
        test_names = [ klass + 'test_condition_id_in_list',
2472
2700
    def test_condition_isinstance(self):
2473
2701
        filtered_suite = tests.filter_suite_by_condition(
2474
2702
            self.suite, tests.condition_isinstance(self.__class__))
2475
 
        class_pattern = 'bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2703
        class_pattern = 'breezy.tests.test_selftest.TestSelftestFiltering.'
2476
2704
        re_filtered = tests.filter_suite_by_re(self.suite, class_pattern)
2477
2705
        self.assertEqual(_test_ids(re_filtered), _test_ids(filtered_suite))
2478
2706
 
2479
2707
    def test_exclude_tests_by_condition(self):
2480
 
        excluded_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2708
        excluded_name = ('breezy.tests.test_selftest.TestSelftestFiltering.'
2481
2709
            'test_exclude_tests_by_condition')
2482
2710
        filtered_suite = tests.exclude_tests_by_condition(self.suite,
2483
2711
            lambda x:x.id() == excluded_name)
2492
2720
        self.all_names = _test_ids(self.suite)
2493
2721
        filtered_suite = tests.exclude_tests_by_re(self.suite,
2494
2722
                                                   'exclude_tests_by_re')
2495
 
        excluded_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2723
        excluded_name = ('breezy.tests.test_selftest.TestSelftestFiltering.'
2496
2724
            'test_exclude_tests_by_re')
2497
2725
        self.assertEqual(len(self.all_names) - 1,
2498
2726
            filtered_suite.countTestCases())
2502
2730
        self.assertEqual(remaining_names, _test_ids(filtered_suite))
2503
2731
 
2504
2732
    def test_filter_suite_by_condition(self):
2505
 
        test_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2733
        test_name = ('breezy.tests.test_selftest.TestSelftestFiltering.'
2506
2734
            'test_filter_suite_by_condition')
2507
2735
        filtered_suite = tests.filter_suite_by_condition(self.suite,
2508
2736
            lambda x:x.id() == test_name)
2512
2740
        filtered_suite = tests.filter_suite_by_re(self.suite,
2513
2741
                                                  'test_filter_suite_by_r')
2514
2742
        filtered_names = _test_ids(filtered_suite)
2515
 
        self.assertEqual(filtered_names, ['bzrlib.tests.test_selftest.'
 
2743
        self.assertEqual(filtered_names, ['breezy.tests.test_selftest.'
2516
2744
            'TestSelftestFiltering.test_filter_suite_by_re'])
2517
2745
 
2518
2746
    def test_filter_suite_by_id_list(self):
2519
 
        test_list = ['bzrlib.tests.test_selftest.'
 
2747
        test_list = ['breezy.tests.test_selftest.'
2520
2748
                     'TestSelftestFiltering.test_filter_suite_by_id_list']
2521
2749
        filtered_suite = tests.filter_suite_by_id_list(
2522
2750
            self.suite, tests.TestIdList(test_list))
2523
2751
        filtered_names = _test_ids(filtered_suite)
2524
2752
        self.assertEqual(
2525
2753
            filtered_names,
2526
 
            ['bzrlib.tests.test_selftest.'
 
2754
            ['breezy.tests.test_selftest.'
2527
2755
             'TestSelftestFiltering.test_filter_suite_by_id_list'])
2528
2756
 
2529
2757
    def test_filter_suite_by_id_startswith(self):
2530
2758
        # By design this test may fail if another test is added whose name also
2531
2759
        # begins with one of the start value used.
2532
 
        klass = 'bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2760
        klass = 'breezy.tests.test_selftest.TestSelftestFiltering.'
2533
2761
        start1 = klass + 'test_filter_suite_by_id_starts'
2534
2762
        start2 = klass + 'test_filter_suite_by_id_li'
2535
2763
        test_list = [klass + 'test_filter_suite_by_id_list',
2566
2794
        self.all_names = _test_ids(self.suite)
2567
2795
        condition = tests.condition_id_re('test_filter_suite_by_r')
2568
2796
        split_suite = tests.split_suite_by_condition(self.suite, condition)
2569
 
        filtered_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2797
        filtered_name = ('breezy.tests.test_selftest.TestSelftestFiltering.'
2570
2798
            'test_filter_suite_by_re')
2571
2799
        self.assertEqual([filtered_name], _test_ids(split_suite[0]))
2572
2800
        self.assertFalse(filtered_name in _test_ids(split_suite[1]))
2578
2806
        self.all_names = _test_ids(self.suite)
2579
2807
        split_suite = tests.split_suite_by_re(self.suite,
2580
2808
                                              'test_filter_suite_by_r')
2581
 
        filtered_name = ('bzrlib.tests.test_selftest.TestSelftestFiltering.'
 
2809
        filtered_name = ('breezy.tests.test_selftest.TestSelftestFiltering.'
2582
2810
            'test_filter_suite_by_re')
2583
2811
        self.assertEqual([filtered_name], _test_ids(split_suite[0]))
2584
2812
        self.assertFalse(filtered_name in _test_ids(split_suite[1]))
2587
2815
        self.assertEqual(remaining_names, _test_ids(split_suite[1]))
2588
2816
 
2589
2817
 
2590
 
class TestCheckInventoryShape(tests.TestCaseWithTransport):
 
2818
class TestCheckTreeShape(tests.TestCaseWithTransport):
2591
2819
 
2592
 
    def test_check_inventory_shape(self):
 
2820
    def test_check_tree_shape(self):
2593
2821
        files = ['a', 'b/', 'b/c']
2594
2822
        tree = self.make_branch_and_tree('.')
2595
2823
        self.build_tree(files)
2596
2824
        tree.add(files)
2597
2825
        tree.lock_read()
2598
2826
        try:
2599
 
            self.check_inventory_shape(tree.inventory, files)
 
2827
            self.check_tree_shape(tree, files)
2600
2828
        finally:
2601
2829
            tree.unlock()
2602
2830
 
2613
2841
            self.run_bzr, ['assert-fail'])
2614
2842
        # make sure we got the real thing, not an error from somewhere else in
2615
2843
        # the test framework
2616
 
        self.assertEquals('always fails', str(e))
 
2844
        self.assertEqual('always fails', str(e))
2617
2845
        # check that there's no traceback in the test log
2618
2846
        self.assertNotContainsRe(self.get_log(), r'Traceback')
2619
2847
 
2629
2857
        out, err = self.run_bzr(["log", "%s/nonexistantpath" % url], retcode=3)
2630
2858
        self.assertEqual(out, '')
2631
2859
        self.assertContainsRe(err,
2632
 
            'bzr: ERROR: Not a branch: ".*nonexistantpath/".\n')
 
2860
            'brz: ERROR: Not a branch: ".*nonexistantpath/".\n')
2633
2861
 
2634
2862
 
2635
2863
class TestTestLoader(tests.TestCase):
2646
2874
            pass
2647
2875
        MyModule.a_class = Stub
2648
2876
        module = MyModule()
 
2877
        module.__name__ = 'fake_module'
2649
2878
        return loader, module
2650
2879
 
2651
2880
    def test_module_no_load_tests_attribute_loads_classes(self):
2654
2883
 
2655
2884
    def test_module_load_tests_attribute_gets_called(self):
2656
2885
        loader, module = self._get_loader_and_module()
2657
 
        # 'self' is here because we're faking the module with a class. Regular
2658
 
        # load_tests do not need that :)
2659
 
        def load_tests(self, standard_tests, module, loader):
 
2886
        def load_tests(loader, standard_tests, pattern):
2660
2887
            result = loader.suiteClass()
2661
2888
            for test in tests.iter_suite_tests(standard_tests):
2662
2889
                result.addTests([test, test])
2663
2890
            return result
2664
2891
        # add a load_tests() method which multiplies the tests from the module.
2665
 
        module.__class__.load_tests = load_tests
2666
 
        self.assertEqual(2, loader.loadTestsFromModule(module).countTestCases())
 
2892
        module.__class__.load_tests = staticmethod(load_tests)
 
2893
        self.assertEqual(
 
2894
            2 * [str(module.a_class('test_foo'))],
 
2895
            list(map(str, loader.loadTestsFromModule(module))))
2667
2896
 
2668
2897
    def test_load_tests_from_module_name_smoke_test(self):
2669
2898
        loader = TestUtil.TestLoader()
2670
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2671
 
        self.assertEquals(['bzrlib.tests.test_sampler.DemoTest.test_nothing'],
 
2899
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
 
2900
        self.assertEqual(['breezy.tests.test_sampler.DemoTest.test_nothing'],
2672
2901
                          _test_ids(suite))
2673
2902
 
2674
2903
    def test_load_tests_from_module_name_with_bogus_module_name(self):
2703
2932
 
2704
2933
    def test_empty_list(self):
2705
2934
        id_list = self._create_id_list([])
2706
 
        self.assertEquals({}, id_list.tests)
2707
 
        self.assertEquals({}, id_list.modules)
 
2935
        self.assertEqual({}, id_list.tests)
 
2936
        self.assertEqual({}, id_list.modules)
2708
2937
 
2709
2938
    def test_valid_list(self):
2710
2939
        id_list = self._create_id_list(
2733
2962
 
2734
2963
    def test_test_suite_matches_id_list_with_unknown(self):
2735
2964
        loader = TestUtil.TestLoader()
2736
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2737
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing',
 
2965
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
 
2966
        test_list = ['breezy.tests.test_sampler.DemoTest.test_nothing',
2738
2967
                     'bogus']
2739
2968
        not_found, duplicates = tests.suite_matches_id_list(suite, test_list)
2740
 
        self.assertEquals(['bogus'], not_found)
2741
 
        self.assertEquals([], duplicates)
 
2969
        self.assertEqual(['bogus'], not_found)
 
2970
        self.assertEqual([], duplicates)
2742
2971
 
2743
2972
    def test_suite_matches_id_list_with_duplicates(self):
2744
2973
        loader = TestUtil.TestLoader()
2745
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
 
2974
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
2746
2975
        dupes = loader.suiteClass()
2747
2976
        for test in tests.iter_suite_tests(suite):
2748
2977
            dupes.addTest(test)
2749
2978
            dupes.addTest(test) # Add it again
2750
2979
 
2751
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing',]
 
2980
        test_list = ['breezy.tests.test_sampler.DemoTest.test_nothing',]
2752
2981
        not_found, duplicates = tests.suite_matches_id_list(
2753
2982
            dupes, test_list)
2754
 
        self.assertEquals([], not_found)
2755
 
        self.assertEquals(['bzrlib.tests.test_sampler.DemoTest.test_nothing'],
 
2983
        self.assertEqual([], not_found)
 
2984
        self.assertEqual(['breezy.tests.test_sampler.DemoTest.test_nothing'],
2756
2985
                          duplicates)
2757
2986
 
2758
2987
 
2763
2992
        # by _test_suite_testmod_names.
2764
2993
        test_list = tests._test_suite_testmod_names()
2765
2994
        self.assertSubset([
2766
 
            'bzrlib.tests.blackbox',
2767
 
            'bzrlib.tests.per_transport',
2768
 
            'bzrlib.tests.test_selftest',
 
2995
            'breezy.tests.blackbox',
 
2996
            'breezy.tests.per_transport',
 
2997
            'breezy.tests.test_selftest',
2769
2998
            ],
2770
2999
            test_list)
2771
3000
 
2778
3007
            self.assertEqual([], test_list)
2779
3008
            return
2780
3009
        self.assertSubset([
2781
 
            'bzrlib.timestamp',
 
3010
            'breezy.timestamp',
2782
3011
            ],
2783
3012
            test_list)
2784
3013
 
2792
3021
        def testmod_names():
2793
3022
            calls.append("testmod_names")
2794
3023
            return [
2795
 
                'bzrlib.tests.blackbox.test_branch',
2796
 
                'bzrlib.tests.per_transport',
2797
 
                'bzrlib.tests.test_selftest',
 
3024
                'breezy.tests.blackbox.test_branch',
 
3025
                'breezy.tests.per_transport',
 
3026
                'breezy.tests.test_selftest',
2798
3027
                ]
2799
3028
        self.overrideAttr(tests, '_test_suite_testmod_names', testmod_names)
2800
3029
        def doctests():
2801
3030
            calls.append("modules_to_doctest")
2802
3031
            if __doc__ is None:
2803
3032
                return []
2804
 
            return ['bzrlib.timestamp']
 
3033
            return ['breezy.timestamp']
2805
3034
        self.overrideAttr(tests, '_test_suite_modules_to_doctest', doctests)
2806
3035
        expected_test_list = [
2807
3036
            # testmod_names
2808
 
            'bzrlib.tests.blackbox.test_branch.TestBranch.test_branch',
2809
 
            ('bzrlib.tests.per_transport.TransportTests'
 
3037
            'breezy.tests.blackbox.test_branch.TestBranch.test_branch',
 
3038
            ('breezy.tests.per_transport.TransportTests'
2810
3039
             '.test_abspath(LocalTransport,LocalURLServer)'),
2811
 
            'bzrlib.tests.test_selftest.TestTestSuite.test_test_suite',
 
3040
            'breezy.tests.test_selftest.TestTestSuite.test_test_suite',
2812
3041
            # plugins can't be tested that way since selftest may be run with
2813
3042
            # --no-plugins
2814
3043
            ]
2815
3044
        if __doc__ is not None:
2816
3045
            expected_test_list.extend([
2817
3046
                # modules_to_doctest
2818
 
                'bzrlib.timestamp.format_highres_date',
 
3047
                'breezy.timestamp.format_highres_date',
2819
3048
                ])
2820
3049
        suite = tests.test_suite()
2821
 
        self.assertEqual(set(["testmod_names", "modules_to_doctest"]),
 
3050
        self.assertEqual({"testmod_names", "modules_to_doctest"},
2822
3051
            set(calls))
2823
3052
        self.assertSubset(expected_test_list, _test_ids(suite))
2824
3053
 
2827
3056
        # to know that starting_with == None works. So a second load is
2828
3057
        # incurred - note that the starting_with parameter causes a partial load
2829
3058
        # rather than a full load so this test should be pretty quick.
2830
 
        test_list = ['bzrlib.tests.test_selftest.TestTestSuite.test_test_suite']
 
3059
        test_list = ['breezy.tests.test_selftest.TestTestSuite.test_test_suite']
2831
3060
        suite = tests.test_suite(test_list,
2832
 
                                 ['bzrlib.tests.test_selftest.TestTestSuite'])
 
3061
                                 ['breezy.tests.test_selftest.TestTestSuite'])
2833
3062
        # test_test_suite_list_and_start is not included 
2834
 
        self.assertEquals(test_list, _test_ids(suite))
 
3063
        self.assertEqual(test_list, _test_ids(suite))
2835
3064
 
2836
3065
 
2837
3066
class TestLoadTestIdList(tests.TestCaseInTempDir):
2850
3079
        self._create_test_list_file(test_list_fname,
2851
3080
                                    'mod1.cl1.meth1\nmod2.cl2.meth2\n')
2852
3081
        tlist = tests.load_test_id_list(test_list_fname)
2853
 
        self.assertEquals(2, len(tlist))
2854
 
        self.assertEquals('mod1.cl1.meth1', tlist[0])
2855
 
        self.assertEquals('mod2.cl2.meth2', tlist[1])
 
3082
        self.assertEqual(2, len(tlist))
 
3083
        self.assertEqual('mod1.cl1.meth1', tlist[0])
 
3084
        self.assertEqual('mod2.cl2.meth2', tlist[1])
2856
3085
 
2857
3086
    def test_load_dirty_file(self):
2858
3087
        test_list_fname = 'test.list'
2860
3089
                                    '  mod1.cl1.meth1\n\nmod2.cl2.meth2  \n'
2861
3090
                                    'bar baz\n')
2862
3091
        tlist = tests.load_test_id_list(test_list_fname)
2863
 
        self.assertEquals(4, len(tlist))
2864
 
        self.assertEquals('mod1.cl1.meth1', tlist[0])
2865
 
        self.assertEquals('', tlist[1])
2866
 
        self.assertEquals('mod2.cl2.meth2', tlist[2])
2867
 
        self.assertEquals('bar baz', tlist[3])
 
3092
        self.assertEqual(4, len(tlist))
 
3093
        self.assertEqual('mod1.cl1.meth1', tlist[0])
 
3094
        self.assertEqual('', tlist[1])
 
3095
        self.assertEqual('mod2.cl2.meth2', tlist[2])
 
3096
        self.assertEqual('bar baz', tlist[3])
2868
3097
 
2869
3098
 
2870
3099
class TestFilteredByModuleTestLoader(tests.TestCase):
2875
3104
        return loader
2876
3105
 
2877
3106
    def test_load_tests(self):
2878
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing']
 
3107
        test_list = ['breezy.tests.test_sampler.DemoTest.test_nothing']
2879
3108
        loader = self._create_loader(test_list)
2880
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2881
 
        self.assertEquals(test_list, _test_ids(suite))
 
3109
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
 
3110
        self.assertEqual(test_list, _test_ids(suite))
2882
3111
 
2883
3112
    def test_exclude_tests(self):
2884
3113
        test_list = ['bogus']
2885
3114
        loader = self._create_loader(test_list)
2886
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2887
 
        self.assertEquals([], _test_ids(suite))
 
3115
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
 
3116
        self.assertEqual([], _test_ids(suite))
2888
3117
 
2889
3118
 
2890
3119
class TestFilteredByNameStartTestLoader(tests.TestCase):
2896
3125
        return loader
2897
3126
 
2898
3127
    def test_load_tests(self):
2899
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing']
2900
 
        loader = self._create_loader('bzrlib.tests.test_samp')
 
3128
        test_list = ['breezy.tests.test_sampler.DemoTest.test_nothing']
 
3129
        loader = self._create_loader('breezy.tests.test_samp')
2901
3130
 
2902
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2903
 
        self.assertEquals(test_list, _test_ids(suite))
 
3131
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
 
3132
        self.assertEqual(test_list, _test_ids(suite))
2904
3133
 
2905
3134
    def test_load_tests_inside_module(self):
2906
 
        test_list = ['bzrlib.tests.test_sampler.DemoTest.test_nothing']
2907
 
        loader = self._create_loader('bzrlib.tests.test_sampler.Demo')
 
3135
        test_list = ['breezy.tests.test_sampler.DemoTest.test_nothing']
 
3136
        loader = self._create_loader('breezy.tests.test_sampler.Demo')
2908
3137
 
2909
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2910
 
        self.assertEquals(test_list, _test_ids(suite))
 
3138
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
 
3139
        self.assertEqual(test_list, _test_ids(suite))
2911
3140
 
2912
3141
    def test_exclude_tests(self):
2913
3142
        test_list = ['bogus']
2914
3143
        loader = self._create_loader('bogus')
2915
3144
 
2916
 
        suite = loader.loadTestsFromModuleName('bzrlib.tests.test_sampler')
2917
 
        self.assertEquals([], _test_ids(suite))
 
3145
        suite = loader.loadTestsFromModuleName('breezy.tests.test_sampler')
 
3146
        self.assertEqual([], _test_ids(suite))
2918
3147
 
2919
3148
 
2920
3149
class TestTestPrefixRegistry(tests.TestCase):
2926
3155
    def test_register_new_prefix(self):
2927
3156
        tpr = self._get_registry()
2928
3157
        tpr.register('foo', 'fff.ooo.ooo')
2929
 
        self.assertEquals('fff.ooo.ooo', tpr.get('foo'))
 
3158
        self.assertEqual('fff.ooo.ooo', tpr.get('foo'))
2930
3159
 
2931
3160
    def test_register_existing_prefix(self):
2932
3161
        tpr = self._get_registry()
2933
3162
        tpr.register('bar', 'bbb.aaa.rrr')
2934
3163
        tpr.register('bar', 'bBB.aAA.rRR')
2935
 
        self.assertEquals('bbb.aaa.rrr', tpr.get('bar'))
 
3164
        self.assertEqual('bbb.aaa.rrr', tpr.get('bar'))
2936
3165
        self.assertThat(self.get_log(),
2937
 
            DocTestMatches("...bar...bbb.aaa.rrr...BB.aAA.rRR", ELLIPSIS))
 
3166
            DocTestMatches("...bar...bbb.aaa.rrr...BB.aAA.rRR",
 
3167
                           doctest.ELLIPSIS))
2938
3168
 
2939
3169
    def test_get_unknown_prefix(self):
2940
3170
        tpr = self._get_registry()
2943
3173
    def test_resolve_prefix(self):
2944
3174
        tpr = self._get_registry()
2945
3175
        tpr.register('bar', 'bb.aa.rr')
2946
 
        self.assertEquals('bb.aa.rr', tpr.resolve_alias('bar'))
 
3176
        self.assertEqual('bb.aa.rr', tpr.resolve_alias('bar'))
2947
3177
 
2948
3178
    def test_resolve_unknown_alias(self):
2949
3179
        tpr = self._get_registry()
2952
3182
 
2953
3183
    def test_predefined_prefixes(self):
2954
3184
        tpr = tests.test_prefix_alias_registry
2955
 
        self.assertEquals('bzrlib', tpr.resolve_alias('bzrlib'))
2956
 
        self.assertEquals('bzrlib.doc', tpr.resolve_alias('bd'))
2957
 
        self.assertEquals('bzrlib.utils', tpr.resolve_alias('bu'))
2958
 
        self.assertEquals('bzrlib.tests', tpr.resolve_alias('bt'))
2959
 
        self.assertEquals('bzrlib.tests.blackbox', tpr.resolve_alias('bb'))
2960
 
        self.assertEquals('bzrlib.plugins', tpr.resolve_alias('bp'))
 
3185
        self.assertEqual('breezy', tpr.resolve_alias('breezy'))
 
3186
        self.assertEqual('breezy.doc', tpr.resolve_alias('bd'))
 
3187
        self.assertEqual('breezy.utils', tpr.resolve_alias('bu'))
 
3188
        self.assertEqual('breezy.tests', tpr.resolve_alias('bt'))
 
3189
        self.assertEqual('breezy.tests.blackbox', tpr.resolve_alias('bb'))
 
3190
        self.assertEqual('breezy.plugins', tpr.resolve_alias('bp'))
 
3191
 
 
3192
 
 
3193
class TestThreadLeakDetection(tests.TestCase):
 
3194
    """Ensure when tests leak threads we detect and report it"""
 
3195
 
 
3196
    class LeakRecordingResult(tests.ExtendedTestResult):
 
3197
        def __init__(self):
 
3198
            tests.ExtendedTestResult.__init__(self, StringIO(), 0, 1)
 
3199
            self.leaks = []
 
3200
        def _report_thread_leak(self, test, leaks, alive):
 
3201
            self.leaks.append((test, leaks))
 
3202
 
 
3203
    def test_testcase_without_addCleanups(self):
 
3204
        """Check old TestCase instances don't break with leak detection"""
 
3205
        class Test(unittest.TestCase):
 
3206
            def runTest(self):
 
3207
                pass
 
3208
        result = self.LeakRecordingResult()
 
3209
        test = Test()
 
3210
        result.startTestRun()
 
3211
        test.run(result)
 
3212
        result.stopTestRun()
 
3213
        self.assertEqual(result._tests_leaking_threads_count, 0)
 
3214
        self.assertEqual(result.leaks, [])
 
3215
        
 
3216
    def test_thread_leak(self):
 
3217
        """Ensure a thread that outlives the running of a test is reported
 
3218
 
 
3219
        Uses a thread that blocks on an event, and is started by the inner
 
3220
        test case. As the thread outlives the inner case's run, it should be
 
3221
        detected as a leak, but the event is then set so that the thread can
 
3222
        be safely joined in cleanup so it's not leaked for real.
 
3223
        """
 
3224
        event = threading.Event()
 
3225
        thread = threading.Thread(name="Leaker", target=event.wait)
 
3226
        class Test(tests.TestCase):
 
3227
            def test_leak(self):
 
3228
                thread.start()
 
3229
        result = self.LeakRecordingResult()
 
3230
        test = Test("test_leak")
 
3231
        self.addCleanup(thread.join)
 
3232
        self.addCleanup(event.set)
 
3233
        result.startTestRun()
 
3234
        test.run(result)
 
3235
        result.stopTestRun()
 
3236
        self.assertEqual(result._tests_leaking_threads_count, 1)
 
3237
        self.assertEqual(result._first_thread_leaker_id, test.id())
 
3238
        self.assertEqual(result.leaks, [(test, {thread})])
 
3239
        self.assertContainsString(result.stream.getvalue(), "leaking threads")
 
3240
 
 
3241
    def test_multiple_leaks(self):
 
3242
        """Check multiple leaks are blamed on the test cases at fault
 
3243
 
 
3244
        Same concept as the previous test, but has one inner test method that
 
3245
        leaks two threads, and one that doesn't leak at all.
 
3246
        """
 
3247
        event = threading.Event()
 
3248
        thread_a = threading.Thread(name="LeakerA", target=event.wait)
 
3249
        thread_b = threading.Thread(name="LeakerB", target=event.wait)
 
3250
        thread_c = threading.Thread(name="LeakerC", target=event.wait)
 
3251
        class Test(tests.TestCase):
 
3252
            def test_first_leak(self):
 
3253
                thread_b.start()
 
3254
            def test_second_no_leak(self):
 
3255
                pass
 
3256
            def test_third_leak(self):
 
3257
                thread_c.start()
 
3258
                thread_a.start()
 
3259
        result = self.LeakRecordingResult()
 
3260
        first_test = Test("test_first_leak")
 
3261
        third_test = Test("test_third_leak")
 
3262
        self.addCleanup(thread_a.join)
 
3263
        self.addCleanup(thread_b.join)
 
3264
        self.addCleanup(thread_c.join)
 
3265
        self.addCleanup(event.set)
 
3266
        result.startTestRun()
 
3267
        unittest.TestSuite(
 
3268
            [first_test, Test("test_second_no_leak"), third_test]
 
3269
            ).run(result)
 
3270
        result.stopTestRun()
 
3271
        self.assertEqual(result._tests_leaking_threads_count, 2)
 
3272
        self.assertEqual(result._first_thread_leaker_id, first_test.id())
 
3273
        self.assertEqual(result.leaks, [
 
3274
            (first_test, {thread_b}),
 
3275
            (third_test, {thread_a, thread_c})])
 
3276
        self.assertContainsString(result.stream.getvalue(), "leaking threads")
 
3277
 
 
3278
 
 
3279
class TestPostMortemDebugging(tests.TestCase):
 
3280
    """Check post mortem debugging works when tests fail or error"""
 
3281
 
 
3282
    class TracebackRecordingResult(tests.ExtendedTestResult):
 
3283
        def __init__(self):
 
3284
            tests.ExtendedTestResult.__init__(self, StringIO(), 0, 1)
 
3285
            self.postcode = None
 
3286
        def _post_mortem(self, tb=None):
 
3287
            """Record the code object at the end of the current traceback"""
 
3288
            tb = tb or sys.exc_info()[2]
 
3289
            if tb is not None:
 
3290
                next = tb.tb_next
 
3291
                while next is not None:
 
3292
                    tb = next
 
3293
                    next = next.tb_next
 
3294
                self.postcode = tb.tb_frame.f_code
 
3295
        def report_error(self, test, err):
 
3296
            pass
 
3297
        def report_failure(self, test, err):
 
3298
            pass
 
3299
 
 
3300
    def test_location_unittest_error(self):
 
3301
        """Needs right post mortem traceback with erroring unittest case"""
 
3302
        class Test(unittest.TestCase):
 
3303
            def runTest(self):
 
3304
                raise RuntimeError
 
3305
        result = self.TracebackRecordingResult()
 
3306
        Test().run(result)
 
3307
        self.assertEqual(result.postcode, Test.runTest.__code__)
 
3308
 
 
3309
    def test_location_unittest_failure(self):
 
3310
        """Needs right post mortem traceback with failing unittest case"""
 
3311
        class Test(unittest.TestCase):
 
3312
            def runTest(self):
 
3313
                raise self.failureException
 
3314
        result = self.TracebackRecordingResult()
 
3315
        Test().run(result)
 
3316
        self.assertEqual(result.postcode, Test.runTest.__code__)
 
3317
 
 
3318
    def test_location_bt_error(self):
 
3319
        """Needs right post mortem traceback with erroring breezy.tests case"""
 
3320
        class Test(tests.TestCase):
 
3321
            def test_error(self):
 
3322
                raise RuntimeError
 
3323
        result = self.TracebackRecordingResult()
 
3324
        Test("test_error").run(result)
 
3325
        self.assertEqual(result.postcode, Test.test_error.__code__)
 
3326
 
 
3327
    def test_location_bt_failure(self):
 
3328
        """Needs right post mortem traceback with failing breezy.tests case"""
 
3329
        class Test(tests.TestCase):
 
3330
            def test_failure(self):
 
3331
                raise self.failureException
 
3332
        result = self.TracebackRecordingResult()
 
3333
        Test("test_failure").run(result)
 
3334
        self.assertEqual(result.postcode, Test.test_failure.__code__)
 
3335
 
 
3336
    def test_env_var_triggers_post_mortem(self):
 
3337
        """Check pdb.post_mortem is called iff BRZ_TEST_PDB is set"""
 
3338
        import pdb
 
3339
        result = tests.ExtendedTestResult(StringIO(), 0, 1)
 
3340
        post_mortem_calls = []
 
3341
        self.overrideAttr(pdb, "post_mortem", post_mortem_calls.append)
 
3342
        self.overrideEnv('BRZ_TEST_PDB', None)
 
3343
        result._post_mortem(1)
 
3344
        self.overrideEnv('BRZ_TEST_PDB', 'on')
 
3345
        result._post_mortem(2)
 
3346
        self.assertEqual([2], post_mortem_calls)
2961
3347
 
2962
3348
 
2963
3349
class TestRunSuite(tests.TestCase):
2976
3362
                                                self.verbosity)
2977
3363
        tests.run_suite(suite, runner_class=MyRunner, stream=StringIO())
2978
3364
        self.assertLength(1, calls)
 
3365
 
 
3366
 
 
3367
class _Selftest(object):
 
3368
    """Mixin for tests needing full selftest output"""
 
3369
 
 
3370
    def _inject_stream_into_subunit(self, stream):
 
3371
        """To be overridden by subclasses that run tests out of process"""
 
3372
 
 
3373
    def _run_selftest(self, **kwargs):
 
3374
        sio = StringIO()
 
3375
        self._inject_stream_into_subunit(sio)
 
3376
        tests.selftest(stream=sio, stop_on_failure=False, **kwargs)
 
3377
        return sio.getvalue()
 
3378
 
 
3379
 
 
3380
class _ForkedSelftest(_Selftest):
 
3381
    """Mixin for tests needing full selftest output with forked children"""
 
3382
 
 
3383
    _test_needs_features = [features.subunit]
 
3384
 
 
3385
    def _inject_stream_into_subunit(self, stream):
 
3386
        """Monkey-patch subunit so the extra output goes to stream not stdout
 
3387
 
 
3388
        Some APIs need rewriting so this kind of bogus hackery can be replaced
 
3389
        by passing the stream param from run_tests down into ProtocolTestCase.
 
3390
        """
 
3391
        from subunit import ProtocolTestCase
 
3392
        _original_init = ProtocolTestCase.__init__
 
3393
        def _init_with_passthrough(self, *args, **kwargs):
 
3394
            _original_init(self, *args, **kwargs)
 
3395
            self._passthrough = stream
 
3396
        self.overrideAttr(ProtocolTestCase, "__init__", _init_with_passthrough)
 
3397
 
 
3398
    def _run_selftest(self, **kwargs):
 
3399
        # GZ 2011-05-26: Add a PosixSystem feature so this check can go away
 
3400
        if getattr(os, "fork", None) is None:
 
3401
            raise tests.TestNotApplicable("Platform doesn't support forking")
 
3402
        # Make sure the fork code is actually invoked by claiming two cores
 
3403
        self.overrideAttr(osutils, "local_concurrency", lambda: 2)
 
3404
        kwargs.setdefault("suite_decorators", []).append(tests.fork_decorator)
 
3405
        return super(_ForkedSelftest, self)._run_selftest(**kwargs)
 
3406
 
 
3407
 
 
3408
class TestParallelFork(_ForkedSelftest, tests.TestCase):
 
3409
    """Check operation of --parallel=fork selftest option"""
 
3410
 
 
3411
    def test_error_in_child_during_fork(self):
 
3412
        """Error in a forked child during test setup should get reported"""
 
3413
        class Test(tests.TestCase):
 
3414
            def testMethod(self):
 
3415
                pass
 
3416
        # We don't care what, just break something that a child will run
 
3417
        self.overrideAttr(tests, "workaround_zealous_crypto_random", None)
 
3418
        out = self._run_selftest(test_suite_factory=Test)
 
3419
        # Lines from the tracebacks of the two child processes may be mixed
 
3420
        # together due to the way subunit parses and forwards the streams,
 
3421
        # so permit extra lines between each part of the error output.
 
3422
        self.assertContainsRe(out,
 
3423
            "Traceback.*:\n"
 
3424
            "(?:.*\n)*"
 
3425
            ".+ in fork_for_tests\n"
 
3426
            "(?:.*\n)*"
 
3427
            "\s*workaround_zealous_crypto_random\(\)\n"
 
3428
            "(?:.*\n)*"
 
3429
            "TypeError:")
 
3430
 
 
3431
 
 
3432
class TestUncollectedWarnings(_Selftest, tests.TestCase):
 
3433
    """Check a test case still alive after being run emits a warning"""
 
3434
 
 
3435
    class Test(tests.TestCase):
 
3436
        def test_pass(self):
 
3437
            pass
 
3438
        def test_self_ref(self):
 
3439
            self.also_self = self.test_self_ref
 
3440
        def test_skip(self):
 
3441
            self.skipTest("Don't need")
 
3442
 
 
3443
    def _get_suite(self):
 
3444
        return TestUtil.TestSuite([
 
3445
            self.Test("test_pass"),
 
3446
            self.Test("test_self_ref"),
 
3447
            self.Test("test_skip"),
 
3448
            ])
 
3449
 
 
3450
    def _run_selftest_with_suite(self, **kwargs):
 
3451
        old_flags = tests.selftest_debug_flags
 
3452
        tests.selftest_debug_flags = old_flags.union(["uncollected_cases"])
 
3453
        gc_on = gc.isenabled()
 
3454
        if gc_on:
 
3455
            gc.disable()
 
3456
        try:
 
3457
            output = self._run_selftest(test_suite_factory=self._get_suite,
 
3458
                **kwargs)
 
3459
        finally:
 
3460
            if gc_on:
 
3461
                gc.enable()
 
3462
            tests.selftest_debug_flags = old_flags
 
3463
        self.assertNotContainsRe(output, "Uncollected test case.*test_pass")
 
3464
        self.assertContainsRe(output, "Uncollected test case.*test_self_ref")
 
3465
        return output
 
3466
 
 
3467
    def test_testsuite(self):
 
3468
        self._run_selftest_with_suite()
 
3469
 
 
3470
    def test_pattern(self):
 
3471
        out = self._run_selftest_with_suite(pattern="test_(?:pass|self_ref)$")
 
3472
        self.assertNotContainsRe(out, "test_skip")
 
3473
 
 
3474
    def test_exclude_pattern(self):
 
3475
        out = self._run_selftest_with_suite(exclude_pattern="test_skip$")
 
3476
        self.assertNotContainsRe(out, "test_skip")
 
3477
 
 
3478
    def test_random_seed(self):
 
3479
        self._run_selftest_with_suite(random_seed="now")
 
3480
 
 
3481
    def test_matching_tests_first(self):
 
3482
        self._run_selftest_with_suite(matching_tests_first=True,
 
3483
            pattern="test_self_ref$")
 
3484
 
 
3485
    def test_starting_with_and_exclude(self):
 
3486
        out = self._run_selftest_with_suite(starting_with=["bt."],
 
3487
            exclude_pattern="test_skip$")
 
3488
        self.assertNotContainsRe(out, "test_skip")
 
3489
 
 
3490
    def test_additonal_decorator(self):
 
3491
        out = self._run_selftest_with_suite(
 
3492
            suite_decorators=[tests.TestDecorator])
 
3493
 
 
3494
 
 
3495
class TestUncollectedWarningsSubunit(TestUncollectedWarnings):
 
3496
    """Check warnings from tests staying alive are emitted with subunit"""
 
3497
 
 
3498
    _test_needs_features = [features.subunit]
 
3499
 
 
3500
    def _run_selftest_with_suite(self, **kwargs):
 
3501
        return TestUncollectedWarnings._run_selftest_with_suite(self,
 
3502
            runner_class=tests.SubUnitBzrRunner, **kwargs)
 
3503
 
 
3504
 
 
3505
class TestUncollectedWarningsForked(_ForkedSelftest, TestUncollectedWarnings):
 
3506
    """Check warnings from tests staying alive are emitted when forking"""
 
3507
 
 
3508
 
 
3509
class TestEnvironHandling(tests.TestCase):
 
3510
 
 
3511
    def test_overrideEnv_None_called_twice_doesnt_leak(self):
 
3512
        self.assertFalse('MYVAR' in os.environ)
 
3513
        self.overrideEnv('MYVAR', '42')
 
3514
        # We use an embedded test to make sure we fix the _captureVar bug
 
3515
        class Test(tests.TestCase):
 
3516
            def test_me(self):
 
3517
                # The first call save the 42 value
 
3518
                self.overrideEnv('MYVAR', None)
 
3519
                self.assertEqual(None, os.environ.get('MYVAR'))
 
3520
                # Make sure we can call it twice
 
3521
                self.overrideEnv('MYVAR', None)
 
3522
                self.assertEqual(None, os.environ.get('MYVAR'))
 
3523
        output = StringIO()
 
3524
        result = tests.TextTestResult(output, 0, 1)
 
3525
        Test('test_me').run(result)
 
3526
        if not result.wasStrictlySuccessful():
 
3527
            self.fail(output.getvalue())
 
3528
        # We get our value back
 
3529
        self.assertEqual('42', os.environ.get('MYVAR'))
 
3530
 
 
3531
 
 
3532
class TestIsolatedEnv(tests.TestCase):
 
3533
    """Test isolating tests from os.environ.
 
3534
 
 
3535
    Since we use tests that are already isolated from os.environ a bit of care
 
3536
    should be taken when designing the tests to avoid bootstrap side-effects.
 
3537
    The tests start an already clean os.environ which allow doing valid
 
3538
    assertions about which variables are present or not and design tests around
 
3539
    these assertions.
 
3540
    """
 
3541
 
 
3542
    class ScratchMonkey(tests.TestCase):
 
3543
 
 
3544
        def test_me(self):
 
3545
            pass
 
3546
 
 
3547
    def test_basics(self):
 
3548
        # Make sure we know the definition of BRZ_HOME: not part of os.environ
 
3549
        # for tests.TestCase.
 
3550
        self.assertTrue('BRZ_HOME' in tests.isolated_environ)
 
3551
        self.assertEqual(None, tests.isolated_environ['BRZ_HOME'])
 
3552
        # Being part of isolated_environ, BRZ_HOME should not appear here
 
3553
        self.assertFalse('BRZ_HOME' in os.environ)
 
3554
        # Make sure we know the definition of LINES: part of os.environ for
 
3555
        # tests.TestCase
 
3556
        self.assertTrue('LINES' in tests.isolated_environ)
 
3557
        self.assertEqual('25', tests.isolated_environ['LINES'])
 
3558
        self.assertEqual('25', os.environ['LINES'])
 
3559
 
 
3560
    def test_injecting_unknown_variable(self):
 
3561
        # BRZ_HOME is known to be absent from os.environ
 
3562
        test = self.ScratchMonkey('test_me')
 
3563
        tests.override_os_environ(test, {'BRZ_HOME': 'foo'})
 
3564
        self.assertEqual('foo', os.environ['BRZ_HOME'])
 
3565
        tests.restore_os_environ(test)
 
3566
        self.assertFalse('BRZ_HOME' in os.environ)
 
3567
 
 
3568
    def test_injecting_known_variable(self):
 
3569
        test = self.ScratchMonkey('test_me')
 
3570
        # LINES is known to be present in os.environ
 
3571
        tests.override_os_environ(test, {'LINES': '42'})
 
3572
        self.assertEqual('42', os.environ['LINES'])
 
3573
        tests.restore_os_environ(test)
 
3574
        self.assertEqual('25', os.environ['LINES'])
 
3575
 
 
3576
    def test_deleting_variable(self):
 
3577
        test = self.ScratchMonkey('test_me')
 
3578
        # LINES is known to be present in os.environ
 
3579
        tests.override_os_environ(test, {'LINES': None})
 
3580
        self.assertTrue('LINES' not in os.environ)
 
3581
        tests.restore_os_environ(test)
 
3582
        self.assertEqual('25', os.environ['LINES'])
 
3583
 
 
3584
 
 
3585
class TestDocTestSuiteIsolation(tests.TestCase):
 
3586
    """Test that `tests.DocTestSuite` isolates doc tests from os.environ.
 
3587
 
 
3588
    Since tests.TestCase alreay provides an isolation from os.environ, we use
 
3589
    the clean environment as a base for testing. To precisely capture the
 
3590
    isolation provided by tests.DocTestSuite, we use doctest.DocTestSuite to
 
3591
    compare against.
 
3592
 
 
3593
    We want to make sure `tests.DocTestSuite` respect `tests.isolated_environ`,
 
3594
    not `os.environ` so each test overrides it to suit its needs.
 
3595
 
 
3596
    """
 
3597
 
 
3598
    def get_doctest_suite_for_string(self, klass, string):
 
3599
        class Finder(doctest.DocTestFinder):
 
3600
 
 
3601
            def find(*args, **kwargs):
 
3602
                test = doctest.DocTestParser().get_doctest(
 
3603
                    string, {}, 'foo', 'foo.py', 0)
 
3604
                return [test]
 
3605
 
 
3606
        suite = klass(test_finder=Finder())
 
3607
        return suite
 
3608
 
 
3609
    def run_doctest_suite_for_string(self, klass, string):
 
3610
        suite = self.get_doctest_suite_for_string(klass, string)
 
3611
        output = StringIO()
 
3612
        result = tests.TextTestResult(output, 0, 1)
 
3613
        suite.run(result)
 
3614
        return result, output
 
3615
 
 
3616
    def assertDocTestStringSucceds(self, klass, string):
 
3617
        result, output = self.run_doctest_suite_for_string(klass, string)
 
3618
        if not result.wasStrictlySuccessful():
 
3619
            self.fail(output.getvalue())
 
3620
 
 
3621
    def assertDocTestStringFails(self, klass, string):
 
3622
        result, output = self.run_doctest_suite_for_string(klass, string)
 
3623
        if result.wasStrictlySuccessful():
 
3624
            self.fail(output.getvalue())
 
3625
 
 
3626
    def test_injected_variable(self):
 
3627
        self.overrideAttr(tests, 'isolated_environ', {'LINES': '42'})
 
3628
        test = """
 
3629
            >>> import os
 
3630
            >>> os.environ['LINES']
 
3631
            '42'
 
3632
            """
 
3633
        # doctest.DocTestSuite fails as it sees '25'
 
3634
        self.assertDocTestStringFails(doctest.DocTestSuite, test)
 
3635
        # tests.DocTestSuite sees '42'
 
3636
        self.assertDocTestStringSucceds(tests.IsolatedDocTestSuite, test)
 
3637
 
 
3638
    def test_deleted_variable(self):
 
3639
        self.overrideAttr(tests, 'isolated_environ', {'LINES': None})
 
3640
        test = """
 
3641
            >>> import os
 
3642
            >>> os.environ.get('LINES')
 
3643
            """
 
3644
        # doctest.DocTestSuite fails as it sees '25'
 
3645
        self.assertDocTestStringFails(doctest.DocTestSuite, test)
 
3646
        # tests.DocTestSuite sees None
 
3647
        self.assertDocTestStringSucceds(tests.IsolatedDocTestSuite, test)
 
3648
 
 
3649
 
 
3650
class TestSelftestExcludePatterns(tests.TestCase):
 
3651
 
 
3652
    def setUp(self):
 
3653
        super(TestSelftestExcludePatterns, self).setUp()
 
3654
        self.overrideAttr(tests, 'test_suite', self.suite_factory)
 
3655
 
 
3656
    def suite_factory(self, keep_only=None, starting_with=None):
 
3657
        """A test suite factory with only a few tests."""
 
3658
        class Test(tests.TestCase):
 
3659
            def id(self):
 
3660
                # We don't need the full class path
 
3661
                return self._testMethodName
 
3662
            def a(self):
 
3663
                pass
 
3664
            def b(self):
 
3665
                pass
 
3666
            def c(self):
 
3667
                pass
 
3668
        return TestUtil.TestSuite([Test("a"), Test("b"), Test("c")])
 
3669
 
 
3670
    def assertTestList(self, expected, *selftest_args):
 
3671
        # We rely on setUp installing the right test suite factory so we can
 
3672
        # test at the command level without loading the whole test suite
 
3673
        out, err = self.run_bzr(('selftest', '--list') + selftest_args)
 
3674
        actual = out.splitlines()
 
3675
        self.assertEqual(expected, actual)
 
3676
 
 
3677
    def test_full_list(self):
 
3678
        self.assertTestList(['a', 'b', 'c'])
 
3679
 
 
3680
    def test_single_exclude(self):
 
3681
        self.assertTestList(['b', 'c'], '-x', 'a')
 
3682
 
 
3683
    def test_mutiple_excludes(self):
 
3684
        self.assertTestList(['c'], '-x', 'a', '-x', 'b')
 
3685
 
 
3686
 
 
3687
class TestCounterHooks(tests.TestCase, SelfTestHelper):
 
3688
 
 
3689
    _test_needs_features = [features.subunit]
 
3690
 
 
3691
    def setUp(self):
 
3692
        super(TestCounterHooks, self).setUp()
 
3693
        class Test(tests.TestCase):
 
3694
 
 
3695
            def setUp(self):
 
3696
                super(Test, self).setUp()
 
3697
                self.hooks = hooks.Hooks()
 
3698
                self.hooks.add_hook('myhook', 'Foo bar blah', (2,4))
 
3699
                self.install_counter_hook(self.hooks, 'myhook')
 
3700
 
 
3701
            def no_hook(self):
 
3702
                pass
 
3703
 
 
3704
            def run_hook_once(self):
 
3705
                for hook in self.hooks['myhook']:
 
3706
                    hook(self)
 
3707
 
 
3708
        self.test_class = Test
 
3709
 
 
3710
    def assertHookCalls(self, expected_calls, test_name):
 
3711
        test = self.test_class(test_name)
 
3712
        result = unittest.TestResult()
 
3713
        test.run(result)
 
3714
        self.assertTrue(hasattr(test, '_counters'))
 
3715
        self.assertTrue('myhook' in test._counters)
 
3716
        self.assertEqual(expected_calls, test._counters['myhook'])
 
3717
 
 
3718
    def test_no_hook(self):
 
3719
        self.assertHookCalls(0, 'no_hook')
 
3720
 
 
3721
    def test_run_hook_once(self):
 
3722
        tt = features.testtools
 
3723
        if tt.module.__version__ < (0, 9, 8):
 
3724
            raise tests.TestSkipped('testtools-0.9.8 required for addDetail')
 
3725
        self.assertHookCalls(1, 'run_hook_once')