Add store mock

xataio · Oct 2, 2024 · fb7ed79 · fb7ed79
1 parent ccf445a
commit fb7ed79
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/pkg/snapshot/store/mocks/mock_snapshot_store.go b/pkg/snapshot/store/mocks/mock_snapshot_store.go
@@ -0,0 +1,25 @@
+// SPDX-License-Identifier: Apache-2.0
+
+package mocks
+
+import (
+	"context"
+
+	"github.com/xataio/pgstream/pkg/snapshot"
+)
+
+type Store struct {
+	CreateSnapshotRequestFn func(context.Context, *snapshot.Snapshot) error
+	UpdateSnapshotRequestFn func(context.Context, *snapshot.Snapshot) error
+	GetSnapshotRequestsFn   func(ctx context.Context, status snapshot.Status) ([]*snapshot.Snapshot, error)
+}
+
+func (m *Store) CreateSnapshotRequest(ctx context.Context, s *snapshot.Snapshot) error {
+	return m.CreateSnapshotRequestFn(ctx, s)
+}
+func (m *Store) UpdateSnapshotRequest(ctx context.Context, s *snapshot.Snapshot) error {
+	return m.UpdateSnapshotRequestFn(ctx, s)
+}
+func (m *Store) GetSnapshotRequests(ctx context.Context, status snapshot.Status) ([]*snapshot.Snapshot, error) {
+	return m.GetSnapshotRequestsFn(ctx, status)
+}