← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_chk_serializer.py

Add bzrlib.pyutils, which has get_named_object, a wrapper around __import__.

This is used to replace various ad hoc implementations of the same logic,
notably the version used in registry's _LazyObjectGetter which had a bug when
getting a module without also getting a member.  And of course, this new
function has unit tests, unlike the replaced code.

This also adds a KnownHooksRegistry subclass to provide a more natural home for
some other logic.

I'm not thrilled about the name of the new module or the new functions, but it's
hard to think of good names for such generic functionality.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_chk_serializer.py
 
1
# Copyright (C) 2009 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
from cStringIO import StringIO
 
18
 
 
19
from bzrlib import (
 
20
    errors,
 
21
    revision,
 
22
    serializer,
 
23
    )
 
24
from bzrlib.chk_serializer import (
 
25
    chk_bencode_serializer,
 
26
    )
 
27
from bzrlib.revision import (
 
28
    Revision,
 
29
    )
 
30
from bzrlib.tests import TestCase
 
31
 
 
32
_working_revision_bencode1 = ('l'
 
33
    'l6:formati10ee'
 
34
    'l9:committer54:Canonical.com Patch Queue Manager <pqm@pqm.ubuntu.com>e'
 
35
    'l8:timezonei3600ee'
 
36
    'l10:propertiesd11:branch-nick6:+trunkee'
 
37
    'l9:timestamp14:1242300770.844e'
 
38
    'l11:revision-id50:pqm@pqm.ubuntu.com-20090514113250-jntkkpminfn3e0tze'
 
39
    'l10:parent-ids'
 
40
        'l'
 
41
        '50:pqm@pqm.ubuntu.com-20090514104039-kggemn7lrretzpvc'
 
42
        '48:jelmer@samba.org-20090510012654-jp9ufxquekaokbeo'
 
43
        'ee'
 
44
    'l14:inventory-sha140:4a2c7fb50e077699242cf6eb16a61779c7b680a7e'
 
45
    'l7:message35:(Jelmer) Move dpush to InterBranch.e'
 
46
    'e')
 
47
 
 
48
_working_revision_bencode1_no_timezone = ('l'
 
49
    'l6:formati10ee'
 
50
    'l9:committer54:Canonical.com Patch Queue Manager <pqm@pqm.ubuntu.com>e'
 
51
    'l9:timestamp14:1242300770.844e'
 
52
    'l10:propertiesd11:branch-nick6:+trunkee'
 
53
    'l11:revision-id50:pqm@pqm.ubuntu.com-20090514113250-jntkkpminfn3e0tze'
 
54
    'l10:parent-ids'
 
55
        'l'
 
56
        '50:pqm@pqm.ubuntu.com-20090514104039-kggemn7lrretzpvc'
 
57
        '48:jelmer@samba.org-20090510012654-jp9ufxquekaokbeo'
 
58
        'ee'
 
59
    'l14:inventory-sha140:4a2c7fb50e077699242cf6eb16a61779c7b680a7e'
 
60
    'l7:message35:(Jelmer) Move dpush to InterBranch.e'
 
61
    'e')
 
62
 
 
63
 
 
64
class TestBEncodeSerializer1(TestCase):
 
65
    """Test BEncode serialization"""
 
66
 
 
67
    def test_unpack_revision(self):
 
68
        """Test unpacking a revision"""
 
69
        rev = chk_bencode_serializer.read_revision_from_string(
 
70
                _working_revision_bencode1)
 
71
        self.assertEquals(rev.committer,
 
72
           "Canonical.com Patch Queue Manager <pqm@pqm.ubuntu.com>")
 
73
        self.assertEquals(rev.inventory_sha1,
 
74
           "4a2c7fb50e077699242cf6eb16a61779c7b680a7")
 
75
        self.assertEquals(["pqm@pqm.ubuntu.com-20090514104039-kggemn7lrretzpvc",
 
76
            "jelmer@samba.org-20090510012654-jp9ufxquekaokbeo"],
 
77
            rev.parent_ids)
 
78
        self.assertEquals("(Jelmer) Move dpush to InterBranch.", rev.message)
 
79
        self.assertEquals("pqm@pqm.ubuntu.com-20090514113250-jntkkpminfn3e0tz",
 
80
           rev.revision_id)
 
81
        self.assertEquals({"branch-nick": u"+trunk"}, rev.properties)
 
82
        self.assertEquals(3600, rev.timezone)
 
83
 
 
84
    def test_written_form_matches(self):
 
85
        rev = chk_bencode_serializer.read_revision_from_string(
 
86
                _working_revision_bencode1)
 
87
        as_str = chk_bencode_serializer.write_revision_to_string(rev)
 
88
        self.assertEqualDiff(_working_revision_bencode1, as_str)
 
89
 
 
90
    def test_unpack_revision_no_timezone(self):
 
91
        rev = chk_bencode_serializer.read_revision_from_string(
 
92
            _working_revision_bencode1_no_timezone)
 
93
        self.assertEquals(None, rev.timezone)
 
94
 
 
95
    def assertRoundTrips(self, serializer, orig_rev):
 
96
        text = serializer.write_revision_to_string(orig_rev)
 
97
        new_rev = serializer.read_revision_from_string(text)
 
98
        self.assertEquals(orig_rev, new_rev)
 
99
 
 
100
    def test_roundtrips_non_ascii(self):
 
101
        rev = Revision("revid1")
 
102
        rev.message = u"\n\xe5me"
 
103
        rev.committer = u'Erik B\xe5gfors'
 
104
        rev.timestamp = 1242385452
 
105
        rev.inventory_sha1 = "4a2c7fb50e077699242cf6eb16a61779c7b680a7"
 
106
        rev.timezone = 3600
 
107
        self.assertRoundTrips(chk_bencode_serializer, rev)
 
108
 
 
109
    def test_roundtrips_xml_invalid_chars(self):
 
110
        rev = Revision("revid1")
 
111
        rev.message = "\t\ue000"
 
112
        rev.committer = u'Erik B\xe5gfors'
 
113
        rev.timestamp = 1242385452
 
114
        rev.timezone = 3600
 
115
        rev.inventory_sha1 = "4a2c7fb50e077699242cf6eb16a61779c7b680a7"
 
116
        self.assertRoundTrips(chk_bencode_serializer, rev)