github.com/paketo-buildpacks/packit@v1.3.2-0.20211206231111-86b75c657449/postal/fakes/transport.go (about)

     1  package fakes
     2  
     3  import (
     4  	"io"
     5  	"sync"
     6  )
     7  
     8  type Transport struct {
     9  	DropCall struct {
    10  		mutex     sync.Mutex
    11  		CallCount int
    12  		Receives  struct {
    13  			Root string
    14  			Uri  string
    15  		}
    16  		Returns struct {
    17  			ReadCloser io.ReadCloser
    18  			Error      error
    19  		}
    20  		Stub func(string, string) (io.ReadCloser, error)
    21  	}
    22  }
    23  
    24  func (f *Transport) Drop(param1 string, param2 string) (io.ReadCloser, error) {
    25  	f.DropCall.mutex.Lock()
    26  	defer f.DropCall.mutex.Unlock()
    27  	f.DropCall.CallCount++
    28  	f.DropCall.Receives.Root = param1
    29  	f.DropCall.Receives.Uri = param2
    30  	if f.DropCall.Stub != nil {
    31  		return f.DropCall.Stub(param1, param2)
    32  	}
    33  	return f.DropCall.Returns.ReadCloser, f.DropCall.Returns.Error
    34  }