예제 #1
0
def testConnection():
    unitTestHelper.printHeader("Testing connection to database");
    unitTestHelper.printInfo("Attempting to connect to connect to database...");
    try:
        con = DB.connectToDB();

        cur = con.cursor();
        cur.execute("SELECT VERSION()");

        ver = cur.fetchone();

        unitTestHelper.printSuccess("Was able to connect to database! Databse Version: %s" % ver);
        con.close();

    except mdb.Error, e:
        unitTestHelper.printFailed("Error %d: %s" % (e.args[0], e.args[1]));
        sys.exit(1);
예제 #2
0
파일: DB.py 프로젝트: badavis/pinkies
def removeTestValues(pidToRemove):
    uth.printInfo("[CLEANUP] Connecting to Database...");
    con = connectToDB();
    cur = con.cursor();
    uth.printSuccess("[CLEANUP] Success!");

    # Cleans up everything related to the given PID. Everything in the DB is linked to that one specific PID.
    try:
        uth.printInfo("[CLEANUP] Removing from SubmittedBy...")
        cur.execute("DELETE FROM SubmittedBy WHERE PID = %s" , (pidToRemove));
        uth.printSuccess("[CLEANUP] Success!");
    except:
        uth.printFailed("[CLEANUP] Failed to remove from SubmittedBy...");
        raise Exception();

    # Commit the changes and close the connection.
    con.commit();
    uth.printSuccess("[CLEANUP] All values removed successfully. Changes have been committed.");
    con.close();
    return;
예제 #3
0
파일: DB.py 프로젝트: badavis/pinkies
def setUpTestValues():
    uth.printInfo("[SETUP] Connecting to Database...");
    con = connectToDB();
    cur = con.cursor();
    uth.printSuccess("[SETUP] Success!");

    # Add to the SubmittedBy table.
    try:
        uth.printInfo("[SETUP] Adding test values to SubmittedBy table...");
        cur.execute("INSERT INTO SubmittedBy (UserID,Date,Time) VALUES (%s,%s,%s)" , ("PinkieTest", "TEST-DATE", "TEST-TIME"));
        uth.printSuccess("[SETUP] Success!");
    except:
        uth.printFailed("[SETUP] Failed to add test values to SubmittedBy table...");
        raise Exception();

    try:
        uth.printInfo("[SETUP] Fetching PID from SubmittedBy table...");
        cur.execute("SELECT * FROM SubmittedBy WHERE UserID = %s AND Date=%s AND Time=%s", ("PinkieTest", "TEST-DATE", "TEST-TIME"));
        pid = int(cur.fetchone()[0]);
        if(pid < 0):
            raise ValueError('Could not get a PID!');

        uth.printSuccess("[SETUP] Success!");
    except:
        uth.printFailed("[SETUP] Failed to fetch PID from SubmittedBy table...");
        raise Exception();

    # Add to the Objects table.
    try:
        uth.printInfo("[SETUP] Adding Test object to the Objects table...");
        cur.execute("INSERT into Objects (PID,Name,Price,Quantity) VALUES (%s,%s,%s,%s)", (pid, "TEST-OBJECT", "1.00", "1"));
        uth.printSuccess("[SETUP] Success!");
    except:
        uth.printFailed("[SETUP] Failed to add Test objects to the Objects table...");
        raise Exception();

    # Add to the Funds table.
    try:
        uth.printInfo("[SETUP] Adding Test fund to the Funds table...");
        cur.execute("INSERT into Funds (PID,FundID,Amount) VALUES (%s,%s,%s)", (pid, "TEST-FUND", "1"));
        uth.printSuccess("[SETUP] Success!");
    except:
        uth.printFailed("[SETUP] Failed to add Test fund to the Funds table...");
        raise Exception();

    # Add to the UploadedFiles table.
    try:
        uth.printInfo("[SETUP] Adding Test file to the UploadedFiles table...");
        cur.execute("INSERT into UploadedFiles (PID,Path,Name) VALUES (%s,%s,%s)", (pid, "TEST-FILE-PATH", "TEST-FILENAME"));
        uth.printSuccess("[SETUP] Success!");
    except:
        uth.printFailed("[SETUP] Failed to add Test file to the UploadedFiles table...");
        raise Exception();

    # Commit our changes to the database.
    con.commit();
    uth.printSuccess("[SETUP] All values added successfully. Changes have been committed");
    # Return the PID for easy removal.
    con.close();
    return pid;