internal/github/edit_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 (
	"context"
	"net/http"
	"slices"
	"strings"
	"testing"

	"golang.org/x/oscar/internal/httprr"
	"golang.org/x/oscar/internal/secret"
	"golang.org/x/oscar/internal/storage"
	"golang.org/x/oscar/internal/testutil"
)

var ctx = context.Background()

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

	// Initial load.
	rr, err := httprr.Open("testdata/tmpedit.httprr", http.DefaultTransport)
	check(err)
	rr.ScrubReq(Scrub)
	sdb := secret.DB(secret.Map{"api.github.com": "user:pass"})
	if rr.Recording() {
		sdb = secret.Netrc()
	}
	c := New(lg, db, sdb, rr.Client())
	check(c.Add("rsc/tmp"))
	c.Sync(ctx)

	var ei, ec *Event
	for e := range c.Events("rsc/tmp", 5, 5) {
		if ei == nil && e.API == "/issues" {
			ei = e
		}
		if ec == nil && e.API == "/issues/comments" {
			ec = e
		}
	}
	if ei == nil {
		t.Fatalf("did not find issue #5")
	}
	if ec == nil {
		t.Fatalf("did not find comment on issue #5")
	}

	issue := ei.Typed.(*Issue)
	issue1, err := c.DownloadIssue(ctx, issue.URL)
	check(err)
	if issue1.Title != issue.Title {
		t.Errorf("DownloadIssue: Title=%q, want %q", issue1.Title, issue.Title)
	}

	comment := ec.Typed.(*IssueComment)
	comment1, err := c.DownloadIssueComment(ctx, comment.URL)
	check(err)
	if comment1.Body != comment.Body {
		t.Errorf("DownloadIssueComment: Body=%q, want %q", comment1.Body, comment.Body)
	}

	c.testing = false // edit github directly (except for the httprr in the way)
	check(c.EditIssueComment(ctx, comment, &IssueCommentChanges{Body: testutil.Rot13(comment.Body)}))
	url, err := c.PostIssueComment(ctx, issue, &IssueCommentChanges{Body: "testing. rot13 is the best."})
	check(err)
	const wantPrefix = "https://github.com/rsc/tmp/issues/5#issuecomment-"
	if !strings.HasPrefix(url, "https://github.com/rsc/tmp/issues/5#issuecomment-") {
		t.Errorf("PostIssueComment: url=%q, want prefix %q", url, wantPrefix)
	}
	check(c.EditIssue(ctx, issue, &IssueChanges{Title: testutil.Rot13(issue.Title)}))
}

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

	c := New(lg, db, nil, nil)
	check(c.Testing().LoadTxtar("../testdata/rsctmp.txt"))

	var ei, ec *Event
	for e := range c.Events("rsc/tmp", 5, 5) {
		if ei == nil && e.API == "/issues" {
			ei = e
		}
		if ec == nil && e.API == "/issues/comments" {
			ec = e
		}
	}
	if ei == nil {
		t.Fatalf("did not find issue #5")
	}
	if ec == nil {
		t.Fatalf("did not find comment on issue #5")
	}

	issue := ei.Typed.(*Issue)
	issue1, err := c.DownloadIssue(ctx, issue.URL)
	check(err)
	if issue1.Title != issue.Title {
		t.Errorf("DownloadIssue: Title=%q, want %q", issue1.Title, issue.Title)
	}

	comment := ec.Typed.(*IssueComment)
	comment1, err := c.DownloadIssueComment(ctx, comment.URL)
	check(err)
	if comment1.Body != comment.Body {
		t.Errorf("DownloadIssueComment: Body=%q, want %q", comment1.Body, comment.Body)
	}

	check(c.EditIssueComment(ctx, comment, &IssueCommentChanges{Body: testutil.Rot13(comment.Body)}))
	_, err = c.PostIssueComment(ctx, issue, &IssueCommentChanges{Body: "testing. rot13 is the best."})
	check(err)
	check(c.EditIssue(ctx, issue, &IssueChanges{Title: testutil.Rot13(issue.Title), Labels: &[]string{"ebg13"}}))

	var edits []string
	for _, e := range c.Testing().Edits() {
		edits = append(edits, e.String())
	}

	want := []string{
		`EditIssueComment(rsc/tmp#5.10000000008, {"body":"Comment!\n"})`,
		`PostIssueComment(rsc/tmp#5, {"body":"testing. rot13 is the best."})`,
		`EditIssue(rsc/tmp#5, {"title":"another new issue","labels":["ebg13"]})`,
	}
	if !slices.Equal(edits, want) {
		t.Fatalf("Testing().Edits():\nhave %s\nwant %s", edits, want)
	}
}