Add initial integration tests for SQL models

This commit is contained in:
Tobie Morgan Hitchcock 2022-07-29 10:48:23 +01:00
parent 58698f58fe
commit ad4393f6ed

56
lib/tests/model.rs Normal file
View file

@ -0,0 +1,56 @@
mod parse;
use parse::Parse;
use surrealdb::sql::Value;
use surrealdb::Datastore;
use surrealdb::Error;
use surrealdb::Session;
#[tokio::test]
async fn model_count() -> Result<(), Error> {
let sql = "
CREATE |test:1000| SET time = time::now();
SELECT count() FROM test GROUP BY 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 BY 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(())
}