home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2010 November / maximum-cd-2010-11.iso / DiscContents / calibre-0.7.13.msi / file_288 (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2010-08-06  |  2.4 KB  |  54 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.6)
  3.  
  4. import unittest
  5. import os
  6. import glob
  7. from test_all import db, test_support, get_new_environment_path, get_new_database_path
  8.  
  9. class pget_bugTestCase(unittest.TestCase):
  10.     db_name = 'test-cursor_pget.db'
  11.     
  12.     def setUp(self):
  13.         self.homeDir = get_new_environment_path()
  14.         self.env = db.DBEnv()
  15.         self.env.open(self.homeDir, db.DB_CREATE | db.DB_INIT_MPOOL)
  16.         self.primary_db = db.DB(self.env)
  17.         self.primary_db.open(self.db_name, 'primary', db.DB_BTREE, db.DB_CREATE)
  18.         self.secondary_db = db.DB(self.env)
  19.         self.secondary_db.set_flags(db.DB_DUP)
  20.         self.secondary_db.open(self.db_name, 'secondary', db.DB_BTREE, db.DB_CREATE)
  21.         self.primary_db.associate(self.secondary_db, (lambda key, data: data))
  22.         self.primary_db.put('salad', 'eggs')
  23.         self.primary_db.put('spam', 'ham')
  24.         self.primary_db.put('omelet', 'eggs')
  25.  
  26.     
  27.     def tearDown(self):
  28.         self.secondary_db.close()
  29.         self.primary_db.close()
  30.         self.env.close()
  31.         del self.secondary_db
  32.         del self.primary_db
  33.         del self.env
  34.         test_support.rmtree(self.homeDir)
  35.  
  36.     
  37.     def test_pget(self):
  38.         cursor = self.secondary_db.cursor()
  39.         self.assertEquals(('eggs', 'salad', 'eggs'), cursor.pget(key = 'eggs', flags = db.DB_SET))
  40.         self.assertEquals(('eggs', 'omelet', 'eggs'), cursor.pget(db.DB_NEXT_DUP))
  41.         self.assertEquals(None, cursor.pget(db.DB_NEXT_DUP))
  42.         self.assertEquals(('ham', 'spam', 'ham'), cursor.pget('ham', 'spam', flags = db.DB_SET))
  43.         self.assertEquals(None, cursor.pget(db.DB_NEXT_DUP))
  44.         cursor.close()
  45.  
  46.  
  47.  
  48. def test_suite():
  49.     return unittest.makeSuite(pget_bugTestCase)
  50.  
  51. if __name__ == '__main__':
  52.     unittest.main(defaultTest = 'test_suite')
  53.  
  54.