my-rook-config/tests/integration/block_createBlock_k8s_test.go

/*
Copyright 2016 The Rook Authors. All rights reserved.

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 integration

import (
	"strings"
	"testing"
	"time"

	"github.com/rook/rook/tests/framework/clients"
	"github.com/rook/rook/tests/framework/contracts"
	"github.com/rook/rook/tests/framework/enums"
	"github.com/rook/rook/tests/framework/installer"
	"github.com/rook/rook/tests/framework/utils"
	"github.com/stretchr/testify/require"
	"github.com/stretchr/testify/suite"
)

// Test K8s Block Image Creation Scenarios. These tests work when platform is set to Kubernetes
var (
	claimName = "test-claim1"
)

func TestK8sBlockCreate(t *testing.T) {
	suite.Run(t, new(K8sBlockImageCreateSuite))
}

type K8sBlockImageCreateSuite struct {
	suite.Suite
	testClient      *clients.TestClient
	bc              contracts.BlockOperator
	kh              *utils.K8sHelper
	initBlockCount  int
	pvcDef          string
	storageclassDef string
	installer       *installer.InstallHelper
}

func (s *K8sBlockImageCreateSuite) SetupSuite() {

	var err error
	s.kh, err = utils.CreatK8sHelper()
	require.NoError(s.T(), err)

	s.installer = installer.NewK8sRookhelper(s.kh.Clientset)

	err = s.installer.InstallRookOnK8s()
	require.NoError(s.T(), err)

	s.testClient, err = clients.CreateTestClient(enums.Kubernetes, s.kh)
	require.Nil(s.T(), err)

	s.bc = s.testClient.GetBlockClient()
	initialBlocks, err := s.bc.BlockList()
	require.Nil(s.T(), err)
	s.initBlockCount = len(initialBlocks)

	s.pvcDef = `apiVersion: v1
kind: PersistentVolumeClaim
metadata:
  name: {{.claimName}}
  annotations:
    volume.beta.kubernetes.io/storage-class: rook-block
spec:
  accessModes:
    - ReadWriteOnce
  resources:
    requests:
      storage: 1M`

	s.storageclassDef = `apiVersion: rook.io/v1alpha1
kind: Pool
metadata:
  name: {{.poolName}}
  namespace: rook
spec:
  replication:
    size: 1
  # For an erasure-coded pool, comment out the replication count above and uncomment the following settings.
  # Make sure you have enough OSDs to support the replica count or erasure code chunks.
  #erasureCode:
  #  codingChunks: 2
  #  dataChunks: 2
---
apiVersion: storage.k8s.io/v1
kind: StorageClass
metadata:
   name: rook-block
provisioner: rook.io/block
parameters:
    pool: {{.poolName}}`
}

// Test case when persistentvolumeclaim is created for a storage class that doesn't exist
func (s *K8sBlockImageCreateSuite) TestCreatePVCWhenNoStorageClassExists() {

	s.T().Log("Test creating PVC(block images) when storage class is not created")
	//Create PVC
	result, err := s.pvcOperation(claimName, "create")
	require.Contains(s.T(), result, "persistentvolumeclaim \"test-claim1\" created", "Make sure pvc is created")
	require.NoError(s.T(), err)

	//check status of PVC
	pvcStatus, err := s.kh.GetPVCStatus(claimName)
	require.Nil(s.T(), err)
	require.Contains(s.T(), pvcStatus, "Pending", "Makes sure PVC is in Pending state")

	//check block image count
	b, _ := s.bc.BlockList()
	require.Equal(s.T(), s.initBlockCount, len(b), "Make sure new block image is not created")

}

// Test case when persistentvolumeclaim is created for a valid storage class
func (s *K8sBlockImageCreateSuite) TestCreatePVCWhenStorageClassExists() {

	s.T().Log("Test creating PVC(block images) when storage class is created")
	//create pool and storageclass
	result1, err1 := s.storageClassOperation("test-pool-1", "create")
	require.Contains(s.T(), result1, "pool \"test-pool-1\" created", "Make sure test pool is created")
	require.Contains(s.T(), result1, "storageclass \"rook-block\" created", "Make sure storageclass is created")
	require.NoError(s.T(), err1)

	//make sure storageclass is created
	present, err := s.kh.IsStorageClassPresent("rook-block")
	require.Nil(s.T(), err)
	require.True(s.T(), present, "Make sure storageclass is present")

	//create pvc
	result2, err2 := s.pvcOperation(claimName, "create")
	require.Contains(s.T(), result2, "persistentvolumeclaim \"test-claim1\" created", "Make sure pvc is created")
	require.NoError(s.T(), err2)

	//check status of PVC
	require.True(s.T(), s.isPVCBound(claimName))

	//check block image count
	b, _ := s.bc.BlockList()
	require.Equal(s.T(), s.initBlockCount+1, len(b), "Make sure new block image is created")

}

// Test case when persistentvolumeclaim is created for a valid storage class twice
func (s *K8sBlockImageCreateSuite) TestCreateSamePVCTwice() {

	s.T().Log("Test creating PVC(create block images) twice")
	//create pool and storageclass
	result1, err1 := s.storageClassOperation("test-pool-1", "create")
	require.Contains(s.T(), result1, "pool \"test-pool-1\" created", "Make sure test pool is created")
	require.Contains(s.T(), result1, "storageclass \"rook-block\" created", "Make sure storageclass is created")
	require.NoError(s.T(), err1)

	//make sure storageclass is created
	present, err := s.kh.IsStorageClassPresent("rook-block")
	require.Nil(s.T(), err)
	require.True(s.T(), present, "Make sure storageclass is present")

	//create pvc
	result2, err2 := s.pvcOperation(claimName, "create")
	require.Contains(s.T(), result2, "persistentvolumeclaim \"test-claim1\" created", "Make sure pvc is created")
	require.NoError(s.T(), err2)

	//check status of PVC
	require.True(s.T(), s.isPVCBound(claimName))

	b1, _ := s.bc.BlockList()
	require.Equal(s.T(), s.initBlockCount+1, len(b1), "Make sure new block image is created")

	//Create same pvc again
	result3, err3 := s.pvcOperation(claimName, "create")
	require.Contains(s.T(), result3, "persistentvolumeclaims \"test-claim1\" already exists", "make sure PVC is not created again")
	require.NoError(s.T(), err3)

	//check status of PVC
	require.True(s.T(), s.isPVCBound(claimName))

	//check bock image count
	b2, _ := s.bc.BlockList()
	require.Equal(s.T(), len(b1), len(b2), "Make sure new block image is created")

}

func (s *K8sBlockImageCreateSuite) TearDownTest() {

	s.pvcOperation(claimName, "delete")
	s.storageClassOperation("test-pool-1", "delete")

}

func (s *K8sBlockImageCreateSuite) storageClassOperation(poolName string, action string) (string, error) {
	config := map[string]string{
		"poolName": poolName,
	}

	result, err := s.kh.ResourceOperationFromTemplate(action, s.storageclassDef, config)

	return result, err

}

func (s *K8sBlockImageCreateSuite) pvcOperation(claimName string, action string) (string, error) {
	config := map[string]string{
		"claimName": claimName,
	}

	result, err := s.kh.ResourceOperationFromTemplate(action, s.pvcDef, config)

	return result, err

}

func (s *K8sBlockImageCreateSuite) isPVCBound(name string) bool {
	inc := 0
	for inc < utils.RetryLoop {
		status, _ := s.kh.GetPVCStatus(claimName)
		if strings.TrimRight(status, "\n") == "'Bound'" {
			return true
		}
		time.Sleep(time.Second * utils.RetryInterval)
		inc++

	}
	return false
}
func (s *K8sBlockImageCreateSuite) TearDownSuite() {

	s.installer.UninstallRookFromK8s()
}