internal/overview/issue_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 overview

import (
	"context"
	"net/http"
	"testing"

	"github.com/google/go-cmp/cmp"
	"github.com/google/go-cmp/cmp/cmpopts"
	"golang.org/x/oscar/internal/github"
	"golang.org/x/oscar/internal/httprr"
	"golang.org/x/oscar/internal/llm"
	"golang.org/x/oscar/internal/llmapp"
	"golang.org/x/oscar/internal/secret"
	"golang.org/x/oscar/internal/storage"
	"golang.org/x/oscar/internal/testutil"
)

func TestIssue(t *testing.T) {
	check := testutil.Checker(t)
	lg := testutil.Slogger(t)
	db := storage.MemDB()

	rr, err := httprr.Open("testdata/ivy.httprr", http.DefaultTransport)
	check(err)
	rr.ScrubReq(github.Scrub)
	sdb := secret.Empty()
	if rr.Recording() {
		sdb = secret.Netrc()
	}
	gh := github.New(lg, db, sdb, rr.Client())
	check(gh.Add("robpike/ivy"))
	ctx := context.Background()
	check(gh.Sync(ctx))

	lc := llmapp.New(lg, llm.EchoContentGenerator(), db)
	c := New(lg, db, gh, lc, "test-name", "test-bot")

	issue := &github.Issue{
		URL:     "https://api.github.com/repos/robpike/ivy/issues/19",
		HTMLURL: "https://github.com/robpike/ivy/issues/19",
		User:    github.User{Login: "xunshicheng"},
		Title:   "cannot get",
		Body: `when i get the source code by the command: go get github.com/robpike/ivy
it print: can't load package: package github.com/robpike/ivy: code in directory D:\gocode\src\github.com\robpike\ivy expects import "robpike.io/ivy"

could you get me a hand！
`,
		Number:    19,
		CreatedAt: "2016-01-05T11:57:53Z",
		UpdatedAt: "2016-01-05T22:39:41Z",
		ClosedAt:  "2016-01-05T22:39:41Z",
		Assignees: []github.User{},
		State:     "closed",
		Labels:    []github.Label{},
	}

	got, err := c.ForIssue(ctx, issue)
	if err != nil {
		t.Fatal(err)
	}

	// This merely checks that the correct call to [llmapp.PostOverview] is made.
	// The internals of [llmapp.Client.PostOverview] are tested in the llmapp package.
	wantOverview, err := lc.PostOverview(ctx,
		&llmapp.Doc{
			Type:   "issue",
			URL:    "https://github.com/robpike/ivy/issues/19",
			Author: "xunshicheng",
			Title:  issue.Title,
			Text:   issue.Body,
		},
		[]*llmapp.Doc{
			{
				Type:   "issue comment",
				URL:    "https://github.com/robpike/ivy/issues/19#issuecomment-169157303",
				Author: "robpike",
				Text: `See the import comment, or listen to the error message. Ivy uses a custom import.

go get robpike.io/ivy

It is a fair point though that this should be explained in the README. I will fix that.
`,
			},
		})
	if err != nil {
		t.Fatal(err)
	}

	want := &IssueResult{
		Overview:      wantOverview,
		LastComment:   169157303,
		TotalComments: 1,
	}

	if diff := cmp.Diff(got, want, cmpopts.IgnoreFields(llmapp.Result{}, "Cached")); diff != "" {
		t.Errorf("IssueOverview() mismatch:\n%s", diff)
	}
}

func TestIssueUpdate(t *testing.T) {
	ctx := context.Background()
	db := storage.MemDB()
	lg := testutil.Slogger(t)
	sdb := secret.Empty()
	gh := github.New(lg, db, sdb, nil)
	lc := llmapp.New(lg, llm.EchoContentGenerator(), db)
	c := New(lg, db, gh, lc, "test-name", "test-bot")
	proj := "hello/world"

	iss := &github.Issue{Number: 1}
	comment := &github.IssueComment{}
	comment2 := &github.IssueComment{}
	comment3 := &github.IssueComment{
		User: github.User{Login: "test-bot"},
	}

	gh.Testing().AddIssue(proj, iss)
	gh.Testing().AddIssueComment(proj, iss.Number, comment)
	gh.Testing().AddIssueComment(proj, iss.Number, comment2)
	gh.Testing().AddIssueComment(proj, iss.Number, comment3)

	got, err := c.ForIssueUpdate(ctx, iss, comment.CommentID())
	if err != nil {
		t.Fatal(err)
	}

	// This merely checks that the correct call to [llmapp.UpdatedPostOverview] is made.
	// The internals of [llmapp.Client.UpdatedPostOverview] are tested in the llmapp package.
	wantOverview, err := lc.UpdatedPostOverview(ctx, &llmapp.Doc{
		Type:  "issue",
		URL:   iss.HTMLURL,
		Title: iss.Title,
		Text:  iss.Body,
	}, []*llmapp.Doc{
		{
			Type: "issue comment",
			URL:  comment.HTMLURL,
			Text: comment.Body,
		},
	}, []*llmapp.Doc{
		{
			Type: "issue comment",
			URL:  comment2.HTMLURL,
			Text: comment2.Body,
		},
	})
	if err != nil {
		t.Fatal(err)
	}

	want := &IssueUpdateResult{
		NewComments:     1,
		TotalComments:   3,
		SkippedComments: 1,
		LastComment:     comment3.CommentID(),
		Overview:        wantOverview,
	}

	if diff := cmp.Diff(want, got, cmpopts.IgnoreFields(llmapp.Result{}, "Cached")); diff != "" {
		t.Errorf("UpdateOverview() mismatch (-want,+got):\n%s", diff)
	}
}