mirror of https://github.com/aldy505/bob.git
refactor: clean up test create table
This commit is contained in:
parent
8f6a475880
commit
5af711e253
|
|
@ -6,67 +6,67 @@ import (
|
|||
"github.com/aldy505/bob"
|
||||
)
|
||||
|
||||
func TestCreate(t *testing.T) {
|
||||
t.Run("should return correct sql string with all columns and types", func(t *testing.T) {
|
||||
sql, _, err := bob.
|
||||
CreateTable("users").
|
||||
UUIDColumn("uuid").
|
||||
StringColumn("string").
|
||||
TextColumn("text").
|
||||
DateColumn("date").
|
||||
BooleanColumn("boolean").
|
||||
IntegerColumn("integer").
|
||||
IntColumn("int").
|
||||
TimeStampColumn("timestamp").
|
||||
TimeColumn("time").
|
||||
DateColumn("date").
|
||||
DateTimeColumn("datetime").
|
||||
JSONColumn("json").
|
||||
JSONBColumn("jsonb").
|
||||
BlobColumn("blob").
|
||||
RealColumn("real").
|
||||
FloatColumn("float").
|
||||
AddColumn(bob.ColumnDef{Name: "custom", Type: "custom"}).
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE \"users\" (\"uuid\" UUID, \"string\" VARCHAR(255), \"text\" TEXT, \"date\" DATE, \"boolean\" BOOLEAN, \"integer\" INTEGER, \"int\" INT, \"timestamp\" TIMESTAMP, \"time\" TIME, \"date\" DATE, \"datetime\" DATETIME, \"json\" JSON, \"jsonb\" JSONB, \"blob\" BLOB, \"real\" REAL, \"float\" FLOAT, \"custom\" custom);"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result:", sql)
|
||||
}
|
||||
})
|
||||
func TestCreateTable(t *testing.T) {
|
||||
sql, _, err := bob.
|
||||
CreateTable("users").
|
||||
UUIDColumn("uuid").
|
||||
StringColumn("string").
|
||||
TextColumn("text").
|
||||
DateColumn("date").
|
||||
BooleanColumn("boolean").
|
||||
IntegerColumn("integer").
|
||||
IntColumn("int").
|
||||
TimeStampColumn("timestamp").
|
||||
TimeColumn("time").
|
||||
DateColumn("date").
|
||||
DateTimeColumn("datetime").
|
||||
JSONColumn("json").
|
||||
JSONBColumn("jsonb").
|
||||
BlobColumn("blob").
|
||||
RealColumn("real").
|
||||
FloatColumn("float").
|
||||
AddColumn(bob.ColumnDef{Name: "custom", Type: "custom"}).
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE \"users\" (\"uuid\" UUID, \"string\" VARCHAR(255), \"text\" TEXT, \"date\" DATE, \"boolean\" BOOLEAN, \"integer\" INTEGER, \"int\" INT, \"timestamp\" TIMESTAMP, \"time\" TIME, \"date\" DATE, \"datetime\" DATETIME, \"json\" JSON, \"jsonb\" JSONB, \"blob\" BLOB, \"real\" REAL, \"float\" FLOAT, \"custom\" custom);"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result:", sql)
|
||||
}
|
||||
}
|
||||
|
||||
t.Run("should return correct sql with extras", func(t *testing.T) {
|
||||
sql, _, err := bob.CreateTable("users").
|
||||
UUIDColumn("id", "PRIMARY KEY").
|
||||
StringColumn("email", "NOT NULL", "UNIQUE").
|
||||
ToSql()
|
||||
func TestCreateTable_Extras(t *testing.T) {
|
||||
sql, _, err := bob.CreateTable("users").
|
||||
UUIDColumn("id", "PRIMARY KEY").
|
||||
StringColumn("email", "NOT NULL", "UNIQUE").
|
||||
ToSql()
|
||||
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE \"users\" (\"id\" UUID PRIMARY KEY, \"email\" VARCHAR(255) NOT NULL UNIQUE);"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result:", sql)
|
||||
}
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE \"users\" (\"id\" UUID PRIMARY KEY, \"email\" VARCHAR(255) NOT NULL UNIQUE);"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result:", sql)
|
||||
}
|
||||
}
|
||||
|
||||
t.Run("should be able to have a schema name", func(t *testing.T) {
|
||||
sql, _, err := bob.
|
||||
CreateTable("users").
|
||||
WithSchema("private").
|
||||
StringColumn("name").
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE \"private\".\"users\" (\"name\" VARCHAR(255));"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result:", sql)
|
||||
}
|
||||
})
|
||||
func TestCreateTable_Schema(t *testing.T) {
|
||||
sql, _, err := bob.
|
||||
CreateTable("users").
|
||||
WithSchema("private").
|
||||
StringColumn("name").
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE \"private\".\"users\" (\"name\" VARCHAR(255));"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result:", sql)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCreateTable_Error(t *testing.T) {
|
||||
t.Run("should emit error on empty table name", func(t *testing.T) {
|
||||
_, _, err := bob.
|
||||
CreateTable("").
|
||||
|
|
@ -85,18 +85,18 @@ func TestCreate(t *testing.T) {
|
|||
t.Fatal("should throw an error, it didn't:", err.Error())
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("should emit create if not exists", func(t *testing.T) {
|
||||
sql, _, err := bob.
|
||||
CreateTableIfNotExists("users").
|
||||
TextColumn("name").
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE IF NOT EXISTS \"users\" (\"name\" TEXT);"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result: ", sql)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestCreateTable_IfNotExists(t *testing.T) {
|
||||
sql, _, err := bob.
|
||||
CreateTableIfNotExists("users").
|
||||
TextColumn("name").
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
result := "CREATE TABLE IF NOT EXISTS \"users\" (\"name\" TEXT);"
|
||||
if sql != result {
|
||||
t.Fatal("sql is not equal to result: ", sql)
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue