2 # Some simple tests for the Python bindings for TDB
3 # Note that this tests the interface of the Python bindings
4 # It does not test tdb itself.
6 # Copyright (C) 2007-2008 Jelmer Vernooij <jelmer@samba.org>
7 # Published under the GNU LGPLv3 or later
10 from unittest
import TestCase
14 class OpenTdbTests(TestCase
):
16 def test_nonexistent_read(self
):
17 self
.assertRaises(IOError, tdb
.Tdb
, "/some/nonexistent/file", 0,
18 tdb
.DEFAULT
, os
.O_RDWR
)
20 class CloseTdbTests(TestCase
):
22 def test_double_close(self
):
23 self
.tdb
= tdb
.Tdb(tempfile
.mkstemp()[1], 0, tdb
.DEFAULT
,
25 self
.assertNotEqual(None, self
.tdb
)
27 # ensure that double close does not crash python
31 # Check that further operations do not crash python
32 self
.assertRaises(RuntimeError, lambda: self
.tdb
.transaction_start())
34 self
.assertRaises(RuntimeError, lambda: self
.tdb
["bar"])
37 class InternalTdbTests(TestCase
):
42 # repr used to crash on internal db
43 self
.assertEquals(repr(self
.tdb
), "Tdb(<internal>)")
46 class SimpleTdbTests(TestCase
):
49 super(SimpleTdbTests
, self
).setUp()
50 self
.tdb
= tdb
.Tdb(tempfile
.mkstemp()[1], 0, tdb
.DEFAULT
,
52 self
.assertNotEqual(None, self
.tdb
)
58 self
.assertTrue(repr(self
.tdb
).startswith("Tdb('"))
60 def test_lockall(self
):
63 def test_max_dead(self
):
64 self
.tdb
.max_dead
= 20
66 def test_unlockall(self
):
70 def test_lockall_read(self
):
71 self
.tdb
.read_lock_all()
72 self
.tdb
.read_unlock_all()
74 def test_reopen(self
):
78 self
.tdb
.store("bar", "bla")
79 self
.assertEquals("bla", self
.tdb
.get("bar"))
81 def test_getitem(self
):
82 self
.tdb
["bar"] = "foo"
84 self
.assertEquals("foo", self
.tdb
["bar"])
86 def test_delete(self
):
87 self
.tdb
["bar"] = "foo"
89 self
.assertRaises(KeyError, lambda: self
.tdb
["bar"])
91 def test_contains(self
):
92 self
.tdb
["bla"] = "bloe"
93 self
.assertTrue("bla" in self
.tdb
)
95 def test_keyerror(self
):
96 self
.assertRaises(KeyError, lambda: self
.tdb
["bla"])
98 def test_hash_size(self
):
101 def test_map_size(self
):
104 def test_freelist_size(self
):
105 self
.tdb
.freelist_size
110 def test_iterator(self
):
111 self
.tdb
["bla"] = "1"
112 self
.tdb
["brainslug"] = "2"
115 self
.assertEquals(["bla", "brainslug"], l
)
117 def test_transaction_cancel(self
):
118 self
.tdb
["bloe"] = "2"
119 self
.tdb
.transaction_start()
120 self
.tdb
["bloe"] = "1"
121 self
.tdb
.transaction_cancel()
122 self
.assertEquals("2", self
.tdb
["bloe"])
124 def test_transaction_commit(self
):
125 self
.tdb
["bloe"] = "2"
126 self
.tdb
.transaction_start()
127 self
.tdb
["bloe"] = "1"
128 self
.tdb
.transaction_commit()
129 self
.assertEquals("1", self
.tdb
["bloe"])
131 def test_transaction_prepare_commit(self
):
132 self
.tdb
["bloe"] = "2"
133 self
.tdb
.transaction_start()
134 self
.tdb
["bloe"] = "1"
135 self
.tdb
.transaction_prepare_commit()
136 self
.tdb
.transaction_commit()
137 self
.assertEquals("1", self
.tdb
["bloe"])
139 def test_iterkeys(self
):
140 self
.tdb
["bloe"] = "2"
141 self
.tdb
["bla"] = "25"
142 i
= self
.tdb
.iterkeys()
143 self
.assertEquals(set(["bloe", "bla"]), set([i
.next(), i
.next()]))
145 def test_clear(self
):
146 self
.tdb
["bloe"] = "2"
147 self
.tdb
["bla"] = "25"
148 self
.assertEquals(2, len(list(self
.tdb
)))
150 self
.assertEquals(0, len(list(self
.tdb
)))
152 def test_repack(self
):
153 self
.tdb
["foo"] = "abc"
154 self
.tdb
["bar"] = "def"
158 def test_seqnum(self
):
159 self
.tdb
.enable_seqnum()
160 seq1
= self
.tdb
.seqnum
161 self
.tdb
.increment_seqnum_nonblock()
162 seq2
= self
.tdb
.seqnum
163 self
.assertEquals(seq2
-seq1
, 1)
166 self
.assertEquals(0, len(list(self
.tdb
)))
167 self
.tdb
["entry"] = "value"
168 self
.assertEquals(1, len(list(self
.tdb
)))
170 def test_add_flags(self
):
171 self
.tdb
.add_flags(tdb
.NOMMAP
)
172 self
.tdb
.remove_flags(tdb
.NOMMAP
)
175 class VersionTests(TestCase
):
177 def test_present(self
):
178 self
.assertTrue(isinstance(tdb
.__version
__, str))
181 if __name__
== '__main__':
183 unittest
.TestProgram()