test/test_metakit.py

   1 #
   2 # Copyright (c) 2001 Bizar Software Pty Ltd (http://www.bizarsoftware.com.au/)
   3 # This module is free software, and you may redistribute it and/or modify
   4 # under the same terms as Python, so long as this copyright message and
   5 # disclaimer are retained in their original form.
   6 #
   7 # IN NO EVENT SHALL BIZAR SOFTWARE PTY LTD BE LIABLE TO ANY PARTY FOR
   8 # DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING
   9 # OUT OF THE USE OF THIS CODE, EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE
  10 # POSSIBILITY OF SUCH DAMAGE.
  11 #
  12 # BIZAR SOFTWARE PTY LTD SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING,
  13 # BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  14 # FOR A PARTICULAR PURPOSE.  THE CODE PROVIDED HEREUNDER IS ON AN "AS IS"
  15 # BASIS, AND THERE IS NO OBLIGATION WHATSOEVER TO PROVIDE MAINTENANCE,
  16 # SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
  17 #
  18 # $Id: test_metakit.py,v 1.4 2004-03-18 01:58:46 richard Exp $
  19 import unittest, os, shutil, time, weakref
  20
  21 from db_test_base import DBTest, ROTest, SchemaTest, ClassicInitTest, config, password
  22
  23 from roundup import backends
  24
  25 class metakitOpener:
  26     if hasattr(backends, 'metakit'):
  27         from roundup.backends import metakit as module
  28         module._instances = weakref.WeakValueDictionary()
  29
  30     def nuke_database(self):
  31         shutil.rmtree(config.DATABASE)
  32
  33 class metakitDBTest(metakitOpener, DBTest):
  34     def testTransactions(self):
  35         # remember the number of items we started
  36         num_issues = len(self.db.issue.list())
  37         self.db.issue.create(title="don't commit me!", status='1')
  38         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  39         self.db.rollback()
  40         self.assertEqual(num_issues, len(self.db.issue.list()))
  41         self.db.issue.create(title="please commit me!", status='1')
  42         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  43         self.db.commit()
  44         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  45         self.db.rollback()
  46         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  47         self.db.file.create(name="test", type="text/plain", content="hi")
  48         self.db.rollback()
  49         num_files = len(self.db.file.list())
  50         for i in range(10):
  51             self.db.file.create(name="test", type="text/plain",
  52                     content="hi %d"%(i))
  53             self.db.commit()
  54         # TODO: would be good to be able to ensure the file is not on disk after
  55         # a rollback...
  56         num_files2 = len(self.db.file.list())
  57         self.assertNotEqual(num_files, num_files2)
  58         self.db.file.create(name="test", type="text/plain", content="hi")
  59         num_rfiles = len(os.listdir(self.db.config.DATABASE + '/files/file/0'))
  60         self.db.rollback()
  61         num_rfiles2 = len(os.listdir(self.db.config.DATABASE + '/files/file/0'))
  62         self.assertEqual(num_files2, len(self.db.file.list()))
  63         self.assertEqual(num_rfiles2, num_rfiles-1)
  64
  65     def testBooleanUnset(self):
  66         # XXX: metakit can't unset Booleans :(
  67         nid = self.db.user.create(username='foo', assignable=1)
  68         self.db.user.set(nid, assignable=None)
  69         self.assertEqual(self.db.user.get(nid, "assignable"), 0)
  70
  71     def testNumberUnset(self):
  72         # XXX: metakit can't unset Numbers :(
  73         nid = self.db.user.create(username='foo', age=1)
  74         self.db.user.set(nid, age=None)
  75         self.assertEqual(self.db.user.get(nid, "age"), 0)
  76
  77     def testPasswordUnset(self):
  78         # XXX: metakit can't unset Numbers (id's) :(
  79         x = password.Password('x')
  80         nid = self.db.user.create(username='foo', password=x)
  81         self.db.user.set(nid, assignable=None)
  82         self.assertEqual(self.db.user.get(nid, "assignable"), 0)
  83
  84 class metakitROTest(metakitOpener, ROTest):
  85     pass
  86
  87 class metakitSchemaTest(metakitOpener, SchemaTest):
  88     pass
  89
  90 class metakitClassicInitTest(ClassicInitTest):
  91     backend = 'metakit'
  92
  93 from session_common import DBMTest
  94 class metakitSessionTest(metakitOpener, DBMTest):
  95     pass
  96
  97 def test_suite():
  98     suite = unittest.TestSuite()
  99     if not hasattr(backends, 'metakit'):
 100         print 'Skipping metakit tests'
 101         return suite
 102     print 'Including metakit tests'
 103     suite.addTest(unittest.makeSuite(metakitDBTest))
 104     suite.addTest(unittest.makeSuite(metakitROTest))
 105     suite.addTest(unittest.makeSuite(metakitSchemaTest))
 106     suite.addTest(unittest.makeSuite(metakitClassicInitTest))
 107     suite.addTest(unittest.makeSuite(metakitSessionTest))
 108     return suite
 109
 110 if __name__ == '__main__':
 111     runner = unittest.TextTestRunner()
 112     unittest.main(testRunner=runner)
 113