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-2013 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, ntdb.Ntdb, "/some/nonexistent/file", 0,
18 ntdb.DEFAULT, os.O_RDWR)
20 class CloseTdbTests(TestCase):
22 def test_double_close(self):
23 self.ntdb = ntdb.Ntdb(tempfile.mkstemp()[1], ntdb.DEFAULT,
25 self.assertNotEqual(None, self.ntdb)
27 # ensure that double close does not crash python
31 # Check that further operations do not crash python
32 self.assertRaises(RuntimeError, lambda: self.ntdb.transaction_start())
34 self.assertRaises(RuntimeError, lambda: self.ntdb["bar"])
37 class InternalTdbTests(TestCase):
40 self.ntdb = ntdb.Ntdb()
42 # repr used to crash on internal db
43 self.assertEquals(repr(self.ntdb), "Ntdb(<internal>)")
46 class SimpleTdbTests(TestCase):
49 super(SimpleTdbTests, self).setUp()
50 self.ntdb = ntdb.Ntdb(tempfile.mkstemp()[1], ntdb.DEFAULT,
52 self.assertNotEqual(None, self.ntdb)
58 self.assertTrue(repr(self.ntdb).startswith("Ntdb('"))
60 def test_lockall(self):
63 def test_unlockall(self):
65 self.ntdb.unlock_all()
67 def test_lockall_read(self):
68 self.ntdb.read_lock_all()
69 self.ntdb.read_unlock_all()
72 self.ntdb.store("bar", "bla")
73 self.assertEquals("bla", self.ntdb.get("bar"))
75 def test_getitem(self):
76 self.ntdb["bar"] = "foo"
77 self.assertEquals("foo", self.ntdb["bar"])
79 def test_delete(self):
80 self.ntdb["bar"] = "foo"
82 self.assertRaises(KeyError, lambda: self.ntdb["bar"])
84 def test_contains(self):
85 self.ntdb["bla"] = "bloe"
86 self.assertTrue("bla" in self.ntdb)
88 def test_keyerror(self):
89 self.assertRaises(KeyError, lambda: self.ntdb["bla"])
94 def test_iterator(self):
95 self.ntdb["bla"] = "1"
96 self.ntdb["brainslug"] = "2"
99 self.assertEquals(["bla", "brainslug"], l)
101 def test_transaction_cancel(self):
102 self.ntdb["bloe"] = "2"
103 self.ntdb.transaction_start()
104 self.ntdb["bloe"] = "1"
105 self.ntdb.transaction_cancel()
106 self.assertEquals("2", self.ntdb["bloe"])
108 def test_transaction_commit(self):
109 self.ntdb["bloe"] = "2"
110 self.ntdb.transaction_start()
111 self.ntdb["bloe"] = "1"
112 self.ntdb.transaction_commit()
113 self.assertEquals("1", self.ntdb["bloe"])
115 def test_transaction_prepare_commit(self):
116 self.ntdb["bloe"] = "2"
117 self.ntdb.transaction_start()
118 self.ntdb["bloe"] = "1"
119 self.ntdb.transaction_prepare_commit()
120 self.ntdb.transaction_commit()
121 self.assertEquals("1", self.ntdb["bloe"])
123 def test_iterkeys(self):
124 self.ntdb["bloe"] = "2"
125 self.ntdb["bla"] = "25"
126 i = self.ntdb.iterkeys()
127 self.assertEquals(set(["bloe", "bla"]), set([i.next(), i.next()]))
129 def test_clear(self):
130 self.ntdb["bloe"] = "2"
131 self.ntdb["bla"] = "25"
132 self.assertEquals(2, len(list(self.ntdb)))
134 self.assertEquals(0, len(list(self.ntdb)))
137 self.assertEquals(0, len(list(self.ntdb)))
138 self.ntdb["entry"] = "value"
139 self.assertEquals(1, len(list(self.ntdb)))
141 def test_add_flags(self):
142 self.ntdb.add_flag(ntdb.NOMMAP)
143 self.ntdb.remove_flag(ntdb.NOMMAP)
146 class VersionTests(TestCase):
148 def test_present(self):
149 self.assertTrue(isinstance(ntdb.__version__, str))
152 if __name__ == '__main__':
154 unittest.TestProgram()