From 47e2b1741cd11200ac0098a118cc6c2e4978ee31 Mon Sep 17 00:00:00 2001
From: lifegpc <g1710431395@gmail.com>
Date: Fri, 23 Sep 2022 04:34:50 +0000
Subject: [PATCH] Make sure folder is already exists and the server database is
 deleted before unit test

---
 src/server/unittest/mod.rs | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/server/unittest/mod.rs b/src/server/unittest/mod.rs
index 33ae657..c48a79e 100644
--- a/src/server/unittest/mod.rs
+++ b/src/server/unittest/mod.rs
@@ -8,6 +8,10 @@ use crate::error::PixivDownloaderError;
 use futures_util::lock::Mutex;
 use hyper::{Body, Request, Response};
 use json::JsonValue;
+#[cfg(test)]
+use std::fs::{create_dir, remove_file};
+#[cfg(test)]
+use std::path::Path;
 use std::sync::Arc;
 
 pub struct UnitTestContext {
@@ -61,6 +65,15 @@ impl UnitTestContext {
 #[proc_macros::async_timeout_test(120s)]
 #[tokio::test(flavor = "multi_thread")]
 async fn test() -> Result<(), PixivDownloaderError> {
+    let p = Path::new("./test");
+    if !p.exists() {
+        let re = create_dir("./test");
+        assert!(re.is_ok() || p.exists());
+    }
+    let t = Path::new("./test/server.db");
+    if t.exists() {
+        remove_file(t)?;
+    }
     let ctx = UnitTestContext::new().await;
     version::test(&ctx).await?;
     Ok(())