github.com/loggregator/cli@v6.33.1-0.20180224010324-82334f081791+incompatible/plugin/rpc/rpcfakes/fake_output_capture.go (about)

     1  // This file was generated by counterfeiter
     2  package rpcfakes
     3  
     4  import (
     5  	"io"
     6  	"sync"
     7  
     8  	"code.cloudfoundry.org/cli/plugin/rpc"
     9  )
    10  
    11  type FakeOutputCapture struct {
    12  	SetOutputBucketStub        func(io.Writer)
    13  	setOutputBucketMutex       sync.RWMutex
    14  	setOutputBucketArgsForCall []struct {
    15  		arg1 io.Writer
    16  	}
    17  	invocations      map[string][][]interface{}
    18  	invocationsMutex sync.RWMutex
    19  }
    20  
    21  func (fake *FakeOutputCapture) SetOutputBucket(arg1 io.Writer) {
    22  	fake.setOutputBucketMutex.Lock()
    23  	fake.setOutputBucketArgsForCall = append(fake.setOutputBucketArgsForCall, struct {
    24  		arg1 io.Writer
    25  	}{arg1})
    26  	fake.recordInvocation("SetOutputBucket", []interface{}{arg1})
    27  	fake.setOutputBucketMutex.Unlock()
    28  	if fake.SetOutputBucketStub != nil {
    29  		fake.SetOutputBucketStub(arg1)
    30  	}
    31  }
    32  
    33  func (fake *FakeOutputCapture) SetOutputBucketCallCount() int {
    34  	fake.setOutputBucketMutex.RLock()
    35  	defer fake.setOutputBucketMutex.RUnlock()
    36  	return len(fake.setOutputBucketArgsForCall)
    37  }
    38  
    39  func (fake *FakeOutputCapture) SetOutputBucketArgsForCall(i int) io.Writer {
    40  	fake.setOutputBucketMutex.RLock()
    41  	defer fake.setOutputBucketMutex.RUnlock()
    42  	return fake.setOutputBucketArgsForCall[i].arg1
    43  }
    44  
    45  func (fake *FakeOutputCapture) Invocations() map[string][][]interface{} {
    46  	fake.invocationsMutex.RLock()
    47  	defer fake.invocationsMutex.RUnlock()
    48  	fake.setOutputBucketMutex.RLock()
    49  	defer fake.setOutputBucketMutex.RUnlock()
    50  	return fake.invocations
    51  }
    52  
    53  func (fake *FakeOutputCapture) recordInvocation(key string, args []interface{}) {
    54  	fake.invocationsMutex.Lock()
    55  	defer fake.invocationsMutex.Unlock()
    56  	if fake.invocations == nil {
    57  		fake.invocations = map[string][][]interface{}{}
    58  	}
    59  	if fake.invocations[key] == nil {
    60  		fake.invocations[key] = [][]interface{}{}
    61  	}
    62  	fake.invocations[key] = append(fake.invocations[key], args)
    63  }
    64  
    65  var _ rpc.OutputCapture = new(FakeOutputCapture)