Fix metadata repository mock to respect metadata.Repository interface

Signed-off-by: Igor Shishkin <me@teran.dev>
teran · Jul 10, 2024 · 5c55f5f · 5c55f5f
1 parent fe27eae
commit 5c55f5f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/repositories/metadata/mock/mock.go b/repositories/metadata/mock/mock.go
@@ -4,8 +4,11 @@ import (
 	"context"
 
 	"github.com/stretchr/testify/mock"
+	"github.com/teran/archived/repositories/metadata"
 )
 
+var _ metadata.Repository = (*Mock)(nil)
+
 type Mock struct {
 	mock.Mock
 }
@@ -73,3 +76,8 @@ func (m *Mock) CreateBLOB(_ context.Context, checksum string, size uint64, mimeT
 	args := m.Called(checksum, size, mimeType)
 	return args.Error(0)
 }
+
+func (m *Mock) GetBlobKeyByObject(_ context.Context, container, version, key string) (string, error) {
+	args := m.Called(container, version, key)
+	return args.String(0), args.Error(1)
+}