Exemplo n.º 1
0
func TestCleanKeys(t *testing.T) {
	pre := map[string]interface{}{
		"/my/cool_val/here":     "test",
		"/this_key":             "foo",
		"/prefix/key":           "test",
		"/my/new-cool-val/here": "bar",
		"prefix/noslash":        "test",
	}
	config.SetPrefix("/prefix")
	clean := cleanKeys(pre, "/prefix")
	if len(clean) != len(pre) {
		t.Fatalf("bad length")
	}
	if _, ok := clean["my_cool_val_here"]; !ok {
		t.Fatalf("bad: %v", clean)
	}
	if _, ok := clean["this_key"]; !ok {
		t.Fatalf("bad: %v", clean)
	}
	if _, ok := clean["key"]; !ok {
		t.Fatalf("bad: %v", clean)
	}
	if _, ok := clean["my_new_cool_val_here"]; !ok {
		t.Fatalf("bad: %v", clean)
	}
	if _, ok := clean["noslash"]; !ok {
		t.Fatalf("bad: %v", clean)
	}
}
Exemplo n.º 2
0
func TestProcessTemplateResources(t *testing.T) {
	log.SetQuiet(true)
	// Setup temporary conf, config, and template directories.
	tempConfDir, err := createTempDirs()
	if err != nil {
		t.Errorf("Failed to create temp dirs: %s", err.Error())
	}
	defer os.RemoveAll(tempConfDir)

	// Create the src template.
	srcTemplateFile := filepath.Join(tempConfDir, "templates", "foo.tmpl")
	err = ioutil.WriteFile(srcTemplateFile, []byte("foo = {{ .foo }}"), 0644)
	if err != nil {
		t.Error(err.Error())
	}

	// Create the dest.
	destFile, err := ioutil.TempFile("", "")
	if err != nil {
		t.Errorf("Failed to create destFile: %s", err.Error())
	}
	defer os.Remove(destFile.Name())

	// Create the template resource configuration file.
	templateResourcePath := filepath.Join(tempConfDir, "conf.d", "foo.toml")
	templateResourceFile, err := os.Create(templateResourcePath)
	if err != nil {
		t.Errorf(err.Error())
	}
	tmpl, err := template.New("templateResourceConfig").Parse(templateResourceConfigTmpl)
	if err != nil {
		t.Errorf("Unable to parse template resource template: %s", err.Error())
	}
	data := make(map[string]string)
	data["src"] = "foo.tmpl"
	data["dest"] = destFile.Name()
	err = tmpl.Execute(templateResourceFile, data)
	if err != nil {
		t.Errorf(err.Error())
	}

	// Load the confd configuration settings.
	if err := config.LoadConfig(""); err != nil {
		t.Errorf(err.Error())
	}
	config.SetPrefix("")
	// Use the temporary tempConfDir from above.
	config.SetConfDir(tempConfDir)

	// Create the stub etcd client.
	c := &MockStore{}
	c.AddKey("/foo", "bar")

	// Process the test template resource.
	runErrors := ProcessTemplateResources(c)
	if len(runErrors) > 0 {
		for _, e := range runErrors {
			t.Errorf(e.Error())
		}
	}
	// Verify the results.
	expected := "foo = bar"
	results, err := ioutil.ReadFile(destFile.Name())
	if err != nil {
		t.Error(err.Error())
	}
	if string(results) != expected {
		t.Errorf("Expected contents of dest == '%s', got %s", expected, string(results))
	}
}