fix cnf/nginx | udpate tests

internal/cnf/nginx.go

@@ -78,6 +78,7 @@ func (d *nginx) browse(_path string, create ...bool) (*lib.Line, bool) {
 					Lines:      make([]*lib.Line, 0),
 				}
 				current.Lines = append(current.Lines, sec)
+				current = sec
 				continue
 			}
 

internal/cnf/nginx_test.go

@@ -102,6 +102,7 @@ func TestNginxSetCreatePath(t *testing.T) {
 		{"ignore xxx;\n", "key", "ignore", "newvalue"},
 		{"ignore xxx;\nsection {\n\tkey value;\n}\n", "section.key", "ignore", "newvalue"},
 		{"section {\n\tkey value;\n\tignore xxx;\n}\n", "section.key", "section.ignore", "newvalue"},
+		{"ignoresec {\n\tignore xxx;\n}\n\nsection {\n}\n", "section.key", "ignoresec.ignore", "newvalue"},
 		{"ignoresec {\n\tignore xxx;\n}\n\nsection {\n\tkey value;\n}\n", "section.key", "ignoresec.ignore", "newvalue"},
 	}
 
@@ -151,3 +152,103 @@ func TestNginxSetCreatePath(t *testing.T) {
 	}
 
 }
+
+func TestNginxSetCreateEncode(t *testing.T) {
+
+	tests := []struct {
+		raw     string
+		key     string
+		value   string
+		encoded string
+	}{
+		{"ignore xxx;\n", "key", "newvalue", "ignore\t\txxx;\nkey\t\tnewvalue;\n"},
+		{"ignore xxx;\nsection {\n\tkey value;\n}\n", "section.key", "newvalue", "ignore\t\txxx;\nsection {\n\tkey\t\tnewvalue;\n}\n\n"},
+		{"section {\n\tkey value;\n\tignore xxx;\n}\n", "section.key", "newvalue", "section {\n\tkey\t\tnewvalue;\n\tignore\t\txxx;\n}\n\n"},
+		{"ignoresec {\n\tignore xxx;\n}\n\nsection {\n}\n", "section.key", "newvalue", "ignoresec {\n\tignore\t\txxx;\n}\n\nsection {\n\tkey\t\tnewvalue;\n}\n\n"},
+		{"ignoresec {\n\tignore xxx;\n}\n\nsection {\n\tkey value;\n}\n", "section.key", "newvalue", "ignoresec {\n\tignore\t\txxx;\n}\n\nsection {\n\tkey\t\tnewvalue;\n}\n\n"},
+
+		{
+			"; comment 1\n\nk1 v1;\n#comment 2\nsec1 {\n}\nsec2 {\nignore xxx;\n sec3 {\nignore2 yyy;\n}\n\n}\n\n",
+			"key",
+			"newvalue",
+			"; comment 1\nk1\t\tv1;\n#comment 2\nsec1 {\n}\n\nsec2 {\n\tignore\t\txxx;\n\tsec3 {\n\t\tignore2\t\tyyy;\n\t}\n\n}\n\nkey\t\tnewvalue;\n"},
+		{
+			"; comment 1\n\nk1 v1;\n#comment 2\nsec1 {\n}\nsec2 {\nignore xxx;\n sec3 {\nignore2 yyy;\n}\n\n}\n\n",
+			"section.key",
+			"newvalue",
+			"; comment 1\nk1\t\tv1;\n#comment 2\nsec1 {\n}\n\nsec2 {\n\tignore\t\txxx;\n\tsec3 {\n\t\tignore2\t\tyyy;\n\t}\n\n}\n\nsection {\n\tkey\t\tnewvalue;\n}\n\n"},
+	}
+
+	for i, test := range tests {
+
+		parser := new(nginx)
+		r, w := bytes.NewBufferString(test.raw), new(bytes.Buffer)
+
+		// try to extract value
+		_, err := parser.ReadFrom(r)
+		if err != nil {
+			t.Errorf("[%d] parse error: %s", i, err)
+			continue
+		}
+
+		// update value
+		if !parser.Set(test.key, test.value) {
+			t.Errorf("[%d] cannot set '%s' to '%s'", i, test.key, test.value)
+			continue
+		}
+
+		// check new value
+		value, found := parser.Get(test.key)
+		if !found {
+			t.Errorf("[%d] expected a result, got none", i)
+			continue
+		}
+
+		// check value
+		if value != test.value {
+			t.Errorf("[%d] expected '%s' got '%s'", i, test.value, value)
+			continue
+		}
+
+		// writeToBuffer
+		_, err = parser.WriteTo(w)
+		if err != nil {
+			t.Errorf("[%d] unexpected write error <%s>", i, err)
+			continue
+		}
+
+		encoded := w.String()
+
+		// check value
+		if encoded != test.encoded {
+			t.Errorf("[%d] wrong encoded value \n-=-=-= HAVE =-=-=-\n%s\n-=-=-= WANT =-=-=-\n%s\n-=-=-=-=-=\n", i, escape(test.encoded), escape(encoded))
+			continue
+		}
+
+		// try to write
+	}
+
+}
+
+func escape(in string) string {
+
+	out := make([]rune, 0)
+
+	for _, char := range in {
+		if char == '\\' {
+			out = append(out, []rune("\\\\")...)
+		} else if char == '\n' {
+			out = append(out, []rune("\\n")...)
+		} else if char == '\r' {
+			out = append(out, []rune("\\r")...)
+		} else if char == '\t' {
+			out = append(out, []rune("\\t")...)
+		} else if char == '\033' {
+			out = append(out, []rune("\\033")...)
+		} else {
+			out = append(out, char)
+		}
+	}
+
+	return string(out)
+}