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

     1  // This file was generated by counterfeiter
     2  package rpcfakes
     3  
     4  import (
     5  	"sync"
     6  
     7  	"code.cloudfoundry.org/cli/cf/commandregistry"
     8  	"code.cloudfoundry.org/cli/plugin/rpc"
     9  )
    10  
    11  type FakeCommandRunner struct {
    12  	CommandStub        func([]string, commandregistry.Dependency, bool) error
    13  	commandMutex       sync.RWMutex
    14  	commandArgsForCall []struct {
    15  		arg1 []string
    16  		arg2 commandregistry.Dependency
    17  		arg3 bool
    18  	}
    19  	commandReturns struct {
    20  		result1 error
    21  	}
    22  	invocations      map[string][][]interface{}
    23  	invocationsMutex sync.RWMutex
    24  }
    25  
    26  func (fake *FakeCommandRunner) Command(arg1 []string, arg2 commandregistry.Dependency, arg3 bool) error {
    27  	var arg1Copy []string
    28  	if arg1 != nil {
    29  		arg1Copy = make([]string, len(arg1))
    30  		copy(arg1Copy, arg1)
    31  	}
    32  	fake.commandMutex.Lock()
    33  	fake.commandArgsForCall = append(fake.commandArgsForCall, struct {
    34  		arg1 []string
    35  		arg2 commandregistry.Dependency
    36  		arg3 bool
    37  	}{arg1Copy, arg2, arg3})
    38  	fake.recordInvocation("Command", []interface{}{arg1Copy, arg2, arg3})
    39  	fake.commandMutex.Unlock()
    40  	if fake.CommandStub != nil {
    41  		return fake.CommandStub(arg1, arg2, arg3)
    42  	} else {
    43  		return fake.commandReturns.result1
    44  	}
    45  }
    46  
    47  func (fake *FakeCommandRunner) CommandCallCount() int {
    48  	fake.commandMutex.RLock()
    49  	defer fake.commandMutex.RUnlock()
    50  	return len(fake.commandArgsForCall)
    51  }
    52  
    53  func (fake *FakeCommandRunner) CommandArgsForCall(i int) ([]string, commandregistry.Dependency, bool) {
    54  	fake.commandMutex.RLock()
    55  	defer fake.commandMutex.RUnlock()
    56  	return fake.commandArgsForCall[i].arg1, fake.commandArgsForCall[i].arg2, fake.commandArgsForCall[i].arg3
    57  }
    58  
    59  func (fake *FakeCommandRunner) CommandReturns(result1 error) {
    60  	fake.CommandStub = nil
    61  	fake.commandReturns = struct {
    62  		result1 error
    63  	}{result1}
    64  }
    65  
    66  func (fake *FakeCommandRunner) Invocations() map[string][][]interface{} {
    67  	fake.invocationsMutex.RLock()
    68  	defer fake.invocationsMutex.RUnlock()
    69  	fake.commandMutex.RLock()
    70  	defer fake.commandMutex.RUnlock()
    71  	return fake.invocations
    72  }
    73  
    74  func (fake *FakeCommandRunner) recordInvocation(key string, args []interface{}) {
    75  	fake.invocationsMutex.Lock()
    76  	defer fake.invocationsMutex.Unlock()
    77  	if fake.invocations == nil {
    78  		fake.invocations = map[string][][]interface{}{}
    79  	}
    80  	if fake.invocations[key] == nil {
    81  		fake.invocations[key] = [][]interface{}{}
    82  	}
    83  	fake.invocations[key] = append(fake.invocations[key], args)
    84  }
    85  
    86  var _ rpc.CommandRunner = new(FakeCommandRunner)