~bzr-pqm/bzr/bzr.dev

1417.1.6 by Robert Collins
introduce transactions for grouping actions done to and with branches
1
# Copyright (C) 2005 by Canonical Ltd
2
#   Authors: Robert Collins <robert.collins@canonical.com>
3
#
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
8
#
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
# GNU General Public License for more details.
13
#
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
18
"""Tests for the IdentityMap class."""
19
20
# import system imports here
21
22
#import bzrlib specific imports here
23
import bzrlib.errors as errors
1185.31.25 by John Arbash Meinel
Renamed all of the tests from selftest/foo.py to tests/test_foo.py
24
from bzrlib.tests import TestCase
1417.1.6 by Robert Collins
introduce transactions for grouping actions done to and with branches
25
import bzrlib.identitymap as identitymap
26
27
28
class TestIdentityMap(TestCase):
29
30
    def test_symbols(self):
31
        from bzrlib.identitymap import IdentityMap
32
33
    def test_construct(self):
34
        identitymap.IdentityMap()
35
36
    def test_add_weave(self):
37
        map = identitymap.IdentityMap()
38
        weave = "foo"
39
        map.add_weave("id", weave)
40
        self.assertEqual(weave, map.find_weave("id"))
41
42
    def test_double_add_weave(self):
43
        map = identitymap.IdentityMap()
44
        weave = "foo"
45
        map.add_weave("id", weave)
46
        self.assertRaises(errors.BzrError, map.add_weave, "id", weave)
47
        self.assertEqual(weave, map.find_weave("id"))
48
 
1417.1.10 by Robert Collins
add a cache bound to Transactions, and a precious facility, so that we keep inventory.weave in memory, but can discard weaves for other such files.
49
    def test_remove_object(self):
50
        map = identitymap.IdentityMap()
51
        weave = "foo"
52
        map.add_weave("id", weave)
53
        map.remove_object(weave)
54
        map.add_weave("id", weave)
1417.1.12 by Robert Collins
cache revision history during read transactions
55
        rev_history = [1]
56
        map.add_revision_history(rev_history)
57
        map.remove_object(rev_history)
58
59
    def test_add_revision_history(self):
60
        map = identitymap.IdentityMap()
61
        rev_history = [1,2,3]
62
        map.add_revision_history(rev_history)
63
        self.assertEqual(rev_history, map.find_revision_history())
64
65
    def test_double_add_revision_history(self):
66
        map = identitymap.IdentityMap()
67
        revision_history = [1]
68
        map.add_revision_history(revision_history)
69
        self.assertRaises(errors.BzrError,
70
                          map.add_revision_history,
71
                          revision_history)
72
        self.assertEqual(revision_history, map.find_revision_history())
1417.1.10 by Robert Collins
add a cache bound to Transactions, and a precious facility, so that we keep inventory.weave in memory, but can discard weaves for other such files.
73
1417.1.6 by Robert Collins
introduce transactions for grouping actions done to and with branches
74
 
75
class TestNullIdentityMap(TestCase):
76
77
    def test_symbols(self):
78
        from bzrlib.identitymap import NullIdentityMap
79
80
    def test_construct(self):
81
        identitymap.NullIdentityMap()
82
83
    def test_add_weave(self):
84
        map = identitymap.NullIdentityMap()
85
        weave = "foo"
86
        map.add_weave("id", weave)
87
        self.assertEqual(None, map.find_weave("id"))
88
89
    def test_double_add_weave(self):
90
        map = identitymap.NullIdentityMap()
91
        weave = "foo"
92
        map.add_weave("id", weave)
93
        map.add_weave("id", weave)
94
        self.assertEqual(None, map.find_weave("id"))
95
        
1417.1.11 by Robert Collins
null identity maps never have objects in them, so someone calling remove_object would be silly - check we do not have a remove_object method
96
    def test_null_identity_map_has_no_remove(self):
97
        map = identitymap.NullIdentityMap()
98
        self.assertEqual(None, getattr(map, 'remove_object', None))
99
1417.1.12 by Robert Collins
cache revision history during read transactions
100
    def test_add_revision_history(self):
101
        map = identitymap.NullIdentityMap()
102
        rev_history = [1,2,3]
103
        map.add_revision_history(rev_history)
104
        self.assertEqual(None, map.find_revision_history())
105
106
    def test_double_add_revision_history(self):
107
        map = identitymap.NullIdentityMap()
108
        revision_history = [1]
109
        map.add_revision_history(revision_history)
110
        map.add_revision_history(revision_history)
111
        self.assertEqual(None, map.find_revision_history())