evilny0/fpdb
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix residual bugs in table droping and creation

Browse Source 
Add Regression test for recreating_tables in mysql
This commit is contained in:
Worros 2008-10-08 02:28:18 +08:00
parent dc81ca854a
commit 5dd9dbc86f
2 changed files with 13 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyfpdb/RegressionTest.py
View File

@ -43,8 +43,14 @@ class TestSequenceFunctions(unittest.TestCase):
def testDatabaseConnection(self):
"""Test all supported DBs"""
result = self.mysql_db.cursor.execute("SHOW TABLES")
self.failUnless(result==13, "Number of tables in database incorrect. Expected 13 got " + str(result))
self.result = self.mysql_db.cursor.execute("SHOW TABLES")
self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result))
def testMySQLRecreateTables(self):
"""Test droping then recreating fpdb table schema"""
self.mysql_db.recreate_tables()
self.result = self.mysql_db.cursor.execute("SHOW TABLES")
self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result))
if __name__ == '__main__':
unittest.main()

21
pyfpdb/fpdb_db.py
View File

@ -61,20 +61,6 @@ class fpdb_db:
self.wrongDbVersion=True
#end def connect
def create_table(self, string):
"""creates a table for the given string
The string should the name of the table followed by the column list
in brackets as if it were an SQL command. Do NOT include the "CREATE TABLES"
bit at the beginning nor the ";" or ENGINE= at the end"""
string="CREATE TABLE "+string
if (self.backend==self.MYSQL_INNODB):
string+=" ENGINE=INNODB"
string+=";"
#print "create_table, string:", string
self.cursor.execute(string)
self.db.commit()
#end def create_table
def disconnect(self, due_to_error=False):
"""Disconnects the DB"""
if due_to_error:
@ -107,17 +93,18 @@ class fpdb_db:
self.cursor.execute(self.sql.query['createHandsActionsTable'])
self.cursor.execute(self.sql.query['createHudCacheTable'])
self.fillDefaultData()
self.db.commit()
#end def disconnect
def drop_tables(self):
"""Drops the fpdb tables from the current db"""
if(self.get_backend_name() == 'MySQL InnoDB'):
# Query the DB to see what tables exist
self.cursor.execute('SHOW TABLES')
#Databases with FOREIGN KEY support need this switched of before you can drop tables
self.drop_referencial_integrity()
# Query the DB to see what tables exist
self.cursor.execute('SHOW TABLES')
for table in self.cursor:
self.cursor.execute(self.sql.query['drop_table'] + table[0])
elif(self.get_backend_name() == 'PostgreSQL'):
@ -126,6 +113,8 @@ class fpdb_db:
elif(self.get_backend_name() == 'SQLite'):
#todo: sqlite version here
print "Empty function here"
self.db.commit()
#end def drop_tables
def drop_referencial_integrity(self):