sqlbuilder/update_test.go

// Copyright 2018 Huan Du. All rights reserved.
// Licensed under the MIT license that can be found in the LICENSE file.

package sqlbuilder

import (
	"fmt"
	"testing"
)

func ExampleUpdateBuilder() {
	ub := NewUpdateBuilder()
	ub.Update("demo.user")
	ub.Set(
		ub.Assign("type", "sys"),
		ub.Incr("credit"),
		"modified_at = UNIX_TIMESTAMP(NOW())", // It's allowed to write arbitrary SQL.
	)
	ub.Where(
		ub.GreaterThan("id", 1234),
		ub.Like("name", "%Du"),
		ub.Or(
			ub.IsNull("id_card"),
			ub.In("status", 1, 2, 5),
		),
		"modified_at > created_at + "+ub.Var(86400), // It's allowed to write arbitrary SQL.
	)

	sql, args := ub.Build()
	fmt.Println(sql)
	fmt.Println(args)

	// Output:
	// UPDATE demo.user SET type = ?, credit = credit + 1, modified_at = UNIX_TIMESTAMP(NOW()) WHERE id > ? AND name LIKE ? AND (id_card IS NULL OR status IN (?, ?, ?)) AND modified_at > created_at + ?
	// [sys 1234 %Du 1 2 5 86400]
}

func TestUpdateAssignments(t *testing.T) {
	cases := map[string]func(ub *UpdateBuilder) string{
		"f = f + 1|[]":     func(ub *UpdateBuilder) string { return ub.Incr("f") },
		"f = f - 1|[]":     func(ub *UpdateBuilder) string { return ub.Decr("f") },
		"f = f + $0|[123]": func(ub *UpdateBuilder) string { return ub.Add("f", 123) },
		"f = f - $0|[123]": func(ub *UpdateBuilder) string { return ub.Sub("f", 123) },
		"f = f * $0|[123]": func(ub *UpdateBuilder) string { return ub.Mul("f", 123) },
		"f = f / $0|[123]": func(ub *UpdateBuilder) string { return ub.Div("f", 123) },
	}

	for expected, f := range cases {
		ub := NewUpdateBuilder()
		s := f(ub)
		ub.Set(s)
		_, args := ub.Build()
		actual := fmt.Sprintf("%v|%v", s, args)

		if actual != expected {
			t.Fatalf("invalid assignment result. [expected:%v] [actual:%v]", expected, actual)
		}
	}
}

func TestUpdateBuilder_SetAppend(t *testing.T) {
	ub := NewUpdateBuilder()
	ub.Set(
		ub.Assign("a", 1),
		ub.Assign("b", 2),
	)
	ub.SetAppend(
		ub.Assign("c", 3),
		ub.Incr("d"),
	)
	sql, args := ub.Build()
	actual := fmt.Sprintf("%v|%v", sql, args)

	if expected := "UPDATE  SET a = ?, b = ?, c = ?, d = d + 1|[1 2 3]"; actual != expected {
		t.Fatalf("invalid assignment result. [expected:%v] [actual:%v]", expected, actual)
	}
}