fix tests

This commit is contained in:
dovholuknf
2023-07-05 17:47:28 -04:00
parent e976c14cbe
commit 0985f63049
5 changed files with 14 additions and 17 deletions
+2 -2
View File
@@ -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)
+1 -3
View File
@@ -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",
+6 -8
View File
@@ -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)
+1 -4
View File
@@ -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",
+4
View File
@@ -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))
}