1
1
package clickhouse_test
2
2
3
3
import (
4
+ "slices"
4
5
"testing"
5
6
6
7
"gorm.io/gorm/utils/tests"
7
8
)
8
9
9
10
func TestCreate (t * testing.T ) {
10
- var user = User {ID : 1 , Name : "create" , FirstName : "zhang" , LastName : "jinzhu" , Age : 18 , Active : true , Salary : 8.8888 }
11
+ user : = User {ID : 1 , Name : "create" , FirstName : "zhang" , LastName : "jinzhu" , Age : 18 , Active : true , Salary : 8.8888 }
11
12
12
13
if err := DB .Create (& user ).Error ; err != nil {
13
14
t .Fatalf ("failed to create user, got error %v" , err )
@@ -27,10 +28,19 @@ func TestCreate(t *testing.T) {
27
28
if err := DB .Raw ("select * from users where id = ?" , user .ID ).Scan (& partialResult ).Error ; err != nil {
28
29
t .Fatalf ("failed to query partial, got error %v" , err )
29
30
}
31
+
32
+ var names []string
33
+ if err := DB .Select ("name" ).Model (& User {}).Find (& names ).Error ; err != nil {
34
+ t .Fatalf ("failed to query user, got error %v" , err )
35
+ }
36
+
37
+ if ! slices .Contains (names , user .Name ) {
38
+ t .Errorf ("name should be included in the result" )
39
+ }
30
40
}
31
41
32
42
func TestBatchCreate (t * testing.T ) {
33
- var users = []User {
43
+ users : = []User {
34
44
{ID : 11 , Name : "batch_create_1" , FirstName : "zhang" , LastName : "jinzhu" , Age : 18 , Active : true , Salary : 6 },
35
45
{ID : 12 , Name : "batch_create_2" , FirstName : "zhang" , LastName : "jinzhu" , Age : 18 , Active : false , Salary : 6.12 },
36
46
{ID : 13 , Name : "batch_create_3" , FirstName : "zhang" , LastName : "jinzhu" , Age : 18 , Active : true , Salary : 6.1234 },
@@ -55,7 +65,7 @@ func TestBatchCreate(t *testing.T) {
55
65
}
56
66
57
67
func TestCreateWithMap (t * testing.T ) {
58
- var user = User {ID : 122 , Name : "create2" , FirstName : "zhang" , LastName : "jinzhu" , Age : 18 , Active : true , Salary : 6.6666 }
68
+ user : = User {ID : 122 , Name : "create2" , FirstName : "zhang" , LastName : "jinzhu" , Age : 18 , Active : true , Salary : 6.6666 }
59
69
60
70
if err := DB .Table ("users" ).Create (& map [string ]interface {}{
61
71
"id" : user .ID , "name" : user .Name , "first_name" : user .FirstName , "last_name" : user .LastName , "age" : user .Age , "active" : user .Active , "salary" : user .Salary ,
0 commit comments