feat: init

pkg/csi/api/client_test.go

@@ -0,0 +1,62 @@
+package api
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/KubelanCloud/kks-csi-plugin/pkg/csi/provisioner"
+)
+
+func TestClientClusterInfo(t *testing.T) {
+	t.Parallel()
+
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path != "/v1/cluster" {
+			t.Fatalf("unexpected path: %s", r.URL.Path)
+		}
+		_ = json.NewEncoder(w).Encode(provisioner.ClusterInfo{
+			StorageID: "abc123",
+		})
+	}))
+	t.Cleanup(srv.Close)
+
+	client := NewClient(ClientConfig{BaseURL: srv.URL})
+	info, err := client.ClusterInfo(context.Background())
+	if err != nil {
+		t.Fatalf("ClusterInfo failed: %v", err)
+	}
+	if info.StorageID != "abc123" {
+		t.Fatalf("unexpected cluster info: %#v", info)
+	}
+}
+
+func TestClientCreateVolume(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/volumes" {
+			t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
+		}
+		w.WriteHeader(http.StatusCreated)
+		_ = json.NewEncoder(w).Encode(provisioner.Volume{
+			VolumeID:  "abc123/k8s-volumes/pvc-1",
+			SizeBytes: 1024,
+		})
+	}))
+	t.Cleanup(srv.Close)
+
+	client := NewClient(ClientConfig{BaseURL: srv.URL})
+	vol, err := client.CreateVolume(context.Background(), provisioner.CreateVolumeRequest{
+		Name:      "pvc-1",
+		SizeBytes: 1024,
+	})
+	if err != nil {
+		t.Fatalf("CreateVolume failed: %v", err)
+	}
+	if vol.VolumeID != "abc123/k8s-volumes/pvc-1" {
+		t.Fatalf("unexpected volume: %#v", vol)
+	}
+}