package common import ( "reflect" "testing" ) func Test_generateQuestionMarks(t *testing.T) { type args struct { x int } tests := []struct { name string args args wantQm string }{ {"-1", args{-1}, ""}, {"0", args{0}, ""}, {"1", args{1}, "?"}, {"2", args{2}, "?, ?"}, } for _, tt := range tests { if gotQm := generateQuestionMarks(tt.args.x); gotQm != tt.wantQm { t.Errorf("%q. generateQuestionMarks() = %v, want %v", tt.name, gotQm, tt.wantQm) } } } func TestWhereClause_In(t *testing.T) { type args struct { initial string fields []string } tests := []struct { name string fields *WhereClause args args want *WhereClause }{ { "simple", &WhereClause{}, args{"users.id", []string{"1", "2", "3"}}, &WhereClause{"WHERE users.id IN (?, ?, ?)", []interface{}{"1", "2", "3"}}, }, { "withExisting", Where("users.username = ?", "Howl").Where("users.xd > ?", "6"), args{"users.id", []string{"1"}}, &WhereClause{ "WHERE users.username = ? AND users.xd > ? AND users.id IN (?)", []interface{}{"Howl", "6", "1"}, }, }, } for _, tt := range tests { w := tt.fields if got := w.In(tt.args.initial, tt.args.fields...); !reflect.DeepEqual(got, tt.want) { t.Errorf("%q. WhereClause.In() = %v, want %v", tt.name, got, tt.want) } } } func TestWhere(t *testing.T) { type args struct { clause string passedParam string allowedValues []string } tests := []struct { name string args args want *WhereClause }{ { "simple", args{"users.id = ?", "5", nil}, &WhereClause{"WHERE users.id = ?", []interface{}{"5"}}, }, { "allowed", args{"users.id = ?", "5", []string{"1", "3", "5"}}, &WhereClause{"WHERE users.id = ?", []interface{}{"5"}}, }, { "notAllowed", args{"users.id = ?", "5", []string{"0"}}, &WhereClause{}, }, } for _, tt := range tests { if got := Where(tt.args.clause, tt.args.passedParam, tt.args.allowedValues...); !reflect.DeepEqual(got, tt.want) { t.Errorf("%q. Where() = %#v, want %#v", tt.name, got, tt.want) } } }