mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
// Copyright 2021 The etcd 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 e2e
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"strings"
|
|
)
|
|
|
|
type CURLReq struct {
|
|
Username string
|
|
Password string
|
|
|
|
IsTLS bool
|
|
Timeout int
|
|
|
|
Endpoint string
|
|
|
|
Value string
|
|
Expected string
|
|
Header string
|
|
|
|
MetricsURLScheme string
|
|
|
|
Ciphers string
|
|
}
|
|
|
|
// CURLPrefixArgs builds the beginning of a curl command for a given key
|
|
// addressed to a random URL in the given cluster.
|
|
func CURLPrefixArgs(clus *EtcdProcessCluster, method string, req CURLReq) []string {
|
|
var (
|
|
cmdArgs = []string{"curl"}
|
|
acurl = clus.Procs[rand.Intn(clus.Cfg.ClusterSize)].Config().Acurl
|
|
)
|
|
if req.MetricsURLScheme != "https" {
|
|
if req.IsTLS {
|
|
if clus.Cfg.ClientTLS != ClientTLSAndNonTLS {
|
|
panic("should not use cURLPrefixArgsUseTLS when serving only TLS or non-TLS")
|
|
}
|
|
cmdArgs = append(cmdArgs, "--cacert", CaPath, "--cert", CertPath, "--key", PrivateKeyPath)
|
|
acurl = ToTLS(clus.Procs[rand.Intn(clus.Cfg.ClusterSize)].Config().Acurl)
|
|
} else if clus.Cfg.ClientTLS == ClientTLS {
|
|
if !clus.Cfg.NoCN {
|
|
cmdArgs = append(cmdArgs, "--cacert", CaPath, "--cert", CertPath, "--key", PrivateKeyPath)
|
|
} else {
|
|
cmdArgs = append(cmdArgs, "--cacert", CaPath, "--cert", CertPath3, "--key", PrivateKeyPath3)
|
|
}
|
|
}
|
|
}
|
|
if req.MetricsURLScheme != "" {
|
|
acurl = clus.Procs[rand.Intn(clus.Cfg.ClusterSize)].EndpointsMetrics()[0]
|
|
}
|
|
ep := acurl + req.Endpoint
|
|
|
|
if req.Username != "" || req.Password != "" {
|
|
cmdArgs = append(cmdArgs, "-L", "-u", fmt.Sprintf("%s:%s", req.Username, req.Password), ep)
|
|
} else {
|
|
cmdArgs = append(cmdArgs, "-L", ep)
|
|
}
|
|
if req.Timeout != 0 {
|
|
cmdArgs = append(cmdArgs, "-m", fmt.Sprintf("%d", req.Timeout))
|
|
}
|
|
|
|
if req.Header != "" {
|
|
cmdArgs = append(cmdArgs, "-H", req.Header)
|
|
}
|
|
|
|
if req.Ciphers != "" {
|
|
cmdArgs = append(cmdArgs, "--ciphers", req.Ciphers)
|
|
}
|
|
|
|
switch method {
|
|
case "POST", "PUT":
|
|
dt := req.Value
|
|
if !strings.HasPrefix(dt, "{") { // for non-JSON value
|
|
dt = "value=" + dt
|
|
}
|
|
cmdArgs = append(cmdArgs, "-X", method, "-d", dt)
|
|
}
|
|
return cmdArgs
|
|
}
|
|
|
|
func CURLPost(clus *EtcdProcessCluster, req CURLReq) error {
|
|
return SpawnWithExpect(CURLPrefixArgs(clus, "POST", req), req.Expected)
|
|
}
|
|
|
|
func CURLPut(clus *EtcdProcessCluster, req CURLReq) error {
|
|
return SpawnWithExpect(CURLPrefixArgs(clus, "PUT", req), req.Expected)
|
|
}
|
|
|
|
func CURLGet(clus *EtcdProcessCluster, req CURLReq) error {
|
|
return SpawnWithExpect(CURLPrefixArgs(clus, "GET", req), req.Expected)
|
|
}
|