← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_upgrade_stacked.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_upgrade_stacked.py
1
 
# Copyright (C) 2008, 2009, 2011 Canonical Ltd
 
1
# Copyright (C) 2008 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
18
18
"""Tests for upgrades of various stacking situations."""
19
19
 
20
20
from bzrlib import (
21
 
    controldir,
 
21
    bzrdir,
22
22
    check,
23
23
    errors,
24
24
    tests,
25
25
    )
26
26
from bzrlib.upgrade import upgrade
27
 
from bzrlib.tests.scenarios import load_tests_apply_scenarios
28
 
 
29
 
 
30
 
def upgrade_scenarios():
31
 
    scenario_pairs = [ # old format, new format, model_change
32
 
#        ('knit', 'rich-root', True),
33
 
        ('knit', '1.6', False),
34
 
#        ('pack-0.92', '1.6', False),
35
 
        ('1.6', '1.6.1-rich-root', True),
36
 
        ]
37
 
    scenarios = []
38
 
    for (old_name, new_name, model_change) in scenario_pairs:
39
 
        name = old_name + ', ' + new_name
40
 
        scenarios.append((name,
41
 
            dict(scenario_old_format=old_name,
42
 
                scenario_new_format=new_name,
43
 
                scenario_model_change=model_change)))
44
 
    return scenarios
45
 
 
46
 
 
47
 
load_tests = load_tests_apply_scenarios
48
27
 
49
28
 
50
29
class TestStackUpgrade(tests.TestCaseWithTransport):
51
30
    # TODO: This should possibly be repeated for all stacking repositories,
52
31
    # pairwise by rich/non-rich format; should possibly also try other kinds
53
32
    # of upgrades like knit->pack. -- mbp 20080804
54
 
    
55
 
    scenarios = upgrade_scenarios()
56
33
 
57
34
    def test_stack_upgrade(self):
58
35
        """Correct checks when stacked-on repository is upgraded.
71
48
        self.assertTrue(stacked.open_branch().get_stacked_on_url())
72
49
        # now we'll upgrade the underlying branch, then upgrade the stacked
73
50
        # branch, and this should still work.
74
 
        new_format = controldir.format_registry.make_bzrdir(
 
51
        new_format = bzrdir.format_registry.make_bzrdir(
75
52
            self.scenario_new_format)
76
53
        upgrade('base', new_format)
77
54
        # in some cases you'll get an error if the underlying model has
81
58
                stacked.open_branch)
82
59
        else:
83
60
            check.check_dwim('stacked', False, True, True)
84
 
        stacked = controldir.ControlDir.open('stacked')
 
61
        stacked = bzrdir.BzrDir.open('stacked')
85
62
        # but we can upgrade the stacked repository
86
63
        upgrade('stacked', new_format)
87
64
        # and now it opens ok
88
 
        stacked = controldir.ControlDir.open('stacked')
 
65
        stacked = bzrdir.BzrDir.open('stacked')
89
66
        # And passes check.
90
67
        check.check_dwim('stacked', False, True, True)
 
68
 
 
69
 
 
70
def load_tests(basic_tests, module, loader):
 
71
    """Generate dynamic scenario tests.
 
72
 
 
73
    Called by the bzrlib test framework.
 
74
    """
 
75
    scenario_pairs = [ # old format, new format, model_change
 
76
#        ('knit', 'rich-root', True),
 
77
        ('knit', '1.6', False),
 
78
#        ('pack-0.92', '1.6', False),
 
79
        ('1.6', '1.6.1-rich-root', True),
 
80
        ]
 
81
    scenarios = []
 
82
    for (old_name, new_name, model_change) in scenario_pairs:
 
83
        name = old_name + ', ' + new_name
 
84
        scenarios.append((name,
 
85
            dict(scenario_old_format=old_name,
 
86
                scenario_new_format=new_name,
 
87
                scenario_model_change=model_change)))
 
88
    suite = loader.suiteClass()
 
89
    return tests.multiply_tests(basic_tests, scenarios, suite)