mirror of
https://github.com/openziti/ziti.git
synced 2026-10-12 15:28:05 +00:00
fix tests
This commit is contained in:
@@ -83,8 +83,8 @@ func NewCmdCreateConfigRouter(routerOptions *CreateConfigRouterOptions) *NewCrea
|
||||
RenderedValues: data,
|
||||
}
|
||||
|
||||
cmd.AddCommand(NewCmdCreateConfigRouterEdge())
|
||||
cmd.AddCommand(NewCmdCreateConfigRouterFabric())
|
||||
cmd.AddCommand(NewCmdCreateConfigRouterEdge(routerOptions, data))
|
||||
cmd.AddCommand(NewCmdCreateConfigRouterFabric(routerOptions, data))
|
||||
|
||||
routerOptions.addCreateFlags(cmd.Command)
|
||||
routerOptions.addFlags(cmd.Command)
|
||||
|
||||
@@ -63,9 +63,7 @@ var (
|
||||
var routerConfigEdgeTemplate string
|
||||
|
||||
// NewCmdCreateConfigRouterEdge creates a command object for the "edge" command
|
||||
func NewCmdCreateConfigRouterEdge() *cobra.Command {
|
||||
data := &ConfigTemplateValues{}
|
||||
routerOptions := &CreateConfigRouterOptions{}
|
||||
func NewCmdCreateConfigRouterEdge(routerOptions *CreateConfigRouterOptions, data *ConfigTemplateValues) *cobra.Command {
|
||||
cmd := &cobra.Command{
|
||||
Use: "edge",
|
||||
Short: "Create an edge router config",
|
||||
|
||||
@@ -15,8 +15,7 @@ var defaultArgs = []string{"edge", "--routerName", "test-router"}
|
||||
var testHostname, _ = os.Hostname()
|
||||
|
||||
func TestEdgeRouterAdvertisedAddress(t *testing.T) {
|
||||
data := &ConfigTemplateValues{}
|
||||
clearEnvAndInitializeTestData()
|
||||
routerOpts := clearEnvAndInitializeTestData()
|
||||
routerAdvHostIp := "192.168.10.10"
|
||||
routerAdvHostDns := "controller01.zitinetwork.example.org"
|
||||
keys := map[string]string{
|
||||
@@ -24,25 +23,24 @@ func TestEdgeRouterAdvertisedAddress(t *testing.T) {
|
||||
"ZITI_EDGE_ROUTER_PORT": "443",
|
||||
}
|
||||
// Defaults to hostname if nothing is set
|
||||
createRouterConfig(defaultArgs, nil, keys)
|
||||
_, data := createRouterConfig(defaultArgs, routerOpts, keys)
|
||||
require.Equal(t, testHostname, data.Router.Edge.AdvertisedHost, nil)
|
||||
|
||||
// If IP override set, uses that value over hostname
|
||||
keys["ZITI_EDGE_ROUTER_IP_OVERRIDE"] = routerAdvHostIp
|
||||
createRouterConfig(defaultArgs, nil, keys)
|
||||
require.Equal(t, routerAdvHostIp, data.Router.Edge.AdvertisedHost, nil)
|
||||
_, data2 := createRouterConfig(defaultArgs, routerOpts, keys)
|
||||
require.Equal(t, routerAdvHostIp, data2.Router.Edge.AdvertisedHost, nil)
|
||||
|
||||
// If advertised address set, uses that over IP override or hostname
|
||||
keys["ZITI_EDGE_ROUTER_ADVERTISED_HOST"] = routerAdvHostDns
|
||||
keys["ZITI_EDGE_ROUTER_IP_OVERRIDE"] = routerAdvHostIp
|
||||
createRouterConfig(defaultArgs, nil, keys)
|
||||
require.Equal(t, routerAdvHostDns, data.Router.Edge.AdvertisedHost, nil)
|
||||
_, data3 := createRouterConfig(defaultArgs, routerOpts, keys)
|
||||
require.Equal(t, routerAdvHostDns, data3.Router.Edge.AdvertisedHost, nil)
|
||||
}
|
||||
|
||||
func TestTunnelerEnabledByDefault(t *testing.T) {
|
||||
// Setup options
|
||||
routerOptions := clearEnvAndInitializeTestData()
|
||||
routerOptions.Output = defaultOutput
|
||||
|
||||
// Create and run the CLI command without the tunnel flag
|
||||
config, _ := createRouterConfig([]string{"edge", "--routerName", "myRouter"}, routerOptions, nil)
|
||||
|
||||
@@ -44,10 +44,7 @@ var (
|
||||
var routerConfigFabricTemplate string
|
||||
|
||||
// NewCmdCreateConfigRouterFabric creates a command object for the "fabric" command
|
||||
func NewCmdCreateConfigRouterFabric() *cobra.Command {
|
||||
data := &ConfigTemplateValues{}
|
||||
routerOptions := &CreateConfigRouterOptions{}
|
||||
|
||||
func NewCmdCreateConfigRouterFabric(routerOptions *CreateConfigRouterOptions, data *ConfigTemplateValues) *cobra.Command {
|
||||
cmd := &cobra.Command{
|
||||
Use: "fabric",
|
||||
Short: "Create a fabric router config",
|
||||
|
||||
@@ -100,6 +100,10 @@ func GetCtrlAdvertisedAddress() string {
|
||||
return getFromEnv(constants.CtrlAdvertisedAddressVarName, HostnameOrNetworkName)
|
||||
}
|
||||
|
||||
func GetEdgeRouterIpOvderride() string {
|
||||
return getFromEnv(constants.ZitiEdgeRouterIPOverrideVarName, defaultValue(""))
|
||||
}
|
||||
|
||||
func GetCtrlAdvertisedPort() string {
|
||||
return getFromEnv(constants.CtrlAdvertisedPortVarName, defaultValue(constants.DefaultCtrlAdvertisedPort))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user