From 4314b3a9e80221b09f23046c410cc78e57606142 Mon Sep 17 00:00:00 2001 From: Ariel Mashraki Date: Tue, 21 Sep 2021 20:31:03 +0300 Subject: [PATCH] entc/gen: add test cases for unique field selection --- entc/integration/integration_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/entc/integration/integration_test.go b/entc/integration/integration_test.go index 64430b754..f43526c5e 100644 --- a/entc/integration/integration_test.go +++ b/entc/integration/integration_test.go @@ -484,6 +484,16 @@ func Select(t *testing.T, client *ent.Client) { require.Empty(a8m.Name) require.Empty(a8m.Nickname) + client.Pet.CreateBulk( + client.Pet.Create().SetName("a"), + client.Pet.Create().SetName("a"), + ).ExecX(ctx) + names = client.Pet.Query().Select(pet.FieldName).StringsX(ctx) + require.Equal([]string{"a", "a"}, names) + names = client.Pet.Query().Unique(true).Select(pet.FieldName).StringsX(ctx) + require.Equal([]string{"a"}, names) + client.Pet.Delete().ExecX(ctx) + pets := client.Pet.CreateBulk( client.Pet.Create().SetName("a"), client.Pet.Create().SetName("b"),