def test_adding_small_integer(self): blueprint = Blueprint('users') blueprint.integer('foo') statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(1, len(statements)) self.assertEqual( 'ALTER TABLE "users" ADD COLUMN "foo" INTEGER NOT NULL', statements[0])
def test_adding_integer(self): blueprint = Blueprint('users') blueprint.integer('foo') statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(1, len(statements)) self.assertEqual( 'ALTER TABLE `users` ADD `foo` INT NOT NULL', statements[0] ) blueprint = Blueprint('users') blueprint.integer('foo', True) statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(1, len(statements)) self.assertEqual( 'ALTER TABLE `users` ADD `foo` INT NOT NULL AUTO_INCREMENT PRIMARY KEY', statements[0] )