func InitConfig(t *testing.T, conf string) {
f, err := ioutil.TempFile("", "skydive_agent")
if err != nil {
t.Fatal(err.Error())
}
var logLevel string
if testing.Verbose() {
logLevel = "DEBUG"
} else {
logLevel = "INFO"
}
conf += fmt.Sprintf("\nlogging:\n default: %s", logLevel)
f.WriteString(conf)
f.Close()
err = config.InitConfigFromFile(f.Name())
if err != nil {
t.Fatal(err.Error())
}
err = logging.InitLogger()
if err != nil {
t.Fatal(err)
}
}
func InitConfig(t *testing.T, conf string) {
f, err := ioutil.TempFile("", "skydive_agent")
if err != nil {
t.Fatal(err.Error())
}
f.WriteString(conf)
f.Close()
err = config.InitConfigFromFile(f.Name())
if err != nil {
t.Fatal(err.Error())
}
err = logging.InitLogger()
if err != nil {
t.Fatal(err)
}
}
func InitConfig(t *testing.T, conf string) {
f, err := ioutil.TempFile("", "skydive_agent")
if err != nil {
t.Fatal(err.Error())
}
param := struct {
AnalyzerPort int
LogLevel string
}{
AnalyzerPort: rand.Intn(400) + 64500,
}
if testing.Verbose() {
param.LogLevel = "DEBUG"
} else {
param.LogLevel = "INFO"
}
tmpl, err := template.New("config").Parse(conf)
if err != nil {
t.Fatal(err.Error())
}
buff := bytes.NewBufferString("")
tmpl.Execute(buff, param)
f.Write(buff.Bytes())
f.Close()
t.Logf("Configuration: %s", buff.String())
err = config.InitConfigFromFile(f.Name())
if err != nil {
t.Fatal(err.Error())
}
err = logging.InitLogger()
if err != nil {
t.Fatal(err)
}
}
func InitConfig(t *testing.T, conf string, params ...HelperParams) {
f, err := ioutil.TempFile("", "skydive_agent")
if err != nil {
t.Fatal(err.Error())
}
if len(params) == 0 {
params = []HelperParams{make(HelperParams)}
}
params[0]["AnalyzerPort"] = 64500
if testing.Verbose() {
params[0]["LogLevel"] = "DEBUG"
} else {
params[0]["LogLevel"] = "INFO"
}
tmpl, err := template.New("config").Parse(conf)
if err != nil {
t.Fatal(err.Error())
}
buff := bytes.NewBufferString("")
tmpl.Execute(buff, params[0])
f.Write(buff.Bytes())
f.Close()
t.Logf("Configuration: %s", buff.String())
err = config.InitConfig("file", f.Name())
if err != nil {
t.Fatal(err.Error())
}
err = logging.InitLogger()
if err != nil {
t.Fatal(err)
}
}