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.6 2003-11-14 00:11:19 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 nodbconfig(config):
37 MYSQL_DATABASE = (config.MYSQL_DBHOST, config.MYSQL_DBUSER, config.MYSQL_DBPASSWORD)
39 class mysqlOpener:
40 if hasattr(backends, 'mysql'):
41 from roundup.backends import mysql as module
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 pass
53 class mysqlROTest(mysqlOpener, ROTest):
54 pass
56 class mysqlSchemaTest(mysqlOpener, SchemaTest):
57 pass
59 class mysqlClassicInitTest(mysqlOpener, ClassicInitTest):
60 backend = 'mysql'
61 extra_config = '''
62 MYSQL_DBHOST = 'localhost'
63 MYSQL_DBUSER = 'rounduptest'
64 MYSQL_DBPASSWORD = 'rounduptest'
65 MYSQL_DBNAME = 'rounduptest'
66 MYSQL_DATABASE = (MYSQL_DBHOST, MYSQL_DBUSER, MYSQL_DBPASSWORD, MYSQL_DBNAME)
67 '''
68 if hasattr(backends, 'mysql'):
69 from roundup.backends import mysql as module
70 def tearDown(self):
71 ClassicInitTest.tearDown(self)
72 self.nuke_database()
74 def test_suite():
75 suite = unittest.TestSuite()
76 if not hasattr(backends, 'mysql'):
77 return suite
79 from roundup.backends import mysql
80 try:
81 # Check if we can run mysql tests
82 import MySQLdb
83 db = mysql.Database(config, 'admin')
84 db.conn.select_db(config.MYSQL_DBNAME)
85 db.sql("SHOW TABLES");
86 tables = db.sql_fetchall()
87 # TODO: reinstate the check here
88 if 0: #tables:
89 # Database should be empty. We don't dare to delete any data
90 raise DatabaseError, "Database %s contains tables"%\
91 config.MYSQL_DBNAME
92 db.sql("DROP DATABASE %s" % config.MYSQL_DBNAME)
93 db.sql("CREATE DATABASE %s" % config.MYSQL_DBNAME)
94 db.close()
95 except (MySQLdb.ProgrammingError, DatabaseError), msg:
96 print "Skipping mysql tests (%s)"%msg
97 else:
98 print 'Including mysql tests'
99 suite.addTest(unittest.makeSuite(mysqlDBTest))
100 suite.addTest(unittest.makeSuite(mysqlROTest))
101 suite.addTest(unittest.makeSuite(mysqlSchemaTest))
102 suite.addTest(unittest.makeSuite(mysqlClassicInitTest))
103 return suite
105 if __name__ == '__main__':
106 runner = unittest.TextTestRunner()
107 unittest.main(testRunner=runner)