40 lines
926 B
Go
40 lines
926 B
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/KubelanCloud/kks-provider-plugin/pkg/lb/provisioner"
|
|
)
|
|
|
|
func TestClientAllocate(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost || r.URL.Path != "/v1/loadbalancers" {
|
|
t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
|
|
}
|
|
w.WriteHeader(http.StatusCreated)
|
|
_ = json.NewEncoder(w).Encode(provisioner.LoadBalancer{
|
|
ID: "abc",
|
|
IP: "172.173.200.1",
|
|
})
|
|
}))
|
|
t.Cleanup(srv.Close)
|
|
|
|
client := NewClient(ClientConfig{BaseURL: srv.URL})
|
|
lb, err := client.Allocate(context.Background(), provisioner.AllocateRequest{
|
|
Namespace: "default",
|
|
Name: "web",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Allocate failed: %v", err)
|
|
}
|
|
if lb.IP != "172.173.200.1" {
|
|
t.Fatalf("unexpected lb: %#v", lb)
|
|
}
|
|
}
|