def test_adding_foreign_key(self): blueprint = Blueprint('users') blueprint.foreign('order_id').references('id').on('orders') statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(1, len(statements)) expected = [ 'ALTER TABLE `users` ADD CONSTRAINT users_order_id_foreign ' 'FOREIGN KEY (`order_id`) REFERENCES `orders` (`id`)' ] self.assertEqual(expected, statements)
def test_adding_foreign_key(self): blueprint = Blueprint('users') blueprint.create() blueprint.string('foo').primary() blueprint.string('order_id') blueprint.foreign('order_id').references('id').on('orders') statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(1, len(statements)) expected = 'CREATE TABLE "users" ("foo" VARCHAR NOT NULL, "order_id" VARCHAR NOT NULL, ' \ 'FOREIGN KEY("order_id") REFERENCES "orders"("id"), PRIMARY KEY ("foo"))' self.assertEqual(expected, statements[0])
def test_adding_foreign_key(self): blueprint = Blueprint('users') blueprint.create() blueprint.string('foo').primary() blueprint.string('order_id') blueprint.foreign('order_id').references('id').on('orders') statements = blueprint.to_sql(self.get_connection(), self.get_grammar()) self.assertEqual(3, len(statements)) expected = [ 'CREATE TABLE "users" ("foo" VARCHAR(255) NOT NULL, "order_id" VARCHAR(255) NOT NULL)', 'ALTER TABLE "users" ADD CONSTRAINT users_order_id_foreign' ' FOREIGN KEY ("order_id") REFERENCES "orders" ("id")', 'ALTER TABLE "users" ADD PRIMARY KEY ("foo")' ] self.assertEqual(expected, statements)