feat: add jobs worker and webhook core
This commit is contained in:
1 parent
7847c95538
commit
065dcd955a
12 files changed
+1664
No files matched your search
@@ -0,0 +1,103 @@
|
||||
package webhook
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"git.nianxx.cn/wangxuming/NianAIGC/backend/internal/jobs"
|
||||
)
|
||||
|
||||
func TestBodyAndSignatureMatchSharedFixture(t *testing.T) {
|
||||
fixture := loadWebhookFixture(t)
|
||||
updatedAt, err := time.Parse(time.RFC3339Nano, fixture.Job.UpdatedAt)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
job := jobs.Job{ID: fixture.Job.ID, Status: jobs.Status(fixture.Job.Status), Capability: fixture.Job.Capability, OutputAssetIDs: fixture.Job.OutputAssetIDs, Error: fixture.Job.Error, UpdatedAt: updatedAt}
|
||||
body, err := Body(job)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if string(body) != fixture.Body {
|
||||
t.Fatalf("body = %s, want %s", body, fixture.Body)
|
||||
}
|
||||
if got := Sign(body, fixture.Secret); got != fixture.Signature {
|
||||
t.Fatalf("signature = %q, want %q", got, fixture.Signature)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDeliverRetriesUntilSuccessAndPreservesExactRequest(t *testing.T) {
|
||||
clock := time.Date(2026, 8, 13, 8, 0, 0, 0, time.UTC)
|
||||
sender := &senderStub{responses: []sendResult{{response: Response{Status: 503}}, {err: errors.New("network")}, {response: Response{Status: 204}}}}
|
||||
deliverer := NewDeliverer(sender, " secret ", func() time.Time { return clock })
|
||||
job := jobs.Job{ID: "job", Status: jobs.StatusSucceeded, Capability: "image.generate", OutputAssetIDs: []string{}, UpdatedAt: clock, WebhookURL: "https://hooks.example.test/job"}
|
||||
|
||||
result, err := deliverer.Deliver(context.Background(), job)
|
||||
if err != nil || result.Attempts != 3 || result.LastStatus == nil || !result.LastStatus.OK || result.LastStatus.Status != 204 {
|
||||
t.Fatalf("Deliver = %#v, %v", result, err)
|
||||
}
|
||||
if len(sender.requests) != 3 || sender.requests[0].Headers["Content-Type"] != "application/json" || sender.requests[0].Headers["User-Agent"] != UserAgent || sender.requests[0].Headers["X-Zhinian-Signature"] == "" {
|
||||
t.Fatalf("requests = %#v", sender.requests)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDeliverRespectsPersistedAttemptCeiling(t *testing.T) {
|
||||
sender := &senderStub{}
|
||||
deliverer := NewDeliverer(sender, "", nil)
|
||||
result, err := deliverer.Deliver(context.Background(), jobs.Job{WebhookURL: "https://hooks.example.test", WebhookAttempts: 3})
|
||||
if err != nil || result.Attempts != 3 || len(sender.requests) != 0 {
|
||||
t.Fatalf("Deliver = %#v, %v requests=%d", result, err, len(sender.requests))
|
||||
}
|
||||
}
|
||||
|
||||
type sendResult struct {
|
||||
response Response
|
||||
err error
|
||||
}
|
||||
type senderStub struct {
|
||||
responses []sendResult
|
||||
requests []Request
|
||||
}
|
||||
|
||||
func (stub *senderStub) Send(_ context.Context, request Request) (Response, error) {
|
||||
stub.requests = append(stub.requests, request)
|
||||
result := stub.responses[len(stub.requests)-1]
|
||||
return result.response, result.err
|
||||
}
|
||||
|
||||
type webhookFixture struct {
|
||||
Secret string `json:"secret"`
|
||||
Body string `json:"body"`
|
||||
Signature string `json:"signature"`
|
||||
Job struct {
|
||||
ID string `json:"id"`
|
||||
Status string `json:"status"`
|
||||
Capability string `json:"capability"`
|
||||
OutputAssetIDs []string `json:"outputAssetIds"`
|
||||
Error *jobs.JobError `json:"error"`
|
||||
UpdatedAt string `json:"updatedAt"`
|
||||
} `json:"job"`
|
||||
}
|
||||
|
||||
func loadWebhookFixture(t *testing.T) webhookFixture {
|
||||
t.Helper()
|
||||
_, filename, _, ok := runtime.Caller(0)
|
||||
if !ok {
|
||||
t.Fatal("locate fixture")
|
||||
}
|
||||
raw, err := os.ReadFile(filepath.Join(filepath.Dir(filename), "..", "..", "..", "contracts", "webhook", "job-webhook-v1.json"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
var fixture webhookFixture
|
||||
if err := json.Unmarshal(raw, &fixture); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return fixture
|
||||
}
|
||||
Reference in new issue
Block a user