mirror of
https://github.com/python/cpython.git
synced 2025-10-28 12:15:13 +00:00
6ba3329c27
Python 2.6 one, since the intention is to keep an unified 2.x/3.x codebase. The Python code is automatically translated using "2to3". Please, do not update this code in Python 3.0 by hand. Update the 2.6 one and then do "2to3".
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
import unittest
|
|
import os, glob
|
|
|
|
from .test_all import db, test_support, get_new_environment_path, \
|
|
get_new_database_path
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class pget_bugTestCase(unittest.TestCase):
|
|
"""Verify that cursor.pget works properly"""
|
|
db_name = 'test-cursor_pget.db'
|
|
|
|
def setUp(self):
|
|
self.homeDir = get_new_environment_path()
|
|
self.env = db.DBEnv()
|
|
self.env.open(self.homeDir, db.DB_CREATE | db.DB_INIT_MPOOL)
|
|
self.primary_db = db.DB(self.env)
|
|
self.primary_db.open(self.db_name, 'primary', db.DB_BTREE, db.DB_CREATE)
|
|
self.secondary_db = db.DB(self.env)
|
|
self.secondary_db.set_flags(db.DB_DUP)
|
|
self.secondary_db.open(self.db_name, 'secondary', db.DB_BTREE, db.DB_CREATE)
|
|
self.primary_db.associate(self.secondary_db, lambda key, data: data)
|
|
self.primary_db.put('salad', 'eggs')
|
|
self.primary_db.put('spam', 'ham')
|
|
self.primary_db.put('omelet', 'eggs')
|
|
|
|
|
|
def tearDown(self):
|
|
self.secondary_db.close()
|
|
self.primary_db.close()
|
|
self.env.close()
|
|
del self.secondary_db
|
|
del self.primary_db
|
|
del self.env
|
|
test_support.rmtree(self.homeDir)
|
|
|
|
def test_pget(self):
|
|
cursor = self.secondary_db.cursor()
|
|
|
|
self.assertEquals(('eggs', 'salad', 'eggs'), cursor.pget(key='eggs', flags=db.DB_SET))
|
|
self.assertEquals(('eggs', 'omelet', 'eggs'), cursor.pget(db.DB_NEXT_DUP))
|
|
self.assertEquals(None, cursor.pget(db.DB_NEXT_DUP))
|
|
|
|
self.assertEquals(('ham', 'spam', 'ham'), cursor.pget('ham', 'spam', flags=db.DB_SET))
|
|
self.assertEquals(None, cursor.pget(db.DB_NEXT_DUP))
|
|
|
|
cursor.close()
|
|
|
|
|
|
def test_suite():
|
|
return unittest.makeSuite(pget_bugTestCase)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='test_suite')
|