ceph-csi/vendor/github.com/kubernetes-csi/drivers/pkg/csi-common/driver_test.go

/*
Copyright 2017 The Kubernetes Authors.

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 csicommon

import (
	"testing"

	"github.com/container-storage-interface/spec/lib/go/csi"
	"github.com/stretchr/testify/assert"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

const (
	fakeDriverName = "fake"
	fakeNodeID     = "fakeNodeID"
)

var (
	fakeVersion           = csi.Version{Major: 5, Minor: 2, Patch: 0}
	fakeVersionsSupported = []*csi.Version{
		{
			Major: 4, Minor: 0, Patch: 0,
		},
		{
			Major: 4, Minor: 1, Patch: 0,
		},
	}
)

func NewFakeDriver() *CSIDriver {
	fakeVersion = csi.Version{Major: 5, Minor: 2, Patch: 0}
	fakeVersionsSupported = []*csi.Version{
		{
			Major: 4, Minor: 0, Patch: 0,
		},
		{
			Major: 4, Minor: 1, Patch: 0,
		},
	}

	driver := NewCSIDriver(fakeDriverName, &fakeVersion, fakeVersionsSupported, fakeNodeID)

	return driver
}

func TestNewFakeDriver(t *testing.T) {
	// Test New fake driver with invalid arguments.
	d := NewCSIDriver("", &fakeVersion, fakeVersionsSupported, fakeNodeID)
	assert.Nil(t, d)
}

func TestCheckVersion(t *testing.T) {

	driver := NewFakeDriver()

	// Exact version
	v := csi.Version{
		Major: 5,
		Minor: 1,
		Patch: 0,
	}
	err := driver.CheckVersion(&v)
	assert.NoError(t, err)

	//Supported version
	v = csi.Version{
		Major: 4,
		Minor: 0,
		Patch: 0,
	}
	err = driver.CheckVersion(&v)
	assert.NoError(t, err)

	// Unsupported version
	v = csi.Version{
		Major: 6,
		Minor: 0,
		Patch: 0,
	}
	err = driver.CheckVersion(&v)
	s, ok := status.FromError(err)
	assert.True(t, ok)
	assert.Equal(t, s.Code(), codes.InvalidArgument)

	// Supported minor version
	v = csi.Version{
		Major: 5,
		Minor: 1,
		Patch: 0,
	}
	err = driver.CheckVersion(&v)
	assert.NoError(t, err)

	// Unsupported minor version
	v = csi.Version{
		Major: 5,
		Minor: 3,
		Patch: 0,
	}
	err = driver.CheckVersion(&v)
	s, ok = status.FromError(err)
	assert.True(t, ok)
	assert.Equal(t, s.Code(), codes.InvalidArgument)
}

func TestGetVolumeCapabilityAccessModes(t *testing.T) {

	d := NewFakeDriver()

	// Test no volume access modes.
	// REVISIT: Do we need to support any default access modes.
	c := d.GetVolumeCapabilityAccessModes()
	assert.Zero(t, len(c))

	// Test driver with access modes.
	d.AddVolumeCapabilityAccessModes([]csi.VolumeCapability_AccessMode_Mode{csi.VolumeCapability_AccessMode_SINGLE_NODE_WRITER})
	modes := d.GetVolumeCapabilityAccessModes()
	assert.Equal(t, 1, len(modes))
	assert.Equal(t, modes[0].GetMode(), csi.VolumeCapability_AccessMode_SINGLE_NODE_WRITER)
}

func TestValidateControllerServiceRequest(t *testing.T) {
	d := NewFakeDriver()

	v := csi.Version{
		Major: 5,
		Minor: 0,
		Patch: 0,
	}

	// Valid requests which require no capabilities
	err := d.ValidateControllerServiceRequest(&v, csi.ControllerServiceCapability_RPC_UNKNOWN)
	assert.NoError(t, err)

	// Test controller service publish/unpublish not supported
	err = d.ValidateControllerServiceRequest(&v, csi.ControllerServiceCapability_RPC_PUBLISH_UNPUBLISH_VOLUME)
	s, ok := status.FromError(err)
	assert.True(t, ok)
	assert.Equal(t, s.Code(), codes.InvalidArgument)

	// Add controller service publish & unpublish request
	d.AddControllerServiceCapabilities(
		[]csi.ControllerServiceCapability_RPC_Type{
			csi.ControllerServiceCapability_RPC_CREATE_DELETE_VOLUME,
			csi.ControllerServiceCapability_RPC_PUBLISH_UNPUBLISH_VOLUME,
			csi.ControllerServiceCapability_RPC_GET_CAPACITY,
			csi.ControllerServiceCapability_RPC_LIST_VOLUMES,
		})

	// Test controller service publish/unpublish is supported
	err = d.ValidateControllerServiceRequest(&v, csi.ControllerServiceCapability_RPC_PUBLISH_UNPUBLISH_VOLUME)
	assert.NoError(t, err)

	// Test controller service create/delete is supported
	err = d.ValidateControllerServiceRequest(&v, csi.ControllerServiceCapability_RPC_CREATE_DELETE_VOLUME)
	assert.NoError(t, err)

	// Test controller service list volumes is supported
	err = d.ValidateControllerServiceRequest(&v, csi.ControllerServiceCapability_RPC_LIST_VOLUMES)
	assert.NoError(t, err)

	// Test controller service get capacity is supported
	err = d.ValidateControllerServiceRequest(&v, csi.ControllerServiceCapability_RPC_GET_CAPACITY)
	assert.NoError(t, err)

}