From 003db29bde7cc4a30f377e1184b3e65ccdb1a275 Mon Sep 17 00:00:00 2001 From: husharp Date: Wed, 11 Jan 2023 10:33:29 +0800 Subject: [PATCH] modify test Signed-off-by: husharp --- tests/client/resource_group_test.go | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/tests/client/resource_group_test.go b/tests/client/resource_group_test.go index bbbe5b10eb08..ec3e064ce814 100644 --- a/tests/client/resource_group_test.go +++ b/tests/client/resource_group_test.go @@ -30,17 +30,27 @@ func (suite *clientTestSuite) TestWatchResourceGroup() { } }() + for i := 0; i < 3; i++ { + group := &rmpb.ResourceGroup{ + Name: "test" + strconv.Itoa(i), + Mode: rmpb.GroupMode_RUMode, + } + value, err := proto.Marshal(group) + _, err = suite.srv.GetClient().Put(suite.srv.Context(), pd.GroupSettingsPathPrefix+strconv.Itoa(i), string(value)) + suite.NoError(err) + } + // TODO: can change to list resource groups after merge watchChan, err := suite.client.WatchResourceGroup(suite.ctx, 0) suite.NoError(err) - for i := 0; i < 3; i++ { + for i := 3; i < 6; i++ { group := &rmpb.ResourceGroup{ Name: "test" + strconv.Itoa(i), Mode: rmpb.GroupMode_RUMode, } value, err := proto.Marshal(group) - _, err = suite.srv.GetClient().Put(suite.srv.Context(), pd.GroupSettingsPathPrefix, string(value)) + _, err = suite.srv.GetClient().Put(suite.srv.Context(), pd.GroupSettingsPathPrefix+strconv.Itoa(i), string(value)) suite.NoError(err) } @@ -48,6 +58,7 @@ func (suite *clientTestSuite) TestWatchResourceGroup() { for { select { case <-time.After(time.Second): + close(watchChan) return case res := <-watchChan: for _, r := range res {