def testConnection(self):
db.reset_connection()
try:
# This should throw an exception because no connection has been made yet.
db.get_connection()
self.fail("get_connection didn't throw an exception.")
except:
pass
db.set_database_type("postgres")
self.assertNotEqual(db.connect_default(), None)
self.assertNotEqual(db.get_connection(), None)
try:
# This should throw an exception because the database type cannot be
# changed while a connection is open.
db.set_database_type("postgres")
self.fail("set_database_type didn't throw an exception.")
except:
pass
db.reset_connection()
try:
# This should throw an exception because the connection has already
# been reset.
db.get_connection()
self.fail("get_connection didn't throw an exception.")
except:
pass
def testDatabaseType(self):
db.set_database_type("postgres")
self.assertEqual(db.get_database_type(), "postgres")
db.set_database_type("mysql")
self.assertEqual(db.get_database_type(), "mysql")
#self.failUnlessRaises(db.set_database_type("error"))
try:
db.set_database_type("error")
self.fail("set_database_type didn't throw an exception.")
except:
pass
self.assertEqual(db.get_database_type(), "mysql")
"test_sub_object.sql",
"interest_rate_grid_points.sql",
"interest_rate_curves.sql"]
parser = argparse.ArgumentParser(description='Upgrades a database to the latest version')
parser.add_argument("-r", "--reset", help="deletes all tables and objects from the database, resetting it back to its initial state", action="store_true")
parser.add_argument("-u", "--user", help="the user name", default="postgres")
parser.add_argument("-p", "--password", help="the user password", default="root")
parser.add_argument("-d", "--database", help="the database name", default="mort")
parser.add_argument("-s", "--server", help="the database host name", default="localhost")
parser.add_argument("-t", "--type", help="the database type (mysql or postgres)", default="postgres")
args = parser.parse_args()
if args.type:
db.set_database_type(args.type)
if args.reset:
if (raw_input("Database Reset: Are you sure you want to delete all tables, objects and associated data from the database? [y/n] ") == "y") :
conn = db.connect(user=args.user, password=args.password, server=args.server)
db.set_isolation_level(conn, 0)
cursor = conn.cursor()
cursor.execute("DROP DATABASE IF EXISTS %s;" % (args.database))
cursor.execute("CREATE DATABASE %s" % (args.database))
db.set_isolation_level(conn, 1)
conn.commit()
cursor.close()
conn.close()
conn = db.connect(user=args.user, password=args.password, server=args.server, database=args.database)
#conn = psycopg2.connect(user=args.user, password=args.password, database=args.database)
def setUp(self):
db.reset_connection()
db.set_database_type("postgres")
self.assertNotEqual(db.connect_default(), None)
self.assertNotEqual(db.get_connection(), None)
pass
