mock_provider.go 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243
  1. // Code generated by MockGen. DO NOT EDIT.
  2. // Source: github.com/kubecost/cost-model/cloud (interfaces: Provider)
  3. // Package mocks is a generated GoMock package.
  4. package mocks
  5. import (
  6. gomock "github.com/golang/mock/gomock"
  7. cloud "github.com/kubecost/cost-model/cloud"
  8. io "io"
  9. v1 "k8s.io/api/core/v1"
  10. url "net/url"
  11. reflect "reflect"
  12. )
  13. // MockProvider is a mock of Provider interface
  14. type MockProvider struct {
  15. ctrl *gomock.Controller
  16. recorder *MockProviderMockRecorder
  17. }
  18. // MockProviderMockRecorder is the mock recorder for MockProvider
  19. type MockProviderMockRecorder struct {
  20. mock *MockProvider
  21. }
  22. // NewMockProvider creates a new mock instance
  23. func NewMockProvider(ctrl *gomock.Controller) *MockProvider {
  24. mock := &MockProvider{ctrl: ctrl}
  25. mock.recorder = &MockProviderMockRecorder{mock}
  26. return mock
  27. }
  28. // EXPECT returns an object that allows the caller to indicate expected use
  29. func (m *MockProvider) EXPECT() *MockProviderMockRecorder {
  30. return m.recorder
  31. }
  32. // AddServiceKey mocks base method
  33. func (m *MockProvider) AddServiceKey(arg0 url.Values) error {
  34. m.ctrl.T.Helper()
  35. ret := m.ctrl.Call(m, "AddServiceKey", arg0)
  36. ret0, _ := ret[0].(error)
  37. return ret0
  38. }
  39. // AddServiceKey indicates an expected call of AddServiceKey
  40. func (mr *MockProviderMockRecorder) AddServiceKey(arg0 interface{}) *gomock.Call {
  41. mr.mock.ctrl.T.Helper()
  42. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddServiceKey", reflect.TypeOf((*MockProvider)(nil).AddServiceKey), arg0)
  43. }
  44. // AllNodePricing mocks base method
  45. func (m *MockProvider) AllNodePricing() (interface{}, error) {
  46. m.ctrl.T.Helper()
  47. ret := m.ctrl.Call(m, "AllNodePricing")
  48. ret0, _ := ret[0].(interface{})
  49. ret1, _ := ret[1].(error)
  50. return ret0, ret1
  51. }
  52. // AllNodePricing indicates an expected call of AllNodePricing
  53. func (mr *MockProviderMockRecorder) AllNodePricing() *gomock.Call {
  54. mr.mock.ctrl.T.Helper()
  55. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AllNodePricing", reflect.TypeOf((*MockProvider)(nil).AllNodePricing))
  56. }
  57. // ClusterInfo mocks base method
  58. func (m *MockProvider) ClusterInfo() (map[string]string, error) {
  59. m.ctrl.T.Helper()
  60. ret := m.ctrl.Call(m, "ClusterInfo")
  61. ret0, _ := ret[0].(map[string]string)
  62. ret1, _ := ret[1].(error)
  63. return ret0, ret1
  64. }
  65. // ClusterInfo indicates an expected call of ClusterInfo
  66. func (mr *MockProviderMockRecorder) ClusterInfo() *gomock.Call {
  67. mr.mock.ctrl.T.Helper()
  68. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClusterInfo", reflect.TypeOf((*MockProvider)(nil).ClusterInfo))
  69. }
  70. // DownloadPricingData mocks base method
  71. func (m *MockProvider) DownloadPricingData() error {
  72. m.ctrl.T.Helper()
  73. ret := m.ctrl.Call(m, "DownloadPricingData")
  74. ret0, _ := ret[0].(error)
  75. return ret0
  76. }
  77. // DownloadPricingData indicates an expected call of DownloadPricingData
  78. func (mr *MockProviderMockRecorder) DownloadPricingData() *gomock.Call {
  79. mr.mock.ctrl.T.Helper()
  80. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadPricingData", reflect.TypeOf((*MockProvider)(nil).DownloadPricingData))
  81. }
  82. // ExternalAllocations mocks base method
  83. func (m *MockProvider) ExternalAllocations(arg0, arg1, arg2 string) ([]*cloud.OutOfClusterAllocation, error) {
  84. m.ctrl.T.Helper()
  85. ret := m.ctrl.Call(m, "ExternalAllocations", arg0, arg1, arg2)
  86. ret0, _ := ret[0].([]*cloud.OutOfClusterAllocation)
  87. ret1, _ := ret[1].(error)
  88. return ret0, ret1
  89. }
  90. // ExternalAllocations indicates an expected call of ExternalAllocations
  91. func (mr *MockProviderMockRecorder) ExternalAllocations(arg0, arg1, arg2 interface{}) *gomock.Call {
  92. mr.mock.ctrl.T.Helper()
  93. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExternalAllocations", reflect.TypeOf((*MockProvider)(nil).ExternalAllocations), arg0, arg1, arg2)
  94. }
  95. // GetConfig mocks base method
  96. func (m *MockProvider) GetConfig() (*cloud.CustomPricing, error) {
  97. m.ctrl.T.Helper()
  98. ret := m.ctrl.Call(m, "GetConfig")
  99. ret0, _ := ret[0].(*cloud.CustomPricing)
  100. ret1, _ := ret[1].(error)
  101. return ret0, ret1
  102. }
  103. // GetConfig indicates an expected call of GetConfig
  104. func (mr *MockProviderMockRecorder) GetConfig() *gomock.Call {
  105. mr.mock.ctrl.T.Helper()
  106. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConfig", reflect.TypeOf((*MockProvider)(nil).GetConfig))
  107. }
  108. // GetDisks mocks base method
  109. func (m *MockProvider) GetDisks() ([]byte, error) {
  110. m.ctrl.T.Helper()
  111. ret := m.ctrl.Call(m, "GetDisks")
  112. ret0, _ := ret[0].([]byte)
  113. ret1, _ := ret[1].(error)
  114. return ret0, ret1
  115. }
  116. // GetDisks indicates an expected call of GetDisks
  117. func (mr *MockProviderMockRecorder) GetDisks() *gomock.Call {
  118. mr.mock.ctrl.T.Helper()
  119. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDisks", reflect.TypeOf((*MockProvider)(nil).GetDisks))
  120. }
  121. // GetKey mocks base method
  122. func (m *MockProvider) GetKey(arg0 map[string]string) cloud.Key {
  123. m.ctrl.T.Helper()
  124. ret := m.ctrl.Call(m, "GetKey", arg0)
  125. ret0, _ := ret[0].(cloud.Key)
  126. return ret0
  127. }
  128. // GetKey indicates an expected call of GetKey
  129. func (mr *MockProviderMockRecorder) GetKey(arg0 interface{}) *gomock.Call {
  130. mr.mock.ctrl.T.Helper()
  131. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetKey", reflect.TypeOf((*MockProvider)(nil).GetKey), arg0)
  132. }
  133. // GetLocalStorageQuery mocks base method
  134. func (m *MockProvider) GetLocalStorageQuery() (string, error) {
  135. m.ctrl.T.Helper()
  136. ret := m.ctrl.Call(m, "GetLocalStorageQuery")
  137. ret0, _ := ret[0].(string)
  138. ret1, _ := ret[1].(error)
  139. return ret0, ret1
  140. }
  141. // GetLocalStorageQuery indicates an expected call of GetLocalStorageQuery
  142. func (mr *MockProviderMockRecorder) GetLocalStorageQuery() *gomock.Call {
  143. mr.mock.ctrl.T.Helper()
  144. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLocalStorageQuery", reflect.TypeOf((*MockProvider)(nil).GetLocalStorageQuery))
  145. }
  146. // GetManagementPlatform mocks base method
  147. func (m *MockProvider) GetManagementPlatform() (string, error) {
  148. m.ctrl.T.Helper()
  149. ret := m.ctrl.Call(m, "GetManagementPlatform")
  150. ret0, _ := ret[0].(string)
  151. ret1, _ := ret[1].(error)
  152. return ret0, ret1
  153. }
  154. // GetManagementPlatform indicates an expected call of GetManagementPlatform
  155. func (mr *MockProviderMockRecorder) GetManagementPlatform() *gomock.Call {
  156. mr.mock.ctrl.T.Helper()
  157. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetManagementPlatform", reflect.TypeOf((*MockProvider)(nil).GetManagementPlatform))
  158. }
  159. // GetPVKey mocks base method
  160. func (m *MockProvider) GetPVKey(arg0 *v1.PersistentVolume, arg1 map[string]string) cloud.PVKey {
  161. m.ctrl.T.Helper()
  162. ret := m.ctrl.Call(m, "GetPVKey", arg0, arg1)
  163. ret0, _ := ret[0].(cloud.PVKey)
  164. return ret0
  165. }
  166. // GetPVKey indicates an expected call of GetPVKey
  167. func (mr *MockProviderMockRecorder) GetPVKey(arg0, arg1 interface{}) *gomock.Call {
  168. mr.mock.ctrl.T.Helper()
  169. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPVKey", reflect.TypeOf((*MockProvider)(nil).GetPVKey), arg0, arg1)
  170. }
  171. // NodePricing mocks base method
  172. func (m *MockProvider) NodePricing(arg0 cloud.Key) (*cloud.Node, error) {
  173. m.ctrl.T.Helper()
  174. ret := m.ctrl.Call(m, "NodePricing", arg0)
  175. ret0, _ := ret[0].(*cloud.Node)
  176. ret1, _ := ret[1].(error)
  177. return ret0, ret1
  178. }
  179. // NodePricing indicates an expected call of NodePricing
  180. func (mr *MockProviderMockRecorder) NodePricing(arg0 interface{}) *gomock.Call {
  181. mr.mock.ctrl.T.Helper()
  182. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NodePricing", reflect.TypeOf((*MockProvider)(nil).NodePricing), arg0)
  183. }
  184. // PVPricing mocks base method
  185. func (m *MockProvider) PVPricing(arg0 cloud.PVKey) (*cloud.PV, error) {
  186. m.ctrl.T.Helper()
  187. ret := m.ctrl.Call(m, "PVPricing", arg0)
  188. ret0, _ := ret[0].(*cloud.PV)
  189. ret1, _ := ret[1].(error)
  190. return ret0, ret1
  191. }
  192. // PVPricing indicates an expected call of PVPricing
  193. func (mr *MockProviderMockRecorder) PVPricing(arg0 interface{}) *gomock.Call {
  194. mr.mock.ctrl.T.Helper()
  195. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PVPricing", reflect.TypeOf((*MockProvider)(nil).PVPricing), arg0)
  196. }
  197. // UpdateConfig mocks base method
  198. func (m *MockProvider) UpdateConfig(arg0 io.Reader, arg1 string) (*cloud.CustomPricing, error) {
  199. m.ctrl.T.Helper()
  200. ret := m.ctrl.Call(m, "UpdateConfig", arg0, arg1)
  201. ret0, _ := ret[0].(*cloud.CustomPricing)
  202. ret1, _ := ret[1].(error)
  203. return ret0, ret1
  204. }
  205. // UpdateConfig indicates an expected call of UpdateConfig
  206. func (mr *MockProviderMockRecorder) UpdateConfig(arg0, arg1 interface{}) *gomock.Call {
  207. mr.mock.ctrl.T.Helper()
  208. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateConfig", reflect.TypeOf((*MockProvider)(nil).UpdateConfig), arg0, arg1)
  209. }