def test_execute(self): database = Mock() select_with_where = ( 'SELECT * FROM some_table WHERE some_column1 = %s', "some_value") select = ('SELECT * FROM some_table') step = SqlStep(database) step.execute((select_with_where)) step.execute(select) self.assert_sql(database, select, select_with_where)
def test_execute(self): self.database.open() sql = SqlStep(self.database) sql.execute(("INSERT INTO %s VALUES('%s')", TABLE_NAME, 1)) sql.execute(("INSERT INTO %s VALUES('%s')", TABLE_NAME, 2)) sql.execute(("INSERT INTO {0} VALUES ('3')".format(TABLE_NAME))) self.database.commit() self.database.open() self.database.execute("SELECT COUNT(*) FROM %s", (AsIs(TABLE_NAME), )) count = int(self.database.cursor.fetchone()[0]) self.assertEqual(3, count) self.database.execute("SELECT * FROM %s", (AsIs(TABLE_NAME), )) values = list(self.database.cursor.fetchall()) self.assertEqual([('1', ), ('2', ), ('3', )], values)
def test_execute(self): self.database.open() sql = SqlStep(self.database) sql.execute(("INSERT INTO %s VALUES('%s')", TABLE_NAME, 1)) sql.execute(("INSERT INTO %s VALUES('%s')", TABLE_NAME, 2)) sql.execute(("INSERT INTO {0} VALUES ('3')".format(TABLE_NAME))) self.database.commit() self.database.open() self.database.execute( "SELECT COUNT(*) FROM %s", (AsIs(TABLE_NAME),)) count = int(self.database.cursor.fetchone()[0]) self.assertEqual(3, count) self.database.execute( "SELECT * FROM %s", (AsIs(TABLE_NAME),)) values = list(self.database.cursor.fetchall()) self.assertEqual([('1',), ('2',), ('3',)], values)