forked from rook/rook
tenzen-y
356f21569f
Fix the following minor bugs. - The wrong description of TEST_BASE_DIR environment variable. - baseTestDir() don't handle error properly Signed-off-by: tenzen-y <toyonomajyutushi@yahoo.co.jp> Signed-off-by: Satoru Takeuchi <satoru.takeuchi@gmail.com>
81 lines
2.3 KiB
Go
81 lines
2.3 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 installer
|
|
|
|
import (
|
|
"os"
|
|
)
|
|
|
|
// testHelmPath gets the helm path
|
|
func testHelmPath() string {
|
|
return getEnvVarWithDefault("TEST_HELM_PATH", "/tmp/rook-tests-scripts-helm/helm")
|
|
}
|
|
|
|
// TestLogCollectionLevel gets whether to collect all logs
|
|
func TestLogCollectionLevel() string {
|
|
return getEnvVarWithDefault("TEST_LOG_COLLECTION_LEVEL", "")
|
|
}
|
|
|
|
// testStorageProvider gets the storage provider for which tests should be run
|
|
func testStorageProvider() string {
|
|
return getEnvVarWithDefault("STORAGE_PROVIDER_TESTS", "")
|
|
}
|
|
|
|
// TestIsOfficialBuild gets the storage provider for which tests should be run
|
|
func TestIsOfficialBuild() bool {
|
|
// PRs will set this to "false", but the official build will not set it, so we compare against "false"
|
|
return getEnvVarWithDefault("TEST_IS_OFFICIAL_BUILD", "") != "false"
|
|
}
|
|
|
|
func StorageClassName() string {
|
|
return getEnvVarWithDefault("TEST_STORAGE_CLASS", "")
|
|
}
|
|
|
|
func UsePVC() bool {
|
|
return StorageClassName() != ""
|
|
}
|
|
|
|
// baseTestDir gets the base test directory
|
|
func baseTestDir() (string, error) {
|
|
// If the base test directory is actively set to WORKING_DIR (as in CI),
|
|
// we use the current working directory.
|
|
val := getEnvVarWithDefault("TEST_BASE_DIR", "/data")
|
|
if val == "WORKING_DIR" {
|
|
var err error
|
|
val, err = os.Getwd()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
return val, nil
|
|
}
|
|
|
|
// TestScratchDevice get the scratch device to be used for OSD
|
|
func TestScratchDevice() string {
|
|
return getEnvVarWithDefault("TEST_SCRATCH_DEVICE", "/dev/nvme0n1")
|
|
}
|
|
|
|
func getEnvVarWithDefault(env, defaultValue string) string {
|
|
val := os.Getenv(env)
|
|
if val == "" {
|
|
logger.Infof("test environment variable (default) %q=%q", env, defaultValue)
|
|
return defaultValue
|
|
}
|
|
logger.Infof("test environment variable %q=%q", env, val)
|
|
return val
|
|
}
|