create_table_test.go

package bob_test

import (
	"testing"

	"github.com/aldy505/bob"
)

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

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

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("").
			StringColumn("name").
			ToSql()
		if err.Error() != "create statements must specify a table" {
			t.Fatal("should throw an error, it didn't:", err.Error())
		}
	})

	t.Run("should emit error if no column were specified", func(t *testing.T) {
		_, _, err := bob.
			CreateTable("users").
			ToSql()
		if err.Error() != "a table should at least have one column" {
			t.Fatal("should throw an error, it didn't:", err.Error())
		}
	})
}

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