from geniusql.test import test

try:
    import _mysql
except ImportError:
    import warnings
    warnings.warn("The _mysql module could not be imported. "
                  "The test for MySQL will not be run.")
else:
    opts = {"host": "localhost",
            "db": "geniusql_test",
            "user": "geniusql_test",
            "passwd": "geniusql_test",
            }
    from geniusql.test import zoo_fixture, test

    db = test.DBConfig('mysql', opts)
    setUp = db.setUp
    tearDown = db.tearDown

    #---------------------------------------------------------------------------
    class Phase1_ZooTests(zoo_fixture.Phase1_ZooTests):
        pass
    db.register_db_test_class(Phase1_ZooTests)

    #---------------------------------------------------------------------------
    class Phase2_ConcurrencyTests(zoo_fixture.Phase2_ConcurrencyTests):
        pass
    db.register_db_test_class(Phase2_ConcurrencyTests)

    #---------------------------------------------------------------------------
    class Phase2_IsolationTests(zoo_fixture.Phase2_IsolationTests):
        pass
    db.register_db_test_class(Phase2_IsolationTests)

    #---------------------------------------------------------------------------
    class Phase2_TransactionTests(zoo_fixture.Phase2_TransactionTests):
        pass
    db.register_db_test_class(Phase2_TransactionTests)

    #---------------------------------------------------------------------------
    # TODO: this isn't ported yet.
    #class Phase2_NumericTests(zoo_fixture.Phase2_NumericTests):
            #pass
    #db.register_db_test_class(Phase2_NumericTests)

    #---------------------------------------------------------------------------
    class Phase2_ConnectionTests(zoo_fixture.Phase2_ConnectionTests):
        pass
    db.register_db_test_class(Phase2_ConnectionTests)

    #---------------------------------------------------------------------------
    class Phase2_SQLInjectionTests(zoo_fixture.Phase2_SQLInjectionTests):
        pass
    db.register_db_test_class(Phase2_SQLInjectionTests)