57 lines
1.2 KiB
Rust
57 lines
1.2 KiB
Rust
|
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(())
|
||
|
|
}
|