rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-05-18 05:30:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixup: massage imports for tests

Browse Source
This commit is contained in:
Vlad Lazar
2025-07-25 19:09:06 +01:00
parent d5cd2de3cf
commit 07396d3fc7
3 changed files with 29 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
storage_controller/src/hadron_queries.rs
View File

@@ -394,8 +394,7 @@ async fn get_pageserver_connection_info(
None => {
// This is never supposed to happen because `SELECT min()` should always return one row.
return Err(DatabaseError::Logical(format!(
"Unexpected empty query result for min(shard_count) query. Tenant ID {}",
tenant_id
"Unexpected empty query result for min(shard_count) query. Tenant ID {tenant_id}"
)));
}
};
@@ -487,9 +486,9 @@ mod test {
fn get_test_sk_node(id: u64) -> SafeKeeperNode {
SafeKeeperNode::new(
NodeId(id),
format!("safekeeper-{}", id),
format!("safekeeper-{id}"),
123,
format!("safekeeper-{}", id),
format!("safekeeper-{id}"),
456,
)
}
@@ -501,13 +500,13 @@ mod test {
let connection_string = pg.settings().url("test");
{
let mut conn = PgConnection::establish(&connection_string)
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
run_migrations(&mut conn).unwrap();
}
let mut connection = AsyncPgConnection::establish(&connection_string)
.await
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
execute_sk_upsert(&mut connection, get_test_sk_node(0).to_database_row())
.await
@@ -565,13 +564,13 @@ mod test {
let connection_string = pg.settings().url("test");
{
let mut conn = PgConnection::establish(&connection_string)
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
run_migrations(&mut conn).unwrap();
}
let mut connection = AsyncPgConnection::establish(&connection_string)
.await
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
// An initial call should insert the timeline safekeepers and return the inserted values.
let timeline1_id = TimelineId::generate();
@@ -629,13 +628,13 @@ mod test {
let connection_string = pg.settings().url("test");
{
let mut conn = PgConnection::establish(&connection_string)
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
run_migrations(&mut conn).unwrap();
}
let mut connection = AsyncPgConnection::establish(&connection_string)
.await
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
// Insert some values
let timeline1_id = TimelineId::generate();
@@ -718,12 +717,12 @@ mod test {
let connection_string = pg.settings().url("test");
{
let mut conn = PgConnection::establish(&connection_string)
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
run_migrations(&mut conn).unwrap();
}
let mut connection = AsyncPgConnection::establish(&connection_string)
.await
.unwrap_or_else(|_| panic!("Error connecting to {}", connection_string));
.unwrap_or_else(|_| panic!("Error connecting to {connection_string}"));
// Insert some values
let safekeeper_ids = vec![
@@ -809,7 +808,7 @@ mod test {
.safekeeper_peers
.push(TimelineSafekeeperPeer {
node_id: NodeId(i),
listen_http_addr: format!("safekeeper-{}", i),
listen_http_addr: format!("safekeeper-{i}"),
http_port: 123,
});
if i < 3 {
@@ -817,7 +816,7 @@ mod test {
.safekeeper_peers
.push(TimelineSafekeeperPeer {
node_id: NodeId(i),
listen_http_addr: format!("safekeeper-{}", i),
listen_http_addr: format!("safekeeper-{i}"),
http_port: 123,
});
expected_responses[3]