surrealpatch/lib/tests/model.rs
2023-01-08 17:38:51 +00:00

56 lines
1.2 KiB
Rust

mod parse;
use parse::Parse;
use surrealdb::dbs::Session;
use surrealdb::err::Error;
use surrealdb::kvs::Datastore;
use surrealdb::sql::Value;
#[tokio::test]
async fn model_count() -> Result<(), Error> {
let sql = "
CREATE |test:1000| SET time = time::now();
SELECT count() FROM test GROUP ALL;
";
let dbs = Datastore::new("memory").await?;
let ses = Session::for_kv().with_ns("test").with_db("test");
let res = &mut dbs.execute(&sql, &ses, None, false).await?;
assert_eq!(res.len(), 2);
//
let tmp = res.remove(0).result;
assert!(tmp.is_ok());
//
let tmp = res.remove(0).result?;
let val = Value::parse(
"[{
count: 1000
}]",
);
assert_eq!(tmp, val);
//
Ok(())
}
#[tokio::test]
async fn model_range() -> Result<(), Error> {
let sql = "
CREATE |test:1..1000| SET time = time::now();
SELECT count() FROM test GROUP ALL;
";
let dbs = Datastore::new("memory").await?;
let ses = Session::for_kv().with_ns("test").with_db("test");
let res = &mut dbs.execute(&sql, &ses, None, false).await?;
assert_eq!(res.len(), 2);
//
let tmp = res.remove(0).result;
assert!(tmp.is_ok());
//
let tmp = res.remove(0).result?;
let val = Value::parse(
"[{
count: 1000
}]",
);
assert_eq!(tmp, val);
//
Ok(())
}