GH-85: pass provider deps to decoder wrapper

yandex · Feb 18, 2018 · 60b4df3 · 60b4df3
1 parent 536169c
commit 60b4df3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/core/provider/json.go b/core/provider/json.go
@@ -25,11 +25,11 @@ func NewJSONProvider(newAmmo func() core.Ammo, conf JSONProviderConfig) core.Pro
 // decode data into intermediate struct, but then transform in into desired ammo.
 // For example, decode {"body":"some data"} into struct { Data string }, and transform it to
 // http.Request.
-func NewCustomJSONProvider(wrapDecoder func(decoder AmmoDecoder) AmmoDecoder, newAmmo func() core.Ammo, conf JSONProviderConfig) core.Provider {
+func NewCustomJSONProvider(wrapDecoder func(deps core.ProviderDeps, decoder AmmoDecoder) AmmoDecoder, newAmmo func() core.Ammo, conf JSONProviderConfig) core.Provider {
 	var newDecoder NewAmmoDecoder = func(deps core.ProviderDeps, source io.Reader) (AmmoDecoder, error) {
 		decoder := NewJSONAmmoDecoder(source, conf.Buffer.BufferSizeOrDefault())
 		if wrapDecoder != nil {
-			decoder = wrapDecoder(decoder)
+			decoder = wrapDecoder(deps, decoder)
 		}
 		return decoder, nil
 	}

diff --git a/core/provider/json_test.go b/core/provider/json_test.go
@@ -63,7 +63,7 @@ func TestCustomJSONProvider(t *testing.T) {
 	newAmmo := func() core.Ammo {
 		return &testJSONAmmo{}
 	}
-	wrapDecoder := func(decoder AmmoDecoder) AmmoDecoder {
+	wrapDecoder := func(_ core.ProviderDeps, decoder AmmoDecoder) AmmoDecoder {
 		return AmmoDecoderFunc(func(ammo core.Ammo) error {
 			err := decoder.Decode(ammo)
 			if err != nil {