|
|
package framework
|
|
|
|
|
|
import (
|
|
|
"flag"
|
|
|
"io"
|
|
|
"math/rand"
|
|
|
"os"
|
|
|
"testing"
|
|
|
"time"
|
|
|
|
|
|
"github.com/onsi/ginkgo"
|
|
|
"github.com/onsi/ginkgo/reporters"
|
|
|
"github.com/onsi/gomega"
|
|
|
)
|
|
|
|
|
|
var DefaultStartTimeout = float64(60 * 60)
|
|
|
|
|
|
type Framework struct {
|
|
|
Config *Config
|
|
|
|
|
|
configFile string // 配置文件的路径
|
|
|
initTimeout float64 // 启动时候，包括安装集群和依赖及本程序的超时时间
|
|
|
}
|
|
|
|
|
|
func NewFramework() *Framework {
|
|
|
return &Framework{}
|
|
|
}
|
|
|
|
|
|
// Flags 解析命令行参数： --config --timeout
|
|
|
func (f *Framework) Flags() *Framework {
|
|
|
flag.StringVar(&f.configFile, "config", "config", "config file to used")
|
|
|
flag.Float64Var(&f.initTimeout, "startup-timeout", DefaultStartTimeout, "startup timeout")
|
|
|
flag.Parse() // 让配置生效，就是将参数赋值到变量中
|
|
|
return f
|
|
|
}
|
|
|
|
|
|
// LoadConfig 加载配置文件到 fmw
|
|
|
func (f *Framework) LoadConfig(writer io.Writer) *Framework {
|
|
|
// 1. 创建 config 对象
|
|
|
config := NewConfig()
|
|
|
|
|
|
// 2. 加载配置文件内容到 config 对象中
|
|
|
if err := config.Load(f.configFile); err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
|
|
|
// 3. 将传入的 writer 应用到 config中
|
|
|
// 4. 将 config 加入到 fmw 中
|
|
|
return f.WithConfig(config.WithWriter(writer))
|
|
|
}
|
|
|
|
|
|
func (f *Framework) SynchronizedBeforeSuite(initFunc func()) *Framework {
|
|
|
if initFunc == nil {
|
|
|
initFunc = func() {
|
|
|
// 1. 安装环境
|
|
|
ginkgo.By("Deploying test environment")
|
|
|
if err := f.DeployTestEnvironment(); err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
|
|
|
// 2. 初始化环境访问的授权，也就是创建kubectl 访问需要的config
|
|
|
ginkgo.By("kubectl switch context")
|
|
|
// TODO
|
|
|
// kubectl config
|
|
|
|
|
|
// 3. 安装依赖和我们的程序
|
|
|
ginkgo.By("Preparing install steps")
|
|
|
installer := NewInstaller(f.Config)
|
|
|
ginkgo.By("Executing install steps")
|
|
|
if err := installer.Install(); err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
ginkgo.SynchronizedBeforeSuite(func() []byte {
|
|
|
initFunc()
|
|
|
return nil
|
|
|
}, func(_ []byte) {}, f.initTimeout)
|
|
|
return f
|
|
|
}
|
|
|
|
|
|
func (f *Framework) SynchronizedAfterSuite(destroyFunc func()) *Framework {
|
|
|
if destroyFunc == nil {
|
|
|
destroyFunc = func() {
|
|
|
// 回收测试环境
|
|
|
if err := f.DestroyTestEnvironment(); err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
ginkgo.SynchronizedAfterSuite(func() {}, destroyFunc, f.initTimeout)
|
|
|
return f
|
|
|
}
|
|
|
|
|
|
func (f *Framework) MRun(m *testing.M) {
|
|
|
rand.Seed(time.Now().UnixNano()) // 优化随机数
|
|
|
os.Exit(m.Run()) // 执行真正的 TestMain
|
|
|
}
|
|
|
|
|
|
func (f *Framework) Run(t *testing.T) {
|
|
|
gomega.RegisterFailHandler(ginkgo.Fail)
|
|
|
var r []ginkgo.Reporter
|
|
|
r = append(r, reporters.NewJUnitReporter("e2e.xml"))
|
|
|
ginkgo.RunSpecsWithDefaultAndCustomReporters(t, "e2e", r)
|
|
|
}
|
|
|
|
|
|
func (f *Framework) WithConfig(config *Config) *Framework {
|
|
|
f.Config = config
|
|
|
return f
|
|
|
}
|
|
|
|
|
|
// TODO
|
|
|
func (f *Framework) DeployTestEnvironment() error {
|
|
|
return nil
|
|
|
|
|
|
}
|
|
|
|
|
|
// TODO
|
|
|
func (f *Framework) DestroyTestEnvironment() error {
|
|
|
return nil
|
|
|
|
|
|
}
|
