github.com/braveheart12/insolar-09-08-19@v0.8.7/pulsar/pulsartestutils/pulsar_storage_mock.go (about) 1 /* 2 * Copyright 2019 Insolar Technologies 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package pulsartestutils 18 19 /* 20 DO NOT EDIT! 21 This code was generated automatically using github.com/gojuno/minimock v1.9 22 The original interface "PulsarStorage" can be found in github.com/insolar/insolar/pulsar/storage 23 */ 24 import ( 25 "sync/atomic" 26 "time" 27 28 "github.com/gojuno/minimock" 29 "github.com/insolar/insolar/core" 30 31 testify_assert "github.com/stretchr/testify/assert" 32 ) 33 34 //PulsarStorageMock implements github.com/insolar/insolar/pulsar/storage.PulsarStorage 35 type PulsarStorageMock struct { 36 t minimock.Tester 37 38 CloseFunc func() (r error) 39 CloseCounter uint64 40 ClosePreCounter uint64 41 CloseMock mPulsarStorageMockClose 42 43 GetLastPulseFunc func() (r *core.Pulse, r1 error) 44 GetLastPulseCounter uint64 45 GetLastPulsePreCounter uint64 46 GetLastPulseMock mPulsarStorageMockGetLastPulse 47 48 SavePulseFunc func(p *core.Pulse) (r error) 49 SavePulseCounter uint64 50 SavePulsePreCounter uint64 51 SavePulseMock mPulsarStorageMockSavePulse 52 53 SetLastPulseFunc func(p *core.Pulse) (r error) 54 SetLastPulseCounter uint64 55 SetLastPulsePreCounter uint64 56 SetLastPulseMock mPulsarStorageMockSetLastPulse 57 } 58 59 //NewPulsarStorageMock returns a mock for github.com/insolar/insolar/pulsar/storage.PulsarStorage 60 func NewPulsarStorageMock(t minimock.Tester) *PulsarStorageMock { 61 m := &PulsarStorageMock{t: t} 62 63 if controller, ok := t.(minimock.MockController); ok { 64 controller.RegisterMocker(m) 65 } 66 67 m.CloseMock = mPulsarStorageMockClose{mock: m} 68 m.GetLastPulseMock = mPulsarStorageMockGetLastPulse{mock: m} 69 m.SavePulseMock = mPulsarStorageMockSavePulse{mock: m} 70 m.SetLastPulseMock = mPulsarStorageMockSetLastPulse{mock: m} 71 72 return m 73 } 74 75 type mPulsarStorageMockClose struct { 76 mock *PulsarStorageMock 77 } 78 79 //Return sets up a mock for PulsarStorage.Close to return Return's arguments 80 func (m *mPulsarStorageMockClose) Return(r error) *PulsarStorageMock { 81 m.mock.CloseFunc = func() error { 82 return r 83 } 84 return m.mock 85 } 86 87 //Set uses given function f as a mock of PulsarStorage.Close method 88 func (m *mPulsarStorageMockClose) Set(f func() (r error)) *PulsarStorageMock { 89 m.mock.CloseFunc = f 90 91 return m.mock 92 } 93 94 //Close implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface 95 func (m *PulsarStorageMock) Close() (r error) { 96 atomic.AddUint64(&m.ClosePreCounter, 1) 97 defer atomic.AddUint64(&m.CloseCounter, 1) 98 99 if m.CloseFunc == nil { 100 m.t.Fatal("Unexpected call to PulsarStorageMock.Close") 101 return 102 } 103 104 return m.CloseFunc() 105 } 106 107 //CloseMinimockCounter returns a count of PulsarStorageMock.CloseFunc invocations 108 func (m *PulsarStorageMock) CloseMinimockCounter() uint64 { 109 return atomic.LoadUint64(&m.CloseCounter) 110 } 111 112 //CloseMinimockPreCounter returns the value of PulsarStorageMock.Close invocations 113 func (m *PulsarStorageMock) CloseMinimockPreCounter() uint64 { 114 return atomic.LoadUint64(&m.ClosePreCounter) 115 } 116 117 type mPulsarStorageMockGetLastPulse struct { 118 mock *PulsarStorageMock 119 } 120 121 //Return sets up a mock for PulsarStorage.GetLastPulse to return Return's arguments 122 func (m *mPulsarStorageMockGetLastPulse) Return(r *core.Pulse, r1 error) *PulsarStorageMock { 123 m.mock.GetLastPulseFunc = func() (*core.Pulse, error) { 124 return r, r1 125 } 126 return m.mock 127 } 128 129 //Set uses given function f as a mock of PulsarStorage.GetLastPulse method 130 func (m *mPulsarStorageMockGetLastPulse) Set(f func() (r *core.Pulse, r1 error)) *PulsarStorageMock { 131 m.mock.GetLastPulseFunc = f 132 133 return m.mock 134 } 135 136 //GetLastPulse implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface 137 func (m *PulsarStorageMock) GetLastPulse() (r *core.Pulse, r1 error) { 138 atomic.AddUint64(&m.GetLastPulsePreCounter, 1) 139 defer atomic.AddUint64(&m.GetLastPulseCounter, 1) 140 141 if m.GetLastPulseFunc == nil { 142 m.t.Fatal("Unexpected call to PulsarStorageMock.GetLastPulse") 143 return 144 } 145 146 return m.GetLastPulseFunc() 147 } 148 149 //GetLastPulseMinimockCounter returns a count of PulsarStorageMock.GetLastPulseFunc invocations 150 func (m *PulsarStorageMock) GetLastPulseMinimockCounter() uint64 { 151 return atomic.LoadUint64(&m.GetLastPulseCounter) 152 } 153 154 //GetLastPulseMinimockPreCounter returns the value of PulsarStorageMock.GetLastPulse invocations 155 func (m *PulsarStorageMock) GetLastPulseMinimockPreCounter() uint64 { 156 return atomic.LoadUint64(&m.GetLastPulsePreCounter) 157 } 158 159 type mPulsarStorageMockSavePulse struct { 160 mock *PulsarStorageMock 161 mockExpectations *PulsarStorageMockSavePulseParams 162 } 163 164 //PulsarStorageMockSavePulseParams represents input parameters of the PulsarStorage.SavePulse 165 type PulsarStorageMockSavePulseParams struct { 166 p *core.Pulse 167 } 168 169 //Expect sets up expected params for the PulsarStorage.SavePulse 170 func (m *mPulsarStorageMockSavePulse) Expect(p *core.Pulse) *mPulsarStorageMockSavePulse { 171 m.mockExpectations = &PulsarStorageMockSavePulseParams{p} 172 return m 173 } 174 175 //Return sets up a mock for PulsarStorage.SavePulse to return Return's arguments 176 func (m *mPulsarStorageMockSavePulse) Return(r error) *PulsarStorageMock { 177 m.mock.SavePulseFunc = func(p *core.Pulse) error { 178 return r 179 } 180 return m.mock 181 } 182 183 //Set uses given function f as a mock of PulsarStorage.SavePulse method 184 func (m *mPulsarStorageMockSavePulse) Set(f func(p *core.Pulse) (r error)) *PulsarStorageMock { 185 m.mock.SavePulseFunc = f 186 m.mockExpectations = nil 187 return m.mock 188 } 189 190 //SavePulse implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface 191 func (m *PulsarStorageMock) SavePulse(p *core.Pulse) (r error) { 192 atomic.AddUint64(&m.SavePulsePreCounter, 1) 193 defer atomic.AddUint64(&m.SavePulseCounter, 1) 194 195 if m.SavePulseMock.mockExpectations != nil { 196 testify_assert.Equal(m.t, *m.SavePulseMock.mockExpectations, PulsarStorageMockSavePulseParams{p}, 197 "PulsarStorage.SavePulse got unexpected parameters") 198 199 if m.SavePulseFunc == nil { 200 201 m.t.Fatal("No results are set for the PulsarStorageMock.SavePulse") 202 203 return 204 } 205 } 206 207 if m.SavePulseFunc == nil { 208 m.t.Fatal("Unexpected call to PulsarStorageMock.SavePulse") 209 return 210 } 211 212 return m.SavePulseFunc(p) 213 } 214 215 //SavePulseMinimockCounter returns a count of PulsarStorageMock.SavePulseFunc invocations 216 func (m *PulsarStorageMock) SavePulseMinimockCounter() uint64 { 217 return atomic.LoadUint64(&m.SavePulseCounter) 218 } 219 220 //SavePulseMinimockPreCounter returns the value of PulsarStorageMock.SavePulse invocations 221 func (m *PulsarStorageMock) SavePulseMinimockPreCounter() uint64 { 222 return atomic.LoadUint64(&m.SavePulsePreCounter) 223 } 224 225 type mPulsarStorageMockSetLastPulse struct { 226 mock *PulsarStorageMock 227 mockExpectations *PulsarStorageMockSetLastPulseParams 228 } 229 230 //PulsarStorageMockSetLastPulseParams represents input parameters of the PulsarStorage.SetLastPulse 231 type PulsarStorageMockSetLastPulseParams struct { 232 p *core.Pulse 233 } 234 235 //Expect sets up expected params for the PulsarStorage.SetLastPulse 236 func (m *mPulsarStorageMockSetLastPulse) Expect(p *core.Pulse) *mPulsarStorageMockSetLastPulse { 237 m.mockExpectations = &PulsarStorageMockSetLastPulseParams{p} 238 return m 239 } 240 241 //Return sets up a mock for PulsarStorage.SetLastPulse to return Return's arguments 242 func (m *mPulsarStorageMockSetLastPulse) Return(r error) *PulsarStorageMock { 243 m.mock.SetLastPulseFunc = func(p *core.Pulse) error { 244 return r 245 } 246 return m.mock 247 } 248 249 //Set uses given function f as a mock of PulsarStorage.SetLastPulse method 250 func (m *mPulsarStorageMockSetLastPulse) Set(f func(p *core.Pulse) (r error)) *PulsarStorageMock { 251 m.mock.SetLastPulseFunc = f 252 m.mockExpectations = nil 253 return m.mock 254 } 255 256 //SetLastPulse implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface 257 func (m *PulsarStorageMock) SetLastPulse(p *core.Pulse) (r error) { 258 atomic.AddUint64(&m.SetLastPulsePreCounter, 1) 259 defer atomic.AddUint64(&m.SetLastPulseCounter, 1) 260 261 if m.SetLastPulseMock.mockExpectations != nil { 262 testify_assert.Equal(m.t, *m.SetLastPulseMock.mockExpectations, PulsarStorageMockSetLastPulseParams{p}, 263 "PulsarStorage.SetLastPulse got unexpected parameters") 264 265 if m.SetLastPulseFunc == nil { 266 267 m.t.Fatal("No results are set for the PulsarStorageMock.SetLastPulse") 268 269 return 270 } 271 } 272 273 if m.SetLastPulseFunc == nil { 274 m.t.Fatal("Unexpected call to PulsarStorageMock.SetLastPulse") 275 return 276 } 277 278 return m.SetLastPulseFunc(p) 279 } 280 281 //SetLastPulseMinimockCounter returns a count of PulsarStorageMock.SetLastPulseFunc invocations 282 func (m *PulsarStorageMock) SetLastPulseMinimockCounter() uint64 { 283 return atomic.LoadUint64(&m.SetLastPulseCounter) 284 } 285 286 //SetLastPulseMinimockPreCounter returns the value of PulsarStorageMock.SetLastPulse invocations 287 func (m *PulsarStorageMock) SetLastPulseMinimockPreCounter() uint64 { 288 return atomic.LoadUint64(&m.SetLastPulsePreCounter) 289 } 290 291 //ValidateCallCounters checks that all mocked methods of the interface have been called at least once 292 //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller 293 func (m *PulsarStorageMock) ValidateCallCounters() { 294 295 if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 { 296 m.t.Fatal("Expected call to PulsarStorageMock.Close") 297 } 298 299 if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 { 300 m.t.Fatal("Expected call to PulsarStorageMock.GetLastPulse") 301 } 302 303 if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 { 304 m.t.Fatal("Expected call to PulsarStorageMock.SavePulse") 305 } 306 307 if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 { 308 m.t.Fatal("Expected call to PulsarStorageMock.SetLastPulse") 309 } 310 311 } 312 313 //CheckMocksCalled checks that all mocked methods of the interface have been called at least once 314 //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller 315 //noinspection GoDeprecation 316 func (m *PulsarStorageMock) CheckMocksCalled() { 317 m.Finish() 318 } 319 320 //Finish checks that all mocked methods of the interface have been called at least once 321 //Deprecated: please use MinimockFinish or use Finish method of minimock.Controller 322 func (m *PulsarStorageMock) Finish() { 323 m.MinimockFinish() 324 } 325 326 //MinimockFinish checks that all mocked methods of the interface have been called at least once 327 func (m *PulsarStorageMock) MinimockFinish() { 328 329 if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 { 330 m.t.Fatal("Expected call to PulsarStorageMock.Close") 331 } 332 333 if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 { 334 m.t.Fatal("Expected call to PulsarStorageMock.GetLastPulse") 335 } 336 337 if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 { 338 m.t.Fatal("Expected call to PulsarStorageMock.SavePulse") 339 } 340 341 if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 { 342 m.t.Fatal("Expected call to PulsarStorageMock.SetLastPulse") 343 } 344 345 } 346 347 //Wait waits for all mocked methods to be called at least once 348 //Deprecated: please use MinimockWait or use Wait method of minimock.Controller 349 func (m *PulsarStorageMock) Wait(timeout time.Duration) { 350 m.MinimockWait(timeout) 351 } 352 353 //MinimockWait waits for all mocked methods to be called at least once 354 //this method is called by minimock.Controller 355 func (m *PulsarStorageMock) MinimockWait(timeout time.Duration) { 356 timeoutCh := time.After(timeout) 357 for { 358 ok := true 359 ok = ok && (m.CloseFunc == nil || atomic.LoadUint64(&m.CloseCounter) > 0) 360 ok = ok && (m.GetLastPulseFunc == nil || atomic.LoadUint64(&m.GetLastPulseCounter) > 0) 361 ok = ok && (m.SavePulseFunc == nil || atomic.LoadUint64(&m.SavePulseCounter) > 0) 362 ok = ok && (m.SetLastPulseFunc == nil || atomic.LoadUint64(&m.SetLastPulseCounter) > 0) 363 364 if ok { 365 return 366 } 367 368 select { 369 case <-timeoutCh: 370 371 if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 { 372 m.t.Error("Expected call to PulsarStorageMock.Close") 373 } 374 375 if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 { 376 m.t.Error("Expected call to PulsarStorageMock.GetLastPulse") 377 } 378 379 if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 { 380 m.t.Error("Expected call to PulsarStorageMock.SavePulse") 381 } 382 383 if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 { 384 m.t.Error("Expected call to PulsarStorageMock.SetLastPulse") 385 } 386 387 m.t.Fatalf("Some mocks were not called on time: %s", timeout) 388 return 389 default: 390 time.Sleep(time.Millisecond) 391 } 392 } 393 } 394 395 //AllMocksCalled returns true if all mocked methods were called before the execution of AllMocksCalled, 396 //it can be used with assert/require, i.e. require.True(mock.AllMocksCalled()) 397 func (m *PulsarStorageMock) AllMocksCalled() bool { 398 399 if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 { 400 return false 401 } 402 403 if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 { 404 return false 405 } 406 407 if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 { 408 return false 409 } 410 411 if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 { 412 return false 413 } 414 415 return true 416 }