internal/github/overview_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 github

import (
	"net/http"
	"testing"

	"github.com/google/go-cmp/cmp"
	"github.com/google/go-cmp/cmp/cmpopts"
	"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 TestIssueOverview(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(Scrub)
	sdb := secret.Empty()
	if rr.Recording() {
		sdb = secret.Netrc()
	}
	c := New(lg, db, sdb, rr.Client())
	check(c.Add("robpike/ivy"))
	check(c.Sync(ctx))

	lc := llmapp.New(lg, llm.EchoTextGenerator(), db)

	got, err := IssueOverview(ctx, lc, db, "robpike/ivy", 19)
	if err != nil {
		t.Fatal(err)
	}

	issue := &Issue{
		URL:     "https://api.github.com/repos/robpike/ivy/issues/19",
		HTMLURL: "https://github.com/robpike/ivy/issues/19",
		User:    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: []User{},
		State:     "closed",
		Labels:    []Label{},
	}

	// 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 := &IssueOverviewResult{
		Issue:         issue,
		Overview:      wantOverview,
		TotalComments: 1,
	}

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

func TestUpdateOverview(t *testing.T) {
	db := storage.MemDB()
	lg := testutil.Slogger(t)
	sdb := secret.Empty()
	gh := New(lg, db, sdb, nil)
	lc := llmapp.New(lg, llm.EchoTextGenerator(), db)
	proj := "hello/world"

	iss := &Issue{Number: 1}
	comment := &IssueComment{}
	comment2 := &IssueComment{}

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

	got, err := UpdateOverview(ctx, lc, db, proj, iss.Number, 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 := &IssueOverviewResult{
		Issue:         iss,
		NewComments:   1,
		TotalComments: 2,
		Overview:      wantOverview,
	}

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