source/lib/tdb/python/tests/simple.py

   1 #!/usr/bin/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 Jelmer Vernooij <jelmer@samba.org>
   7 # Published under the GNU LGPL
   8
   9 import tdb
  10 from unittest import TestCase
  11 import os
  12
  13 class SimpleTdbTests(TestCase):
  14     def setUp(self):
  15         super(SimpleTdbTests, self).setUp()
  16         self.tdb = tdb.Tdb(os.tmpnam(), 0, tdb.DEFAULT, os.O_CREAT|os.O_RDWR)
  17         self.assertNotEqual(None, self.tdb)
  18
  19     def tearDown(self):
  20         del self.tdb
  21
  22     def test_lockall(self):
  23         self.tdb.lock_all()
  24
  25     def test_max_dead(self):
  26         self.tdb.max_dead = 20
  27
  28     def test_unlockall(self):
  29         self.tdb.lock_all()
  30         self.tdb.unlock_all()
  31
  32     def test_lockall_read(self):
  33         self.tdb.read_lock_all()
  34         self.tdb.read_unlock_all()
  35
  36     def test_reopen(self):
  37         self.tdb.reopen()
  38
  39     def test_store(self):
  40         self.tdb.store("bar", "bla")
  41         self.assertEquals("bla", self.tdb.fetch("bar"))
  42
  43     def test_fetch(self):
  44         self.tdb["bar"] = "foo"
  45         self.tdb.reopen()
  46         self.assertEquals("foo", self.tdb["bar"])
  47
  48     def test_delete(self):
  49         self.tdb["bar"] = "foo"
  50         del self.tdb["bar"]
  51         self.assertRaises(KeyError, lambda: self.tdb["bar"])
  52
  53     def test_contains(self):
  54         self.tdb["bla"] = "bloe"
  55         self.assertTrue("bla" in self.tdb)
  56
  57     def test_keyerror(self):
  58         self.assertRaises(KeyError, lambda: self.tdb["bla"])
  59
  60     def test_hash_size(self):
  61         self.tdb.hash_size
  62
  63     def test_map_size(self):
  64         self.tdb.map_size
  65
  66     def test_name(self):
  67         self.tdb.name
  68
  69     def test_iterator(self):
  70         self.tdb["bla"] = "1"
  71         self.tdb["brainslug"] = "2"
  72         self.assertEquals(["bla", "brainslug"], list(self.tdb))
  73
  74     def test_items(self):
  75         self.tdb["bla"] = "1"
  76         self.tdb["brainslug"] = "2"
  77         self.assertEquals([("bla", "1"), ("brainslug", "2")], self.tdb.items())
  78
  79     def test_transaction_cancel(self):
  80         self.tdb["bloe"] = "2"
  81         self.tdb.transaction_start()
  82         self.tdb["bloe"] = "1"
  83         self.tdb.transaction_cancel()
  84         self.assertEquals("2", self.tdb["bloe"])
  85
  86     def test_transaction_commit(self):
  87         self.tdb["bloe"] = "2"
  88         self.tdb.transaction_start()
  89         self.tdb["bloe"] = "1"
  90         self.tdb.transaction_commit()
  91         self.assertEquals("1", self.tdb["bloe"])
  92
  93     def test_iterator(self):
  94         self.tdb["bloe"] = "2"
  95         self.tdb["bla"] = "hoi"
  96         i = iter(self.tdb)
  97         self.assertEquals(set(["bloe", "bla"]), set([i.next(), i.next()]))
  98
  99     def test_keys(self):
 100         self.tdb["bloe"] = "2"
 101         self.tdb["bla"] = "25"
 102         self.assertEquals(["bla", "bloe"], self.tdb.keys())
 103
 104     def test_values(self):
 105         self.tdb["bloe"] = "2"
 106         self.tdb["bla"] = "25"
 107         self.assertEquals(["25", "2"], self.tdb.values())
 108
 109     def test_clear(self):
 110         self.tdb["bloe"] = "2"
 111         self.tdb["bla"] = "25"
 112         self.assertEquals(2, len(self.tdb))
 113         self.tdb.clear()
 114         self.assertEquals(0, len(self.tdb))
 115
 116     def test_len(self):
 117         self.assertEquals(0, len(self.tdb))
 118         self.tdb["entry"] = "value"
 119         self.assertEquals(1, len(self.tdb))