Divesh Angula
8b2ec14240
- added new tests flag `--load_size` to control amount of data being written -S,M or L(less than 1M,1M and 10M respectively) - added new test flag `--collect_metrics` - if this flag is present, the framework will start prometheus and grafana services and monitor rook cluster - added new test flag `--enable_chaos` - if this flag is present, random pods in rook cluster will be killed peridiocally - update longhaul test pipeline (added new flags) - updated docs
108 lines
3.4 KiB
Go
108 lines
3.4 KiB
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 longhaul
|
|
|
|
import (
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/rook/rook/tests/framework/clients"
|
|
"github.com/rook/rook/tests/framework/contracts"
|
|
"github.com/rook/rook/tests/framework/installer"
|
|
"github.com/rook/rook/tests/framework/utils"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
// Rook Object Store longhaul test
|
|
// Start Rook, create an object store and object user
|
|
// Create object store bucket and perform CURD operations.
|
|
// This Test creates 1 to n object stores - first object store is present throughout the run,
|
|
// but all other object stores are deleted after each run.
|
|
//NOTE: This tests doesn't clean up the cluster or volume after the run, the tests is designed
|
|
//to reuse the same cluster and volume for multiple runs or over a period of time.
|
|
// It is recommended to run this test with -count test param (to repeat th test n number of times)
|
|
// along with --load_parallel_runs params(number of concurrent operations per test) and
|
|
//--load_volumes(number of volumes that are created per test
|
|
func TestObjectLongHaul(t *testing.T) {
|
|
suite.Run(t, new(ObjectLongHaulSuite))
|
|
}
|
|
|
|
type ObjectLongHaulSuite struct {
|
|
suite.Suite
|
|
kh *utils.K8sHelper
|
|
installer *installer.InstallHelper
|
|
tc *clients.TestClient
|
|
namespace string
|
|
op contracts.Setup
|
|
}
|
|
|
|
func (s *ObjectLongHaulSuite) SetupSuite() {
|
|
var err error
|
|
s.namespace = "longhaul-ns"
|
|
s.op, s.kh, s.installer = NewBaseLoadTestOperations(s.T, s.namespace)
|
|
s.tc, err = clients.CreateTestClient(s.kh, s.namespace)
|
|
require.Nil(s.T(), err)
|
|
|
|
}
|
|
|
|
func (s *ObjectLongHaulSuite) TestObjectLonghaulRun() {
|
|
var wg sync.WaitGroup
|
|
storeName := "longhaulstore"
|
|
wg.Add(s.installer.Env.LoadVolumeNumber)
|
|
for i := 1; i <= s.installer.Env.LoadVolumeNumber; i++ {
|
|
if i == 1 {
|
|
go ObjectStoreOperations(s, &wg, s.namespace, storeName+strconv.Itoa(i), false)
|
|
} else {
|
|
go ObjectStoreOperations(s, &wg, s.namespace, storeName+strconv.Itoa(i), randomBool())
|
|
}
|
|
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func ObjectStoreOperations(s *ObjectLongHaulSuite, wg *sync.WaitGroup, namespace string, storeName string, deleteStore bool) {
|
|
defer wg.Done()
|
|
bucketName := "loadbucket"
|
|
s3 := createObjectStoreAndUser(s.T, s.kh, s.tc, s.namespace, storeName, "longhaul", "LongHaulTest")
|
|
isFound, err := s3.IsBucketPresent(bucketName)
|
|
if err == nil {
|
|
if !isFound {
|
|
s3.CreateBucket(bucketName)
|
|
}
|
|
} else {
|
|
logger.Infof("Error which check if %s bucket exists, err -> %v", bucketName, err)
|
|
s.T().Fail()
|
|
}
|
|
|
|
performObjectStoreOperations(s.installer, s3, bucketName)
|
|
if deleteStore {
|
|
delOpts := metav1.DeleteOptions{}
|
|
s.tc.GetObjectClient().ObjectDelete(namespace, storeName, 3, false, s.kh)
|
|
s.kh.Clientset.CoreV1().Services(namespace).Delete("rgw-external-"+storeName, &delOpts)
|
|
}
|
|
s3 = nil
|
|
}
|
|
|
|
func (s *ObjectLongHaulSuite) TearDownSuite() {
|
|
s.tc = nil
|
|
s.kh = nil
|
|
s = nil
|
|
}
|