mirror of
https://github.com/multica-ai/multica.git
synced 2026-06-17 11:48:42 +02:00
41cb91abd9
* feat: add cloud runtime fleet proxy API Co-authored-by: multica-agent <github@multica.ai> * test: cover cloud runtime handler nits Co-authored-by: multica-agent <github@multica.ai> --------- Co-authored-by: Eve <eve@multica-ai.local> Co-authored-by: multica-agent <github@multica.ai>
91 lines
2.7 KiB
Go
91 lines
2.7 KiB
Go
package cloudruntime
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
)
|
|
|
|
func TestClientDoForwardsFleetRequest(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
t.Fatalf("method = %s, want POST", r.Method)
|
|
}
|
|
if r.URL.Path != "/base/api/v1/nodes" {
|
|
t.Fatalf("path = %s, want /base/api/v1/nodes", r.URL.Path)
|
|
}
|
|
if r.URL.RawQuery != "limit=20&offset=0" {
|
|
t.Fatalf("query = %s, want limit=20&offset=0", r.URL.RawQuery)
|
|
}
|
|
if got := r.Header.Get("Accept"); got != "application/json" {
|
|
t.Fatalf("Accept = %q", got)
|
|
}
|
|
if got := r.Header.Get("Content-Type"); got != "application/json" {
|
|
t.Fatalf("Content-Type = %q", got)
|
|
}
|
|
if got := r.Header.Get("X-User-ID"); got != "01972f7e-7e8d-77ef-a13d-1b0ce3e9c001" {
|
|
t.Fatalf("X-User-ID = %q", got)
|
|
}
|
|
if got := r.Header.Get("X-User-PAT"); got != "mul_test_pat" {
|
|
t.Fatalf("X-User-PAT = %q", got)
|
|
}
|
|
if got := r.Header.Get("X-Request-ID"); got != "request-123" {
|
|
t.Fatalf("X-Request-ID = %q", got)
|
|
}
|
|
body, err := io.ReadAll(r.Body)
|
|
if err != nil {
|
|
t.Fatalf("read body: %v", err)
|
|
}
|
|
if got := string(body); got != `{"instance_type":"g5.xlarge"}` {
|
|
t.Fatalf("body = %s", got)
|
|
}
|
|
w.Header().Set("X-Request-ID", "fleet-request-456")
|
|
w.WriteHeader(http.StatusCreated)
|
|
w.Write([]byte(`{"status":"launching"}`))
|
|
}))
|
|
defer srv.Close()
|
|
|
|
client := NewClient(Config{BaseURL: srv.URL + "/base/"})
|
|
resp, err := client.Do(context.Background(), Request{
|
|
Method: http.MethodPost,
|
|
Path: "/api/v1/nodes",
|
|
Query: url.Values{"limit": []string{"20"}, "offset": []string{"0"}},
|
|
Body: []byte(`{"instance_type":"g5.xlarge"}`),
|
|
UserID: "01972f7e-7e8d-77ef-a13d-1b0ce3e9c001",
|
|
UserPAT: "mul_test_pat",
|
|
RequestID: "request-123",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Do: %v", err)
|
|
}
|
|
if resp.StatusCode != http.StatusCreated {
|
|
t.Fatalf("status = %d", resp.StatusCode)
|
|
}
|
|
if got := resp.Header.Get("X-Request-ID"); got != "fleet-request-456" {
|
|
t.Fatalf("response X-Request-ID = %q", got)
|
|
}
|
|
if got := string(resp.Body); got != `{"status":"launching"}` {
|
|
t.Fatalf("response body = %s", got)
|
|
}
|
|
}
|
|
|
|
func TestClientDoDisabled(t *testing.T) {
|
|
client := NewClient(Config{})
|
|
_, err := client.Do(context.Background(), Request{Method: http.MethodGet, Path: "/healthz"})
|
|
if !errors.Is(err, ErrDisabled) {
|
|
t.Fatalf("err = %v, want ErrDisabled", err)
|
|
}
|
|
}
|
|
|
|
func TestClientDoInvalidBaseURL(t *testing.T) {
|
|
client := NewClient(Config{BaseURL: "http://%"})
|
|
_, err := client.Do(context.Background(), Request{Method: http.MethodGet, Path: "/healthz"})
|
|
if !errors.Is(err, ErrInvalidBaseURL) {
|
|
t.Fatalf("err = %v, want ErrInvalidBaseURL", err)
|
|
}
|
|
}
|