← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_xml.py

  • Committer: Martin Pool
  • Date: 2005-08-18 05:44:39 UTC
  • Revision ID: mbp@sourcefrog.net-20050818054439-ba0873b87a8c1671
- add code to run weave utility under profiler

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_xml.py
1
 
# Copyright (C) 2005 by 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
from cStringIO import StringIO
18
 
 
19
 
from bzrlib.tests import TestCase
20
 
from bzrlib.inventory import Inventory, InventoryEntry
21
 
from bzrlib.xml4 import serializer_v4
22
 
from bzrlib.xml5 import serializer_v5
23
 
 
24
 
_working_inventory_v4 = """<inventory file_id="TREE_ROOT">
25
 
<entry file_id="bar-20050901064931-73b4b1138abc9cd2" kind="file" name="bar" parent_id="TREE_ROOT" />
26
 
<entry file_id="foo-20050801201819-4139aa4a272f4250" kind="directory" name="foo" parent_id="TREE_ROOT" />
27
 
<entry file_id="bar-20050824000535-6bc48cfad47ed134" kind="file" name="bar" parent_id="foo-20050801201819-4139aa4a272f4250" />
28
 
</inventory>"""
29
 
 
30
 
 
31
 
_revision_v4 = """<revision committer="Martin Pool &lt;mbp@sourcefrog.net&gt;"
32
 
    inventory_id="mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9"
33
 
    inventory_sha1="e79c31c1deb64c163cf660fdedd476dd579ffd41"
34
 
    revision_id="mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9"
35
 
    timestamp="1125907235.211783886"
36
 
    timezone="36000">
37
 
<message>- start splitting code for xml (de)serialization away from objects
38
 
  preparatory to supporting multiple formats by a single library
39
 
</message>
40
 
<parents>
41
 
<revision_ref revision_id="mbp@sourcefrog.net-20050905063503-43948f59fa127d92" revision_sha1="7bdf4cc8c5bdac739f8cf9b10b78cf4b68f915ff" />
42
 
</parents>
43
 
</revision>
44
 
"""
45
 
 
46
 
_revision_v5 = """<revision committer="Martin Pool &lt;mbp@sourcefrog.net&gt;"
47
 
    inventory_sha1="e79c31c1deb64c163cf660fdedd476dd579ffd41"
48
 
    revision_id="mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9"
49
 
    timestamp="1125907235.211783886"
50
 
    timezone="36000">
51
 
<message>- start splitting code for xml (de)serialization away from objects
52
 
  preparatory to supporting multiple formats by a single library
53
 
</message>
54
 
<parents>
55
 
<revision_ref revision_id="mbp@sourcefrog.net-20050905063503-43948f59fa127d92"/>
56
 
</parents>
57
 
</revision>
58
 
"""
59
 
 
60
 
_committed_inv_v5 = """<inventory>
61
 
<file file_id="bar-20050901064931-73b4b1138abc9cd2" 
62
 
      name="bar" parent_id="TREE_ROOT" 
63
 
      revision="mbp@foo-123123"/>
64
 
<directory name="subdir"
65
 
           file_id="foo-20050801201819-4139aa4a272f4250"
66
 
           parent_id="TREE_ROOT" 
67
 
           revision="mbp@foo-00"/>
68
 
<file file_id="bar-20050824000535-6bc48cfad47ed134" 
69
 
      name="bar" parent_id="foo-20050801201819-4139aa4a272f4250" 
70
 
      revision="mbp@foo-00"/>
71
 
</inventory>
72
 
"""
73
 
 
74
 
_basis_inv_v5 = """<inventory revision_id="mbp@sourcefrog.net-20050905063503-43948f59fa127d92">
75
 
<file file_id="bar-20050901064931-73b4b1138abc9cd2" 
76
 
      name="bar" parent_id="TREE_ROOT" 
77
 
      revision="mbp@foo-123123"/>
78
 
<directory name="subdir"
79
 
           file_id="foo-20050801201819-4139aa4a272f4250"
80
 
           parent_id="TREE_ROOT" 
81
 
           revision="mbp@foo-00"/>
82
 
<file file_id="bar-20050824000535-6bc48cfad47ed134" 
83
 
      name="bar" parent_id="foo-20050801201819-4139aa4a272f4250" 
84
 
      revision="mbp@foo-00"/>
85
 
</inventory>
86
 
"""
87
 
 
88
 
class TestSerializer(TestCase):
89
 
    """Test XML serialization"""
90
 
    def test_canned_inventory(self):
