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

        squeak = Squeak(self.db, table)
        squeak.replace_definition(column, new_definition)

        was_replaced = False
        for field in squeak.fields:
            m = re.match(r'^["\']?(%s)["\']?\s(.*)' % column, field)
            if m and m.group(1) == column and m.group(2) == new_definition:
                was_replaced = 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_replaced)