mirror of
https://github.com/goharbor/harbor.git
synced 2024-09-12 05:58:23 +02:00
142 lines
3.1 KiB
Go
142 lines
3.1 KiB
Go
package envs
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/goharbor/harbor/tests/apitests/api-testing/client"
|
|
)
|
|
|
|
//Environment keeps the testing env info
|
|
type Environment struct {
|
|
Protocol string //env var: HTTP_PROTOCOL
|
|
Hostname string //env var: TESTING_ENV_HOSTNAME
|
|
Account string //env var: TESTING_ENV_ACCOUNT
|
|
Password string //env var: TESTING_ENV_PASSWORD
|
|
Admin string //env var: TESTING_ENV_ADMIN
|
|
AdminPass string //env var: TESTING_ENV_ADMIN_PASS
|
|
TestingProject string //env var: TESTING_PROJECT_NAME
|
|
ImageName string //env var: TESTING_IMAGE_NAME
|
|
ImageTag string //env var: TESTING_IMAGE_TAG
|
|
CAFile string //env var: CA_FILE_PATH
|
|
CertFile string //env var: CERT_FILE_PATH
|
|
KeyFile string //env var: KEY_FILE_PATH
|
|
ProxyURL string //env var: http_proxy, https_proxy, HTTP_PROXY, HTTPS_PROXY
|
|
|
|
//API client
|
|
HTTPClient *client.APIClient
|
|
|
|
//Docker client
|
|
DockerClient *client.DockerClient
|
|
|
|
//Initialize status
|
|
loaded bool
|
|
}
|
|
|
|
//Load test env info
|
|
func (env *Environment) Load() error {
|
|
host := os.Getenv("TESTING_ENV_HOSTNAME")
|
|
if isNotEmpty(host) {
|
|
env.Hostname = host
|
|
}
|
|
|
|
account := os.Getenv("TESTING_ENV_ACCOUNT")
|
|
if isNotEmpty(account) {
|
|
env.Account = account
|
|
}
|
|
|
|
pwd := os.Getenv("TESTING_ENV_PASSWORD")
|
|
if isNotEmpty(pwd) {
|
|
env.Password = pwd
|
|
}
|
|
|
|
admin := os.Getenv("TESTING_ENV_ADMIN")
|
|
if isNotEmpty(admin) {
|
|
env.Admin = admin
|
|
}
|
|
|
|
adminPwd := os.Getenv("TESTING_ENV_ADMIN_PASS")
|
|
if isNotEmpty(adminPwd) {
|
|
env.AdminPass = adminPwd
|
|
}
|
|
|
|
pro := os.Getenv("TESTING_PROJECT_NAME")
|
|
if isNotEmpty(pro) {
|
|
env.TestingProject = pro
|
|
}
|
|
|
|
imgName := os.Getenv("TESTING_IMAGE_NAME")
|
|
if isNotEmpty(imgName) {
|
|
env.ImageName = imgName
|
|
}
|
|
|
|
imgTag := os.Getenv("TESTING_IMAGE_TAG")
|
|
if isNotEmpty(imgTag) {
|
|
env.ImageTag = imgTag
|
|
}
|
|
|
|
protocol := os.Getenv("HTTP_PROTOCOL")
|
|
if isNotEmpty(protocol) {
|
|
env.Protocol = protocol
|
|
}
|
|
|
|
caFile := os.Getenv("CA_FILE_PATH")
|
|
if isNotEmpty(caFile) {
|
|
env.CAFile = caFile
|
|
}
|
|
|
|
keyFile := os.Getenv("KEY_FILE_PATH")
|
|
if isNotEmpty(keyFile) {
|
|
env.KeyFile = keyFile
|
|
}
|
|
|
|
certFile := os.Getenv("CERT_FILE_PATH")
|
|
if isNotEmpty(certFile) {
|
|
env.CertFile = certFile
|
|
}
|
|
|
|
proxyEnvVar := "https_proxy"
|
|
if env.Protocol == "http" {
|
|
proxyEnvVar = "http_proxy"
|
|
}
|
|
proxyURL := os.Getenv(proxyEnvVar)
|
|
if !isNotEmpty(proxyURL) {
|
|
proxyURL = os.Getenv(strings.ToUpper(proxyEnvVar))
|
|
}
|
|
if isNotEmpty(proxyURL) {
|
|
env.ProxyURL = proxyURL
|
|
}
|
|
|
|
if !env.loaded {
|
|
cfg := client.APIClientConfig{
|
|
Username: env.Admin,
|
|
Password: env.AdminPass,
|
|
CaFile: env.CAFile,
|
|
CertFile: env.CertFile,
|
|
KeyFile: env.KeyFile,
|
|
Proxy: env.ProxyURL,
|
|
}
|
|
|
|
httpClient, err := client.NewAPIClient(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
env.HTTPClient = httpClient
|
|
env.DockerClient = &client.DockerClient{}
|
|
|
|
env.loaded = true
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
//RootURI : The root URI like https://<hostname>
|
|
func (env *Environment) RootURI() string {
|
|
return fmt.Sprintf("%s://%s", env.Protocol, env.Hostname)
|
|
}
|
|
|
|
func isNotEmpty(str string) bool {
|
|
return len(strings.TrimSpace(str)) > 0
|
|
}
|