Function TestListCategories

pkg/cmd/discussion/client/client_test.go:942–1010

(t *testing.T)

}

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

	tests := []struct {
		name      string
		httpStubs func(*testing.T, *httpmock.Registry)
		wantErr   string
		wantCats  []DiscussionCategory
	}{
		{
			name: "maps all fields",
			httpStubs: func(t *testing.T, reg *httpmock.Registry) {
				reg.Register(
					httpmock.GraphQL(`query DiscussionCategoryList\b`),
					httpmock.StringResponse(`{"data":{"repository":{
						"hasDiscussionsEnabled":true,
						"discussionCategories":{"nodes":[
							{"id":"C1","name":"General","slug":"general","emoji":":speech_balloon:","isAnswerable":false},
							{"id":"C2","name":"Q&A","slug":"q-a","emoji":":question:","isAnswerable":true}
						]}
					}}}`),
				)
			},
			wantCats: []DiscussionCategory{
				{ID: "C1", Name: "General", Slug: "general", Emoji: ":speech_balloon:", IsAnswerable: false},
				{ID: "C2", Name: "Q&A", Slug: "q-a", Emoji: ":question:", IsAnswerable: true},
			},
		},
		{
			name: "discussions disabled",
			httpStubs: func(t *testing.T, reg *httpmock.Registry) {
				reg.Register(
					httpmock.GraphQL(`query DiscussionCategoryList\b`),
					httpmock.StringResponse(`{"data":{"repository":{
						"hasDiscussionsEnabled":false,
						"discussionCategories":{"nodes":[]}
					}}}`),
				)
			},
			wantErr: "discussions disabled",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			reg := &httpmock.Registry{}
			defer reg.Verify(t)

			if tt.httpStubs != nil {
				tt.httpStubs(t, reg)
			}

			c := newTestDiscussionClient(reg)
			categories, err := c.ListCategories(repo)

			if tt.wantErr != "" {
				require.Error(t, err)
				assert.Contains(t, err.Error(), tt.wantErr)