Skip to content

Commit 0619dbc

Browse files
committed
fix: init in-memory files and shard repositories as defaults
Fixes: #238
1 parent dabbf9c commit 0619dbc

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

internal/environment.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,11 +189,16 @@ func NewEnvironment(env *Environment) (*Environment, error) {
189189
fileRepository = files.NewRepository(env.DB)
190190
shardRepository = shards.NewRepository(env.DB, env.Config.Sharding.Mappings)
191191
}
192-
193192
if env.SpannerClient != nil {
194193
fileRepository = files.NewSpannerRepository(env.SpannerClient)
195194
shardRepository = shards.NewSpannerRepository(env.SpannerClient, env.Config.Sharding.Mappings)
196195
}
196+
if fileRepository == nil {
197+
fileRepository = files.NewMockRepository()
198+
}
199+
if shardRepository == nil {
200+
shardRepository = shards.NewInMemoryRepository()
201+
}
197202

198203
fileReceiver, err := pipeline.Start(ctx, env.Logger, env.Config, shardRepository, fileRepository, httpSub)
199204
if err != nil {

internal/files/mock_repo_files.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,10 @@ type MockRepository struct {
66
Err error
77
}
88

9+
func NewMockRepository() Repository {
10+
return &MockRepository{}
11+
}
12+
913
func (r *MockRepository) Record(_ context.Context, file AcceptedFile) error {
1014
return r.Err
1115
}

0 commit comments

Comments
 (0)