1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
# Copyright 2010 Lars Wirzenius
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import os
import shutil
import tempfile
import unittest
import larch
import larch.nodestore_disk
class DummyNodeStore(object):
def __init__(self, dirname):
self.dirname = dirname
self.journal = self
def makedirs(self, dirname):
if not os.path.exists(dirname):
os.makedirs(dirname)
def cat(self, filename):
return file(filename).read()
def overwrite_file(self, filename, contents):
file(filename, 'w').write(contents)
def exists(self, filename):
return os.path.exists(filename)
def rename(self, old, new):
os.rename(old, new)
def remove(self, filename):
os.remove(filename)
class RefcountStoreTests(unittest.TestCase):
def setUp(self):
self.dirname = tempfile.mkdtemp()
self.rs = self.new_rs()
def tearDown(self):
shutil.rmtree(self.dirname)
def new_rs(self):
return larch.RefcountStore(DummyNodeStore(self.dirname))
def test_returns_zero_for_unset_refcount(self):
self.assertEqual(self.rs.get_refcount(123), 0)
def test_sets_refcount(self):
self.rs.set_refcount(123, 1)
self.assertEqual(self.rs.get_refcount(123), 1)
def test_updates_refcount(self):
self.rs.set_refcount(123, 1)
self.rs.set_refcount(123, 2)
self.assertEqual(self.rs.get_refcount(123), 2)
def test_refcounts_are_not_saved_automatically(self):
self.rs.set_refcount(123, 1)
rs2 = self.new_rs()
self.assertEqual(rs2.get_refcount(123), 0)
def test_saves_refcounts(self):
self.rs.set_refcount(123, 1)
self.rs.save_refcounts()
rs2 = self.new_rs()
self.assertEqual(rs2.get_refcount(123), 1)
def test_save_refcounts_works_without_changes(self):
self.assertEqual(self.rs.save_refcounts(), None)
def test_refcount_group_encode_decode_round_trip_works(self):
refs = range(2048)
for ref in refs:
self.rs.set_refcount(ref, ref)
encoded = larch.refcountstore.encode_refcounts(
self.rs.refcounts, 0, 1024, range(1024))
decoded = larch.refcountstore.decode_refcounts(encoded)
self.assertEqual(decoded, [(x, x) for x in refs[:1024]])
def test_group_returns_correct_start_id_for_node_zero(self):
self.assertEqual(self.rs._start_id(0), 0)
def test_group_returns_correct_start_id_for_last_id_in_group(self):
self.assertEqual(self.rs._start_id(self.rs.per_group - 1), 0)
def test_group_returns_correct_start_id_for_first_in_second_group(self):
self.assertEqual(self.rs._start_id(self.rs.per_group),
self.rs.per_group)
def test_group_returns_correct_start_id_for_second_in_second_group(self):
self.assertEqual(self.rs._start_id(self.rs.per_group + 1),
self.rs.per_group)
|
