def test_adding_enum(self): blueprint = Blueprint('users') blueprint.enum('foo', ['bar', 'baz']) statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(1, len(statements)) self.assertEqual( 'ALTER TABLE `users` ADD `foo` ENUM(\'bar\', \'baz\') NOT NULL', statements[0] )
def test_adding_enum(self): blueprint = Blueprint('users') blueprint.enum('foo', ['bar', 'baz']) statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(1, len(statements)) self.assertEqual( 'ALTER TABLE "users" ADD COLUMN "foo" VARCHAR NOT NULL', statements[0])