test: Ask ingesters to persist before asking catalog for their partition keys

test_helpers_end_to_end/src/mini_cluster.rs

@@ -563,16 +563,20 @@ impl MiniCluster {
         Ok(IngesterResponse { partitions })
     }
 
-    /// Ask all of the ingesters to persist their data.
+    /// Ask all of the ingesters to persist their data for the cluster namespace.
     pub async fn persist_ingesters(&self) {
+        self.persist_ingesters_by_namespace(None).await;
+    }
+
+    /// Ask all of the ingesters to persist their data for a specified namespace, or the cluster
+    /// namespace if none specified.
+    pub async fn persist_ingesters_by_namespace(&self, namespace: Option<String>) {
+        let namespace = namespace.unwrap_or_else(|| self.namespace().into());
         for ingester in &self.ingesters {
             let mut ingester_client =
                 influxdb_iox_client::ingester::Client::new(ingester.ingester_grpc_connection());
 
-            ingester_client
+            ingester_client.persist(namespace.clone()).await.unwrap();
-                .persist(self.namespace().into())
-                .await
-                .unwrap();
         }
     }
 

test_helpers_end_to_end/src/steps.rs

@@ -606,6 +606,12 @@ where
                     namespace_name,
                     expected,
                 } => {
+                    info!("====Persist ingesters to ensure catalog partition records exist");
+                    state
+                        .cluster()
+                        .persist_ingesters_by_namespace(namespace_name.clone())
+                        .await;
+
                     info!("====Begin reading partition keys for table: {}", table_name);
                     let partition_keys = state
                         .cluster()