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.9 2004-03-24 06:18:59 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 mysqlOpener:
  37     if hasattr(backends, 'mysql'):
  38         from roundup.backends import mysql as module
  39
  40     def setUp(self):
  41         self.module.db_nuke(config)
  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     def setUp(self):
  52         mysqlOpener.setUp(self)
  53         DBTest.setUp(self)
  54
  55 class mysqlROTest(mysqlOpener, ROTest):
  56     def setUp(self):
  57         mysqlOpener.setUp(self)
  58         ROTest.setUp(self)
  59
  60 class mysqlSchemaTest(mysqlOpener, SchemaTest):
  61     def setUp(self):
  62         mysqlOpener.setUp(self)
  63         SchemaTest.setUp(self)
  64
  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()
  80
  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)
  89
  90 def test_suite():
  91     suite = unittest.TestSuite()
  92     if not hasattr(backends, 'mysql'):
  93         print "Skipping mysql tests"
  94         return suite
  95
  96     from roundup.backends import mysql
  97     try:
  98         # Check if we can run mysql tests
  99         import MySQLdb
 100         db = mysql.Database(config, 'admin')
 101         db.close()
 102     except (MySQLdb.ProgrammingError, DatabaseError), msg:
 103         print "Skipping mysql tests (%s)"%msg
 104     else:
 105         print 'Including mysql tests'
 106         suite.addTest(unittest.makeSuite(mysqlDBTest))
 107         suite.addTest(unittest.makeSuite(mysqlROTest))
 108         suite.addTest(unittest.makeSuite(mysqlSchemaTest))
 109         suite.addTest(unittest.makeSuite(mysqlClassicInitTest))
 110         suite.addTest(unittest.makeSuite(mysqlSessionTest))
 111     return suite
 112
 113 if __name__ == '__main__':
 114     runner = unittest.TextTestRunner()
 115     unittest.main(testRunner=runner)
 116