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.4 2003-11-05 21:54:57 jlgijsbers 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, nodbconfig, \
26 ClassicInitTest
28 class mysqlOpener:
29 if hasattr(backends, 'mysql'):
30 from roundup.backends import mysql as module
32 def tearDown(self):
33 self.db.close()
34 self.module.db_nuke(config)
36 class mysqlDBTest(mysqlOpener, DBTest):
37 pass
39 class mysqlROTest(mysqlOpener, ROTest):
40 pass
42 class mysqlSchemaTest(mysqlOpener, SchemaTest):
43 pass
45 class mysqlClassicInitTest(ClassicInitTest):
46 backend = 'mysql'
47 extra_config = '''
48 MYSQL_DBHOST = 'localhost'
49 MYSQL_DBUSER = 'rounduptest'
50 MYSQL_DBPASSWORD = 'rounduptest'
51 MYSQL_DBNAME = 'rounduptest'
52 MYSQL_DATABASE = (MYSQL_DBHOST, MYSQL_DBUSER, MYSQL_DBPASSWORD, MYSQL_DBNAME)
53 '''
54 if hasattr(backends, 'mysql'):
55 from roundup.backends import mysql as module
56 def tearDown(self):
57 ClassicInitTest.tearDown(self)
58 self.module.db_nuke(config)
60 def test_suite():
61 suite = unittest.TestSuite()
62 if not hasattr(backends, 'mysql'):
63 return suite
65 from roundup.backends import mysql
66 try:
67 # Check if we can run mysql tests
68 import MySQLdb
69 db = mysql.Database(nodbconfig, 'admin')
70 db.conn.select_db(config.MYSQL_DBNAME)
71 db.sql("SHOW TABLES");
72 tables = db.sql_fetchall()
73 if 0: #tables:
74 # Database should be empty. We don't dare to delete any data
75 raise DatabaseError, "Database %s contains tables"%\
76 config.MYSQL_DBNAME
77 db.sql("DROP DATABASE %s" % config.MYSQL_DBNAME)
78 db.sql("CREATE DATABASE %s" % config.MYSQL_DBNAME)
79 db.close()
80 except (MySQLdb.ProgrammingError, DatabaseError), msg:
81 print "Skipping mysql tests (%s)"%msg
82 else:
83 print 'Including mysql tests'
84 suite.addTest(unittest.makeSuite(mysqlDBTest))
85 suite.addTest(unittest.makeSuite(mysqlROTest))
86 suite.addTest(unittest.makeSuite(mysqlSchemaTest))
87 suite.addTest(unittest.makeSuite(mysqlClassicInitTest))
88 return suite
90 if __name__ == '__main__':
91 runner = unittest.TextTestRunner()
92 unittest.main(testRunner=runner)