76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
package githubclient
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestReadLimitedGitHubBodyRejectsOversizedResponse(t *testing.T) {
|
|
_, err := readLimitedGitHubBody(strings.NewReader("abcd"), 3)
|
|
if err == nil {
|
|
t.Fatal("expected oversized response to be rejected")
|
|
}
|
|
if !strings.Contains(err.Error(), "response body exceeds") {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestReadLimitedGitHubBodyAllowsResponseAtLimit(t *testing.T) {
|
|
got, err := readLimitedGitHubBody(strings.NewReader("abc"), 3)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if string(got) != "abc" {
|
|
t.Fatalf("body = %q, want abc", string(got))
|
|
}
|
|
}
|
|
|
|
func TestEndpointURLEscapesRepoPathSegments(t *testing.T) {
|
|
client := New("https://ghe.example.com/api/v3", "")
|
|
|
|
got, err := client.endpointURL("repos", "owner", "repo", "commits", "feature/a b")
|
|
if err != nil {
|
|
t.Fatalf("endpointURL() error = %v", err)
|
|
}
|
|
want := "https://ghe.example.com/api/v3/repos/owner/repo/commits/feature%2Fa%20b"
|
|
if got != want {
|
|
t.Fatalf("endpointURL() = %q, want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestSplitRepoFullNameRejectsExtraSegments(t *testing.T) {
|
|
if _, _, err := splitRepoFullName("owner/repo/../../user"); err == nil {
|
|
t.Fatal("expected malformed repository name to be rejected")
|
|
}
|
|
}
|
|
|
|
func TestCopyLimitedArchiveRejectsOversizedStream(t *testing.T) {
|
|
var dst bytes.Buffer
|
|
n, err := copyLimitedArchive(&dst, strings.NewReader("abcd"), 3)
|
|
if err == nil {
|
|
t.Fatal("expected oversized archive stream to be rejected")
|
|
}
|
|
if n != 4 {
|
|
t.Fatalf("copied = %d, want 4", n)
|
|
}
|
|
if !strings.Contains(err.Error(), "archive exceeds") {
|
|
t.Fatalf("error = %q, want archive exceeds", err.Error())
|
|
}
|
|
}
|
|
|
|
func TestCopyLimitedArchiveAllowsStreamAtLimit(t *testing.T) {
|
|
var dst bytes.Buffer
|
|
n, err := copyLimitedArchive(&dst, strings.NewReader("abc"), 3)
|
|
if err != nil {
|
|
t.Fatalf("copyLimitedArchive() error = %v", err)
|
|
}
|
|
if n != 3 {
|
|
t.Fatalf("copied = %d, want 3", n)
|
|
}
|
|
if got, err := io.ReadAll(&dst); err != nil || string(got) != "abc" {
|
|
t.Fatalf("dst = %q, err = %v; want abc", string(got), err)
|
|
}
|
|
}
|