NexVeridian/wikidata-to-surrealdb
1
0
Fork 0
mirror of https://github.com/NexVeridian/wikidata-to-surrealdb.git synced 2025-09-02 01:49:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

swap to rstest

Browse source
This commit is contained in:
Elijah McMorris 2024-08-25 20:30:42 -07:00
parent 58938e8f71
commit d508518beb
Signed by: NexVeridian
SSH key fingerprint: SHA256:bsA1SKZxuEcEVHAy3gY1HUeM5ykRJl0U0kQHQn0hMg8
5 changed files with 535 additions and 581 deletions
Download patch file Download diff file Expand all files Collapse all files

0
tests/data/filter.surql → tests/data/test_filter.surql
View file

49
tests/integration.rs
View file

@ -1,4 +1,5 @@
use anyhow::{Error, Ok, Result};
use rstest::rstest;
use std::{env, io::BufRead};
use surrealdb::{engine::local::Db, Surreal};
@ -47,24 +48,18 @@ async fn entity() {
assert_eq!(51.0, entity_query(&db).await.unwrap().unwrap())
}
async fn entity_threaded_insert(create_version: CreateVersion) -> Result<Surreal<Db>, Error> {
#[rstest]
#[case(CreateVersion::Single)]
#[case(CreateVersion::Bulk)]
#[tokio::test]
async fn entity_threaded(#[case] version: CreateVersion) -> Result<(), Error> {
let db = inti_db().await?;
let reader = File_Format::new("json").reader("tests/data/Entity.json")?;
let reader = init_reader("json", "Entity");
create_db_entities_threaded(Some(db.clone()), reader, None, 1_000, 100, create_version).await?;
Ok(db)
}
create_db_entities_threaded(Some(db.clone()), reader, None, 1_000, 100, version).await?;
#[tokio::test]
async fn entity_threaded() {
let db = entity_threaded_insert(CreateVersion::Single).await.unwrap();
assert_eq!(51.0, entity_query(&db).await.unwrap().unwrap())
}
#[tokio::test]
async fn entity_threaded_bulk_insert() {
let db = entity_threaded_insert(CreateVersion::Bulk).await.unwrap();
assert_eq!(51.0, entity_query(&db).await.unwrap().unwrap())
assert_eq!(51.0, entity_query(&db).await?.unwrap());
Ok(())
}
async fn property_query(db: &Surreal<Db>) -> Result<Option<f32>, Error> {
@ -89,24 +84,16 @@ async fn property() {
assert_eq!(2.0, property_query(&db).await.unwrap().unwrap())
}
async fn property_threaded_insert(create_version: CreateVersion) -> Result<Surreal<Db>, Error> {
#[rstest]
#[case(CreateVersion::Single)]
#[case(CreateVersion::Bulk)]
#[tokio::test]
async fn property_threaded(#[case] version: CreateVersion) -> Result<(), Error> {
let db = inti_db().await?;
let reader = init_reader("json", "Property");
create_db_entities_threaded(Some(db.clone()), reader, None, 1_000, 100, create_version).await?;
Ok(db)
}
create_db_entities_threaded(Some(db.clone()), reader, None, 1_000, 100, version).await?;
#[tokio::test]
async fn property_threaded_single_insert() {
let db = property_threaded_insert(CreateVersion::Single)
.await
.unwrap();
assert_eq!(2.0, property_query(&db).await.unwrap().unwrap())
}
#[tokio::test]
async fn property_threaded_bulk_insert() {
let db = property_threaded_insert(CreateVersion::Bulk).await.unwrap();
assert_eq!(2.0, property_query(&db).await.unwrap().unwrap())
assert_eq!(2.0, property_query(&db).await?.unwrap());
Ok(())
}