Add function returns to info structure (#4642)

Co-authored-by: Tobie Morgan Hitchcock <tobie@surrealdb.com>
This commit is contained in:
Julian 2024-09-03 15:02:09 +02:00 committed by GitHub
parent b41b28f099
commit 507dd931f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 25 additions and 0 deletions

View file

@ -120,6 +120,7 @@ impl InfoStructure for DefineFunctionStatement {
"block".to_string() => self.block.structure(), "block".to_string() => self.block.structure(),
"permissions".to_string() => self.permissions.structure(), "permissions".to_string() => self.permissions.structure(),
"comment".to_string(), if let Some(v) = self.comment => v.into(), "comment".to_string(), if let Some(v) = self.comment => v.into(),
"returns".to_string(), if let Some(v) = self.returns => v.structure(),
}) })
} }
} }

View file

@ -635,3 +635,27 @@ async fn access_info_redacted_structure() {
); );
} }
} }
#[tokio::test]
async fn function_info_structure() {
let sql = r#"
DEFINE FUNCTION fn::example($name: string) -> string { RETURN "Hello, " + $name + "!"; };
INFO FOR DB STRUCTURE;
"#;
let dbs = new_ds().await.unwrap();
let ses = Session::owner().with_ns("ns").with_db("db");
let mut res = dbs.execute(sql, &ses, None).await.unwrap();
assert_eq!(res.len(), 2);
let out = res.pop().unwrap().output();
assert!(out.is_ok(), "Unexpected error: {:?}", out);
let out_expected =
r#"{ accesses: [], analyzers: [], functions: [{ args: [['name', 'string']], block: "{ RETURN 'Hello, ' + $name + '!'; }", name: 'example', permissions: true, returns: 'string' }], models: [], params: [], tables: [], users: [] }"#.to_string();
let out_str = out.unwrap().to_string();
assert_eq!(
out_str, out_expected,
"Output '{out_str}' doesn't match expected output '{out_expected}'",
);
}