Function TestEditDiscussionLabels

pkg/cmd/discussion/client/client_test.go:2856–3124

(t *testing.T)

}

func TestEditDiscussionLabels(t *testing.T) {
	repo := ghrepo.New("OWNER", "REPO")

	baseNode := func() discussionListNode {
		return discussionListNode{
			ID:     "D_1",
			Number: 5,
			Title:  "T",
			Body:   "B",
			URL:    "https://github.com/OWNER/REPO/discussions/5",
			Author: actorNode{
				TypeName: "User",
				Login:    "alice",
				User:     struct{ ID, Name string }{ID: "U1", Name: "Alice"},
				Bot:      struct{ ID string }{ID: "U1"},
			},
			Category: struct {
				ID           string
				Name         string
				Slug         string
				Emoji        string
				IsAnswerable bool
			}{ID: "CAT_1", Name: "General", Slug: "general"},
			ReactionGroups: []struct {
				Content string
				Users   struct{ TotalCount int }
			}{},
			CreatedAt: time.Date(2025, 6, 1, 0, 0, 0, 0, time.UTC),
			UpdatedAt: time.Date(2025, 6, 1, 0, 0, 0, 0, time.UTC),
		}
	}

	tests := []struct {
		name      string
		addIDs    []string
		removeIDs []string
		setupMock func(reg *httpmock.Registry)
		wantErr   string
		wantNode  func() discussionListNode
	}{
		{
			name:      "adds and removes labels",
			addIDs:    []string{"L_bug", "L_enh"},
			removeIDs: []string{"L_old"},
			setupMock: func(reg *httpmock.Registry) {
				reg.Register(
					httpmock.GraphQLMutationMatcher(`mutation RemoveLabelsFromDiscussion\b`, func(input map[string]interface{}) bool {
						assert.Equal(t, "D_1", input["labelableId"])
						assert.Equal(t, []interface{}{"L_old"}, input["labelIds"])
						return true
					}),
					// This response is superseded by the subsequent add mutation so we don't need all fields.
					httpmock.StringResponse(`{"data":{"removeLabelsFromLabelable":{"labelable":{"id": "D_1"}}}}`),
				)
				reg.Register(
					httpmock.GraphQLMutationMatcher(`mutation AddLabelsToDiscussion\b`, func(input map[string]interface{}) bool {
						assert.Equal(t, "D_1", input["labelableId"])
						assert.Equal(t, []interface{}{"L_bug", "L_enh"}, input["labelIds"])