tests/framework/e2e/flags.go - main - githubmirror/etcd

tests/framework/e2e/flags.go
99 строк3 KB
Siyuan Zhang
e2e: add flag to pass specific binary path for last release.
21 май 2024, 19:57
21 май 2024, 19:57cde6cd0
// 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 (
	"flag"
	"os"
	"runtime"

	"go.etcd.io/etcd/tests/v3/framework/testutils"
)

var (
	CertDir string

	CertPath       string
	PrivateKeyPath string
	CaPath         string

	CertPath2       string
	PrivateKeyPath2 string

	CertPath3       string
	PrivateKeyPath3 string

	CrlPath               string
	RevokedCertPath       string
	RevokedPrivateKeyPath string

	BinPath     binPath
	FixturesDir = testutils.MustAbsPath("../fixtures")
)

type binPath struct {
	Etcd            string
	EtcdLastRelease string
	Etcdctl         string
	Etcdutl         string
	LazyFS          string
}

func (bp *binPath) LazyFSAvailable() bool {
	_, err := os.Stat(bp.LazyFS)
	if err != nil {
		if !os.IsNotExist(err) {
			panic(err)
		}
		return false
	}
	return true
}

func InitFlags() {
	os.Setenv("ETCD_UNSUPPORTED_ARCH", runtime.GOARCH)

	binDirDef := testutils.MustAbsPath("../../bin")
	certDirDef := FixturesDir

	binDir := flag.String("bin-dir", binDirDef, "The directory for store etcd and etcdctl binaries.")
	binLastRelease := flag.String("bin-last-release", "", "The path for the last release etcd binary.")

	flag.StringVar(&CertDir, "cert-dir", certDirDef, "The directory for store certificate files.")
	flag.Parse()

	BinPath = binPath{
		Etcd:            *binDir + "/etcd",
		EtcdLastRelease: *binDir + "/etcd-last-release",
		Etcdctl:         *binDir + "/etcdctl",
		Etcdutl:         *binDir + "/etcdutl",
		LazyFS:          *binDir + "/lazyfs",
	}
	if *binLastRelease != "" {
		BinPath.EtcdLastRelease = *binLastRelease
	}
	CertPath = CertDir + "/server.crt"
	PrivateKeyPath = CertDir + "/server.key.insecure"
	CaPath = CertDir + "/ca.crt"
	RevokedCertPath = CertDir + "/server-revoked.crt"
	RevokedPrivateKeyPath = CertDir + "/server-revoked.key.insecure"
	CrlPath = CertDir + "/revoke.crl"

	CertPath2 = CertDir + "/server2.crt"
	PrivateKeyPath2 = CertDir + "/server2.key.insecure"

	CertPath3 = CertDir + "/server3.crt"
	PrivateKeyPath3 = CertDir + "/server3.key.insecure"
}