diff options
author | James Liu <jliu@gitlab.com> | 2023-12-13 09:03:58 +0300 |
---|---|---|
committer | James Liu <jliu@gitlab.com> | 2024-01-15 05:40:54 +0300 |
commit | 91e9c2e1e0c3f26db2b3a8b2acf19082c0b79bd8 (patch) | |
tree | e2124b96e7116127b6b2beb7b7124c5c7f3bfbc6 /internal/backup/pipeline_test.go | |
parent | ada54fc25380ad5b6844f46df793a62075b4bb31 (diff) |
backup: Add ListRepositories to the strategy
Adds a new method to the Strategy interface used by regular and
server-side backups for performing repository backups and restores. This
new method calls the internal WalkRepos() RPC to fetch a list of repos
in a given storage.
Diffstat (limited to 'internal/backup/pipeline_test.go')
-rw-r--r-- | internal/backup/pipeline_test.go | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/internal/backup/pipeline_test.go b/internal/backup/pipeline_test.go index cd77fb5a7..22dd65a83 100644 --- a/internal/backup/pipeline_test.go +++ b/internal/backup/pipeline_test.go @@ -126,6 +126,7 @@ type MockStrategy struct { RestoreFunc func(context.Context, *RestoreRequest) error RemoveAllRepositoriesFunc func(context.Context, *RemoveAllRepositoriesRequest) error RemoveRepositoryFunc func(context.Context, *RemoveRepositoryRequest) error + ListRepositoriesFunc func(context.Context, *ListRepositoriesRequest) ([]*gitalypb.Repository, error) } func (s MockStrategy) Create(ctx context.Context, req *CreateRequest) error { @@ -156,6 +157,13 @@ func (s MockStrategy) RemoveRepository(ctx context.Context, req *RemoveRepositor return nil } +func (s MockStrategy) ListRepositories(ctx context.Context, req *ListRepositoriesRequest) ([]*gitalypb.Repository, error) { + if s.ListRepositoriesFunc != nil { + return s.ListRepositoriesFunc(ctx, req) + } + return nil, nil +} + func testPipeline(t *testing.T, init func() *Pipeline) { strategy := MockStrategy{ CreateFunc: func(_ context.Context, req *CreateRequest) error { |