github.com/IBM-Blockchain/fabric-operator@v1.0.4/pkg/initializer/common/enroller/mocks/cryptoenroller.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package mocks
     3  
     4  import (
     5  	"sync"
     6  	"time"
     7  
     8  	"github.com/IBM-Blockchain/fabric-operator/api/v1beta1"
     9  	"github.com/IBM-Blockchain/fabric-operator/pkg/initializer/common/config"
    10  	"github.com/IBM-Blockchain/fabric-operator/pkg/initializer/common/enroller"
    11  )
    12  
    13  type CryptoEnroller struct {
    14  	EnrollStub        func() (*config.Response, error)
    15  	enrollMutex       sync.RWMutex
    16  	enrollArgsForCall []struct {
    17  	}
    18  	enrollReturns struct {
    19  		result1 *config.Response
    20  		result2 error
    21  	}
    22  	enrollReturnsOnCall map[int]struct {
    23  		result1 *config.Response
    24  		result2 error
    25  	}
    26  	GetEnrollmentRequestStub        func() *v1beta1.Enrollment
    27  	getEnrollmentRequestMutex       sync.RWMutex
    28  	getEnrollmentRequestArgsForCall []struct {
    29  	}
    30  	getEnrollmentRequestReturns struct {
    31  		result1 *v1beta1.Enrollment
    32  	}
    33  	getEnrollmentRequestReturnsOnCall map[int]struct {
    34  		result1 *v1beta1.Enrollment
    35  	}
    36  	PingCAStub        func(time.Duration) error
    37  	pingCAMutex       sync.RWMutex
    38  	pingCAArgsForCall []struct {
    39  		arg1 time.Duration
    40  	}
    41  	pingCAReturns struct {
    42  		result1 error
    43  	}
    44  	pingCAReturnsOnCall map[int]struct {
    45  		result1 error
    46  	}
    47  	invocations      map[string][][]interface{}
    48  	invocationsMutex sync.RWMutex
    49  }
    50  
    51  func (fake *CryptoEnroller) Enroll() (*config.Response, error) {
    52  	fake.enrollMutex.Lock()
    53  	ret, specificReturn := fake.enrollReturnsOnCall[len(fake.enrollArgsForCall)]
    54  	fake.enrollArgsForCall = append(fake.enrollArgsForCall, struct {
    55  	}{})
    56  	stub := fake.EnrollStub
    57  	fakeReturns := fake.enrollReturns
    58  	fake.recordInvocation("Enroll", []interface{}{})
    59  	fake.enrollMutex.Unlock()
    60  	if stub != nil {
    61  		return stub()
    62  	}
    63  	if specificReturn {
    64  		return ret.result1, ret.result2
    65  	}
    66  	return fakeReturns.result1, fakeReturns.result2
    67  }
    68  
    69  func (fake *CryptoEnroller) EnrollCallCount() int {
    70  	fake.enrollMutex.RLock()
    71  	defer fake.enrollMutex.RUnlock()
    72  	return len(fake.enrollArgsForCall)
    73  }
    74  
    75  func (fake *CryptoEnroller) EnrollCalls(stub func() (*config.Response, error)) {
    76  	fake.enrollMutex.Lock()
    77  	defer fake.enrollMutex.Unlock()
    78  	fake.EnrollStub = stub
    79  }
    80  
    81  func (fake *CryptoEnroller) EnrollReturns(result1 *config.Response, result2 error) {
    82  	fake.enrollMutex.Lock()
    83  	defer fake.enrollMutex.Unlock()
    84  	fake.EnrollStub = nil
    85  	fake.enrollReturns = struct {
    86  		result1 *config.Response
    87  		result2 error
    88  	}{result1, result2}
    89  }
    90  
    91  func (fake *CryptoEnroller) EnrollReturnsOnCall(i int, result1 *config.Response, result2 error) {
    92  	fake.enrollMutex.Lock()
    93  	defer fake.enrollMutex.Unlock()
    94  	fake.EnrollStub = nil
    95  	if fake.enrollReturnsOnCall == nil {
    96  		fake.enrollReturnsOnCall = make(map[int]struct {
    97  			result1 *config.Response
    98  			result2 error
    99  		})
   100  	}
   101  	fake.enrollReturnsOnCall[i] = struct {
   102  		result1 *config.Response
   103  		result2 error
   104  	}{result1, result2}
   105  }
   106  
   107  func (fake *CryptoEnroller) GetEnrollmentRequest() *v1beta1.Enrollment {
   108  	fake.getEnrollmentRequestMutex.Lock()
   109  	ret, specificReturn := fake.getEnrollmentRequestReturnsOnCall[len(fake.getEnrollmentRequestArgsForCall)]
   110  	fake.getEnrollmentRequestArgsForCall = append(fake.getEnrollmentRequestArgsForCall, struct {
   111  	}{})
   112  	stub := fake.GetEnrollmentRequestStub
   113  	fakeReturns := fake.getEnrollmentRequestReturns
   114  	fake.recordInvocation("GetEnrollmentRequest", []interface{}{})
   115  	fake.getEnrollmentRequestMutex.Unlock()
   116  	if stub != nil {
   117  		return stub()
   118  	}
   119  	if specificReturn {
   120  		return ret.result1
   121  	}
   122  	return fakeReturns.result1
   123  }
   124  
   125  func (fake *CryptoEnroller) GetEnrollmentRequestCallCount() int {
   126  	fake.getEnrollmentRequestMutex.RLock()
   127  	defer fake.getEnrollmentRequestMutex.RUnlock()
   128  	return len(fake.getEnrollmentRequestArgsForCall)
   129  }
   130  
   131  func (fake *CryptoEnroller) GetEnrollmentRequestCalls(stub func() *v1beta1.Enrollment) {
   132  	fake.getEnrollmentRequestMutex.Lock()
   133  	defer fake.getEnrollmentRequestMutex.Unlock()
   134  	fake.GetEnrollmentRequestStub = stub
   135  }
   136  
   137  func (fake *CryptoEnroller) GetEnrollmentRequestReturns(result1 *v1beta1.Enrollment) {
   138  	fake.getEnrollmentRequestMutex.Lock()
   139  	defer fake.getEnrollmentRequestMutex.Unlock()
   140  	fake.GetEnrollmentRequestStub = nil
   141  	fake.getEnrollmentRequestReturns = struct {
   142  		result1 *v1beta1.Enrollment
   143  	}{result1}
   144  }
   145  
   146  func (fake *CryptoEnroller) GetEnrollmentRequestReturnsOnCall(i int, result1 *v1beta1.Enrollment) {
   147  	fake.getEnrollmentRequestMutex.Lock()
   148  	defer fake.getEnrollmentRequestMutex.Unlock()
   149  	fake.GetEnrollmentRequestStub = nil
   150  	if fake.getEnrollmentRequestReturnsOnCall == nil {
   151  		fake.getEnrollmentRequestReturnsOnCall = make(map[int]struct {
   152  			result1 *v1beta1.Enrollment
   153  		})
   154  	}
   155  	fake.getEnrollmentRequestReturnsOnCall[i] = struct {
   156  		result1 *v1beta1.Enrollment
   157  	}{result1}
   158  }
   159  
   160  func (fake *CryptoEnroller) PingCA(arg1 time.Duration) error {
   161  	fake.pingCAMutex.Lock()
   162  	ret, specificReturn := fake.pingCAReturnsOnCall[len(fake.pingCAArgsForCall)]
   163  	fake.pingCAArgsForCall = append(fake.pingCAArgsForCall, struct {
   164  		arg1 time.Duration
   165  	}{arg1})
   166  	stub := fake.PingCAStub
   167  	fakeReturns := fake.pingCAReturns
   168  	fake.recordInvocation("PingCA", []interface{}{arg1})
   169  	fake.pingCAMutex.Unlock()
   170  	if stub != nil {
   171  		return stub(arg1)
   172  	}
   173  	if specificReturn {
   174  		return ret.result1
   175  	}
   176  	return fakeReturns.result1
   177  }
   178  
   179  func (fake *CryptoEnroller) PingCACallCount() int {
   180  	fake.pingCAMutex.RLock()
   181  	defer fake.pingCAMutex.RUnlock()
   182  	return len(fake.pingCAArgsForCall)
   183  }
   184  
   185  func (fake *CryptoEnroller) PingCACalls(stub func(time.Duration) error) {
   186  	fake.pingCAMutex.Lock()
   187  	defer fake.pingCAMutex.Unlock()
   188  	fake.PingCAStub = stub
   189  }
   190  
   191  func (fake *CryptoEnroller) PingCAArgsForCall(i int) time.Duration {
   192  	fake.pingCAMutex.RLock()
   193  	defer fake.pingCAMutex.RUnlock()
   194  	argsForCall := fake.pingCAArgsForCall[i]
   195  	return argsForCall.arg1
   196  }
   197  
   198  func (fake *CryptoEnroller) PingCAReturns(result1 error) {
   199  	fake.pingCAMutex.Lock()
   200  	defer fake.pingCAMutex.Unlock()
   201  	fake.PingCAStub = nil
   202  	fake.pingCAReturns = struct {
   203  		result1 error
   204  	}{result1}
   205  }
   206  
   207  func (fake *CryptoEnroller) PingCAReturnsOnCall(i int, result1 error) {
   208  	fake.pingCAMutex.Lock()
   209  	defer fake.pingCAMutex.Unlock()
   210  	fake.PingCAStub = nil
   211  	if fake.pingCAReturnsOnCall == nil {
   212  		fake.pingCAReturnsOnCall = make(map[int]struct {
   213  			result1 error
   214  		})
   215  	}
   216  	fake.pingCAReturnsOnCall[i] = struct {
   217  		result1 error
   218  	}{result1}
   219  }
   220  
   221  func (fake *CryptoEnroller) Invocations() map[string][][]interface{} {
   222  	fake.invocationsMutex.RLock()
   223  	defer fake.invocationsMutex.RUnlock()
   224  	fake.enrollMutex.RLock()
   225  	defer fake.enrollMutex.RUnlock()
   226  	fake.getEnrollmentRequestMutex.RLock()
   227  	defer fake.getEnrollmentRequestMutex.RUnlock()
   228  	fake.pingCAMutex.RLock()
   229  	defer fake.pingCAMutex.RUnlock()
   230  	copiedInvocations := map[string][][]interface{}{}
   231  	for key, value := range fake.invocations {
   232  		copiedInvocations[key] = value
   233  	}
   234  	return copiedInvocations
   235  }
   236  
   237  func (fake *CryptoEnroller) recordInvocation(key string, args []interface{}) {
   238  	fake.invocationsMutex.Lock()
   239  	defer fake.invocationsMutex.Unlock()
   240  	if fake.invocations == nil {
   241  		fake.invocations = map[string][][]interface{}{}
   242  	}
   243  	if fake.invocations[key] == nil {
   244  		fake.invocations[key] = [][]interface{}{}
   245  	}
   246  	fake.invocations[key] = append(fake.invocations[key], args)
   247  }
   248  
   249  var _ enroller.CryptoEnroller = new(CryptoEnroller)