@@ -527,18 +527,30 @@ func ConvertEnvDatabaseConfigToFaultInjectorConfig(envConfig EnvDatabaseConfig,
527527	randomPortOffset  :=  1  +  rand .Intn (10 ) // Random port offset to avoid conflicts 
528528
529529	// Build the database config for fault injector 
530+ 	// TODO: Make this configurable 
531+ 	// IT is the defaults for a sharded database at the moment 
530532	dbConfig  :=  DatabaseConfig {
531- 		Name :            name ,
532- 		Port :            port  +  randomPortOffset ,
533- 		MemorySize :      268435456 , // 256MB default 
534- 		Replication :     true ,
535- 		EvictionPolicy :  "noeviction" ,
536- 		ProxyPolicy :     "single" ,
537- 		AutoUpgrade :     true ,
538- 		Sharding :        true ,
539- 		ShardsCount :     2 ,
533+ 		Name :           name ,
534+ 		Port :           port  +  randomPortOffset ,
535+ 		MemorySize :     268435456 , // 256MB default 
536+ 		Replication :    true ,
537+ 		EvictionPolicy : "noeviction" ,
538+ 		ProxyPolicy :    "single" ,
539+ 		AutoUpgrade :    true ,
540+ 		Sharding :       true ,
541+ 		ShardsCount :    2 ,
542+ 		ShardKeyRegex : []ShardKeyRegexPattern {
543+ 			{Regex : ".*\\ {(?<tag>.*)\\ }.*" },
544+ 			{Regex : "(?<tag>.*)" },
545+ 		},
540546		ShardsPlacement : "dense" ,
541- 		OSSCluster :      false ,
547+ 		ModuleList : []DatabaseModule {
548+ 			{ModuleArgs : "" , ModuleName : "ReJSON" },
549+ 			{ModuleArgs : "" , ModuleName : "search" },
550+ 			{ModuleArgs : "" , ModuleName : "timeseries" },
551+ 			{ModuleArgs : "" , ModuleName : "bf" },
552+ 		},
553+ 		OSSCluster : false ,
542554	}
543555
544556	// If we have raw_endpoints with cluster info, configure for cluster 
0 commit comments