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