40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type roundTripFunc func(*http.Request) (*http.Response, error)
|
||
|
|
|
||
|
|
func (f roundTripFunc) RoundTrip(r *http.Request) (*http.Response, error) { return f(r) }
|
||
|
|
|
||
|
|
func TestAWXClientLaunchAndGetJob(t *testing.T) {
|
||
|
|
client := NewAWXClient("https://awx.example", "token")
|
||
|
|
client.client.Transport = roundTripFunc(func(r *http.Request) (*http.Response, error) {
|
||
|
|
if r.Header.Get("Authorization") != "Bearer token" {
|
||
|
|
t.Fatalf("missing auth header")
|
||
|
|
}
|
||
|
|
job := AWXJob{ID: 42, Status: "pending"}
|
||
|
|
if r.URL.Path == "/api/v2/jobs/42/" {
|
||
|
|
job.Status = "successful"
|
||
|
|
} else if r.URL.Path != "/api/v2/job_templates/7/launch/" {
|
||
|
|
t.Fatalf("unexpected path %s", r.URL.Path)
|
||
|
|
}
|
||
|
|
raw, _ := json.Marshal(job)
|
||
|
|
return &http.Response{StatusCode: http.StatusOK, Status: "200 OK", Body: io.NopCloser(bytes.NewReader(raw)), Header: make(http.Header)}, nil
|
||
|
|
})
|
||
|
|
job, err := client.Launch(context.Background(), 7, AWXLaunchRequest{InventoryID: 3, ExtraVars: map[string]any{"task_id": "t"}})
|
||
|
|
if err != nil || job.ID != 42 {
|
||
|
|
t.Fatalf("launch = %#v, err=%v", job, err)
|
||
|
|
}
|
||
|
|
job, err = client.GetJob(context.Background(), "42")
|
||
|
|
if err != nil || job.Status != "successful" {
|
||
|
|
t.Fatalf("get job = %#v, err=%v", job, err)
|
||
|
|
}
|
||
|
|
}
|