def test_create_sql_with_primary_key(self):
table = Table("users")
table.add_column("id", "integer")
table.add_column("name", "string")
table.set_primary_key("id")
sql = 'CREATE TABLE "users" ("id" INTEGER NOT NULL, "name" VARCHAR NOT NULL)'
self.assertEqual([sql], self.platform.compile_create_sql(table))
def test_create_sql_with_primary_key(self):
table = Table("users")
table.add_column("id", "integer")
table.add_column("name", "string")
table.set_primary_key("id")
sql = 'CREATE TABLE "users" (id INTEGER PRIMARY KEY, name VARCHAR)'
self.assertEqual(sql, self.platform.compile_create_sql(table))
def test_create_sql_with_unique_constraint(self):
table = Table("users")
table.add_column("id", "integer")
table.add_column("name", "string")
table.add_constraint("name", "unique", ["name"])
table.set_primary_key("id")
sql = 'CREATE TABLE "users" ("id" INTEGER NOT NULL, "name" VARCHAR NOT NULL, UNIQUE(name))'
self.platform.constraintize(table.get_added_constraints())
self.assertEqual(self.platform.compile_create_sql(table), [sql])
def test_create_sql_with_multiple_unique_constraint(self):
table = Table("users")
table.add_column("id", "integer")
table.add_column("email", "string")
table.add_column("name", "string")
table.add_constraint("name", "unique", ["name", "email"])
table.set_primary_key("id")
sql = 'CREATE TABLE "users" (id INTEGER PRIMARY KEY, email VARCHAR, name VARCHAR, UNIQUE(name, email))'
self.platform.constraintize(table.get_added_constraints())
self.assertEqual(self.platform.compile_create_sql(table), sql)
def test_create_sql_with_foreign_key_constraint(self):
table = Table("users")
table.add_column("id", "integer")
table.add_column("profile_id", "integer")
table.add_column("comment_id", "integer")
table.add_foreign_key("profile_id", "profiles", "id")
table.add_foreign_key("comment_id", "comments", "id")
table.set_primary_key("id")
sql = (
'CREATE TABLE "users" ('
'"id" INTEGER NOT NULL, "profile_id" INTEGER NOT NULL, "comment_id" INTEGER NOT NULL, '
'CONSTRAINT users_profile_id_foreign FOREIGN KEY ("profile_id") REFERENCES "profiles"("id"), '
'CONSTRAINT users_comment_id_foreign FOREIGN KEY ("comment_id") REFERENCES "comments"("id"))'
)
self.platform.constraintize(table.get_added_constraints())
self.assertEqual(self.platform.compile_create_sql(table), [sql])
def test_create_sql_with_foreign_key_constraint(self):
table = Table("users")
table.add_column("id", "integer")
table.add_column("profile_id", "integer")
table.add_column("comment_id", "integer")
table.add_foreign_key("profile_id", "profiles", "id")
table.add_foreign_key("comment_id", "comments", "id")
table.set_primary_key("id")
sql = (
'CREATE TABLE "users" ('
"id INTEGER PRIMARY KEY, profile_id INTEGER, comment_id INTEGER, "
"CONSTRAINT users_profile_id_foreign FOREIGN KEY (profile_id) REFERENCES profiles(id), "
"CONSTRAINT users_comment_id_foreign FOREIGN KEY (comment_id) REFERENCES comments(id))"
)
self.platform.constraintize(table.get_added_constraints())
self.assertEqual(self.platform.compile_create_sql(table), sql)
def test_primary_key(self):
table = Table("users")
table.add_column("id", "integer")
table.set_primary_key("id")
self.assertEqual(table.primary_key, "id")