fix tests

vektra · Feb 13, 2024 · 1805cf2 · 1805cf2
1 parent 587f197
commit 1805cf2
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/pkg/config/config_test.go b/pkg/config/config_test.go
@@ -908,6 +908,7 @@ func TestNewConfigFromViper(t *testing.T) {
 				Case:   "camel",
 				Dir:    ".",
 				Output: "./mocks",
+				Style:  "mockery",
 			},
 		},
 		{
@@ -919,11 +920,13 @@ packages:
 			want: &Config{
 				Dir:                  "mocks/{{.PackagePath}}",
 				FileName:             "mock_{{.InterfaceName}}.go",
+				Formatter:            "goimports",
 				IncludeAutoGenerated: true,
 				MockName:             "Mock{{.InterfaceName}}",
 				Outpkg:               "{{.PackageName}}",
 				WithExpecter:         true,
 				LogLevel:             "info",
+				Style:                "mockery",
 			},
 		},
 		{
@@ -937,11 +940,13 @@ packages:
 			want: &Config{
 				Dir:                  "barfoo",
 				FileName:             "foobar.go",
+				Formatter:            "goimports",
 				IncludeAutoGenerated: true,
 				MockName:             "Mock{{.InterfaceName}}",
 				Outpkg:               "{{.PackageName}}",
 				WithExpecter:         true,
 				LogLevel:             "info",
+				Style:                "mockery",
 			},
 		},
 	}
@@ -974,9 +979,7 @@ packages:
 			got._cfgAsMap = nil
 			tt.want._cfgAsMap = nil
 
-			if !reflect.DeepEqual(got, tt.want) {
-				t.Errorf("NewConfigFromViper() = %v, want %v", got, tt.want)
-			}
+			assert.Equal(t, tt.want, got)
 		})
 	}
 }

diff --git a/pkg/fixtures/variadic.go b/pkg/fixtures/variadic.go
@@ -5,4 +5,3 @@ type VariadicFunction = func(args1 string, args2 ...interface{}) interface{}
 type Variadic interface {
 	VariadicFunction(str string, vFunc VariadicFunction) error
 }
-
diff --git a/pkg/outputter_test.go b/pkg/outputter_test.go
@@ -226,7 +226,9 @@ func TestOutputter_Generate(t *testing.T) {
 	}
 	for _, tt := range tests {
 		if tt.fields.config == nil {
-			tt.fields.config = &config.Config{}
+			tt.fields.config = &config.Config{
+				Style: "mockery",
+			}
 		}
 		tt.fields.config.Dir = t.TempDir()
 		tt.fields.config.MockName = "Mock{{.InterfaceName}}"
Original file line number	Diff line number	Diff line change
Expand Up		@@ -5,4 +5,3 @@ type VariadicFunction = func(args1 string, args2 ...interface{}) interface{}
		type Variadic interface {
		VariadicFunction(str string, vFunc VariadicFunction) error
		}