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_nonexistant_read(self):
17 self.assertRaises(IOError, tdb.Tdb, "/some/nonexistant/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
32 class InternalTdbTests(TestCase):
37 # repr used to crash on internal db
38 self.assertEquals(repr(self.tdb), "Tdb(<internal>)")
41 class SimpleTdbTests(TestCase):
44 super(SimpleTdbTests, self).setUp()
45 self.tdb = tdb.Tdb(tempfile.mkstemp()[1], 0, tdb.DEFAULT,
47 self.assertNotEqual(None, self.tdb)
53 self.assertTrue(repr(self.tdb).startswith("Tdb('"))
55 def test_lockall(self):
58 def test_max_dead(self):
59 self.tdb.max_dead = 20
61 def test_unlockall(self):
65 def test_lockall_read(self):
66 self.tdb.read_lock_all()
67 self.tdb.read_unlock_all()
69 def test_reopen(self):
73 self.tdb.store("bar", "bla")
74 self.assertEquals("bla", self.tdb.get("bar"))
76 def test_getitem(self):
77 self.tdb["bar"] = "foo"
79 self.assertEquals("foo", self.tdb["bar"])
81 def test_delete(self):
82 self.tdb["bar"] = "foo"
84 self.assertRaises(KeyError, lambda: self.tdb["bar"])
86 def test_contains(self):
87 self.tdb["bla"] = "bloe"
88 self.assertTrue("bla" in self.tdb)
90 def test_keyerror(self):
91 self.assertRaises(KeyError, lambda: self.tdb["bla"])
93 def test_hash_size(self):
96 def test_map_size(self):
99 def test_freelist_size(self):
100 self.tdb.freelist_size
105 def test_iterator(self):
106 self.tdb["bla"] = "1"
107 self.tdb["brainslug"] = "2"
110 self.assertEquals(["bla", "brainslug"], l)
112 def test_transaction_cancel(self):
113 self.tdb["bloe"] = "2"
114 self.tdb.transaction_start()
115 self.tdb["bloe"] = "1"
116 self.tdb.transaction_cancel()
117 self.assertEquals("2", self.tdb["bloe"])
119 def test_transaction_commit(self):
120 self.tdb["bloe"] = "2"
121 self.tdb.transaction_start()
122 self.tdb["bloe"] = "1"
123 self.tdb.transaction_commit()
124 self.assertEquals("1", self.tdb["bloe"])
126 def test_transaction_prepare_commit(self):
127 self.tdb["bloe"] = "2"
128 self.tdb.transaction_start()
129 self.tdb["bloe"] = "1"
130 self.tdb.transaction_prepare_commit()
131 self.tdb.transaction_commit()
132 self.assertEquals("1", self.tdb["bloe"])
134 def test_iterkeys(self):
135 self.tdb["bloe"] = "2"
136 self.tdb["bla"] = "25"
137 i = self.tdb.iterkeys()
138 self.assertEquals(set(["bloe", "bla"]), set([i.next(), i.next()]))
140 def test_clear(self):
141 self.tdb["bloe"] = "2"
142 self.tdb["bla"] = "25"
143 self.assertEquals(2, len(list(self.tdb)))
145 self.assertEquals(0, len(list(self.tdb)))
147 def test_repack(self):
148 self.tdb["foo"] = "abc"
149 self.tdb["bar"] = "def"
153 def test_seqnum(self):
154 self.tdb.enable_seqnum()
155 seq1 = self.tdb.seqnum
156 self.tdb.increment_seqnum_nonblock()
157 seq2 = self.tdb.seqnum
158 self.assertEquals(seq2-seq1, 1)
161 self.assertEquals(0, len(list(self.tdb)))
162 self.tdb["entry"] = "value"
163 self.assertEquals(1, len(list(self.tdb)))
165 def test_add_flags(self):
166 self.tdb.add_flags(tdb.NOMMAP)
167 self.tdb.remove_flags(tdb.NOMMAP)
170 class VersionTests(TestCase):
172 def test_present(self):
173 self.assertTrue(isinstance(tdb.__version__, str))
176 if __name__ == '__main__':
178 unittest.TestProgram()