add ut

Signed-off-by: he1pa <18012015693@163.com>

pkg/tools/gen/gentoml_test.go

@@ -1,6 +1,7 @@
 package gen
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/goccy/go-yaml"
@@ -100,12 +101,24 @@ c_key = "value3"
 `,
 			expectErr: nil,
 		},
+
+		{
+			name: "Table Test",
+			data: &yaml.MapSlice{
+				{Key: "a_key", Value: map[string]string{
+					"a_a_key": "value2",
+				}},
+				{Key: "b_key", Value: "value1"},
+			},
+			expectedTOML: "",
+			expectErr:    fmt.Errorf("unsupported to define 'b_key' after a table, ref: https://toml.io/en/v1.0.0#table"),
+		},
 	}
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			tomlData, err := toml.Marshal(tt.data)
-			if err != tt.expectErr {
+			if err.Error() != tt.expectErr.Error() {
 				t.Fatalf("expected error: %v, got: %v", tt.expectErr, err)
 			}