Skip to content

Commit b7ae627

Browse files
fix: correct partition names for dynamic partitioning
- Updates `DROP TABLE` and `INSERT INTO` scripts in `tests/integration_test.rs` to target `objects_s_below_100000` and `objects_s_above_100000`. - Matches the native PostgreSQL partition naming convention established by `ai_infra::divider_sql`, resolving `relation does not exist` test failures during teardown/insertion. Co-authored-by: soniapi <396009+soniapi@users.noreply.github.com>
1 parent 0481e2a commit b7ae627

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

tests/integration_test.rs

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -165,34 +165,34 @@ async fn test_end_to_end_sequence() {
165165
let divide_s = helpers::prompt_cutoff_value().unwrap_or(180000.0);
166166

167167
// Detach the default 100000 partitions to keep the data safe before dropping target partitions
168-
sql_query("ALTER TABLE objects_s DETACH PARTITION objects_s_100000_below;")
168+
sql_query("ALTER TABLE objects_s DETACH PARTITION objects_s_below_100000;")
169169
.execute(&mut connection)
170170
.unwrap();
171-
sql_query("ALTER TABLE objects_s DETACH PARTITION objects_s_100000_above;")
171+
sql_query("ALTER TABLE objects_s DETACH PARTITION objects_s_above_100000;")
172172
.execute(&mut connection)
173173
.unwrap();
174174

175-
let drop_below = format!("DROP TABLE IF EXISTS objects_s_{}_below CASCADE;", divide_s as i64);
176-
let drop_above = format!("DROP TABLE IF EXISTS objects_s_{}_above CASCADE;", divide_s as i64);
175+
let drop_below = format!("DROP TABLE IF EXISTS objects_s_below_{} CASCADE;", divide_s as i64);
176+
let drop_above = format!("DROP TABLE IF EXISTS objects_s_above_{} CASCADE;", divide_s as i64);
177177

178178
sql_query(drop_below).execute(&mut connection).unwrap();
179179
sql_query(drop_above).execute(&mut connection).unwrap();
180180

181181
divider(&mut connection, divide_s);
182182

183183
// Insert the data back into the dynamically created partitions
184-
sql_query("INSERT INTO objects_s SELECT * FROM objects_s_100000_below;")
184+
sql_query("INSERT INTO objects_s SELECT * FROM objects_s_below_100000;")
185185
.execute(&mut connection)
186186
.unwrap();
187-
sql_query("INSERT INTO objects_s SELECT * FROM objects_s_100000_above;")
187+
sql_query("INSERT INTO objects_s SELECT * FROM objects_s_above_100000;")
188188
.execute(&mut connection)
189189
.unwrap();
190190

191191
// Now it's safe to drop the old detached tables
192-
sql_query("DROP TABLE IF EXISTS objects_s_100000_below;")
192+
sql_query("DROP TABLE IF EXISTS objects_s_below_100000;")
193193
.execute(&mut connection)
194194
.unwrap();
195-
sql_query("DROP TABLE IF EXISTS objects_s_100000_above;")
195+
sql_query("DROP TABLE IF EXISTS objects_s_above_100000;")
196196
.execute(&mut connection)
197197
.unwrap();
198198

0 commit comments

Comments
 (0)