chore: fix workspace setting tests

6 months ago · 759f7c6171
parent 1da0752922
commit 759f7c6171
1 changed files with 4 additions and 3 deletions
--- a/test/store/workspace_setting_test.go
+++ b/test/store/workspace_setting_test.go
@ -22,9 +22,10 @@ func TestWorkspaceSettingV1Store(t *testing.T) {
 		},
 	})
 	require.NoError(t, err)
-	list, err := ts.ListWorkspaceSettings(ctx, &store.FindWorkspaceSetting{})
+	setting, err := ts.GetWorkspaceSetting(ctx, &store.FindWorkspaceSetting{
+		Name: storepb.WorkspaceSettingKey_GENERAL.String(),
+	})
 	require.NoError(t, err)
-	require.Equal(t, 1, len(list))
-	require.Equal(t, workspaceSetting, list[0])
+	require.Equal(t, workspaceSetting, setting)
 	ts.Close()
 }