@@ -88,6 +88,10 @@ func TestExtraManifests_Load(t *testing.T) {
8888 },
8989 }
9090
91+ yamlPattern := "extra-manifests/*.yaml"
92+ ymlPattern := "extra-manifests/*.yml"
93+ jsonPattern := "extra-manifests/*.json"
94+
9195 for _ , tc := range cases {
9296 t .Run (tc .name , func (t * testing.T ) {
9397 yamlFiles := []* asset.File {}
@@ -115,33 +119,22 @@ func TestExtraManifests_Load(t *testing.T) {
115119 defer mockCtrl .Finish ()
116120
117121 fileFetcher := mock .NewMockFileFetcher (mockCtrl )
122+
118123 if tc .yamlFetchError != nil {
119- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.yaml" ).Return (
120- []* asset.File {},
121- tc .yamlFetchError ,
122- )
123- } else if tc .ymlFetchError != nil {
124- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.yaml" ).Return (yamlFiles , nil )
124+ fileFetcher .EXPECT ().FetchByPattern (yamlPattern ).Return ([]* asset.File {}, tc .yamlFetchError )
125+ } else {
126+ fileFetcher .EXPECT ().FetchByPattern (yamlPattern ).Return (yamlFiles , nil )
125127
126128 if tc .ymlFetchError != nil {
127- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.yml" ).Return (
128- []* asset.File {},
129- tc .ymlFetchError ,
130- )
131- } else {
132- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.yml" ).Return (ymlFiles , nil )
133- }
134- } else {
135- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.yaml" ).Return (yamlFiles , nil )
136- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.yml" ).Return (ymlFiles , nil )
137-
138- if tc .jsonFetchError != nil {
139- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.json" ).Return (
140- []* asset.File {},
141- tc .jsonFetchError ,
142- )
129+ fileFetcher .EXPECT ().FetchByPattern (ymlPattern ).Return ([]* asset.File {}, tc .ymlFetchError )
143130 } else {
144- fileFetcher .EXPECT ().FetchByPattern ("extra-manifests/*.json" ).Return (jsonFiles , nil )
131+ fileFetcher .EXPECT ().FetchByPattern (ymlPattern ).Return (ymlFiles , nil )
132+
133+ if tc .jsonFetchError != nil {
134+ fileFetcher .EXPECT ().FetchByPattern (jsonPattern ).Return ([]* asset.File {}, tc .jsonFetchError )
135+ } else {
136+ fileFetcher .EXPECT ().FetchByPattern (jsonPattern ).Return (jsonFiles , nil )
137+ }
145138 }
146139 }
147140
0 commit comments