github.com/yongjacky/phoenix-go-orm-builder@v0.3.5/builder_update_test.go (about)

     1  // Copyright 2018 The Xorm Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package builder
     6  
     7  import (
     8  	"testing"
     9  
    10  	"github.com/stretchr/testify/assert"
    11  )
    12  
    13  func TestBuilderUpdate(t *testing.T) {
    14  	sql, args, err := Update(Eq{"a": 2}).From("table1").Where(Eq{"a": 1}).ToSQL()
    15  	assert.NoError(t, err)
    16  	assert.EqualValues(t, "UPDATE table1 SET a=? WHERE a=?", sql)
    17  	assert.EqualValues(t, []interface{}{2, 1}, args)
    18  
    19  	sql, args, err = Update(Eq{"a": 2, "b": 1}).From("table1").Where(Eq{"a": 1}).ToSQL()
    20  	assert.NoError(t, err)
    21  	assert.EqualValues(t, "UPDATE table1 SET a=?,b=? WHERE a=?", sql)
    22  	assert.EqualValues(t, []interface{}{2, 1, 1}, args)
    23  
    24  	sql, args, err = Update(Eq{"a": 2}, Eq{"b": 1}).From("table1").Where(Eq{"a": 1}).ToSQL()
    25  	assert.NoError(t, err)
    26  	assert.EqualValues(t, "UPDATE table1 SET a=?,b=? WHERE a=?", sql)
    27  	assert.EqualValues(t, []interface{}{2, 1, 1}, args)
    28  
    29  	sql, args, err = Update(Eq{"a": 2, "b": Incr(1)}).From("table2").Where(Eq{"a": 1}).ToSQL()
    30  	assert.NoError(t, err)
    31  	assert.EqualValues(t, "UPDATE table2 SET a=?,b=b+? WHERE a=?", sql)
    32  	assert.EqualValues(t, []interface{}{2, 1, 1}, args)
    33  
    34  	sql, args, err = Update(Eq{"a": 2, "b": Incr(1), "c": Decr(1), "d": Expr("select count(*) from table2")}).From("table2").Where(Eq{"a": 1}).ToSQL()
    35  	assert.NoError(t, err)
    36  	assert.EqualValues(t, "UPDATE table2 SET a=?,b=b+?,c=c-?,d=(select count(*) from table2) WHERE a=?", sql)
    37  	assert.EqualValues(t, []interface{}{2, 1, 1, 1}, args)
    38  
    39  	sql, args, err = Update(Eq{"a": 2}).Where(Eq{"a": 1}).ToSQL()
    40  	assert.Error(t, err)
    41  	assert.EqualValues(t, ErrNoTableName, err)
    42  
    43  	sql, args, err = Update(Eq{}).From("table1").Where(Eq{"a": 1}).ToSQL()
    44  	assert.Error(t, err)
    45  	assert.EqualValues(t, ErrNoColumnToUpdate, err)
    46  
    47  	var builder = Builder{cond: NewCond()}
    48  	sql, args, err = builder.Update(Eq{"a": 2, "b": 1}).From("table1").Where(Eq{"a": 1}).ToSQL()
    49  	assert.NoError(t, err)
    50  	assert.EqualValues(t, "UPDATE table1 SET a=?,b=? WHERE a=?", sql)
    51  	assert.EqualValues(t, []interface{}{2, 1, 1}, args)
    52  
    53  	sql, args, err = Update(Eq{"a": 1}, Expr("c = c+1")).From("table1").Where(Eq{"b": 2}).ToSQL()
    54  	assert.NoError(t, err)
    55  	assert.EqualValues(t, "UPDATE table1 SET a=?,c = c+1 WHERE b=?", sql)
    56  	assert.EqualValues(t, []interface{}{1, 2}, args)
    57  
    58  	sql, args, err = Update(Eq{"a": 2}).From("table1").ToSQL()
    59  	assert.NoError(t, err)
    60  	assert.EqualValues(t, "UPDATE table1 SET a=?", sql)
    61  	assert.EqualValues(t, []interface{}{2}, args)
    62  }