コード例 #1
0
ファイル: tests.py プロジェクト: goldenboy/squeak
    def _rename_column(self, table, column):
        cursor = sqlite3.connect(self.db).cursor()
        start = len(cursor.execute("SELECT * from %s" % table).fetchone())
        cursor.close()

        squeak = Squeak(self.db, table)
        squeak.rename_column(column, "%s_renamed" % column)

        was_renamed = False
        for field in squeak.fields:
            if re.match("[\"']%s_renamed" % column, field):
                was_renamed = True
                break

        cursor = sqlite3.connect(self.db).cursor()
        end = len(cursor.execute("SELECT * from %s" % table).fetchone())
        cursor.close()

        return self.assert_(start == end and was_renamed)