github.com/koko1123/flow-go-1@v0.29.6/module/mock/dkg_controller.go (about)

     1  // Code generated by mockery v2.13.1. DO NOT EDIT.
     2  
     3  package mock
     4  
     5  import (
     6  	crypto "github.com/onflow/flow-go/crypto"
     7  	flow "github.com/koko1123/flow-go-1/model/flow"
     8  
     9  	mock "github.com/stretchr/testify/mock"
    10  )
    11  
    12  // DKGController is an autogenerated mock type for the DKGController type
    13  type DKGController struct {
    14  	mock.Mock
    15  }
    16  
    17  // End provides a mock function with given fields:
    18  func (_m *DKGController) End() error {
    19  	ret := _m.Called()
    20  
    21  	var r0 error
    22  	if rf, ok := ret.Get(0).(func() error); ok {
    23  		r0 = rf()
    24  	} else {
    25  		r0 = ret.Error(0)
    26  	}
    27  
    28  	return r0
    29  }
    30  
    31  // EndPhase1 provides a mock function with given fields:
    32  func (_m *DKGController) EndPhase1() error {
    33  	ret := _m.Called()
    34  
    35  	var r0 error
    36  	if rf, ok := ret.Get(0).(func() error); ok {
    37  		r0 = rf()
    38  	} else {
    39  		r0 = ret.Error(0)
    40  	}
    41  
    42  	return r0
    43  }
    44  
    45  // EndPhase2 provides a mock function with given fields:
    46  func (_m *DKGController) EndPhase2() error {
    47  	ret := _m.Called()
    48  
    49  	var r0 error
    50  	if rf, ok := ret.Get(0).(func() error); ok {
    51  		r0 = rf()
    52  	} else {
    53  		r0 = ret.Error(0)
    54  	}
    55  
    56  	return r0
    57  }
    58  
    59  // GetArtifacts provides a mock function with given fields:
    60  func (_m *DKGController) GetArtifacts() (crypto.PrivateKey, crypto.PublicKey, []crypto.PublicKey) {
    61  	ret := _m.Called()
    62  
    63  	var r0 crypto.PrivateKey
    64  	if rf, ok := ret.Get(0).(func() crypto.PrivateKey); ok {
    65  		r0 = rf()
    66  	} else {
    67  		if ret.Get(0) != nil {
    68  			r0 = ret.Get(0).(crypto.PrivateKey)
    69  		}
    70  	}
    71  
    72  	var r1 crypto.PublicKey
    73  	if rf, ok := ret.Get(1).(func() crypto.PublicKey); ok {
    74  		r1 = rf()
    75  	} else {
    76  		if ret.Get(1) != nil {
    77  			r1 = ret.Get(1).(crypto.PublicKey)
    78  		}
    79  	}
    80  
    81  	var r2 []crypto.PublicKey
    82  	if rf, ok := ret.Get(2).(func() []crypto.PublicKey); ok {
    83  		r2 = rf()
    84  	} else {
    85  		if ret.Get(2) != nil {
    86  			r2 = ret.Get(2).([]crypto.PublicKey)
    87  		}
    88  	}
    89  
    90  	return r0, r1, r2
    91  }
    92  
    93  // GetIndex provides a mock function with given fields:
    94  func (_m *DKGController) GetIndex() int {
    95  	ret := _m.Called()
    96  
    97  	var r0 int
    98  	if rf, ok := ret.Get(0).(func() int); ok {
    99  		r0 = rf()
   100  	} else {
   101  		r0 = ret.Get(0).(int)
   102  	}
   103  
   104  	return r0
   105  }
   106  
   107  // Poll provides a mock function with given fields: blockReference
   108  func (_m *DKGController) Poll(blockReference flow.Identifier) error {
   109  	ret := _m.Called(blockReference)
   110  
   111  	var r0 error
   112  	if rf, ok := ret.Get(0).(func(flow.Identifier) error); ok {
   113  		r0 = rf(blockReference)
   114  	} else {
   115  		r0 = ret.Error(0)
   116  	}
   117  
   118  	return r0
   119  }
   120  
   121  // Run provides a mock function with given fields:
   122  func (_m *DKGController) Run() error {
   123  	ret := _m.Called()
   124  
   125  	var r0 error
   126  	if rf, ok := ret.Get(0).(func() error); ok {
   127  		r0 = rf()
   128  	} else {
   129  		r0 = ret.Error(0)
   130  	}
   131  
   132  	return r0
   133  }
   134  
   135  // Shutdown provides a mock function with given fields:
   136  func (_m *DKGController) Shutdown() {
   137  	_m.Called()
   138  }
   139  
   140  // SubmitResult provides a mock function with given fields:
   141  func (_m *DKGController) SubmitResult() error {
   142  	ret := _m.Called()
   143  
   144  	var r0 error
   145  	if rf, ok := ret.Get(0).(func() error); ok {
   146  		r0 = rf()
   147  	} else {
   148  		r0 = ret.Error(0)
   149  	}
   150  
   151  	return r0
   152  }
   153  
   154  type mockConstructorTestingTNewDKGController interface {
   155  	mock.TestingT
   156  	Cleanup(func())
   157  }
   158  
   159  // NewDKGController creates a new instance of DKGController. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   160  func NewDKGController(t mockConstructorTestingTNewDKGController) *DKGController {
   161  	mock := &DKGController{}
   162  	mock.Mock.Test(t)
   163  
   164  	t.Cleanup(func() { mock.AssertExpectations(t) })
   165  
   166  	return mock
   167  }