91
 
        """Test unpacked a canned inventory v4 file."""
92
 
        inp = StringIO(_working_inventory_v4)
93
 
        inv = serializer_v4.read_inventory(inp)
94
 
        self.assertEqual(len(inv), 4)
95
 
        self.assert_('bar-20050901064931-73b4b1138abc9cd2' in inv)
96
 
 
97
 
    def test_unpack_revision(self):
98
 
        """Test unpacking a canned revision v4"""
99
 
        inp = StringIO(_revision_v4)
100
 
        rev = serializer_v4.read_revision(inp)
101
 
        eq = self.assertEqual
102
 
        eq(rev.committer,
103
 
           "Martin Pool <mbp@sourcefrog.net>")
104
 
        eq(rev.inventory_id,
105
 
           "mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9")
106
 
        eq(len(rev.parent_ids), 1)
107
 
        eq(rev.parent_ids[0],
108
 
           "mbp@sourcefrog.net-20050905063503-43948f59fa127d92")
109
 
 
110
 
    def test_unpack_revision_5(self):
111
 
        """Test unpacking a canned revision v5"""
112
 
        inp = StringIO(_revision_v5)
113
 
        rev = serializer_v5.read_revision(inp)
114
 
        eq = self.assertEqual
115
 
        eq(rev.committer,
116
 
           "Martin Pool <mbp@sourcefrog.net>")
117
 
        eq(len(rev.parent_ids), 1)
118
 
        eq(rev.timezone, 36000)
119
 
        eq(rev.parent_ids[0],
120
 
           "mbp@sourcefrog.net-20050905063503-43948f59fa127d92")
121
 
 
122
 
    def test_unpack_inventory_5(self):
123
 
        """Unpack canned new-style inventory"""
124
 
        inp = StringIO(_committed_inv_v5)
125
 
        inv = serializer_v5.read_inventory(inp)
126
 
        eq = self.assertEqual
127
 
        eq(len(inv), 4)
128
 
        ie = inv['bar-20050824000535-6bc48cfad47ed134']
129
 
        eq(ie.kind, 'file')
130
 
        eq(ie.revision, 'mbp@foo-00')
131
 
        eq(ie.name, 'bar')
132
 
        eq(inv[ie.parent_id].kind, 'directory')
133
 
 
134
 
    def test_unpack_basis_inventory_5(self):
135
 
        """Unpack canned new-style inventory"""
136
 
        inp = StringIO(_basis_inv_v5)
137
 
        inv = serializer_v5.read_inventory(inp)
138
 
        eq = self.assertEqual
139
 
        eq(len(inv), 4)
140
 
        eq(inv.revision_id, 'mbp@sourcefrog.net-20050905063503-43948f59fa127d92')
141
 
        ie = inv['bar-20050824000535-6bc48cfad47ed134']
142
 
        eq(ie.kind, 'file')
143
 
        eq(ie.revision, 'mbp@foo-00')
144
 
        eq(ie.name, 'bar')
145
 
        eq(inv[ie.parent_id].kind, 'directory')
146
 
 
147
 
    def test_repack_inventory_5(self):
148
 
        inp = StringIO(_committed_inv_v5)
149
 
        inv = serializer_v5.read_inventory(inp)
150
 
        outp = StringIO()
151
 
        serializer_v5.write_inventory(inv, outp)
152
 
        inv2 = serializer_v5.read_inventory(StringIO(outp.getvalue()))
153
 
        self.assertEqual(inv, inv2)
154
 
 
155
 
    def test_repack_revision_5(self):
156
 
        """Round-trip revision to XML v5"""
157
 
        inp = StringIO(_revision_v5)
158
 
        rev = serializer_v5.read_revision(inp)
159
 
        outp = StringIO()
160
 
        serializer_v5.write_revision(rev, outp)
161
 
        outfile_contents = outp.getvalue()
162
 
        rev2 = serializer_v5.read_revision(StringIO(outfile_contents))
163
 
        self.assertEqual(rev, rev2)
164
 
 
165
 
    def test_pack_revision_5(self):
166
 
        """Pack revision to XML v5"""
167
 
        # fixed 20051025, revisions should have final newline
168
 
        rev = serializer_v5.read_revision_from_string(_revision_v5)
169
 
        outp = StringIO()
170
 
        serializer_v5.write_revision(rev, outp)
171
 
        outfile_contents = outp.getvalue()
172
 
        self.assertEqual(outfile_contents[-1], '\n')
173
 
        self.assertEqualDiff(outfile_contents, serializer_v5.write_revision_to_string(rev))