surrealpatch/lib/tests/return.rs

mod parse;
use parse::Parse;
mod helpers;
use helpers::new_ds;
use surrealdb::dbs::Session;
use surrealdb::err::Error;
use surrealdb::sql::Value;

#[tokio::test]
async fn return_subquery_only() -> Result<(), Error> {
	let sql = "
		CREATE person:tobie SET name = 'Tobie';
		CREATE person:jaime SET name = 'Jaime';
		LET $single = person:tobie;
		--
		SELECT name FROM person;
		SELECT VALUE name FROM person;
		SELECT name FROM ONLY person;
		SELECT VALUE name FROM ONLY person;
		SELECT name FROM person:tobie;
		SELECT VALUE name FROM person:tobie;
		SELECT name FROM ONLY person:tobie;
		SELECT VALUE name FROM ONLY person:tobie;
		SELECT name FROM $single;
		SELECT VALUE name FROM $single;
		SELECT name FROM ONLY $single;
		SELECT VALUE name FROM ONLY $single;
		--
		RETURN SELECT name FROM person;
		RETURN SELECT VALUE name FROM person;
		RETURN SELECT name FROM ONLY person;
		RETURN SELECT VALUE name FROM ONLY person;
		RETURN SELECT name FROM person:tobie;
		RETURN SELECT VALUE name FROM person:tobie;
		RETURN SELECT name FROM ONLY person:tobie;
		RETURN SELECT VALUE name FROM ONLY person:tobie;
		RETURN SELECT name FROM $single;
		RETURN SELECT VALUE name FROM $single;
		RETURN SELECT name FROM ONLY $single;
		RETURN SELECT VALUE name FROM ONLY $single;
	";
	let dbs = new_ds().await?;
	let ses = Session::owner().with_ns("test").with_db("test");
	let res = &mut dbs.execute(sql, &ses, None).await?;
	assert_eq!(res.len(), 27);
	//
	let tmp = res.remove(0).result;
	assert!(tmp.is_ok());
	//
	let tmp = res.remove(0).result;
	assert!(tmp.is_ok());
	//
	let tmp = res.remove(0).result;
	assert!(tmp.is_ok());
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("[{ name: 'Jaime' }, { name: 'Tobie' }]");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("['Jaime', 'Tobie']");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result;
	assert!(matches!(
		tmp.err(),
		Some(e) if e.to_string() == r#"Expected a single result output when using the ONLY keyword"#
	));
	//
	let tmp = res.remove(0).result;
	assert!(matches!(
		tmp.err(),
		Some(e) if e.to_string() == r#"Expected a single result output when using the ONLY keyword"#
	));
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("[{ name: 'Tobie' }]");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("['Tobie']");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("{ name: 'Tobie' }");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::from("Tobie");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("[{ name: 'Tobie' }]");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("['Tobie']");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("{ name: 'Tobie' }");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::from("Tobie");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("[{ name: 'Jaime' }, { name: 'Tobie' }]");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("['Jaime', 'Tobie']");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result;
	assert!(matches!(
		tmp.err(),
		Some(e) if e.to_string() == r#"Expected a single result output when using the ONLY keyword"#
	));
	//
	let tmp = res.remove(0).result;
	assert!(matches!(
		tmp.err(),
		Some(e) if e.to_string() == r#"Expected a single result output when using the ONLY keyword"#
	));
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("[{ name: 'Tobie' }]");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("['Tobie']");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("{ name: 'Tobie' }");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::from("Tobie");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("[{ name: 'Tobie' }]");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("['Tobie']");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("{ name: 'Tobie' }");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::from("Tobie");
	assert_eq!(tmp, val);
	//
	Ok(())
}

#[tokio::test]
async fn return_breaks_nested_execution() -> Result<(), Error> {
	let sql = "
		DEFINE FUNCTION fn::test() {
		    {
				RETURN 1;
			};
			RETURN 2;
		};

		RETURN fn::test();

		BEGIN;
		CREATE ONLY a:1;
		RETURN 1;
		CREATE ONLY a:2;
		COMMIT;

		{
            RETURN 1;
        };

        SELECT VALUE {
            IF $this % 2 == 0 {
            RETURN $this;
            } ELSE {
                RETURN $this + 1;
            }
        } FROM [1, 2, 3, 4];
	";
	let dbs = new_ds().await?;
	let ses = Session::owner().with_ns("test").with_db("test");
	let res = &mut dbs.execute(sql, &ses, None).await?;
	assert_eq!(res.len(), 5);
	//
	let tmp = res.remove(0).result;
	assert!(tmp.is_ok());
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("1");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("1");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("1");
	assert_eq!(tmp, val);
	//
	let tmp = res.remove(0).result?;
	let val = Value::parse("[2, 2, 4, 4]");
	assert_eq!(tmp, val);
	//
	Ok(())
}