Function TestChainableAPI

tests/chainable_api_test.go:43–133

(t *testing.T)

}

func TestChainableAPI(t *testing.T) {
	db := newTestDB()

	// Model
	m := &struct{ ID int }{}
	tx := db.Model(m)
	if tx.Statement.Model != m {
		t.Fatalf("Model not set, got %v", tx.Statement.Model)
	}

	// Table
	tx = tx.Table("users")
	if tx.Statement.Table != "users" {
		t.Fatalf("Table not set, got %v", tx.Statement.Table)
	}
	if tx.Statement.TableExpr == nil {
		t.Fatalf("TableExpr expected to be set")
	}

	// Distinct + Select
	tx = tx.Distinct("name", "age")
	if !tx.Statement.Distinct {
		t.Fatalf("Distinct expected true")
	}
	if len(tx.Statement.Selects) != 2 || tx.Statement.Selects[0] != "name" {
		t.Fatalf("Selects expected [name age], got %v", tx.Statement.Selects)
	}

	// Where
	tx = tx.Where("age = ?", 20)
	c, ok := tx.Statement.Clauses["WHERE"]
	if !ok {
		t.Fatalf("WHERE clause expected")
	}
	if where, ok := c.Expression.(clause.Where); !ok || len(where.Exprs) == 0 {
		t.Fatalf("WHERE expressions expected, got %v", c.Expression)
	}

	// Order
	tx = tx.Order("name DESC")
	if _, ok := tx.Statement.Clauses["ORDER BY"]; !ok {
		t.Fatalf("ORDER BY clause expected")
	}

	// Limit / Offset
	tx = tx.Limit(10).Offset(5)
	if cl, ok := tx.Statement.Clauses["LIMIT"]; !ok {
		t.Fatalf("LIMIT clause expected")
	} else {
		if limit, ok := cl.Expression.(clause.Limit); !ok || limit.Limit == nil || *limit.Limit != 10 || limit.Offset != 5 {
			t.Fatalf("LIMIT/Offset values unexpected: %v", cl.Expression)
		}
	}

	// Joins
	tx = tx.Joins("JOIN accounts ON accounts.user_id = users.id")
	if len(tx.Statement.Joins) == 0 {
		t.Fatalf("Joins expected")