← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/per_workingtree/test_content_filters.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-01-26 10:49:57 UTC
  • mfrom: (4987.1.1 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100126104957-dmtqnc0pckuruyla
(vila,
        jam) Implement TestCase.overrideAttr to simplify tests setUp/cleanup

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/per_workingtree/test_content_filters.py
123
123
        # the easiest way currently given we don't have easy access to the
124
124
        # WorkingTree after it is created but before the filter stack is used
125
125
        # to populate content.
126
 
        self.real_content_filter_stack = WorkingTree._content_filter_stack
127
 
        def restore_real_content_filter_stack():
128
 
            WorkingTree._content_filter_stack = self.real_content_filter_stack
129
 
        self.addCleanup(restore_real_content_filter_stack)
130
 
        def _content_filter_stack(tree, path=None, file_id=None):
 
126
        def new_stack(tree, path=None, file_id=None):
131
127
            if path.endswith('.txt'):
132
128
                return [ContentFilter(_swapcase, _swapcase)]
133
129
            else:
134
130
                return []
135
 
        WorkingTree._content_filter_stack = _content_filter_stack
 
131
        self.overrideAttr(WorkingTree, '_content_filter_stack', new_stack)
136
132
 
137
133
    def assert_basis_content(self, expected_content, branch, file_id):
138
134
        # Note: We need to use try/finally here instead of addCleanup()