@@ -19,14 +19,17 @@ package driver
1919import (
2020 "net"
2121
22- "github.com/kubernetes-csi/csi-test/v5/utils "
22+ "github.com/container-storage-interface/spec/lib/go/csi "
2323 "google.golang.org/grpc"
24+
25+ "github.com/kubernetes-csi/csi-test/v5/utils"
2426)
2527
2628type MockCSIDriverServers struct {
27- Controller * MockControllerServer
28- Identity * MockIdentityServer
29- Node * MockNodeServer
29+ Controller * MockControllerServer
30+ Identity * MockIdentityServer
31+ Node * MockNodeServer
32+ SnapshotMetadata * MockSnapshotMetadataServer
3033}
3134
3235type MockCSIDriver struct {
@@ -38,9 +41,22 @@ func NewMockCSIDriver(servers *MockCSIDriverServers) *MockCSIDriver {
3841 return & MockCSIDriver {
3942 CSIDriver : CSIDriver {
4043 servers : & CSIDriverServers {
41- Controller : servers .Controller ,
42- Node : servers .Node ,
43- Identity : servers .Identity ,
44+ Controller : struct {
45+ csi.UnsafeControllerServer
46+ * MockControllerServer
47+ }{MockControllerServer : servers .Controller },
48+ Node : struct {
49+ csi.UnsafeNodeServer
50+ * MockNodeServer
51+ }{MockNodeServer : servers .Node },
52+ Identity : struct {
53+ csi.UnsafeIdentityServer
54+ * MockIdentityServer
55+ }{MockIdentityServer : servers .Identity },
56+ SnapshotMetadata : struct {
57+ csi.UnsafeSnapshotMetadataServer
58+ * MockSnapshotMetadataServer
59+ }{MockSnapshotMetadataServer : servers .SnapshotMetadata },
4460 },
4561 },
4662 }
0 commit comments