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.3 2004-01-27 18:16:50 wc2so1 Exp $
19 import unittest, os, shutil, time, weakref
21 from db_test_base import DBTest, ROTest, SchemaTest, ClassicInitTest, config, password
23 from roundup import backends
25 class metakitOpener:
26 if hasattr(backends, 'metakit'):
27 from roundup.backends import metakit as module
28 module._instances = weakref.WeakValueDictionary()
30 def nuke_database(self):
31 shutil.rmtree(config.DATABASE)
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)
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)
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)
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)
84 class metakitROTest(metakitOpener, ROTest):
85 pass
87 class metakitSchemaTest(metakitOpener, SchemaTest):
88 pass
90 class metakitClassicInitTest(ClassicInitTest):
91 backend = 'metakit'
93 def test_suite():
94 suite = unittest.TestSuite()
95 if not hasattr(backends, 'metakit'):
96 print 'Skipping metakit tests'
97 return suite
98 print 'Including metakit tests'
99 suite.addTest(unittest.makeSuite(metakitDBTest))
100 suite.addTest(unittest.makeSuite(metakitROTest))
101 suite.addTest(unittest.makeSuite(metakitSchemaTest))
102 suite.addTest(unittest.makeSuite(metakitClassicInitTest))
103 return suite
105 if __name__ == '__main__':
106 runner = unittest.TextTestRunner()
107 unittest.main(testRunner=runner)