feat(pierre): sanity check
This commit is contained in:
80
internal/gitadapters/bitbucket/adapter_test.go
Normal file
80
internal/gitadapters/bitbucket/adapter_test.go
Normal file
@@ -0,0 +1,80 @@
|
||||
package bitbucket
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestBitbucketGetFileContentSuccess(t *testing.T) {
|
||||
const expected = "file content"
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
// Verify path structure
|
||||
if r.URL.Path != "/rest/api/1.0/projects/owner/repos/repo/raw/path/to/file.go" {
|
||||
t.Fatalf("unexpected URL path: %s", r.URL.Path)
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(expected))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
// Trim trailing slash handling done in NewBitbucket
|
||||
adapter := NewBitbucket(server.URL, "")
|
||||
content, err := adapter.GetFileContent(context.Background(), "owner", "repo", "path/to/file.go", "")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if content != expected {
|
||||
t.Fatalf("expected %q, got %q", expected, content)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBitbucketGetFileContentError(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
w.Write([]byte("not found"))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
adapter := NewBitbucket(server.URL, "")
|
||||
_, err := adapter.GetFileContent(context.Background(), "owner", "repo", "missing.go", "")
|
||||
if err == nil {
|
||||
t.Fatalf("expected error for non‑200 response")
|
||||
}
|
||||
}
|
||||
|
||||
func TestBitbucketGetPRHeadSHASuccess(t *testing.T) {
|
||||
const sha = "deadbeef"
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path != "/rest/api/1.0/projects/owner/repos/repo/pull-requests/42" {
|
||||
t.Fatalf("unexpected URL: %s", r.URL.Path)
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(`{"toRef":{"latestCommit":"` + sha + `"}}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
adapter := NewBitbucket(server.URL, "")
|
||||
got, err := adapter.GetPRHeadSHA(context.Background(), "owner", "repo", 42)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if got != sha {
|
||||
t.Fatalf("expected sha %s, got %s", sha, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBitbucketGetPRHeadSHAError(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte("error"))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
adapter := NewBitbucket(server.URL, "")
|
||||
_, err := adapter.GetPRHeadSHA(context.Background(), "owner", "repo", 1)
|
||||
if err == nil {
|
||||
t.Fatalf("expected error for non‑200 response")
|
||||
}
|
||||
}
|
||||
@@ -47,10 +47,10 @@ func (b *BitbucketAdapter) GetPR(ctx context.Context, projectKey, repositorySlug
|
||||
)
|
||||
|
||||
response, err := http.DefaultClient.Do(r)
|
||||
defer response.Body.Close() // Add this
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer response.Body.Close() // Add this
|
||||
|
||||
err = json.NewDecoder(response.Body).Decode(&pr)
|
||||
|
||||
@@ -86,10 +86,10 @@ func (b *BitbucketAdapter) AddComment(ctx context.Context, owner, repo string, p
|
||||
}
|
||||
|
||||
response, err := http.DefaultClient.Do(r)
|
||||
defer response.Body.Close() // Add this
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer response.Body.Close() // Add this
|
||||
|
||||
if response.StatusCode >= 300 || response.StatusCode < 200 {
|
||||
sb := &strings.Builder{}
|
||||
|
||||
@@ -1,6 +1,10 @@
|
||||
package bitbucket
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"git.schreifuchs.ch/schreifuchs/pierre-bot/internal/gitadapters/baseadapter"
|
||||
@@ -18,3 +22,43 @@ func NewBitbucket(baseURL string, bearerToken string) *BitbucketAdapter {
|
||||
Rest: baseadapter.NewRest(baseURL, bearerToken),
|
||||
}
|
||||
}
|
||||
|
||||
// GetFileContent returns the raw file content at the given ref (commit SHA) or HEAD if ref is empty.
|
||||
func (b *BitbucketAdapter) GetFileContent(ctx context.Context, projectKey, repositorySlug, path, ref string) (string, error) {
|
||||
// Use the Rest helper to build the base URL, then add the "at" query param if needed.
|
||||
r, err := b.CreateRequest(ctx, http.MethodGet, nil,
|
||||
"/projects/", projectKey, "repos", repositorySlug, "raw", path)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if ref != "" {
|
||||
q := r.URL.Query()
|
||||
q.Set("at", ref)
|
||||
r.URL.RawQuery = q.Encode()
|
||||
}
|
||||
|
||||
resp, err := http.DefaultClient.Do(r)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
sb := &strings.Builder{}
|
||||
io.Copy(sb, resp.Body)
|
||||
return "", fmt.Errorf("error fetching file %s status %d, body %s", path, resp.StatusCode, sb.String())
|
||||
}
|
||||
content, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return string(content), nil
|
||||
}
|
||||
|
||||
// GetPRHeadSHA fetches the PR and returns the SHA of the source (to) branch.
|
||||
func (b *BitbucketAdapter) GetPRHeadSHA(ctx context.Context, projectKey, repositorySlug string, pullRequestID int) (string, error) {
|
||||
pr, err := b.GetPR(ctx, projectKey, repositorySlug, pullRequestID)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return pr.ToRef.LatestCommit, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user