fix ut

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

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pkg/3rdparty/toml/encode.go b/pkg/3rdparty/toml/encode.go
@@ -394,7 +394,6 @@ 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.
@@ -407,7 +406,7 @@ func (enc *Encoder) eTable(key Key, rv reflect.Value) {
 		enc.newline()
 	}
 	enc.eMapOrStruct(key, rv, false)
-	enc.inTable = inTable
+	enc.inTable = false
 }
 
 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
@@ -127,9 +127,9 @@ c_key = "value3"
 			} else if err == nil && tt.expectErr == nil {
 				if got := string(tomlData); got != tt.expectedTOML {
 					t.Errorf("expected:\n%s\ngot:\n%s", tt.expectedTOML, got)
-				} else {
-					t.Fatalf("expected error: %v, got: %v", tt.expectErr, err)
 				}
+			} else {
+				t.Fatalf("expected error: %v, got: %v", tt.expectErr, err)
 			}
 
 		})