Function TestHandler

secrets/api_test.go:12–56

(t *testing.T)

)

func TestHandler(t *testing.T) {
	p := &testPlugin{}
	h := NewHandler(p)
	l := sockets.NewInmemSocket("test", 0)
	go h.Serve(l)
	defer l.Close()

	client := &http.Client{Transport: &http.Transport{
		Dial: l.Dial,
	}}

	resp, err := pluginRequest(client, getPath, Request{SecretName: "my-secret"})
	if err != nil {
		t.Fatal(err)
	}
	if resp.Err != "" {
		t.Fatalf("error while getting secret: %v", resp.Err)
	}
	if p.get != 1 {
		t.Fatalf("expected get 1, got %d", p.get)
	}
	if !bytes.EqualFold(secret, resp.Value) {
		t.Fatalf("expecting secret value %s, got %s", secret, resp.Value)
	}
	resp, err = pluginRequest(client, getPath, Request{SecretName: ""})
	if err != nil {
		t.Fatal(err)
	}
	if p.get != 2 {
		t.Fatalf("expected get 2, got %d", p.get)
	}
	if resp.Err == "" {
		t.Fatalf("expected missing secret")
	}
	resp, err = pluginRequest(client, getPath, Request{SecretName: "another-secret", SecretLabels: map[string]string{"prefix": "p-"}})
	if err != nil {
		t.Fatal(err)
	}
	if resp.Err != "" {
		t.Fatalf("error while getting secret: %v", resp.Err)
	}
	if !bytes.EqualFold(append([]byte("p-"), secret...), resp.Value) {
		t.Fatalf("expecting secret value %s, got %s", secret, resp.Value)
	}
}

func pluginRequest(client *http.Client, method string, req Request) (*Response, error) {
	b, err := json.Marshal(req)