| // Code generated by mockery v1.0.0. DO NOT EDIT. |
| |
| package mocks |
| |
| import ( |
| context "context" |
| |
| diff "go.skia.org/infra/golden/go/diff" |
| |
| mock "github.com/stretchr/testify/mock" |
| |
| types "go.skia.org/infra/golden/go/types" |
| ) |
| |
| // FailureStore is an autogenerated mock type for the FailureStore type |
| type FailureStore struct { |
| mock.Mock |
| } |
| |
| // AddDigestFailure provides a mock function with given fields: ctx, failure |
| func (_m *FailureStore) AddDigestFailure(ctx context.Context, failure *diff.DigestFailure) error { |
| ret := _m.Called(ctx, failure) |
| |
| var r0 error |
| if rf, ok := ret.Get(0).(func(context.Context, *diff.DigestFailure) error); ok { |
| r0 = rf(ctx, failure) |
| } else { |
| r0 = ret.Error(0) |
| } |
| |
| return r0 |
| } |
| |
| // PurgeDigestFailures provides a mock function with given fields: ctx, digests |
| func (_m *FailureStore) PurgeDigestFailures(ctx context.Context, digests types.DigestSlice) error { |
| ret := _m.Called(ctx, digests) |
| |
| var r0 error |
| if rf, ok := ret.Get(0).(func(context.Context, types.DigestSlice) error); ok { |
| r0 = rf(ctx, digests) |
| } else { |
| r0 = ret.Error(0) |
| } |
| |
| return r0 |
| } |
| |
| // UnavailableDigests provides a mock function with given fields: ctx |
| func (_m *FailureStore) UnavailableDigests(ctx context.Context) (map[types.Digest]*diff.DigestFailure, error) { |
| ret := _m.Called(ctx) |
| |
| var r0 map[types.Digest]*diff.DigestFailure |
| if rf, ok := ret.Get(0).(func(context.Context) map[types.Digest]*diff.DigestFailure); ok { |
| r0 = rf(ctx) |
| } else { |
| if ret.Get(0) != nil { |
| r0 = ret.Get(0).(map[types.Digest]*diff.DigestFailure) |
| } |
| } |
| |
| var r1 error |
| if rf, ok := ret.Get(1).(func(context.Context) error); ok { |
| r1 = rf(ctx) |
| } else { |
| r1 = ret.Error(1) |
| } |
| |
| return r0, r1 |
| } |