2020-06-02 22:48:55 +08:00
|
|
|
package e2e
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"flag"
|
|
|
|
|
"fmt"
|
|
|
|
|
"os"
|
|
|
|
|
"testing"
|
|
|
|
|
|
2020-09-23 13:49:14 +08:00
|
|
|
"github.com/fatedier/frp/pkg/util/log"
|
2020-06-02 22:48:55 +08:00
|
|
|
"github.com/fatedier/frp/test/e2e/framework"
|
2020-09-07 14:57:23 +08:00
|
|
|
|
|
|
|
|
// test source
|
|
|
|
|
_ "github.com/fatedier/frp/test/e2e/basic"
|
|
|
|
|
_ "github.com/fatedier/frp/test/e2e/plugin"
|
|
|
|
|
|
|
|
|
|
_ "github.com/onsi/ginkgo"
|
2020-06-02 22:48:55 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// handleFlags sets up all flags and parses the command line.
|
|
|
|
|
func handleFlags() {
|
|
|
|
|
framework.RegisterCommonFlags(flag.CommandLine)
|
|
|
|
|
flag.Parse()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestMain(m *testing.M) {
|
|
|
|
|
// Register test flags, then parse flags.
|
|
|
|
|
handleFlags()
|
|
|
|
|
|
|
|
|
|
if err := framework.ValidateTestContext(&framework.TestContext); err != nil {
|
|
|
|
|
fmt.Println(err)
|
|
|
|
|
os.Exit(1)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
log.InitLog("console", "", framework.TestContext.LogLevel, 0, true)
|
|
|
|
|
os.Exit(m.Run())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestE2E(t *testing.T) {
|
|
|
|
|
RunE2ETests(t)
|
|
|
|
|
}
|
