fix: init in-memory files and shard repositories as defaults

Fixes: #238
moov-io · Aug 21, 2024 · 0619dbc · 0619dbc
1 parent dabbf9c
commit 0619dbc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/internal/environment.go b/internal/environment.go
@@ -189,11 +189,16 @@ func NewEnvironment(env *Environment) (*Environment, error) {
 		fileRepository = files.NewRepository(env.DB)
 		shardRepository = shards.NewRepository(env.DB, env.Config.Sharding.Mappings)
 	}
-
 	if env.SpannerClient != nil {
 		fileRepository = files.NewSpannerRepository(env.SpannerClient)
 		shardRepository = shards.NewSpannerRepository(env.SpannerClient, env.Config.Sharding.Mappings)
 	}
+	if fileRepository == nil {
+		fileRepository = files.NewMockRepository()
+	}
+	if shardRepository == nil {
+		shardRepository = shards.NewInMemoryRepository()
+	}
 
 	fileReceiver, err := pipeline.Start(ctx, env.Logger, env.Config, shardRepository, fileRepository, httpSub)
 	if err != nil {

diff --git a/internal/files/mock_repo_files.go b/internal/files/mock_repo_files.go
@@ -6,6 +6,10 @@ type MockRepository struct {
 	Err error
 }
 
+func NewMockRepository() Repository {
+	return &MockRepository{}
+}
+
 func (r *MockRepository) Record(_ context.Context, file AcceptedFile) error {
 	return r.Err
 }