5bfafcd6fc
These packages are still imported by the CLI. Reverting the premature extraction so forge consumers can resolve them. Co-Authored-By: Virgil <virgil@lethean.io>
84 lines
2 KiB
Go
84 lines
2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"forge.lthn.ai/core/go-scm/jobrunner"
|
|
)
|
|
|
|
func TestPublishDraft_Match_Good(t *testing.T) {
|
|
h := NewPublishDraftHandler(nil)
|
|
sig := &jobrunner.PipelineSignal{
|
|
IsDraft: true,
|
|
PRState: "OPEN",
|
|
CheckStatus: "SUCCESS",
|
|
}
|
|
assert.True(t, h.Match(sig))
|
|
}
|
|
|
|
func TestPublishDraft_Match_Bad_NotDraft(t *testing.T) {
|
|
h := NewPublishDraftHandler(nil)
|
|
sig := &jobrunner.PipelineSignal{
|
|
IsDraft: false,
|
|
PRState: "OPEN",
|
|
CheckStatus: "SUCCESS",
|
|
}
|
|
assert.False(t, h.Match(sig))
|
|
}
|
|
|
|
func TestPublishDraft_Match_Bad_ChecksFailing(t *testing.T) {
|
|
h := NewPublishDraftHandler(nil)
|
|
sig := &jobrunner.PipelineSignal{
|
|
IsDraft: true,
|
|
PRState: "OPEN",
|
|
CheckStatus: "FAILURE",
|
|
}
|
|
assert.False(t, h.Match(sig))
|
|
}
|
|
|
|
func TestPublishDraft_Execute_Good(t *testing.T) {
|
|
var capturedMethod string
|
|
var capturedPath string
|
|
var capturedBody string
|
|
|
|
srv := httptest.NewServer(withVersion(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
capturedMethod = r.Method
|
|
capturedPath = r.URL.Path
|
|
b, _ := io.ReadAll(r.Body)
|
|
capturedBody = string(b)
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte(`{}`))
|
|
})))
|
|
defer srv.Close()
|
|
|
|
client := newTestForgeClient(t, srv.URL)
|
|
|
|
h := NewPublishDraftHandler(client)
|
|
sig := &jobrunner.PipelineSignal{
|
|
RepoOwner: "host-uk",
|
|
RepoName: "core-php",
|
|
PRNumber: 42,
|
|
IsDraft: true,
|
|
PRState: "OPEN",
|
|
}
|
|
|
|
result, err := h.Execute(context.Background(), sig)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, http.MethodPatch, capturedMethod)
|
|
assert.Equal(t, "/api/v1/repos/host-uk/core-php/pulls/42", capturedPath)
|
|
assert.Contains(t, capturedBody, `"draft":false`)
|
|
|
|
assert.True(t, result.Success)
|
|
assert.Equal(t, "publish_draft", result.Action)
|
|
assert.Equal(t, "host-uk", result.RepoOwner)
|
|
assert.Equal(t, "core-php", result.RepoName)
|
|
assert.Equal(t, 42, result.PRNumber)
|
|
}
|