add csi-test to vendor

vendor/github.com/kubernetes-csi/csi-test/test/co_test.go

@@ -0,0 +1,188 @@
+/*
+Copyright 2017 Luis Pabón luis@portworx.com
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+package test
+
+import (
+	"context"
+	"fmt"
+	"reflect"
+	"testing"
+
+	"github.com/container-storage-interface/spec/lib/go/csi/v0"
+	"github.com/golang/mock/gomock"
+	"github.com/golang/protobuf/proto"
+	mock_driver "github.com/kubernetes-csi/csi-test/driver"
+	mock_utils "github.com/kubernetes-csi/csi-test/utils"
+)
+
+func TestPluginInfoResponse(t *testing.T) {
+
+	// Setup mock
+	m := gomock.NewController(t)
+	defer m.Finish()
+	driver := mock_driver.NewMockIdentityServer(m)
+
+	// Setup input
+	in := &csi.GetPluginInfoRequest{}
+
+	// Setup mock outout
+	out := &csi.GetPluginInfoResponse{
+		Name:          "mock",
+		VendorVersion: "0.1.1",
+		Manifest: map[string]string{
+			"hello": "world",
+		},
+	}
+
+	// Setup expectation
+	driver.EXPECT().GetPluginInfo(nil, in).Return(out, nil).Times(1)
+
+	// Actual call
+	r, err := driver.GetPluginInfo(nil, in)
+	name := r.GetName()
+	if err != nil {
+		t.Errorf("Error: %s", err.Error())
+	}
+	if name != "mock" {
+		t.Errorf("Unknown name: %s\n", name)
+	}
+}
+
+type pbMatcher struct {
+	x proto.Message
+}
+
+func (p pbMatcher) Matches(x interface{}) bool {
+	y := x.(proto.Message)
+	return proto.Equal(p.x, y)
+}
+
+func (p pbMatcher) String() string {
+	return fmt.Sprintf("pb equal to %v", p.x)
+}
+
+func pbMatch(x interface{}) gomock.Matcher {
+	v := x.(proto.Message)
+	return &pbMatcher{v}
+}
+
+func TestGRPCGetPluginInfoReponse(t *testing.T) {
+
+	// Setup mock
+	m := gomock.NewController(&mock_utils.SafeGoroutineTester{})
+	defer m.Finish()
+	driver := mock_driver.NewMockIdentityServer(m)
+
+	// Setup input
+	in := &csi.GetPluginInfoRequest{}
+
+	// Setup mock outout
+	out := &csi.GetPluginInfoResponse{
+		Name:          "mock",
+		VendorVersion: "0.1.1",
+		Manifest: map[string]string{
+			"hello": "world",
+		},
+	}
+
+	// Setup expectation
+	// !IMPORTANT!: Must set context expected value to gomock.Any() to match any value
+	driver.EXPECT().GetPluginInfo(gomock.Any(), pbMatch(in)).Return(out, nil).Times(1)
+
+	// Create a new RPC
+	server := mock_driver.NewMockCSIDriver(&mock_driver.MockCSIDriverServers{
+		Identity: driver,
+	})
+	conn, err := server.Nexus()
+	if err != nil {
+		t.Errorf("Error: %s", err.Error())
+	}
+	defer server.Close()
+
+	// Make call
+	c := csi.NewIdentityClient(conn)
+	r, err := c.GetPluginInfo(context.Background(), in)
+	if err != nil {
+		t.Errorf("Error: %s", err.Error())
+	}
+
+	name := r.GetName()
+	if name != "mock" {
+		t.Errorf("Unknown name: %s\n", name)
+	}
+}
+
+func TestGRPCAttach(t *testing.T) {
+
+	// Setup mock
+	m := gomock.NewController(&mock_utils.SafeGoroutineTester{})
+	defer m.Finish()
+	driver := mock_driver.NewMockControllerServer(m)
+
+	// Setup input
+	defaultVolumeID := "myname"
+	defaultNodeID := "MyNodeID"
+	defaultCaps := &csi.VolumeCapability{
+		AccessType: &csi.VolumeCapability_Mount{
+			Mount: &csi.VolumeCapability_MountVolume{},
+		},
+		AccessMode: &csi.VolumeCapability_AccessMode{
+			Mode: csi.VolumeCapability_AccessMode_MULTI_NODE_MULTI_WRITER,
+		},
+	}
+	publishVolumeInfo := map[string]string{
+		"first":  "foo",
+		"second": "bar",
+		"third":  "baz",
+	}
+	defaultRequest := &csi.ControllerPublishVolumeRequest{
+		VolumeId:         defaultVolumeID,
+		NodeId:           defaultNodeID,
+		VolumeCapability: defaultCaps,
+		Readonly:         false,
+	}
+
+	// Setup mock outout
+	out := &csi.ControllerPublishVolumeResponse{
+		PublishInfo: publishVolumeInfo,
+	}
+
+	// Setup expectation
+	// !IMPORTANT!: Must set context expected value to gomock.Any() to match any value
+	driver.EXPECT().ControllerPublishVolume(gomock.Any(), pbMatch(defaultRequest)).Return(out, nil).Times(1)
+
+	// Create a new RPC
+	server := mock_driver.NewMockCSIDriver(&mock_driver.MockCSIDriverServers{
+		Controller: driver,
+	})
+	conn, err := server.Nexus()
+	if err != nil {
+		t.Errorf("Error: %s", err.Error())
+	}
+	defer server.Close()
+
+	// Make call
+	c := csi.NewControllerClient(conn)
+	r, err := c.ControllerPublishVolume(context.Background(), defaultRequest)
+	if err != nil {
+		t.Errorf("Error: %s", err.Error())
+	}
+
+	info := r.GetPublishInfo()
+	if !reflect.DeepEqual(info, publishVolumeInfo) {
+		t.Errorf("Invalid publish info: %v", info)
+	}
+}