test/test_mysql.py

   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 $
  19
  20 import unittest, os, shutil, time, imp
  21
  22 from roundup.hyperdb import DatabaseError
  23 from roundup import init, backends
  24
  25 from db_test_base import DBTest, ROTest, config, SchemaTest, ClassicInitTest
  26
  27
  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)
  35
  36 class nodbconfig(config):
  37     MYSQL_DATABASE = (config.MYSQL_DBHOST, config.MYSQL_DBUSER, config.MYSQL_DBPASSWORD)
  38
  39 class mysqlOpener:
  40     if hasattr(backends, 'mysql'):
  41         from roundup.backends import mysql as module
  42
  43     def tearDown(self):
  44         self.db.close()
  45         self.nuke_database()
  46
  47     def nuke_database(self):
  48         self.module.db_nuke(config)
  49
  50 class mysqlDBTest(mysqlOpener, DBTest):
  51     pass
  52
  53 class mysqlROTest(mysqlOpener, ROTest):
  54     pass
  55
  56 class mysqlSchemaTest(mysqlOpener, SchemaTest):
  57     pass
  58
  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()
  73
  74 def test_suite():
  75     suite = unittest.TestSuite()
  76     if not hasattr(backends, 'mysql'):
  77         return suite
  78
  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
 104
 105 if __name__ == '__main__':
 106     runner = unittest.TextTestRunner()
 107     unittest.main(testRunner=runner)
 108