fix: rename option method

Signed-off-by: Vladislav Sukhin <[email protected]>
kubeshop · Jul 23, 2024 · e0b39a4 · e0b39a4
1 parent cb97cd2
commit e0b39a4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/cmd/api-server/main.go b/cmd/api-server/main.go
@@ -292,9 +292,9 @@ func main() {
 		isDocDb := cfg.APIMongoDBType == storage.TypeDocDB
 		sequenceRepository := sequence.NewMongoRepository(db)
 		mongoResultsRepository := result.NewMongoRepository(db, cfg.APIMongoAllowDiskUse, isDocDb, result.WithFeatureFlags(features),
-			result.WithLogsClient(logGrpcClient), result.WithMongoRepositorySequenceCollection(sequenceRepository))
+			result.WithLogsClient(logGrpcClient), result.WithMongoRepositorySequence(sequenceRepository))
 		resultsRepository = mongoResultsRepository
-		testResultsRepository = testresult.NewMongoRepository(db, cfg.APIMongoAllowDiskUse, isDocDb)
+		testResultsRepository = testresult.NewMongoRepository(db, cfg.APIMongoAllowDiskUse, isDocDb, testresult.WithMongoRepositorySequence(sequenceRepository))
 		testWorkflowResultsRepository = testworkflow2.NewMongoRepository(db, cfg.APIMongoAllowDiskUse)
 		configRepository = configrepository.NewMongoRepository(db)
 		triggerLeaseBackend = triggers.NewMongoLeaseBackend(db)

diff --git a/pkg/repository/result/mongo.go b/pkg/repository/result/mongo.go
@@ -121,7 +121,7 @@ func WithMongoRepositoryResultCollection(collection *mongo.Collection) MongoRepo
 	}
 }
 
-func WithMongoRepositorySequenceCollection(sequenceRepository sequence.Repository) MongoRepositoryOpt {
+func WithMongoRepositorySequence(sequenceRepository sequence.Repository) MongoRepositoryOpt {
 	return func(r *MongoRepository) {
 		r.sequenceRepository = sequenceRepository
 	}

diff --git a/pkg/repository/testresult/mongo.go b/pkg/repository/testresult/mongo.go
@@ -51,7 +51,7 @@ func WithMongoRepositoryCollection(collection *mongo.Collection) MongoRepository
 	}
 }
 
-func WithMongoRepositorySequenceCollection(sequenceRepository sequence.Repository) MongoRepositoryOpt {
+func WithMongoRepositorySequence(sequenceRepository sequence.Repository) MongoRepositoryOpt {
 	return func(r *MongoRepository) {
 		r.sequenceRepository = sequenceRepository
 	}