1fdc9417dfc628fa6f05eb5a32c3b9d415450eb9
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_mysql.py,v 1.8 2004-03-18 01:58:46 richard Exp $
20 import unittest, os, shutil, time, imp
22 from roundup.hyperdb import DatabaseError
23 from roundup import init, backends
25 from db_test_base import DBTest, ROTest, config, SchemaTest, ClassicInitTest
28 # Mysql connection data
29 config.MYSQL_DBHOST = 'localhost'
30 config.MYSQL_DBUSER = 'rounduptest'
31 config.MYSQL_DBPASSWORD = 'rounduptest'
32 config.MYSQL_DBNAME = 'rounduptest'
33 config.MYSQL_DATABASE = (config.MYSQL_DBHOST, config.MYSQL_DBUSER,
34 config.MYSQL_DBPASSWORD, config.MYSQL_DBNAME)
36 class mysqlOpener:
37 if hasattr(backends, 'mysql'):
38 from roundup.backends import mysql as module
40 def setUp(self):
41 self.module.db_nuke(config)
43 def tearDown(self):
44 self.db.close()
45 self.nuke_database()
47 def nuke_database(self):
48 self.module.db_nuke(config)
50 class mysqlDBTest(mysqlOpener, DBTest):
51 def setUp(self):
52 mysqlOpener.setUp(self)
53 DBTest.setUp(self)
55 class mysqlROTest(mysqlOpener, ROTest):
56 def setUp(self):
57 mysqlOpener.setUp(self)
58 ROTest.setUp(self)
60 class mysqlSchemaTest(mysqlOpener, SchemaTest):
61 def setUp(self):
62 mysqlOpener.setUp(self)
63 SchemaTest.setUp(self)
65 class mysqlClassicInitTest(mysqlOpener, ClassicInitTest):
66 backend = 'mysql'
67 extra_config = '''
68 MYSQL_DBHOST = 'localhost'
69 MYSQL_DBUSER = 'rounduptest'
70 MYSQL_DBPASSWORD = 'rounduptest'
71 MYSQL_DBNAME = 'rounduptest'
72 MYSQL_DATABASE = (MYSQL_DBHOST, MYSQL_DBUSER, MYSQL_DBPASSWORD, MYSQL_DBNAME)
73 '''
74 def setUp(self):
75 mysqlOpener.setUp(self)
76 ClassicInitTest.setUp(self)
77 def tearDown(self):
78 ClassicInitTest.tearDown(self)
79 self.nuke_database()
81 from session_common import RDBMSTest
82 class mysqlSessionTest(mysqlOpener, RDBMSTest):
83 def setUp(self):
84 mysqlOpener.setUp(self)
85 RDBMSTest.setUp(self)
86 def tearDown(self):
87 RDBMSTest.tearDown(self)
88 mysqlOpener.tearDown(self)
90 def test_suite():
91 suite = unittest.TestSuite()
92 if not hasattr(backends, 'mysql'):
93 return suite
95 from roundup.backends import mysql
96 try:
97 # Check if we can run mysql tests
98 import MySQLdb
99 db = mysql.Database(config, 'admin')
100 db.close()
101 except (MySQLdb.ProgrammingError, DatabaseError), msg:
102 print "Skipping mysql tests (%s)"%msg
103 else:
104 print 'Including mysql tests'
105 suite.addTest(unittest.makeSuite(mysqlDBTest))
106 suite.addTest(unittest.makeSuite(mysqlROTest))
107 suite.addTest(unittest.makeSuite(mysqlSchemaTest))
108 suite.addTest(unittest.makeSuite(mysqlClassicInitTest))
109 suite.addTest(unittest.makeSuite(mysqlSessionTest))
110 return suite
112 if __name__ == '__main__':
113 runner = unittest.TextTestRunner()
114 unittest.main(testRunner=runner)