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_postgresql.py,v 1.13 2006-08-23 12:57:10 schlatterbeck Exp $
20 import unittest
22 from roundup.hyperdb import DatabaseError
24 from db_test_base import DBTest, ROTest, config, SchemaTest, ClassicInitTest
26 from roundup.backends import get_backend, have_backend
28 class postgresqlOpener:
29 if have_backend('postgresql'):
30 module = get_backend('postgresql')
32 def setUp(self):
33 pass
35 def tearDown(self):
36 self.nuke_database()
38 def nuke_database(self):
39 # clear out the database - easiest way is to nuke and re-create it
40 self.module.db_nuke(config)
42 class postgresqlDBTest(postgresqlOpener, DBTest):
43 def setUp(self):
44 postgresqlOpener.setUp(self)
45 DBTest.setUp(self)
47 def tearDown(self):
48 DBTest.tearDown(self)
49 postgresqlOpener.tearDown(self)
51 class postgresqlROTest(postgresqlOpener, ROTest):
52 def setUp(self):
53 postgresqlOpener.setUp(self)
54 ROTest.setUp(self)
56 def tearDown(self):
57 ROTest.tearDown(self)
58 postgresqlOpener.tearDown(self)
60 class postgresqlSchemaTest(postgresqlOpener, SchemaTest):
61 def setUp(self):
62 postgresqlOpener.setUp(self)
63 SchemaTest.setUp(self)
65 def tearDown(self):
66 SchemaTest.tearDown(self)
67 postgresqlOpener.tearDown(self)
69 class postgresqlClassicInitTest(postgresqlOpener, ClassicInitTest):
70 backend = 'postgresql'
71 def setUp(self):
72 postgresqlOpener.setUp(self)
73 ClassicInitTest.setUp(self)
75 def tearDown(self):
76 ClassicInitTest.tearDown(self)
77 postgresqlOpener.tearDown(self)
79 from session_common import RDBMSTest
80 class postgresqlSessionTest(postgresqlOpener, RDBMSTest):
81 def setUp(self):
82 postgresqlOpener.setUp(self)
83 RDBMSTest.setUp(self)
84 def tearDown(self):
85 RDBMSTest.tearDown(self)
86 postgresqlOpener.tearDown(self)
88 def test_suite():
89 suite = unittest.TestSuite()
90 if not have_backend('postgresql'):
91 print "Skipping postgresql tests"
92 return suite
94 # make sure we start with a clean slate
95 if postgresqlOpener.module.db_exists(config):
96 postgresqlOpener.module.db_nuke(config, 1)
98 # TODO: Check if we can run postgresql tests
99 print 'Including postgresql tests'
100 suite.addTest(unittest.makeSuite(postgresqlDBTest))
101 suite.addTest(unittest.makeSuite(postgresqlROTest))
102 suite.addTest(unittest.makeSuite(postgresqlSchemaTest))
103 suite.addTest(unittest.makeSuite(postgresqlClassicInitTest))
104 suite.addTest(unittest.makeSuite(postgresqlSessionTest))
105 return suite
107 # vim: set et sts=4 sw=4 :