record enc.inTable statu

Signed-off-by: he1pa <18012015693@163.com>
kcl-lang · Jan 6, 2025 · 8c75530 · 8c75530
1 parent f32088d
commit 8c75530
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pkg/3rdparty/toml/encode.go b/pkg/3rdparty/toml/encode.go
@@ -394,6 +394,7 @@ func (enc *Encoder) eArrayOfTables(key Key, rv reflect.Value) {
 }
 
 func (enc *Encoder) eTable(key Key, rv reflect.Value) {
+	inTable := enc.inTable
 	enc.inTable = true
 	if len(key) == 1 {
 		// Output an extra newline between top-level tables.
@@ -406,7 +407,7 @@ func (enc *Encoder) eTable(key Key, rv reflect.Value) {
 		enc.newline()
 	}
 	enc.eMapOrStruct(key, rv, false)
-	enc.inTable = false
+	enc.inTable = inTable
 }
 
 func (enc *Encoder) eMapOrStruct(key Key, rv reflect.Value, inline bool) {

diff --git a/pkg/tools/gen/gentoml_test.go b/pkg/tools/gen/gentoml_test.go
@@ -121,6 +121,8 @@ c_key = "value3"
 			if err != nil && tt.expectErr != nil {
 				if err.Error() != tt.expectErr.Error() {
 					t.Fatalf("expected error: %v, got: %v", tt.expectErr, err)
+				} else {
+					return
 				}
 			}