// getConfigs returns a list of test configs based on the passed in flags.
func getConfigs(t *testing.T) []cluster.TestConfig {
// If a config not supplied, just read the flags.
if (flagConfig == nil || len(*flagConfig) == 0) &&
(flagTestConfigs == nil || !*flagTestConfigs) {
return []cluster.TestConfig{readConfigFromFlags()}
}
var configs []cluster.TestConfig
if flagTestConfigs != nil && *flagTestConfigs {
configs = append(configs, cluster.DefaultConfigs()...)
}
if flagConfig != nil && len(*flagConfig) > 0 {
// Read the passed in config from the command line.
var config cluster.TestConfig
if err := json.Unmarshal([]byte(*flagConfig), &config); err != nil {
t.Error(err)
}
configs = append(configs, config)
}
// Override duration in all configs if the flags are set.
for i := 0; i < len(configs); i++ {
// Override values.
if flagDuration != nil && *flagDuration != cluster.DefaultDuration {
configs[i].Duration = *flagDuration
}
// Set missing defaults.
if configs[i].Duration == 0 {
configs[i].Duration = cluster.DefaultDuration
}
}
return configs
}
// getConfigs returns a list of test configs based on the passed in flags.
func getConfigs() []cluster.TestConfig {
if *flagTestConfigs {
return cluster.DefaultConfigs()
}
return []cluster.TestConfig{readConfigFromFlags()}
}
