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