← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/test_xml.py

  • Committer: Martin Pool
  • Date: 2005-07-07 10:22:02 UTC
  • Revision ID: mbp@sourcefrog.net-20050707102201-2d2a13a25098b101
- rearrange and clear up merged weave

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/selftest/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.selftest 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
 
class TestSerializer(TestCase):
75
 
    """Test XML serialization"""
76
 
    def test_canned_inventory(self):
77
 
        """Test unpacked a canned inventory v4 file."""
78
 
        inp = StringIO(_working_inventory_v4)
79
 
        inv = serializer_v4.read_inventory(inp)
80
 
        self.assertEqual(len(inv), 4)
81
 
        self.assert_('bar-20050901064931-73b4b1138abc9cd2' in inv)
82
 
 
83
 
    def test_unpack_revision(self):
84
 
        """Test unpacking a canned revision v4"""
85
 
        inp = StringIO(_revision_v4)
86
 
        rev = serializer_v4.read_revision(inp)
87
 
        eq = self.assertEqual
88
 
        eq(rev.committer,
89
 
           "Martin Pool <mbp@sourcefrog.net>")
90
 
        eq(rev.inventory_id,
91
 
           "mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9")
92
 
        eq(len(rev.parent_ids), 1)
93
 
        eq(rev.parent_ids[0],
94
 
           "mbp@sourcefrog.net-20050905063503-43948f59fa127d92")
95
 
 
96
 
    def test_unpack_revision_5(self):
97
 
        """Test unpacking a canned revision v5"""
98
 
        inp = StringIO(_revision_v5)
99
 
        rev = serializer_v5.read_revision(inp)
100
 
        eq = self.assertEqual
101
 
        eq(rev.committer,
102
 
           "Martin Pool <mbp@sourcefrog.net>")
103
 
        eq(len(rev.parent_ids), 1)
104
 
        eq(rev.timezone, 36000)
105
 
        eq(rev.parent_ids[0],
106
 
           "mbp@sourcefrog.net-20050905063503-43948f59fa127d92")
107
 
 
108
 
    def test_unpack_inventory_5(self):
109
 
        """Unpack canned new-style inventory"""
110
 
        inp = StringIO(_committed_inv_v5)
111
 
        inv = serializer_v5.read_inventory(inp)
112
 
        eq = self.assertEqual
113
 
        eq(len(inv), 4)
114
 
        ie = inv['bar-20050824000535-6bc48cfad47ed134']
115
 
        eq(ie.kind, 'file')
116
 
        eq(ie.revision, 'mbp@foo-00')
117
 
        eq(ie.name, 'bar')
118
 
        eq(inv[ie.parent_id].kind, 'directory')
119
 
 
120
 
    def test_repack_inventory_5(self):
121
 
        inp = StringIO(_committed_inv_v5)
122
 
        inv = serializer_v5.read_inventory(inp)
123
 
        outp = StringIO()
124
 
        serializer_v5.write_inventory(inv, outp)
125
 
        inv2 = serializer_v5.read_inventory(StringIO(outp.getvalue()))
126
 
        self.assertEqual(inv, inv2)
127
 
 
128
 
    def test_repack_revision_5(self):
129
 
        """Round-trip revision to XML v5"""
130
 
        inp = StringIO(_revision_v5)
131
 
        rev = serializer_v5.read_revision(inp)
132
 
        outp = StringIO()
133
 
        serializer_v5.write_revision(rev, outp)
134
 
        outfile_contents = outp.getvalue()
135
 
        rev2 = serializer_v5.read_revision(StringIO(outfile_contents))
136
 
        self.assertEqual(rev, rev2)
137
 
 
138
 
    def test_pack_revision_5(self):
139
 
        """Pack revision to XML v5"""
140
 
        # fixed 20051025, revisions should have final newline
141
 
        rev = serializer_v5.read_revision_from_string(_revision_v5)
142
 
        outp = StringIO()
143
 
        serializer_v5.write_revision(rev, outp)
144
 
        outfile_contents = outp.getvalue()
145
 
        self.assertEqual(outfile_contents[-1], '\n')
146
 
        self.assertEqualDiff(outfile_contents, serializer_v5.write_revision_to_string(rev))