From 5c529f60e0d279c62487923ca793cb94bf76ce3b Mon Sep 17 00:00:00 2001 From: steebchen Date: Thu, 9 Nov 2023 14:51:07 +0700 Subject: [PATCH] add test for In --- test/features/enums/enums_test.go | 51 +++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) diff --git a/test/features/enums/enums_test.go b/test/features/enums/enums_test.go index 0a4c93862..6d454a6fd 100644 --- a/test/features/enums/enums_test.go +++ b/test/features/enums/enums_test.go @@ -181,6 +181,57 @@ func TestEnums(t *testing.T) { // }, //}, actual) }, + }, { + name: "in", + run: func(t *testing.T, client *PrismaClient, ctx cx) { + _, err := client.User.CreateOne( + User.Role.Set(RoleAdmin), + User.ID.Set("123"), + ).Exec(ctx) + if err != nil { + t.Fatalf("fail %s", err) + } + + _, err = client.User.CreateOne( + User.Role.Set(RoleModerator), + User.ID.Set("456"), + ).Exec(ctx) + if err != nil { + t.Fatalf("fail %s", err) + } + + _, err = client.User.CreateOne( + User.Role.Set(RoleUser), + User.ID.Set("789"), + ).Exec(ctx) + if err != nil { + t.Fatalf("fail %s", err) + } + + actual, err := client.User.FindMany( + User.Role.In([]Role{RoleUser, RoleAdmin}), + ).OrderBy( + User.ID.Order(SortOrderAsc), + ).Exec(ctx) + if err != nil { + t.Fatalf("fail %s", err) + } + + massert.Equal(t, []UserModel{ + { + InnerUser: InnerUser{ + ID: "123", + Role: RoleAdmin, + }, + }, + { + InnerUser: InnerUser{ + ID: "789", + Role: RoleUser, + }, + }, + }, actual) + }, }} for _, tt := range tests { tt := tt