← Back to branch summary

~malept/ubuntu/lucid/python2.6/dev-dependency-fix

« back to all changes in this revision

Viewing changes to Lib/bsddb/test/test_get_none.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-02-13 12:51:00 UTC
  • Revision ID: james.westby@ubuntu.com-20090213125100-uufgcb9yeqzujpqw
Tags: upstream-2.6.1
Import upstream version 2.6.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
Lib/bsddb/test/test_get_none.py
 
1
"""
 
2
TestCases for checking set_get_returns_none.
 
3
"""
 
4
 
 
5
import os, string
 
6
import unittest
 
7
 
 
8
from test_all import db, verbose, get_new_database_path
 
9
 
 
10
 
 
11
#----------------------------------------------------------------------
 
12
 
 
13
class GetReturnsNoneTestCase(unittest.TestCase):
 
14
    def setUp(self):
 
15
        self.filename = get_new_database_path()
 
16
 
 
17
    def tearDown(self):
 
18
        try:
 
19
            os.remove(self.filename)
 
20
        except os.error:
 
21
            pass
 
22
 
 
23
 
 
24
    def test01_get_returns_none(self):
 
25
        d = db.DB()
 
26
        d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
 
27
        d.set_get_returns_none(1)
 
28
 
 
29
        for x in string.letters:
 
30
            d.put(x, x * 40)
 
31
 
 
32
        data = d.get('bad key')
 
33
        self.assertEqual(data, None)
 
34
 
 
35
        data = d.get(string.letters[0])
 
36
        self.assertEqual(data, string.letters[0]*40)
 
37
 
 
38
        count = 0
 
39
        c = d.cursor()
 
40
        rec = c.first()
 
41
        while rec:
 
42
            count = count + 1
 
43
            rec = c.next()
 
44
 
 
45
        self.assertEqual(rec, None)
 
46
        self.assertEqual(count, len(string.letters))
 
47
 
 
48
        c.close()
 
49
        d.close()
 
50
 
 
51
 
 
52
    def test02_get_raises_exception(self):
 
53
        d = db.DB()
 
54
        d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
 
55
        d.set_get_returns_none(0)
 
56
 
 
57
        for x in string.letters:
 
58
            d.put(x, x * 40)
 
59
 
 
60
        self.assertRaises(db.DBNotFoundError, d.get, 'bad key')
 
61
        self.assertRaises(KeyError, d.get, 'bad key')
 
62
 
 
63
        data = d.get(string.letters[0])
 
64
        self.assertEqual(data, string.letters[0]*40)
 
65
 
 
66
        count = 0
 
67
        exceptionHappened = 0
 
68
        c = d.cursor()
 
69
        rec = c.first()
 
70
        while rec:
 
71
            count = count + 1
 
72
            try:
 
73
                rec = c.next()
 
74
            except db.DBNotFoundError:  # end of the records
 
75
                exceptionHappened = 1
 
76
                break
 
77
 
 
78
        self.assertNotEqual(rec, None)
 
79
        self.assert_(exceptionHappened)
 
80
        self.assertEqual(count, len(string.letters))
 
81
 
 
82
        c.close()
 
83
        d.close()
 
84
 
 
85
#----------------------------------------------------------------------
 
86
 
 
87
def test_suite():
 
88
    return unittest.makeSuite(GetReturnsNoneTestCase)
 
89
 
 
90
 
 
91
if __name__ == '__main__':
 
92
    unittest.main(defaultTest='test_suite')