def database_reset_confirmation(): db_info = mysql_util.get_db_info() user_input = raw_input('You are testing using database "' + db_info['db'] + \ '".\nThis testing will effect the tables in this database.\n' + \ 'Do you want to continue (n/y)? ') if user_input != 'y' and user_input != 'Y': print 'Testing is cancelled' sys.exit()
def setUp(self): self.db_info = mysql_util.get_db_info() self.db = MySQLdb.connect(host=self.db_info['host'], user=self.db_info['user'], \ passwd=self.db_info['passwd'], unix_socket=self.db_info['socket']) self.cursor = self.db.cursor() if mysql_util.does_db_exist(self.db_info['db']): self.cursor.execute("DROP DATABASE " + self.db_info['db']) self.cursor.execute("CREATE DATABASE " + self.db_info['db'] + " DEFAULT CHARACTER SET 'utf8'") self.db.select_db(self.db_info['db']) self._create_test_tables() self.db.select_db(self.db_info['db'])
def import_papers_and_authors_tbl(): print 'Importing papers and authors table' db_info = mysql_util.get_db_info() ngram_import_cmd = 'mysql -u' + db_info['user'] + ' -p' + db_info['passwd'] + \ ' ' + db_info['db'] + ' < ./settings/test_sql/testseers.sql' os.system(ngram_import_cmd)
def import_ngrams_tbl(): print 'Importing ngrams table' db_info = mysql_util.get_db_info() ngram_import_cmd = 'mysql -u' + db_info['user'] + ' -p' + db_info['passwd'] + \ ' ' + db_info['db'] + ' < ./settings/test_sql/ngrams.sql' os.system(ngram_import_cmd)