[sim][sled agent] Register storage after sled

sled-agent/src/sim/server.rs

@@ -9,6 +9,7 @@
 use super::config::Config;
 use super::http_entrypoints::api as http_api;
 use super::sled_agent::SledAgent;
+use crucible_agent_client::types::State as RegionState;
 
 use nexus_client::Client as NexusClient;
 use omicron_common::backoff::{
@@ -66,16 +67,6 @@ impl Server {
         .map_err(|error| format!("initializing server: {}", error))?
         .start();
 
-        // Create all the Zpools requested by the config, and allocate a single
-        // Crucible dataset for each. This emulates the setup we expect to have
-        // on the physical rack.
-        for zpool in &config.storage.zpools {
-            let zpool_id = uuid::Uuid::new_v4();
-            sled_agent.create_zpool(zpool_id, zpool.size).await;
-            let dataset_id = uuid::Uuid::new_v4();
-            sled_agent.create_crucible_dataset(zpool_id, dataset_id).await;
-        }
-
         /*
          * Notify the control plane that we're up, and continue trying this
          * until it succeeds. We retry with an randomized, capped exponential
@@ -108,6 +99,25 @@ impl Server {
         )
         .await
         .expect("Expected an infinite retry loop contacting Nexus");
+
+        // Create all the Zpools requested by the config, and allocate a single
+        // Crucible dataset for each. This emulates the setup we expect to have
+        // on the physical rack.
+        for zpool in &config.storage.zpools {
+            let zpool_id = uuid::Uuid::new_v4();
+            sled_agent.create_zpool(zpool_id, zpool.size).await;
+            let dataset_id = uuid::Uuid::new_v4();
+            sled_agent.create_crucible_dataset(zpool_id, dataset_id).await;
+
+            // Whenever Nexus tries to allocate a region, it should complete
+            // immediately. What efficiency!
+            let crucible =
+                sled_agent.get_crucible_dataset(zpool_id, dataset_id).await;
+            crucible
+                .set_create_callback(Box::new(|_| RegionState::Created))
+                .await;
+        }
+
         Ok(Server { sled_agent, http_server })
     }