1
# Copyright (C) 2006 Canonical Ltd
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.
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.
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
17
"""Revision store tests."""
20
import bzrlib.errors as errors
21
from bzrlib.revision import Revision
22
from bzrlib.store.revision import RevisionStore
23
from bzrlib.tests import TestCaseWithTransport
24
from bzrlib.transactions import PassThroughTransaction
27
class TestFactory(TestCaseWithTransport):
29
def test_factory_keeps_smoke_in(self):
30
s = self.store_factory.create(self.get_url('.'))
31
self.assertTrue(isinstance(s, RevisionStore))
34
class TestAll(TestCaseWithTransport):
37
super(TestAll, self).setUp()
38
self.store = self.store_factory.create(self.get_url('.'))
39
self.transaction = PassThroughTransaction()
40
self.store.get_scope = self.get_transaction
42
def get_transaction(self):
43
return self.transaction
45
def test_add_has_get(self):
46
rev = self.add_sample_rev()
47
self.assertTrue(self.store.has_revision_id('A', self.transaction))
48
rev2 = self.store.get_revision('A', self.transaction)
49
self.assertEqual(rev, rev2)
51
def add_sample_rev(self):
52
rev = Revision(timestamp=0,
54
committer="Foo Bar <foo@example.com>",
56
inventory_sha1='aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',
58
self.store.add_revision(rev, self.transaction)
61
def test_has_missing(self):
62
# has of a non present id -> False
63
self.assertFalse(self.store.has_revision_id('missing', self.transaction))
65
def test_has_null(self):
67
self.assertTrue(self.store.has_revision_id('null:', self.transaction))
69
def test_get_revision_missing(self):
70
# get_revision('B') -> raises NoSuchRevision
71
self.assertRaises(errors.NoSuchRevision,
72
self.store.get_revision,
76
def test_get_revision_null(self):
77
# get_revision(null) -> raises ReservedId
78
self.assertRaises(errors.ReservedId,
79
self.store.get_revision,
83
def test_add_signature_text_missing(self):
84
# add of a text signature for a missing revision must work, to allow
85
# revisions to be added after the signature.
86
self.store.add_revision_signature_text('A', 'foo\nbar', self.transaction)
87
# but must not be visible
88
self.assertRaises(errors.NoSuchRevision,
89
self.store.has_signature,
93
self.assertRaises(errors.NoSuchRevision,
94
self.store.get_signature_text,
97
# until the revision is added
99
self.assertTrue(self.store.has_signature('A', self.transaction))
100
self.assertEqual('foo\nbar',
101
self.store.get_signature_text('A', self.transaction))
103
def test_add_signature_text(self):
104
# add a signature to a existing revision works.
105
self.add_sample_rev()
106
self.assertFalse(self.store.has_signature('A', self.transaction))
107
self.assertRaises(errors.NoSuchRevision,
108
self.store.get_signature_text,
111
self.store.add_revision_signature_text('A', 'foo\nbar', self.transaction)
112
self.assertTrue(self.store.has_signature('A', self.transaction))
113
self.assertEqual('foo\nbar',
114
self.store.get_signature_text('A', self.transaction))
116
def test_total_size(self):
117
# we get a revision count and a numeric size figure from total_size().
118
count, bytes = self.store.total_size(self.transaction)
119
self.assertEqual(0, count)
120
# some stores use disk immediately that they are created so we just
121
# check that its an int.
122
self.assertIsInstance(bytes, (int, long))
123
self.add_sample_rev()
124
count, bytes = self.store.total_size(self.transaction)
125
self.assertEqual(1, count)
126
self.assertNotEqual(0, bytes)
128
def test_all_revision_ids(self):
129
self.assertEqual([], self.store.all_revision_ids(self.transaction))
130
self.add_sample_rev()
131
self.assertEqual(['A'], self.store.all_revision_ids(self.transaction))
added
removed
bzrlib/tests/revisionstore_implementations/test_all.py
