internal/labels/labels_test.go

// Copyright 2024 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package labels

import (
	"context"
	"testing"

	"golang.org/x/oscar/internal/github"
	"golang.org/x/oscar/internal/llm"
)

func TestIssueLabels(t *testing.T) {
	ctx := context.Background()
	llm := kindTestGenerator()

	iss := &github.Issue{
		Title: "title",
		Body:  "body",
	}

	cat, exp, err := IssueCategory(ctx, llm, "golang/go", iss)
	if err != nil {
		t.Fatal(err)
	}
	got := response{cat.Name, exp}
	want := response{CategoryName: "other", Explanation: "whatever"}
	if got != want {
		t.Errorf("got %v, want %v", got, want)
	}
}

func kindTestGenerator() llm.ContentGenerator {
	return llm.TestContentGenerator(
		"kindTestGenerator",
		func(_ context.Context, schema *llm.Schema, promptParts []llm.Part) (string, error) {
			return `{"CategoryName":"other","Explanation":"whatever"}`, nil
		})
}

func TestCleanIssueBody(t *testing.T) {
	for _, tc := range []struct {
		in   string
		want string
	}{
		{"", ""},
		{"# H\nword\nword2\n", "# H\n\nword\nword2\n"},
		{
			"<!-- comment -->\n### H3\n<!-- another --> done",
			"\n\n### H3\n\n done\n",
		},
		{
			"<!--\ncomment\n-->\n### H3\n<!-- another -->\ndone",
			"\n\n### H3\n\n\n\ndone\n",
		},
		{
			"<!-- a --> b -->",
			" b -->\n",
		},
	} {
		got := cleanIssueBody(github.ParseMarkdown(tc.in))
		if got != tc.want {
			t.Errorf("%q:\ngot  %q\nwant %q", tc.in, got, tc.want)
		}
	}
}

func TestHasText(t *testing.T) {
	for _, tc := range []struct {
		in   string
		want bool
	}{
		{"", false},
		{"something", true},
		{"# just a heading", false},
		{"## head\nx", true},
	} {
		got := hasText(github.ParseMarkdown(tc.in))
		if got != tc.want {
			t.Errorf("%q: got %t, want %t", tc.in, got, tc.want)
		}
	}
}