hub/internal/repo/mock.go

package repo

import (
	"context"

	"github.com/artifacthub/hub/internal/hub"
	"github.com/stretchr/testify/mock"
	"helm.sh/helm/v3/pkg/repo"
)

// ClonerMock is a mock implementation of the RepositoryCloner interface.
type ClonerMock struct {
	mock.Mock
}

// CloneRepository implements the RepositoryCloner interface.
func (m *ClonerMock) CloneRepository(ctx context.Context, r *hub.Repository) (string, string, error) {
	args := m.Called(ctx, r)
	return args.String(0), args.String(1), args.Error(2)
}

// ErrorsCollectorMock is mock ErrorsCollector implementation.
type ErrorsCollectorMock struct {
	mock.Mock
}

// Append implements the ErrorsCollector interface.
func (m *ErrorsCollectorMock) Append(repositoryID string, err string) {
	m.Called(repositoryID, err)
}

// Flush implements the ErrorsCollector interface.
func (m *ErrorsCollectorMock) Flush() {
	m.Called()
}

// Init implements the ErrorsCollector interface.
func (m *ErrorsCollectorMock) Init(repositoryID string) {
	m.Called(repositoryID)
}

// HelmIndexLoaderMock is a mock implementation of the HelmIndexLoader
// interface.
type HelmIndexLoaderMock struct {
	mock.Mock
}

// LoadIndex implements the HelmIndexLoader interface.
func (m *HelmIndexLoaderMock) LoadIndex(r *hub.Repository) (*repo.IndexFile, string, error) {
	args := m.Called(r)
	indexFile, _ := args.Get(0).(*repo.IndexFile)
	return indexFile, args.String(1), args.Error(2)
}

// ManagerMock is a mock implementation of the RepositoryManager interface.
type ManagerMock struct {
	mock.Mock
}

// Add implements the RepositoryManager interface.
func (m *ManagerMock) Add(ctx context.Context, orgName string, r *hub.Repository) error {
	args := m.Called(ctx, orgName, r)
	return args.Error(0)
}

// CheckAvailability implements the RepositoryManager interface.
func (m *ManagerMock) CheckAvailability(ctx context.Context, resourceKind, value string) (bool, error) {
	args := m.Called(ctx, resourceKind, value)
	return args.Bool(0), args.Error(1)
}

// ClaimOwnership implements the RepositoryManager interface.
func (m *ManagerMock) ClaimOwnership(ctx context.Context, name, orgName string) error {
	args := m.Called(ctx, name, orgName)
	return args.Error(0)
}

// Delete implements the RepositoryManager interface.
func (m *ManagerMock) Delete(ctx context.Context, name string) error {
	args := m.Called(ctx, name)
	return args.Error(0)
}

// GetByID implements the RepositoryManager interface.
func (m *ManagerMock) GetByID(
	ctx context.Context,
	repositoryID string,
	includeCredentials bool,
) (*hub.Repository, error) {
	args := m.Called(ctx, repositoryID, includeCredentials)
	data, _ := args.Get(0).(*hub.Repository)
	return data, args.Error(1)
}

// GetByName implements the RepositoryManager interface.
func (m *ManagerMock) GetByName(
	ctx context.Context,
	name string,
	includeCredentials bool,
) (*hub.Repository, error) {
	args := m.Called(ctx, name, includeCredentials)
	data, _ := args.Get(0).(*hub.Repository)
	return data, args.Error(1)
}

// GetMetadata implements the RepositoryManager interface.
func (m *ManagerMock) GetMetadata(r *hub.Repository, basePath string) (*hub.RepositoryMetadata, error) {
	args := m.Called(r, basePath)
	data, _ := args.Get(0).(*hub.RepositoryMetadata)
	return data, args.Error(1)
}

// GetPackagesDigest implements the RepositoryManager interface.
func (m *ManagerMock) GetPackagesDigest(
	ctx context.Context,
	repositoryID string,
) (map[string]string, error) {
	args := m.Called(ctx, repositoryID)
	data, _ := args.Get(0).(map[string]string)
	return data, args.Error(1)
}

// GetRemoteDigest implements the RepositoryManager interface.
func (m *ManagerMock) GetRemoteDigest(ctx context.Context, r *hub.Repository) (string, error) {
	args := m.Called(ctx, r)
	return args.String(0), args.Error(1)
}

// Search implements the RepositoryManager interface.
func (m *ManagerMock) Search(
	ctx context.Context,
	input *hub.SearchRepositoryInput,
) (*hub.SearchRepositoryResult, error) {
	args := m.Called(ctx, input)
	data, _ := args.Get(0).(*hub.SearchRepositoryResult)
	return data, args.Error(1)
}

// SearchJSON implements the RepositoryManager interface.
func (m *ManagerMock) SearchJSON(
	ctx context.Context,
	input *hub.SearchRepositoryInput,
) (*hub.JSONQueryResult, error) {
	args := m.Called(ctx, input)
	data, _ := args.Get(0).(*hub.JSONQueryResult)
	return data, args.Error(1)
}

// SetLastScanningResults implements the RepositoryManager interface.
func (m *ManagerMock) SetLastScanningResults(ctx context.Context, repositoryID, errs string) error {
	args := m.Called(ctx, repositoryID, errs)
	return args.Error(0)
}

// SetLastTrackingResults implements the RepositoryManager interface.
func (m *ManagerMock) SetLastTrackingResults(ctx context.Context, repositoryID, errs string) error {
	args := m.Called(ctx, repositoryID, errs)
	return args.Error(0)
}

// SetVerifiedPublisher implements the RepositoryManager interface.
func (m *ManagerMock) SetVerifiedPublisher(ctx context.Context, repositoryID string, verified bool) error {
	args := m.Called(ctx, repositoryID, verified)
	return args.Error(0)
}

// Transfer implements the RepositoryManager interface.
func (m *ManagerMock) Transfer(ctx context.Context, name, orgName string, ownershipClaim bool) error {
	args := m.Called(ctx, name, orgName, ownershipClaim)
	return args.Error(0)
}

// Update implements the RepositoryManager interface.
func (m *ManagerMock) Update(ctx context.Context, r *hub.Repository) error {
	args := m.Called(ctx, r)
	return args.Error(0)
}

// UpdateDigest implements the RepositoryManager interface.
func (m *ManagerMock) UpdateDigest(ctx context.Context, repositoryID, digest string) error {
	args := m.Called(ctx, repositoryID, digest)
	return args.Error(0)
}

// OLMOCIExporterMock is a mock implementation of the OLMOCIExporter interface.
type OLMOCIExporterMock struct {
	mock.Mock
}

// ExportRepository implements the OLMOCIExporter interface.
func (m *OLMOCIExporterMock) ExportRepository(ctx context.Context, r *hub.Repository) (string, error) {
	args := m.Called(ctx, r)
	return args.String(0), args.Error(1)
}