validgen/internal/codegenerator/build_func_validator_test.go at d7c4ae75c8a32a40c7d22692eb3d1889ef22ffcc · opencodeco/validgen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package codegenerator

import (
	"testing"

	"github.com/opencodeco/validgen/internal/analyzer"
	"github.com/opencodeco/validgen/internal/common"
	"github.com/opencodeco/validgen/internal/parser"
	"github.com/sergi/go-diff/diffmatchpatch"
)

func TestBuildFuncValidatorCodeFieldOperations(t *testing.T) {
	type fields struct {
		Struct *analyzer.Struct
	}
	tests := []struct {
		name   string
		fields fields
		want   string
	}{
		{
			name: "Field inner op",
			fields: fields{
				Struct: &analyzer.Struct{
					Struct: parser.Struct{
						PackageName: "main",
						StructName:  "TestStruct",
						Fields: []parser.Field{
							{
								FieldName: "Field1",
								Type:      common.FieldType{BaseType: "string"},
								Tag:       ``,
							},
							{
								FieldName: "Field2",
								Type:      common.FieldType{BaseType: "string"},
								Tag:       `validate:"neqfield=Field1"`,
							},
						},
					},
					FieldsValidations: []analyzer.FieldValidations{
						{
							Validations: []*analyzer.Validation{},
						},
						{
							Validations: []*analyzer.Validation{AssertParserValidation(t, "neqfield=Field1")},
						},
					},
				},
			},
			want: `func TestStructValidate(obj *TestStruct) []error {
var errs []error
if !(obj.Field2 != obj.Field1) {
errs = append(errs, types.NewValidationError("Field2 must not be equal to Field1"))
}
return errs
}
`,
		},
		{
			name: "Field nested op",
			fields: fields{
				Struct: &analyzer.Struct{
					Struct: parser.Struct{
						PackageName: "main",
						StructName:  "TestStruct",
						Fields: []parser.Field{
							{
								FieldName: "Field1",
								Type:      common.FieldType{BaseType: "string"},
								Tag:       `validate:"neqfield=Nested.Field2"`,
							},
							{
								FieldName: "Nested",
								Type:      common.FieldType{BaseType: "NestedStruct"},
								Tag:       ``,
							},
						},
					},
					FieldsValidations: []analyzer.FieldValidations{
						{
							Validations: []*analyzer.Validation{AssertParserValidation(t, "neqfield=Nested.Field2")},
						},
						{
							Validations: []*analyzer.Validation{},
						},
					},
				},
			},
			want: `func TestStructValidate(obj *TestStruct) []error {
var errs []error
if !(obj.Field1 != obj.Nested.Field2) {
errs = append(errs, types.NewValidationError("Field1 must not be equal to Nested.Field2"))
}
return errs
}
`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gv := GenValidations{
				Struct: tt.fields.Struct,
			}
			got, err := gv.BuildFuncValidatorCode()
			if err != nil {
				t.Errorf("FileValidator.GenerateValidator() error = %v, wantErr %v", err, nil)
				return
			}
			if got != tt.want {
				t.Errorf("FileValidator.GenerateValidator() = %v, want %v", got, tt.want)
				dmp := diffmatchpatch.New()
				diffs := dmp.DiffMain(tt.want, got, false)
				if len(diffs) > 1 {
					t.Errorf("FileValidator.GenerateValidator() diff = \n%v", dmp.DiffPrettyText(diffs))
				}
			}
		})
	}
}