lib/tdb/python/tests/simple.py

   1 #!/usr/bin/env python
   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.
   5 #
   6 # Copyright (C) 2007-2008 Jelmer Vernooij <jelmer@samba.org>
   7 # Published under the GNU LGPLv3 or later
   8
   9 import tdb
  10 from unittest import TestCase
  11 import os, tempfile
  12
  13
  14 class OpenTdbTests(TestCase):
  15
  16     def test_nonexistent_read(self):
  17         self.assertRaises(IOError, tdb.Tdb, "/some/nonexistent/file", 0,
  18                 tdb.DEFAULT, os.O_RDWR)
  19
  20 class CloseTdbTests(TestCase):
  21
  22     def test_double_close(self):
  23         # No hash size in tdb2.
  24         if tdb.__version__.startswith("2"):
  25             self.tdb = tdb.Tdb(tempfile.mkstemp()[1], tdb.DEFAULT,
  26                                os.O_CREAT|os.O_RDWR)
  27         else:
  28             self.tdb = tdb.Tdb(tempfile.mkstemp()[1], 0, tdb.DEFAULT,
  29                                os.O_CREAT|os.O_RDWR)
  30         self.assertNotEqual(None, self.tdb)
  31
  32         # ensure that double close does not crash python
  33         self.tdb.close()
  34         self.tdb.close()
  35
  36         # Check that further operations do not crash python
  37         self.assertRaises(RuntimeError, lambda: self.tdb.transaction_start())
  38
  39         self.assertRaises(RuntimeError, lambda: self.tdb["bar"])
  40
  41
  42 class InternalTdbTests(TestCase):
  43
  44     def test_repr(self):
  45         self.tdb = tdb.Tdb()
  46
  47         # repr used to crash on internal db
  48         self.assertEquals(repr(self.tdb), "Tdb(<internal>)")
  49
  50
  51 class SimpleTdbTests(TestCase):
  52
  53     def setUp(self):
  54         super(SimpleTdbTests, self).setUp()
  55         if tdb.__version__.startswith("2"):
  56             self.tdb = tdb.Tdb(tempfile.mkstemp()[1], tdb.DEFAULT,
  57                                os.O_CREAT|os.O_RDWR)
  58         else:
  59             self.tdb = tdb.Tdb(tempfile.mkstemp()[1], 0, tdb.DEFAULT,
  60                                os.O_CREAT|os.O_RDWR)
  61         self.assertNotEqual(None, self.tdb)
  62
  63     def tearDown(self):
  64         del self.tdb
  65
  66     def test_repr(self):
  67         self.assertTrue(repr(self.tdb).startswith("Tdb('"))
  68
  69     def test_lockall(self):
  70         self.tdb.lock_all()
  71
  72     def test_max_dead(self):
  73         if not tdb.__version__.startswith("2"):
  74             self.tdb.max_dead = 20
  75
  76     def test_unlockall(self):
  77         self.tdb.lock_all()
  78         self.tdb.unlock_all()
  79
  80     def test_lockall_read(self):
  81         self.tdb.read_lock_all()
  82         self.tdb.read_unlock_all()
  83
  84     def test_reopen(self):
  85         if not tdb.__version__.startswith("2"):
  86             self.tdb.reopen()
  87
  88     def test_store(self):
  89         self.tdb.store("bar", "bla")
  90         self.assertEquals("bla", self.tdb.get("bar"))
  91
  92     def test_getitem(self):
  93         self.tdb["bar"] = "foo"
  94         if not tdb.__version__.startswith("2"):
  95             self.tdb.reopen()
  96         self.assertEquals("foo", self.tdb["bar"])
  97
  98     def test_delete(self):
  99         self.tdb["bar"] = "foo"
 100         del self.tdb["bar"]
 101         self.assertRaises(KeyError, lambda: self.tdb["bar"])
 102
 103     def test_contains(self):
 104         self.tdb["bla"] = "bloe"
 105         self.assertTrue("bla" in self.tdb)
 106
 107     def test_keyerror(self):
 108         self.assertRaises(KeyError, lambda: self.tdb["bla"])
 109
 110     def test_hash_size(self):
 111         if not tdb.__version__.startswith("2"):
 112             self.tdb.hash_size
 113
 114     def test_map_size(self):
 115         if not tdb.__version__.startswith("2"):
 116             self.tdb.map_size
 117
 118     def test_freelist_size(self):
 119         if not tdb.__version__.startswith("2"):
 120             self.tdb.freelist_size
 121
 122     def test_name(self):
 123         self.tdb.filename
 124
 125     def test_iterator(self):
 126         self.tdb["bla"] = "1"
 127         self.tdb["brainslug"] = "2"
 128         l = list(self.tdb)
 129         l.sort()
 130         self.assertEquals(["bla", "brainslug"], l)
 131
 132     def test_transaction_cancel(self):
 133         self.tdb["bloe"] = "2"
 134         self.tdb.transaction_start()
 135         self.tdb["bloe"] = "1"
 136         self.tdb.transaction_cancel()
 137         self.assertEquals("2", self.tdb["bloe"])
 138
 139     def test_transaction_commit(self):
 140         self.tdb["bloe"] = "2"
 141         self.tdb.transaction_start()
 142         self.tdb["bloe"] = "1"
 143         self.tdb.transaction_commit()
 144         self.assertEquals("1", self.tdb["bloe"])
 145
 146     def test_transaction_prepare_commit(self):
 147         self.tdb["bloe"] = "2"
 148         self.tdb.transaction_start()
 149         self.tdb["bloe"] = "1"
 150         self.tdb.transaction_prepare_commit()
 151         self.tdb.transaction_commit()
 152         self.assertEquals("1", self.tdb["bloe"])
 153
 154     def test_iterkeys(self):
 155         self.tdb["bloe"] = "2"
 156         self.tdb["bla"] = "25"
 157         i = self.tdb.iterkeys()
 158         self.assertEquals(set(["bloe", "bla"]), set([i.next(), i.next()]))
 159
 160     def test_clear(self):
 161         self.tdb["bloe"] = "2"
 162         self.tdb["bla"] = "25"
 163         self.assertEquals(2, len(list(self.tdb)))
 164         self.tdb.clear()
 165         self.assertEquals(0, len(list(self.tdb)))
 166
 167     def test_repack(self):
 168         if not tdb.__version__.startswith("2"):
 169             self.tdb["foo"] = "abc"
 170             self.tdb["bar"] = "def"
 171             del self.tdb["foo"]
 172             self.tdb.repack()
 173
 174     def test_seqnum(self):
 175         if not tdb.__version__.startswith("2"):
 176             self.tdb.enable_seqnum()
 177             seq1 = self.tdb.seqnum
 178             self.tdb.increment_seqnum_nonblock()
 179             seq2 = self.tdb.seqnum
 180             self.assertEquals(seq2-seq1, 1)
 181
 182     def test_len(self):
 183         self.assertEquals(0, len(list(self.tdb)))
 184         self.tdb["entry"] = "value"
 185         self.assertEquals(1, len(list(self.tdb)))
 186
 187     def test_add_flags(self):
 188         if tdb.__version__.startswith("2"):
 189             self.tdb.add_flag(tdb.NOMMAP)
 190             self.tdb.remove_flag(tdb.NOMMAP)
 191         else:
 192             self.tdb.add_flags(tdb.NOMMAP)
 193             self.tdb.remove_flags(tdb.NOMMAP)
 194
 195
 196 class VersionTests(TestCase):
 197
 198     def test_present(self):
 199         self.assertTrue(isinstance(tdb.__version__, str))
 200
 201
 202 if __name__ == '__main__':
 203     import unittest
 204     unittest.TestProgram()