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.2 2003-11-14 00:11:19 richard Exp $
  19
  20 import unittest, os, shutil, time, weakref
  21
  22 from db_test_base import DBTest, ROTest, SchemaTest, ClassicInitTest, config
  23
  24 from roundup import backends
  25
  26 class metakitOpener:
  27     if hasattr(backends, 'metakit'):
  28         from roundup.backends import metakit as module
  29         module._instances = weakref.WeakValueDictionary()
  30
  31     def nuke_database(self):
  32         shutil.rmtree(config.DATABASE)
  33
  34 class metakitDBTest(metakitOpener, DBTest):
  35     def testTransactions(self):
  36         # remember the number of items we started
  37         num_issues = len(self.db.issue.list())
  38         self.db.issue.create(title="don't commit me!", status='1')
  39         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  40         self.db.rollback()
  41         self.assertEqual(num_issues, len(self.db.issue.list()))
  42         self.db.issue.create(title="please commit me!", status='1')
  43         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  44         self.db.commit()
  45         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  46         self.db.rollback()
  47         self.assertNotEqual(num_issues, len(self.db.issue.list()))
  48         self.db.file.create(name="test", type="text/plain", content="hi")
  49         self.db.rollback()
  50         num_files = len(self.db.file.list())
  51         for i in range(10):
  52             self.db.file.create(name="test", type="text/plain",
  53                     content="hi %d"%(i))
  54             self.db.commit()
  55         # TODO: would be good to be able to ensure the file is not on disk after
  56         # a rollback...
  57         num_files2 = len(self.db.file.list())
  58         self.assertNotEqual(num_files, num_files2)
  59         self.db.file.create(name="test", type="text/plain", content="hi")
  60         num_rfiles = len(os.listdir(self.db.config.DATABASE + '/files/file/0'))
  61         self.db.rollback()
  62         num_rfiles2 = len(os.listdir(self.db.config.DATABASE + '/files/file/0'))
  63         self.assertEqual(num_files2, len(self.db.file.list()))
  64         self.assertEqual(num_rfiles2, num_rfiles-1)
  65
  66     def testBooleanUnset(self):
  67         # XXX: metakit can't unset Booleans :(
  68         nid = self.db.user.create(username='foo', assignable=1)
  69         self.db.user.set(nid, assignable=None)
  70         self.assertEqual(self.db.user.get(nid, "assignable"), 0)
  71
  72     def testNumberUnset(self):
  73         # XXX: metakit can't unset Numbers :(
  74         nid = self.db.user.create(username='foo', age=1)
  75         self.db.user.set(nid, age=None)
  76         self.assertEqual(self.db.user.get(nid, "age"), 0)
  77
  78 class metakitROTest(metakitOpener, ROTest):
  79     pass
  80
  81 class metakitSchemaTest(metakitOpener, SchemaTest):
  82     pass
  83
  84 class metakitClassicInitTest(ClassicInitTest):
  85     backend = 'metakit'
  86
  87 def test_suite():
  88     suite = unittest.TestSuite()
  89     if not hasattr(backends, 'metakit'):
  90         print 'Skipping metakit tests'
  91         return suite
  92     print 'Including metakit tests'
  93     suite.addTest(unittest.makeSuite(metakitDBTest))
  94     suite.addTest(unittest.makeSuite(metakitROTest))
  95     suite.addTest(unittest.makeSuite(metakitSchemaTest))
  96     suite.addTest(unittest.makeSuite(metakitClassicInitTest))
  97     return suite
  98
  99 if __name__ == '__main__':
 100     runner = unittest.TextTestRunner()
 101     unittest.main(testRunner=runner)
 102