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.4 2003-11-05 21:54:57 jlgijsbers 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, nodbconfig, \
  26     ClassicInitTest
  27
  28 class mysqlOpener:
  29     if hasattr(backends, 'mysql'):
  30         from roundup.backends import mysql as module
  31
  32     def tearDown(self):
  33         self.db.close()
  34         self.module.db_nuke(config)
  35
  36 class mysqlDBTest(mysqlOpener, DBTest):
  37     pass
  38
  39 class mysqlROTest(mysqlOpener, ROTest):
  40     pass
  41
  42 class mysqlSchemaTest(mysqlOpener, SchemaTest):
  43     pass
  44
  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)
  59
  60 def test_suite():
  61     suite = unittest.TestSuite()
  62     if not hasattr(backends, 'mysql'):
  63         return suite
  64
  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
  89
  90 if __name__ == '__main__':
  91     runner = unittest.TextTestRunner()
  92     unittest.main(testRunner=runner)
  93