@@ -175,8 +175,8 @@ func (s MongoClickhouseSuite) Test_Simple_Flow_Partitioned_StringID() {
175175 adminClient := s .Source ().(* MongoSource ).AdminClient ()
176176 collection := adminClient .Database (srcDatabase ).Collection (srcTable )
177177
178- docs := make ([]any , 1000 )
179- for i := range 1000 {
178+ docs := make ([]any , 100 )
179+ for i := range 100 {
180180 docs [i ] = bson.D {
181181 {Key : "_id" , Value : fmt .Sprintf ("id-%05d" , i )},
182182 {Key : fmt .Sprintf ("init_key_%d" , i ), Value : fmt .Sprintf ("init_value_%d" , i )},
@@ -196,7 +196,7 @@ func (s MongoClickhouseSuite) Test_Simple_Flow_Partitioned_StringID() {
196196 require .NoError (t , catalogPool .QueryRow (t .Context (),
197197 `SELECT COUNT(*) FROM peerdb_stats.qrep_partitions WHERE parent_mirror_name = $1` ,
198198 flowConnConfig .FlowJobName ).Scan (& partitionCount ))
199- require .Greater (t , partitionCount , 1 , "expected multiple partitions for 1000 string-id rows" )
199+ require .Greater (t , partitionCount , 1 , "expected multiple partitions for 100 string-id rows" )
200200
201201 SetupCDCFlowStatusQuery (t , env , flowConnConfig )
202202 cdcDocs := make ([]any , 10 )
@@ -232,8 +232,8 @@ func (s MongoClickhouseSuite) Test_Simple_Flow_Partitioned_NumericID() {
232232 adminClient := s .Source ().(* MongoSource ).AdminClient ()
233233 collection := adminClient .Database (srcDatabase ).Collection (srcTable )
234234
235- docs := make ([]any , 1000 )
236- for i := range 1000 {
235+ docs := make ([]any , 100 )
236+ for i := range 100 {
237237 docs [i ] = bson.D {
238238 {Key : "_id" , Value : int32 (i + 1 )},
239239 {Key : fmt .Sprintf ("init_key_%d" , i ), Value : fmt .Sprintf ("init_value_%d" , i )},
@@ -253,7 +253,7 @@ func (s MongoClickhouseSuite) Test_Simple_Flow_Partitioned_NumericID() {
253253 require .NoError (t , catalogPool .QueryRow (t .Context (),
254254 `SELECT COUNT(*) FROM peerdb_stats.qrep_partitions WHERE parent_mirror_name = $1` ,
255255 flowConnConfig .FlowJobName ).Scan (& partitionCount ))
256- require .Greater (t , partitionCount , 1 , "expected multiple partitions for 1000 numeric-id rows" )
256+ require .Greater (t , partitionCount , 1 , "expected multiple partitions for 100 numeric-id rows" )
257257
258258 SetupCDCFlowStatusQuery (t , env , flowConnConfig )
259259 cdcDocs := make ([]any , 10 )
0 commit comments