more error handling and overwrite option

This commit is contained in:
Elijah McMorris 2023-12-18 02:38:46 -08:00
parent e37d413372
commit 8905c88819
Signed by: NexVeridian
SSH key fingerprint: SHA256:bsA1SKZxuEcEVHAy3gY1HUeM5ykRJl0U0kQHQn0hMg8
8 changed files with 76 additions and 23 deletions

View file

@ -9,6 +9,7 @@ use wikidata_to_surrealdb::utils::*;
async fn inti_db() -> Result<Surreal<Db>, Error> {
env::set_var("WIKIDATA_LANG", "en");
env::set_var("OVERWRITE_DB", "true");
let db = Surreal::new::<Mem>(()).await?;
db.use_ns("wikidata").use_db("wikidata").await?;
@ -40,7 +41,7 @@ async fn entity() {
.unwrap();
for line in reader.lines() {
create_db_entity(&db, line.unwrap()).await.unwrap();
create_db_entity(&db, &line.unwrap()).await.unwrap();
}
assert_eq!(51.0, entity_query(&db).await.unwrap().unwrap())
@ -82,7 +83,7 @@ async fn property() {
.unwrap();
for line in reader.lines() {
create_db_entity(&db, line.unwrap()).await.unwrap();
create_db_entity(&db, &line.unwrap()).await.unwrap();
}
assert_eq!(2.0, property_query(&db).await.unwrap().unwrap())