← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_filter_tree.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_filter_tree.py
1
 
# Copyright (C) 2011 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
"""Tests for ContentFilterTree"""
18
 
 
19
 
import tarfile
20
 
import zipfile
21
 
 
22
 
from bzrlib import (
23
 
    export,
24
 
    filter_tree,
25
 
    tests,
26
 
    )
27
 
from bzrlib.tests import (
28
 
    fixtures,
29
 
    )
30
 
from bzrlib.tests.test_filters import _stack_1
31
 
 
32
 
 
33
 
class TestFilterTree(tests.TestCaseWithTransport):
34
 
 
35
 
    def make_tree(self):
36
 
        self.underlying_tree = fixtures.make_branch_and_populated_tree(
37
 
            self)
38
 
        def stack_callback(path):
39
 
            return _stack_1
40
 
        self.filter_tree = filter_tree.ContentFilterTree(
41
 
            self.underlying_tree, stack_callback)
42
 
        return self.filter_tree
43
 
 
44
 
    def test_get_file_text(self):
45
 
        self.make_tree()
46
 
        self.assertEquals(
47
 
            self.underlying_tree.get_file_text('hello-id'),
48
 
            'hello world')
49
 
        self.assertEquals(
50
 
            self.filter_tree.get_file_text('hello-id'),
51
 
            'HELLO WORLD')
52
 
 
53
 
    def test_tar_export_content_filter_tree(self):
54
 
        # TODO: this could usefully be run generically across all exporters.
55
 
        self.make_tree()
56
 
        export.export(self.filter_tree, "out.tgz")
57
 
        ball = tarfile.open("out.tgz", "r:gz")
58
 
        self.assertEquals(
59
 
            'HELLO WORLD',
60
 
            ball.extractfile('out/hello').read())
61
 
 
62
 
    def test_zip_export_content_filter_tree(self):
63
 
        self.make_tree()
64
 
        export.export(self.filter_tree, 'out.zip')
65
 
        zipf = zipfile.ZipFile('out.zip', 'r')
66
 
        self.assertEquals(
67
 
            'HELLO WORLD',
68
 
            zipf.read('out/hello'))