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]
)
