github.com/cloudfoundry-attic/ltc@v0.0.0-20151123212628-098adc7919fc/task_runner/fake_task_runner/fake_task_runner.go (about)

     1  // This file was generated by counterfeiter
     2  package fake_task_runner
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/cloudfoundry-incubator/ltc/task_runner"
     8  )
     9  
    10  type FakeTaskRunner struct {
    11  	CreateTaskStub        func(createTaskParams task_runner.CreateTaskParams) error
    12  	createTaskMutex       sync.RWMutex
    13  	createTaskArgsForCall []struct {
    14  		createTaskParams task_runner.CreateTaskParams
    15  	}
    16  	createTaskReturns struct {
    17  		result1 error
    18  	}
    19  	SubmitTaskStub        func(submitTaskJson []byte) (string, error)
    20  	submitTaskMutex       sync.RWMutex
    21  	submitTaskArgsForCall []struct {
    22  		submitTaskJson []byte
    23  	}
    24  	submitTaskReturns struct {
    25  		result1 string
    26  		result2 error
    27  	}
    28  	DeleteTaskStub        func(taskGuid string) error
    29  	deleteTaskMutex       sync.RWMutex
    30  	deleteTaskArgsForCall []struct {
    31  		taskGuid string
    32  	}
    33  	deleteTaskReturns struct {
    34  		result1 error
    35  	}
    36  	CancelTaskStub        func(taskGuid string) error
    37  	cancelTaskMutex       sync.RWMutex
    38  	cancelTaskArgsForCall []struct {
    39  		taskGuid string
    40  	}
    41  	cancelTaskReturns struct {
    42  		result1 error
    43  	}
    44  }
    45  
    46  func (fake *FakeTaskRunner) CreateTask(createTaskParams task_runner.CreateTaskParams) error {
    47  	fake.createTaskMutex.Lock()
    48  	fake.createTaskArgsForCall = append(fake.createTaskArgsForCall, struct {
    49  		createTaskParams task_runner.CreateTaskParams
    50  	}{createTaskParams})
    51  	fake.createTaskMutex.Unlock()
    52  	if fake.CreateTaskStub != nil {
    53  		return fake.CreateTaskStub(createTaskParams)
    54  	} else {
    55  		return fake.createTaskReturns.result1
    56  	}
    57  }
    58  
    59  func (fake *FakeTaskRunner) CreateTaskCallCount() int {
    60  	fake.createTaskMutex.RLock()
    61  	defer fake.createTaskMutex.RUnlock()
    62  	return len(fake.createTaskArgsForCall)
    63  }
    64  
    65  func (fake *FakeTaskRunner) CreateTaskArgsForCall(i int) task_runner.CreateTaskParams {
    66  	fake.createTaskMutex.RLock()
    67  	defer fake.createTaskMutex.RUnlock()
    68  	return fake.createTaskArgsForCall[i].createTaskParams
    69  }
    70  
    71  func (fake *FakeTaskRunner) CreateTaskReturns(result1 error) {
    72  	fake.CreateTaskStub = nil
    73  	fake.createTaskReturns = struct {
    74  		result1 error
    75  	}{result1}
    76  }
    77  
    78  func (fake *FakeTaskRunner) SubmitTask(submitTaskJson []byte) (string, error) {
    79  	fake.submitTaskMutex.Lock()
    80  	fake.submitTaskArgsForCall = append(fake.submitTaskArgsForCall, struct {
    81  		submitTaskJson []byte
    82  	}{submitTaskJson})
    83  	fake.submitTaskMutex.Unlock()
    84  	if fake.SubmitTaskStub != nil {
    85  		return fake.SubmitTaskStub(submitTaskJson)
    86  	} else {
    87  		return fake.submitTaskReturns.result1, fake.submitTaskReturns.result2
    88  	}
    89  }
    90  
    91  func (fake *FakeTaskRunner) SubmitTaskCallCount() int {
    92  	fake.submitTaskMutex.RLock()
    93  	defer fake.submitTaskMutex.RUnlock()
    94  	return len(fake.submitTaskArgsForCall)
    95  }
    96  
    97  func (fake *FakeTaskRunner) SubmitTaskArgsForCall(i int) []byte {
    98  	fake.submitTaskMutex.RLock()
    99  	defer fake.submitTaskMutex.RUnlock()
   100  	return fake.submitTaskArgsForCall[i].submitTaskJson
   101  }
   102  
   103  func (fake *FakeTaskRunner) SubmitTaskReturns(result1 string, result2 error) {
   104  	fake.SubmitTaskStub = nil
   105  	fake.submitTaskReturns = struct {
   106  		result1 string
   107  		result2 error
   108  	}{result1, result2}
   109  }
   110  
   111  func (fake *FakeTaskRunner) DeleteTask(taskGuid string) error {
   112  	fake.deleteTaskMutex.Lock()
   113  	fake.deleteTaskArgsForCall = append(fake.deleteTaskArgsForCall, struct {
   114  		taskGuid string
   115  	}{taskGuid})
   116  	fake.deleteTaskMutex.Unlock()
   117  	if fake.DeleteTaskStub != nil {
   118  		return fake.DeleteTaskStub(taskGuid)
   119  	} else {
   120  		return fake.deleteTaskReturns.result1
   121  	}
   122  }
   123  
   124  func (fake *FakeTaskRunner) DeleteTaskCallCount() int {
   125  	fake.deleteTaskMutex.RLock()
   126  	defer fake.deleteTaskMutex.RUnlock()
   127  	return len(fake.deleteTaskArgsForCall)
   128  }
   129  
   130  func (fake *FakeTaskRunner) DeleteTaskArgsForCall(i int) string {
   131  	fake.deleteTaskMutex.RLock()
   132  	defer fake.deleteTaskMutex.RUnlock()
   133  	return fake.deleteTaskArgsForCall[i].taskGuid
   134  }
   135  
   136  func (fake *FakeTaskRunner) DeleteTaskReturns(result1 error) {
   137  	fake.DeleteTaskStub = nil
   138  	fake.deleteTaskReturns = struct {
   139  		result1 error
   140  	}{result1}
   141  }
   142  
   143  func (fake *FakeTaskRunner) CancelTask(taskGuid string) error {
   144  	fake.cancelTaskMutex.Lock()
   145  	fake.cancelTaskArgsForCall = append(fake.cancelTaskArgsForCall, struct {
   146  		taskGuid string
   147  	}{taskGuid})
   148  	fake.cancelTaskMutex.Unlock()
   149  	if fake.CancelTaskStub != nil {
   150  		return fake.CancelTaskStub(taskGuid)
   151  	} else {
   152  		return fake.cancelTaskReturns.result1
   153  	}
   154  }
   155  
   156  func (fake *FakeTaskRunner) CancelTaskCallCount() int {
   157  	fake.cancelTaskMutex.RLock()
   158  	defer fake.cancelTaskMutex.RUnlock()
   159  	return len(fake.cancelTaskArgsForCall)
   160  }
   161  
   162  func (fake *FakeTaskRunner) CancelTaskArgsForCall(i int) string {
   163  	fake.cancelTaskMutex.RLock()
   164  	defer fake.cancelTaskMutex.RUnlock()
   165  	return fake.cancelTaskArgsForCall[i].taskGuid
   166  }
   167  
   168  func (fake *FakeTaskRunner) CancelTaskReturns(result1 error) {
   169  	fake.CancelTaskStub = nil
   170  	fake.cancelTaskReturns = struct {
   171  		result1 error
   172  	}{result1}
   173  }
   174  
   175  var _ task_runner.TaskRunner = new(FakeTaskRunner)