← Back to branch summary

~ubuntu-branches/ubuntu/karmic/pypy/karmic

« back to all changes in this revision

Viewing changes to lib-python/2.4.1/test/test_dumbdbm.py

  • Committer: Bazaar Package Importer
  • Author(s): Alexandre Fayolle
  • Date: 2007-04-13 09:33:09 UTC
  • Revision ID: james.westby@ubuntu.com-20070413093309-yoojh4jcoocu2krz
Tags: upstream-1.0.0
Import upstream version 1.0.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib-python/2.4.1/test/test_dumbdbm.py
 
1
#! /usr/bin/env python
 
2
"""Test script for the dumbdbm module
 
3
   Original by Roger E. Masse
 
4
"""
 
5
 
 
6
import os
 
7
import unittest
 
8
import dumbdbm
 
9
from test import test_support
 
10
 
 
11
_fname = test_support.TESTFN
 
12
 
 
13
def _delete_files():
 
14
    for ext in [".dir", ".dat", ".bak"]:
 
15
        try:
 
16
            os.unlink(_fname + ext)
 
17
        except OSError:
 
18
            pass
 
19
 
 
20
class DumbDBMTestCase(unittest.TestCase):
 
21
    _dict = {'0': '',
 
22
             'a': 'Python:',
 
23
             'b': 'Programming',
 
24
             'c': 'the',
 
25
             'd': 'way',
 
26
             'f': 'Guido',
 
27
             'g': 'intended'
 
28
             }
 
29
 
 
30
    def __init__(self, *args):
 
31
        unittest.TestCase.__init__(self, *args)
 
32
 
 
33
    def test_dumbdbm_creation(self):
 
34
        f = dumbdbm.open(_fname, 'c')
 
35
        self.assertEqual(f.keys(), [])
 
36
        for key in self._dict:
 
37
            f[key] = self._dict[key]
 
38
        self.read_helper(f)
 
39
        f.close()
 
40
 
 
41
    def test_close_twice(self):
 
42
        f = dumbdbm.open(_fname)
 
43
        f['a'] = 'b'
 
44
        self.assertEqual(f['a'], 'b')
 
45
        f.close()
 
46
        f.close()
 
47
 
 
48
    def test_dumbdbm_modification(self):
 
49
        self.init_db()
 
50
        f = dumbdbm.open(_fname, 'w')
 
51
        self._dict['g'] = f['g'] = "indented"
 
52
        self.read_helper(f)
 
53
        f.close()
 
54
 
 
55
    def test_dumbdbm_read(self):
 
56
        self.init_db()
 
57
        f = dumbdbm.open(_fname, 'r')
 
58
        self.read_helper(f)
 
59
        f.close()
 
60
 
 
61
    def test_dumbdbm_keys(self):
 
62
        self.init_db()
 
63
        f = dumbdbm.open(_fname)
 
64
        keys = self.keys_helper(f)
 
65
        f.close()
 
66
 
 
67
    def test_write_write_read(self):
 
68
        # test for bug #482460
 
69
        f = dumbdbm.open(_fname)
 
70
        f['1'] = 'hello'
 
71
        f['1'] = 'hello2'
 
72
        f.close()
 
73
        f = dumbdbm.open(_fname)
 
74
        self.assertEqual(f['1'], 'hello2')
 
75
        f.close()
 
76
 
 
77
    def read_helper(self, f):
 
78
        keys = self.keys_helper(f)
 
79
        for key in self._dict:
 
80
            self.assertEqual(self._dict[key], f[key])
 
81
 
 
82
    def init_db(self):
 
83
        f = dumbdbm.open(_fname, 'w')
 
84
        for k in self._dict:
 
85
            f[k] = self._dict[k]
 
86
        f.close()
 
87
 
 
88
    def keys_helper(self, f):
 
89
        keys = f.keys()
 
90
        keys.sort()
 
91
        dkeys = self._dict.keys()
 
92
        dkeys.sort()
 
93
        self.assertEqual(keys, dkeys)
 
94
        return keys
 
95
 
 
96
    # Perform randomized operations.  This doesn't make assumptions about
 
97
    # what *might* fail.
 
98
    def test_random(self):
 
99
        import random
 
100
        d = {}  # mirror the database
 
101
        for dummy in range(5):
 
102
            f = dumbdbm.open(_fname)
 
103
            for dummy in range(100):
 
104
                k = random.choice('abcdefghijklm')
 
105
                if random.random() < 0.2:
 
106
                    if k in d:
 
107
                        del d[k]
 
108
                        del f[k]
 
109
                else:
 
110
                    v = random.choice('abc') * random.randrange(10000)
 
111
                    d[k] = v
 
112
                    f[k] = v
 
113
                    self.assertEqual(f[k], v)
 
114
            f.close()
 
115
 
 
116
            f = dumbdbm.open(_fname)
 
117
            expected = d.items()
 
118
            expected.sort()
 
119
            got = f.items()
 
120
            got.sort()
 
121
            self.assertEqual(expected, got)
 
122
            f.close()
 
123
 
 
124
    def tearDown(self):
 
125
        _delete_files()
 
126
 
 
127
    def setUp(self):
 
128
        _delete_files()
 
129
 
 
130
def test_main():
 
131
    try:
 
132
        test_support.run_unittest(DumbDBMTestCase)
 
133
    finally:
 
134
        _delete_files()
 
135
 
 
136
if __name__ == "__main__":
 
137
    test_main()