mirror of
https://github.com/getcompanion-ai/computer-host.git
synced 2026-04-15 14:03:47 +00:00
55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package firecracker
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
func TestPutSnapshotLoadIncludesNetworkOverrides(t *testing.T) {
|
|
var (
|
|
gotPath string
|
|
gotBody string
|
|
)
|
|
|
|
socketPath, shutdown := startUnixSocketServer(t, func(w http.ResponseWriter, r *http.Request) {
|
|
body, err := io.ReadAll(r.Body)
|
|
if err != nil {
|
|
t.Fatalf("read request body: %v", err)
|
|
}
|
|
gotPath = r.URL.Path
|
|
gotBody = string(body)
|
|
w.WriteHeader(http.StatusNoContent)
|
|
})
|
|
defer shutdown()
|
|
|
|
client := newAPIClient(socketPath)
|
|
err := client.PutSnapshotLoad(context.Background(), SnapshotLoadParams{
|
|
SnapshotPath: "vmstate.bin",
|
|
MemBackend: &MemBackend{
|
|
BackendType: "File",
|
|
BackendPath: "memory.bin",
|
|
},
|
|
ResumeVm: false,
|
|
NetworkOverrides: []NetworkOverride{
|
|
{
|
|
IfaceID: "net0",
|
|
HostDevName: "fctap7",
|
|
},
|
|
},
|
|
VsockOverride: &VsockOverride{UDSPath: "/run/microagent-personalizer.vsock"},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("put snapshot load: %v", err)
|
|
}
|
|
|
|
if gotPath != "/snapshot/load" {
|
|
t.Fatalf("request path mismatch: got %q want %q", gotPath, "/snapshot/load")
|
|
}
|
|
|
|
want := "{\"snapshot_path\":\"vmstate.bin\",\"mem_backend\":{\"backend_type\":\"File\",\"backend_path\":\"memory.bin\"},\"resume_vm\":false,\"network_overrides\":[{\"iface_id\":\"net0\",\"host_dev_name\":\"fctap7\"}],\"vsock_override\":{\"uds_path\":\"/run/microagent-personalizer.vsock\"}}"
|
|
if gotBody != want {
|
|
t.Fatalf("request body mismatch:\n got: %s\nwant: %s", gotBody, want)
|
|
}
|
|
}
|