mirror of https://github.com/artifacthub/hub.git
40 lines
1017 B
Go
40 lines
1017 B
Go
package notification
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/artifacthub/hub/internal/hub"
|
|
"github.com/jackc/pgx/v4"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
// ManagerMock is a mock implementation of the NotificationManager interface.
|
|
type ManagerMock struct {
|
|
mock.Mock
|
|
}
|
|
|
|
// Add implements the NotificationManager interface.
|
|
func (m *ManagerMock) Add(ctx context.Context, tx pgx.Tx, n *hub.Notification) error {
|
|
args := m.Called(ctx, tx, n)
|
|
return args.Error(0)
|
|
}
|
|
|
|
// GetPending implements the NotificationManager interface.
|
|
func (m *ManagerMock) GetPending(ctx context.Context, tx pgx.Tx) (*hub.Notification, error) {
|
|
args := m.Called(ctx, tx)
|
|
data, _ := args.Get(0).(*hub.Notification)
|
|
return data, args.Error(1)
|
|
}
|
|
|
|
// UpdateStatus implements the NotificationManager interface.
|
|
func (m *ManagerMock) UpdateStatus(
|
|
ctx context.Context,
|
|
tx pgx.Tx,
|
|
notificationID string,
|
|
processed bool,
|
|
processedErr error,
|
|
) error {
|
|
args := m.Called(ctx, tx, notificationID, processed, processedErr)
|
|
return args.Error(0)
|
|
}
|