Mock driver: Return probe ready status

CSI 0.3.0 introduced probe ready status. This change adds the Ready value in ProbeResponse.
kubernetes-csi · Jun 15, 2018 · 07b5a06 · 07b5a06
1 parent 1bf94ed
commit 07b5a06
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/mock/service/identity.go b/mock/service/identity.go
@@ -1,6 +1,7 @@
 package service
 
 import (
+	"github.com/golang/protobuf/ptypes/wrappers"
 	"golang.org/x/net/context"
 
 	"github.com/container-storage-interface/spec/lib/go/csi/v0"
@@ -23,7 +24,9 @@ func (s *service) Probe(
 	req *csi.ProbeRequest) (
 	*csi.ProbeResponse, error) {
 
-	return &csi.ProbeResponse{}, nil
+	return &csi.ProbeResponse{
+		Ready: &wrappers.BoolValue{Value: true},
+	}, nil
 }
 
 func (s *service) GetPluginCapabilities(

diff --git a/pkg/sanity/identity.go b/pkg/sanity/identity.go
@@ -79,6 +79,8 @@ var _ = Describe("Probe [Identity Service]", func() {
 		Expect(ok).To(BeTrue())
 		Expect(serverError.Code() == codes.FailedPrecondition ||
 			serverError.Code() == codes.OK).To(BeTrue())
+
+		Expect(res.GetReady().GetValue()).To(BeTrue())
 	})
 })