Method query_named_rows

nodedb-test-support/src/pgwire_harness/query.rs:60–85

Execute a SQL statement, returning every row as a `HashMap` keyed by the column name reported in the row description. Useful for tests that need to assert on specific projected columns regardless of projection order. NULL columns are stored as the empty string.

(
        &self,
        sql: &str,
    )

    /// that need to assert on specific projected columns regardless of
    /// projection order. NULL columns are stored as the empty string.
    pub async fn query_named_rows(
        &self,
        sql: &str,
    ) -> Result<Vec<std::collections::HashMap<String, String>>, String> {
        let client = self.client.as_ref();
        match client.simple_query(sql).await {
            Ok(msgs) => {
                let mut rows: Vec<std::collections::HashMap<String, String>> = Vec::new();
                for msg in msgs {
                    if let tokio_postgres::SimpleQueryMessage::Row(row) = msg {
                        // `SimpleColumn` is not `Clone`; collect names by
                        // borrowing the column slice directly.
                        let names: Vec<String> =
                            row.columns().iter().map(|c| c.name().to_string()).collect();
                        let mut map = std::collections::HashMap::with_capacity(names.len());
                        for (i, name) in names.into_iter().enumerate() {
                            map.insert(name, row.get(i).unwrap_or("").to_string());
                        }
                        rows.push(map);
                    }
                }
                Ok(rows)
            }
            Err(e) => Err(pg_error_detail(&e)),
        }
    }

    /// Execute a SQL statement, returning every row as a Vec of its column
    /// values (in projection order). Column count is taken from the first