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 Jelmer Vernooij <jelmer@samba.org>
7 # Published under the GNU LGPL
10 from unittest import TestCase
14 class OpenTdbTests(TestCase):
15 def test_nonexistant_read(self):
16 self.assertRaises(IOError, tdb.Tdb, "/some/nonexistant/file", 0, tdb.DEFAULT, os.O_RDWR)
19 class SimpleTdbTests(TestCase):
21 super(SimpleTdbTests, self).setUp()
22 self.tdb = tdb.Tdb(tempfile.mkstemp()[1], 0, tdb.DEFAULT, os.O_CREAT|os.O_RDWR)
23 self.assertNotEqual(None, self.tdb)
28 def test_lockall(self):
31 def test_max_dead(self):
32 self.tdb.max_dead = 20
34 def test_unlockall(self):
38 def test_lockall_read(self):
39 self.tdb.read_lock_all()
40 self.tdb.read_unlock_all()
42 def test_reopen(self):
46 self.tdb.store("bar", "bla")
47 self.assertEquals("bla", self.tdb.get("bar"))
49 def test_getitem(self):
50 self.tdb["bar"] = "foo"
52 self.assertEquals("foo", self.tdb["bar"])
54 def test_delete(self):
55 self.tdb["bar"] = "foo"
57 self.assertRaises(KeyError, lambda: self.tdb["bar"])
59 def test_contains(self):
60 self.tdb["bla"] = "bloe"
61 self.assertTrue("bla" in self.tdb)
63 def test_keyerror(self):
64 self.assertRaises(KeyError, lambda: self.tdb["bla"])
66 def test_hash_size(self):
69 def test_map_size(self):
75 def test_iterator(self):
77 self.tdb["brainslug"] = "2"
78 self.assertEquals(["bla", "brainslug"], list(self.tdb))
82 self.tdb["brainslug"] = "2"
83 self.assertEquals([("bla", "1"), ("brainslug", "2")], self.tdb.items())
85 def test_iteritems(self):
86 self.tdb["bloe"] = "2"
87 self.tdb["bla"] = "25"
88 i = self.tdb.iteritems()
89 self.assertEquals(set([("bla", "25"), ("bloe", "2")]),
90 set([i.next(), i.next()]))
92 def test_transaction_cancel(self):
93 self.tdb["bloe"] = "2"
94 self.tdb.transaction_start()
95 self.tdb["bloe"] = "1"
96 self.tdb.transaction_cancel()
97 self.assertEquals("2", self.tdb["bloe"])
99 def test_transaction_commit(self):
100 self.tdb["bloe"] = "2"
101 self.tdb.transaction_start()
102 self.tdb["bloe"] = "1"
103 self.tdb.transaction_commit()
104 self.assertEquals("1", self.tdb["bloe"])
106 def test_iterator(self):
107 self.tdb["bloe"] = "2"
108 self.tdb["bla"] = "hoi"
110 self.assertEquals(set(["bloe", "bla"]), set([i.next(), i.next()]))
113 self.tdb["bloe"] = "2"
114 self.tdb["bla"] = "25"
115 self.assertEquals(["bla", "bloe"], self.tdb.keys())
117 def test_iterkeys(self):
118 self.tdb["bloe"] = "2"
119 self.tdb["bla"] = "25"
120 i = self.tdb.iterkeys()
121 self.assertEquals(set(["bloe", "bla"]), set([i.next(), i.next()]))
123 def test_values(self):
124 self.tdb["bloe"] = "2"
125 self.tdb["bla"] = "25"
126 self.assertEquals(["25", "2"], self.tdb.values())
128 def test_itervalues(self):
129 self.tdb["bloe"] = "2"
130 self.tdb["bla"] = "25"
131 i = self.tdb.itervalues()
132 self.assertEquals(set(["25", "2"]), set([i.next(), i.next()]))
134 def test_clear(self):
135 self.tdb["bloe"] = "2"
136 self.tdb["bla"] = "25"
137 self.assertEquals(2, len(self.tdb))
139 self.assertEquals(0, len(self.tdb))
142 self.assertEquals(0, len(self.tdb))
143 self.tdb["entry"] = "value"
144 self.assertEquals(1, len(self.tdb))
147 if __name__ == '__main__':
149 unittest.TestProgram()