Skip to content

Commit 05b2d5a

Browse files
authored
[nexus-test-interface] make get_*_address non-async (#9431)
No need for these to be async, they just return values from memory.
1 parent ef9510d commit 05b2d5a

File tree

11 files changed

+33
-40
lines changed

11 files changed

+33
-40
lines changed

dev-tools/omicron-dev/src/main.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,11 +114,11 @@ impl RunAllArgs {
114114
println!("omicron-dev: nexus external API: {:?}", addr);
115115
println!(
116116
"omicron-dev: nexus internal API: {:?}",
117-
cptestctx.server.get_http_server_internal_address().await,
117+
cptestctx.server.get_http_server_internal_address(),
118118
);
119119
println!(
120120
"omicron-dev: nexus lockstep API: {:?}",
121-
cptestctx.server.get_http_server_lockstep_address().await,
121+
cptestctx.server.get_http_server_lockstep_address(),
122122
);
123123
println!(
124124
"omicron-dev: cockroachdb pid: {}",

nexus/src/app/mod.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -836,7 +836,7 @@ impl Nexus {
836836
Ok(())
837837
}
838838

839-
pub(crate) async fn get_external_server_address(
839+
pub(crate) fn get_external_server_address(
840840
&self,
841841
) -> Option<std::net::SocketAddr> {
842842
self.external_server
@@ -846,7 +846,7 @@ impl Nexus {
846846
.map(|server| server.local_addr())
847847
}
848848

849-
pub(crate) async fn get_techport_server_address(
849+
pub(crate) fn get_techport_server_address(
850850
&self,
851851
) -> Option<std::net::SocketAddr> {
852852
self.techport_external_server
@@ -856,7 +856,7 @@ impl Nexus {
856856
.map(|server| server.local_addr())
857857
}
858858

859-
pub(crate) async fn get_internal_server_address(
859+
pub(crate) fn get_internal_server_address(
860860
&self,
861861
) -> Option<std::net::SocketAddr> {
862862
self.internal_server
@@ -866,7 +866,7 @@ impl Nexus {
866866
.map(|server| server.local_addr())
867867
}
868868

869-
pub(crate) async fn get_lockstep_server_address(
869+
pub(crate) fn get_lockstep_server_address(
870870
&self,
871871
) -> Option<std::net::SocketAddr> {
872872
self.lockstep_server

nexus/src/app/quiesce.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -631,7 +631,7 @@ mod test {
631631
let log = &cptestctx.logctx.log;
632632
let nexus_lockstep_url = format!(
633633
"http://{}",
634-
cptestctx.server.get_http_server_lockstep_address().await
634+
cptestctx.server.get_http_server_lockstep_address(),
635635
);
636636
let nexus_client = nexus_lockstep_client::Client::new(
637637
&nexus_lockstep_url,
@@ -663,7 +663,7 @@ mod test {
663663
let log = &cptestctx.logctx.log;
664664
let nexus_lockstep_url = format!(
665665
"http://{}",
666-
cptestctx.server.get_http_server_lockstep_address().await
666+
cptestctx.server.get_http_server_lockstep_address(),
667667
);
668668
let nexus_client = nexus_lockstep_client::Client::new(
669669
&nexus_lockstep_url,

nexus/src/lib.rs

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -418,20 +418,20 @@ impl nexus_test_interface::NexusServer for Server {
418418
self.apictx.context.nexus.inventory_load_rx()
419419
}
420420

421-
async fn get_http_server_external_address(&self) -> SocketAddr {
422-
self.apictx.context.nexus.get_external_server_address().await.unwrap()
421+
fn get_http_server_external_address(&self) -> SocketAddr {
422+
self.apictx.context.nexus.get_external_server_address().unwrap()
423423
}
424424

425-
async fn get_http_server_techport_address(&self) -> SocketAddr {
426-
self.apictx.context.nexus.get_techport_server_address().await.unwrap()
425+
fn get_http_server_techport_address(&self) -> SocketAddr {
426+
self.apictx.context.nexus.get_techport_server_address().unwrap()
427427
}
428428

429-
async fn get_http_server_internal_address(&self) -> SocketAddr {
430-
self.apictx.context.nexus.get_internal_server_address().await.unwrap()
429+
fn get_http_server_internal_address(&self) -> SocketAddr {
430+
self.apictx.context.nexus.get_internal_server_address().unwrap()
431431
}
432432

433-
async fn get_http_server_lockstep_address(&self) -> SocketAddr {
434-
self.apictx.context.nexus.get_lockstep_server_address().await.unwrap()
433+
fn get_http_server_lockstep_address(&self) -> SocketAddr {
434+
self.apictx.context.nexus.get_lockstep_server_address().unwrap()
435435
}
436436

437437
async fn upsert_test_dataset(

nexus/test-interface/src/lib.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -88,10 +88,10 @@ pub trait NexusServer: Send + Sync + 'static {
8888

8989
fn inventory_load_rx(&self) -> watch::Receiver<Option<Arc<Collection>>>;
9090

91-
async fn get_http_server_external_address(&self) -> SocketAddr;
92-
async fn get_http_server_techport_address(&self) -> SocketAddr;
93-
async fn get_http_server_internal_address(&self) -> SocketAddr;
94-
async fn get_http_server_lockstep_address(&self) -> SocketAddr;
91+
fn get_http_server_external_address(&self) -> SocketAddr;
92+
fn get_http_server_techport_address(&self) -> SocketAddr;
93+
fn get_http_server_internal_address(&self) -> SocketAddr;
94+
fn get_http_server_lockstep_address(&self) -> SocketAddr;
9595

9696
// Previously, as a dataset was created (within the sled agent),
9797
// we'd use an internal API from Nexus to record that the dataset

nexus/test-utils/src/lib.rs

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1171,14 +1171,11 @@ impl<'a, N: NexusServer> ControlPlaneTestContextBuilder<'a, N> {
11711171
)
11721172
.await;
11731173

1174-
let external_server_addr =
1175-
server.get_http_server_external_address().await;
1174+
let external_server_addr = server.get_http_server_external_address();
11761175
let techport_external_server_addr =
1177-
server.get_http_server_techport_address().await;
1178-
let internal_server_addr =
1179-
server.get_http_server_internal_address().await;
1180-
let lockstep_server_addr =
1181-
server.get_http_server_lockstep_address().await;
1176+
server.get_http_server_techport_address();
1177+
let internal_server_addr = server.get_http_server_internal_address();
1178+
let lockstep_server_addr = server.get_http_server_lockstep_address();
11821179
let testctx_external = ClientTestContext::new(
11831180
external_server_addr,
11841181
self.logctx

nexus/tests/integration_tests/demo_saga.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ async fn test_demo_saga(cptestctx: &ControlPlaneTestContext) {
2323
let log = &cptestctx.logctx.log;
2424
let nexus_lockstep_url = format!(
2525
"http://{}",
26-
cptestctx.server.get_http_server_lockstep_address().await
26+
cptestctx.server.get_http_server_lockstep_address(),
2727
);
2828
let nexus_client =
2929
nexus_lockstep_client::Client::new(&nexus_lockstep_url, log.clone());

nexus/tests/integration_tests/instances.rs

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1129,8 +1129,7 @@ async fn test_instance_migration_compatible_cpu_platforms(
11291129
// Set up a second sled-agent representing a sled with a Turin processor.
11301130
// The instance itself requires only Milan, so it should be able to migrate
11311131
// both directions.
1132-
let nexus_address =
1133-
cptestctx.server.get_http_server_internal_address().await;
1132+
let nexus_address = cptestctx.server.get_http_server_internal_address();
11341133

11351134
let config = omicron_sled_agent::sim::Config::for_testing(
11361135
SledUuid::new_v4(),
@@ -1319,8 +1318,7 @@ async fn test_instance_migration_incompatible_cpu_platforms(
13191318

13201319
// Set up a second sled-agent representing a sled with a Turin processor.
13211320
// The instance will require Turin, so it will be placed here.
1322-
let nexus_address =
1323-
cptestctx.server.get_http_server_internal_address().await;
1321+
let nexus_address = cptestctx.server.get_http_server_internal_address();
13241322

13251323
let config = omicron_sled_agent::sim::Config::for_testing(
13261324
SledUuid::new_v4(),
@@ -1397,8 +1395,7 @@ async fn test_instance_migration_unknown_sled_type(
13971395

13981396
// Set up a second sled-agent representing a sled with unknown processor
13991397
// type. We won't be able to migrate to (or from) here.
1400-
let nexus_address =
1401-
cptestctx.server.get_http_server_internal_address().await;
1398+
let nexus_address = cptestctx.server.get_http_server_internal_address();
14021399

14031400
let config = omicron_sled_agent::sim::Config::for_testing(
14041401
SledUuid::new_v4(),
@@ -6737,8 +6734,7 @@ async fn test_can_start_instance_with_cpu_platform(
67376734
1
67386735
);
67396736

6740-
let nexus_address =
6741-
cptestctx.server.get_http_server_internal_address().await;
6737+
let nexus_address = cptestctx.server.get_http_server_internal_address();
67426738

67436739
let config = omicron_sled_agent::sim::Config::for_testing(
67446740
SledUuid::new_v4(),

nexus/tests/integration_tests/oximeter.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ async fn test_oximeter_reregistration() {
226226
// Restart the producer, and verify that we have _more_ data than before
227227
// Set up a test metric producer server
228228
context.producer = nexus_test_utils::start_producer_server(
229-
context.server.get_http_server_internal_address().await,
229+
context.server.get_http_server_internal_address(),
230230
nexus_test_utils::PRODUCER_UUID.parse().unwrap(),
231231
)
232232
.expect("Failed to restart metric producer server");
@@ -303,7 +303,7 @@ async fn test_oximeter_reregistration() {
303303
// Restart oximeter again, and verify that we have even more new data.
304304
context.oximeter = nexus_test_utils::start_oximeter(
305305
context.logctx.log.new(o!("component" => "oximeter")),
306-
context.server.get_http_server_internal_address().await,
306+
context.server.get_http_server_internal_address(),
307307
context.clickhouse.native_address().port(),
308308
oximeter_id,
309309
)

nexus/tests/integration_tests/quiesce.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ async fn test_quiesce(cptestctx: &ControlPlaneTestContext) {
3030
let opctx = OpContext::for_tests(log.clone(), datastore.clone());
3131
let nexus_lockstep_url = format!(
3232
"http://{}",
33-
cptestctx.server.get_http_server_lockstep_address().await
33+
cptestctx.server.get_http_server_lockstep_address(),
3434
);
3535
let nexus_client =
3636
nexus_lockstep_client::Client::new(&nexus_lockstep_url, log.clone());

0 commit comments

Comments
 (0)