github.com/cloudfoundry-incubator/stembuild@v0.0.0-20211223202937-5b61d62226c6/construct/constructfakes/fake_guest_manager.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package constructfakes
     3  
     4  import (
     5  	"context"
     6  	"io"
     7  	"sync"
     8  
     9  	"github.com/cloudfoundry-incubator/stembuild/construct"
    10  )
    11  
    12  type FakeGuestManager struct {
    13  	DownloadFileInGuestStub        func(context.Context, string) (io.Reader, int64, error)
    14  	downloadFileInGuestMutex       sync.RWMutex
    15  	downloadFileInGuestArgsForCall []struct {
    16  		arg1 context.Context
    17  		arg2 string
    18  	}
    19  	downloadFileInGuestReturns struct {
    20  		result1 io.Reader
    21  		result2 int64
    22  		result3 error
    23  	}
    24  	downloadFileInGuestReturnsOnCall map[int]struct {
    25  		result1 io.Reader
    26  		result2 int64
    27  		result3 error
    28  	}
    29  	ExitCodeForProgramInGuestStub        func(context.Context, int64) (int32, error)
    30  	exitCodeForProgramInGuestMutex       sync.RWMutex
    31  	exitCodeForProgramInGuestArgsForCall []struct {
    32  		arg1 context.Context
    33  		arg2 int64
    34  	}
    35  	exitCodeForProgramInGuestReturns struct {
    36  		result1 int32
    37  		result2 error
    38  	}
    39  	exitCodeForProgramInGuestReturnsOnCall map[int]struct {
    40  		result1 int32
    41  		result2 error
    42  	}
    43  	StartProgramInGuestStub        func(context.Context, string, string) (int64, error)
    44  	startProgramInGuestMutex       sync.RWMutex
    45  	startProgramInGuestArgsForCall []struct {
    46  		arg1 context.Context
    47  		arg2 string
    48  		arg3 string
    49  	}
    50  	startProgramInGuestReturns struct {
    51  		result1 int64
    52  		result2 error
    53  	}
    54  	startProgramInGuestReturnsOnCall map[int]struct {
    55  		result1 int64
    56  		result2 error
    57  	}
    58  	invocations      map[string][][]interface{}
    59  	invocationsMutex sync.RWMutex
    60  }
    61  
    62  func (fake *FakeGuestManager) DownloadFileInGuest(arg1 context.Context, arg2 string) (io.Reader, int64, error) {
    63  	fake.downloadFileInGuestMutex.Lock()
    64  	ret, specificReturn := fake.downloadFileInGuestReturnsOnCall[len(fake.downloadFileInGuestArgsForCall)]
    65  	fake.downloadFileInGuestArgsForCall = append(fake.downloadFileInGuestArgsForCall, struct {
    66  		arg1 context.Context
    67  		arg2 string
    68  	}{arg1, arg2})
    69  	fake.recordInvocation("DownloadFileInGuest", []interface{}{arg1, arg2})
    70  	fake.downloadFileInGuestMutex.Unlock()
    71  	if fake.DownloadFileInGuestStub != nil {
    72  		return fake.DownloadFileInGuestStub(arg1, arg2)
    73  	}
    74  	if specificReturn {
    75  		return ret.result1, ret.result2, ret.result3
    76  	}
    77  	fakeReturns := fake.downloadFileInGuestReturns
    78  	return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3
    79  }
    80  
    81  func (fake *FakeGuestManager) DownloadFileInGuestCallCount() int {
    82  	fake.downloadFileInGuestMutex.RLock()
    83  	defer fake.downloadFileInGuestMutex.RUnlock()
    84  	return len(fake.downloadFileInGuestArgsForCall)
    85  }
    86  
    87  func (fake *FakeGuestManager) DownloadFileInGuestCalls(stub func(context.Context, string) (io.Reader, int64, error)) {
    88  	fake.downloadFileInGuestMutex.Lock()
    89  	defer fake.downloadFileInGuestMutex.Unlock()
    90  	fake.DownloadFileInGuestStub = stub
    91  }
    92  
    93  func (fake *FakeGuestManager) DownloadFileInGuestArgsForCall(i int) (context.Context, string) {
    94  	fake.downloadFileInGuestMutex.RLock()
    95  	defer fake.downloadFileInGuestMutex.RUnlock()
    96  	argsForCall := fake.downloadFileInGuestArgsForCall[i]
    97  	return argsForCall.arg1, argsForCall.arg2
    98  }
    99  
   100  func (fake *FakeGuestManager) DownloadFileInGuestReturns(result1 io.Reader, result2 int64, result3 error) {
   101  	fake.downloadFileInGuestMutex.Lock()
   102  	defer fake.downloadFileInGuestMutex.Unlock()
   103  	fake.DownloadFileInGuestStub = nil
   104  	fake.downloadFileInGuestReturns = struct {
   105  		result1 io.Reader
   106  		result2 int64
   107  		result3 error
   108  	}{result1, result2, result3}
   109  }
   110  
   111  func (fake *FakeGuestManager) DownloadFileInGuestReturnsOnCall(i int, result1 io.Reader, result2 int64, result3 error) {
   112  	fake.downloadFileInGuestMutex.Lock()
   113  	defer fake.downloadFileInGuestMutex.Unlock()
   114  	fake.DownloadFileInGuestStub = nil
   115  	if fake.downloadFileInGuestReturnsOnCall == nil {
   116  		fake.downloadFileInGuestReturnsOnCall = make(map[int]struct {
   117  			result1 io.Reader
   118  			result2 int64
   119  			result3 error
   120  		})
   121  	}
   122  	fake.downloadFileInGuestReturnsOnCall[i] = struct {
   123  		result1 io.Reader
   124  		result2 int64
   125  		result3 error
   126  	}{result1, result2, result3}
   127  }
   128  
   129  func (fake *FakeGuestManager) ExitCodeForProgramInGuest(arg1 context.Context, arg2 int64) (int32, error) {
   130  	fake.exitCodeForProgramInGuestMutex.Lock()
   131  	ret, specificReturn := fake.exitCodeForProgramInGuestReturnsOnCall[len(fake.exitCodeForProgramInGuestArgsForCall)]
   132  	fake.exitCodeForProgramInGuestArgsForCall = append(fake.exitCodeForProgramInGuestArgsForCall, struct {
   133  		arg1 context.Context
   134  		arg2 int64
   135  	}{arg1, arg2})
   136  	fake.recordInvocation("ExitCodeForProgramInGuest", []interface{}{arg1, arg2})
   137  	fake.exitCodeForProgramInGuestMutex.Unlock()
   138  	if fake.ExitCodeForProgramInGuestStub != nil {
   139  		return fake.ExitCodeForProgramInGuestStub(arg1, arg2)
   140  	}
   141  	if specificReturn {
   142  		return ret.result1, ret.result2
   143  	}
   144  	fakeReturns := fake.exitCodeForProgramInGuestReturns
   145  	return fakeReturns.result1, fakeReturns.result2
   146  }
   147  
   148  func (fake *FakeGuestManager) ExitCodeForProgramInGuestCallCount() int {
   149  	fake.exitCodeForProgramInGuestMutex.RLock()
   150  	defer fake.exitCodeForProgramInGuestMutex.RUnlock()
   151  	return len(fake.exitCodeForProgramInGuestArgsForCall)
   152  }
   153  
   154  func (fake *FakeGuestManager) ExitCodeForProgramInGuestCalls(stub func(context.Context, int64) (int32, error)) {
   155  	fake.exitCodeForProgramInGuestMutex.Lock()
   156  	defer fake.exitCodeForProgramInGuestMutex.Unlock()
   157  	fake.ExitCodeForProgramInGuestStub = stub
   158  }
   159  
   160  func (fake *FakeGuestManager) ExitCodeForProgramInGuestArgsForCall(i int) (context.Context, int64) {
   161  	fake.exitCodeForProgramInGuestMutex.RLock()
   162  	defer fake.exitCodeForProgramInGuestMutex.RUnlock()
   163  	argsForCall := fake.exitCodeForProgramInGuestArgsForCall[i]
   164  	return argsForCall.arg1, argsForCall.arg2
   165  }
   166  
   167  func (fake *FakeGuestManager) ExitCodeForProgramInGuestReturns(result1 int32, result2 error) {
   168  	fake.exitCodeForProgramInGuestMutex.Lock()
   169  	defer fake.exitCodeForProgramInGuestMutex.Unlock()
   170  	fake.ExitCodeForProgramInGuestStub = nil
   171  	fake.exitCodeForProgramInGuestReturns = struct {
   172  		result1 int32
   173  		result2 error
   174  	}{result1, result2}
   175  }
   176  
   177  func (fake *FakeGuestManager) ExitCodeForProgramInGuestReturnsOnCall(i int, result1 int32, result2 error) {
   178  	fake.exitCodeForProgramInGuestMutex.Lock()
   179  	defer fake.exitCodeForProgramInGuestMutex.Unlock()
   180  	fake.ExitCodeForProgramInGuestStub = nil
   181  	if fake.exitCodeForProgramInGuestReturnsOnCall == nil {
   182  		fake.exitCodeForProgramInGuestReturnsOnCall = make(map[int]struct {
   183  			result1 int32
   184  			result2 error
   185  		})
   186  	}
   187  	fake.exitCodeForProgramInGuestReturnsOnCall[i] = struct {
   188  		result1 int32
   189  		result2 error
   190  	}{result1, result2}
   191  }
   192  
   193  func (fake *FakeGuestManager) StartProgramInGuest(arg1 context.Context, arg2 string, arg3 string) (int64, error) {
   194  	fake.startProgramInGuestMutex.Lock()
   195  	ret, specificReturn := fake.startProgramInGuestReturnsOnCall[len(fake.startProgramInGuestArgsForCall)]
   196  	fake.startProgramInGuestArgsForCall = append(fake.startProgramInGuestArgsForCall, struct {
   197  		arg1 context.Context
   198  		arg2 string
   199  		arg3 string
   200  	}{arg1, arg2, arg3})
   201  	fake.recordInvocation("StartProgramInGuest", []interface{}{arg1, arg2, arg3})
   202  	fake.startProgramInGuestMutex.Unlock()
   203  	if fake.StartProgramInGuestStub != nil {
   204  		return fake.StartProgramInGuestStub(arg1, arg2, arg3)
   205  	}
   206  	if specificReturn {
   207  		return ret.result1, ret.result2
   208  	}
   209  	fakeReturns := fake.startProgramInGuestReturns
   210  	return fakeReturns.result1, fakeReturns.result2
   211  }
   212  
   213  func (fake *FakeGuestManager) StartProgramInGuestCallCount() int {
   214  	fake.startProgramInGuestMutex.RLock()
   215  	defer fake.startProgramInGuestMutex.RUnlock()
   216  	return len(fake.startProgramInGuestArgsForCall)
   217  }
   218  
   219  func (fake *FakeGuestManager) StartProgramInGuestCalls(stub func(context.Context, string, string) (int64, error)) {
   220  	fake.startProgramInGuestMutex.Lock()
   221  	defer fake.startProgramInGuestMutex.Unlock()
   222  	fake.StartProgramInGuestStub = stub
   223  }
   224  
   225  func (fake *FakeGuestManager) StartProgramInGuestArgsForCall(i int) (context.Context, string, string) {
   226  	fake.startProgramInGuestMutex.RLock()
   227  	defer fake.startProgramInGuestMutex.RUnlock()
   228  	argsForCall := fake.startProgramInGuestArgsForCall[i]
   229  	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
   230  }
   231  
   232  func (fake *FakeGuestManager) StartProgramInGuestReturns(result1 int64, result2 error) {
   233  	fake.startProgramInGuestMutex.Lock()
   234  	defer fake.startProgramInGuestMutex.Unlock()
   235  	fake.StartProgramInGuestStub = nil
   236  	fake.startProgramInGuestReturns = struct {
   237  		result1 int64
   238  		result2 error
   239  	}{result1, result2}
   240  }
   241  
   242  func (fake *FakeGuestManager) StartProgramInGuestReturnsOnCall(i int, result1 int64, result2 error) {
   243  	fake.startProgramInGuestMutex.Lock()
   244  	defer fake.startProgramInGuestMutex.Unlock()
   245  	fake.StartProgramInGuestStub = nil
   246  	if fake.startProgramInGuestReturnsOnCall == nil {
   247  		fake.startProgramInGuestReturnsOnCall = make(map[int]struct {
   248  			result1 int64
   249  			result2 error
   250  		})
   251  	}
   252  	fake.startProgramInGuestReturnsOnCall[i] = struct {
   253  		result1 int64
   254  		result2 error
   255  	}{result1, result2}
   256  }
   257  
   258  func (fake *FakeGuestManager) Invocations() map[string][][]interface{} {
   259  	fake.invocationsMutex.RLock()
   260  	defer fake.invocationsMutex.RUnlock()
   261  	fake.downloadFileInGuestMutex.RLock()
   262  	defer fake.downloadFileInGuestMutex.RUnlock()
   263  	fake.exitCodeForProgramInGuestMutex.RLock()
   264  	defer fake.exitCodeForProgramInGuestMutex.RUnlock()
   265  	fake.startProgramInGuestMutex.RLock()
   266  	defer fake.startProgramInGuestMutex.RUnlock()
   267  	copiedInvocations := map[string][][]interface{}{}
   268  	for key, value := range fake.invocations {
   269  		copiedInvocations[key] = value
   270  	}
   271  	return copiedInvocations
   272  }
   273  
   274  func (fake *FakeGuestManager) recordInvocation(key string, args []interface{}) {
   275  	fake.invocationsMutex.Lock()
   276  	defer fake.invocationsMutex.Unlock()
   277  	if fake.invocations == nil {
   278  		fake.invocations = map[string][][]interface{}{}
   279  	}
   280  	if fake.invocations[key] == nil {
   281  		fake.invocations[key] = [][]interface{}{}
   282  	}
   283  	fake.invocations[key] = append(fake.invocations[key], args)
   284  }
   285  
   286  var _ construct.GuestManager = new(FakeGuestManager)