File size: 1,143 Bytes
d6f631f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | package adapter
import (
"context"
"github.com/stretchr/testify/mock"
"github.com/openmeterio/openmeter/openmeter/progressmanager"
progressmanagerentity "github.com/openmeterio/openmeter/openmeter/progressmanager/entity"
)
var _ progressmanager.Service = &MockProgressManager{}
func NewMockProgressManager() *MockProgressManager {
return &MockProgressManager{}
}
// MockProgressManager is a mock for the ProgressManager
type MockProgressManager struct {
mock.Mock
}
func (m *MockProgressManager) GetProgress(ctx context.Context, input progressmanagerentity.GetProgressInput) (*progressmanagerentity.Progress, error) {
args := m.Called(ctx, input)
result := args.Get(0)
if result == nil {
return nil, args.Error(1)
}
return result.(*progressmanagerentity.Progress), args.Error(1)
}
func (m *MockProgressManager) DeleteProgressByRuntimeID(ctx context.Context, runtimeID string) error {
args := m.Called(ctx, runtimeID)
return args.Error(0)
}
func (m *MockProgressManager) UpsertProgress(ctx context.Context, input progressmanagerentity.UpsertProgressInput) error {
args := m.Called(ctx, input)
return args.Error(0)
}
|