Add SQLite Support for Testing

Author: SakaDream

.env.sample

@@ -1,3 +1,6 @@
 APP_HOST=localhost
 APP_PORT=8000
-DATABASE_URL=postgres://username:password@server/db_name
+# For Postgres
+DATABASE_URL=postgres://username:password@server/db_name
+# For SQLite
+DATABASE_URL=file://path/to/database/file.db

Cargo.toml

@@ -33,3 +33,7 @@ features = ["serde"]
 [dependencies.uuid]
 version = "0.8.1"
 features = ["v4"]
+
+[dev-dependencies.diesel]
+version = "1.4.5"
+features = ["sqlite", "r2d2", "chrono"]

rustfmt.toml

@@ -0,0 +1 @@
+reorder_imports = true

src/config/db.rs

@@ -1,18 +1,62 @@
+#[allow(unused_imports)]
 use diesel::{
     pg::PgConnection,
+    sqlite::SqliteConnection,
+    sql_query,
     r2d2::{self, ConnectionManager},
 };
 
 embed_migrations!();
 
+#[cfg(not(test))]
 pub type Connection = PgConnection;
+
+#[cfg(test)]
+pub type Connection = SqliteConnection;
+
 pub type Pool = r2d2::Pool<ConnectionManager<Connection>>;
 
+#[cfg(not(test))]
 pub fn migrate_and_config_db(url: &str) -> Pool {
-    info!("Migrating and configurating database...");    
+    info!("Migrating and configurating database..."); 
     let manager = ConnectionManager::<Connection>::new(url);
     let pool = r2d2::Pool::builder().build(manager).expect("Failed to create pool.");
     embedded_migrations::run(&pool.get().expect("Failed to migrate."));
 
     pool
 }
+
+#[cfg(test)]
+pub fn migrate_and_config_db(url: &str) -> Pool {
+    use crate::diesel::RunQueryDsl;
+    info!("Migrating and configurating database..."); 
+    let manager = ConnectionManager::<Connection>::new(url);
+    let pool = r2d2::Pool::builder().build(manager).expect("Failed to create pool.");
+    
+    sql_query(r#"DROP TABLE IF EXISTS login_history;"#).execute(&pool.get().unwrap());
+    sql_query(r#"DROP TABLE IF EXISTS users;"#).execute(&pool.get().unwrap());
+    sql_query(r#"DROP TABLE IF EXISTS people;"#).execute(&pool.get().unwrap());
+    sql_query(r#"CREATE TABLE people (
+        id INTEGER PRIMARY KEY NOT NULL,
+        name TEXT NOT NULL,
+        gender BOOLEAN NOT NULL,
+        age INTEGER NOT NULL,
+        address TEXT NOT NULL,
+        phone TEXT NOT NULL,
+        email TEXT NOT NULL
+    );"#).execute(&pool.get().unwrap());
+    sql_query(r#"CREATE TABLE users (
+        id INTEGER PRIMARY KEY NOT NULL,
+        username TEXT NOT NULL,
+        email TEXT NOT NULL,
+        password TEXT NOT NULL,
+        login_session TEXT NOT NULL DEFAULT ''
+    );"#).execute(&pool.get().unwrap());
+    sql_query(r#"CREATE TABLE login_history (
+        id INTEGER PRIMARY KEY NOT NULL,
+        user_id INTEGER NOT NULL REFERENCES users(id),
+        login_timestamp INTEGER NOT NULL
+    );"#).execute(&pool.get().unwrap());
+
+    pool
+}

src/main.rs

@@ -52,8 +52,6 @@ async fn main() -> io::Result<()> {
     let app_url = format!("{}:{}", &app_host, &app_port);
     let db_url = env::var("DATABASE_URL").expect("DATABASE_URL not found.");
 
-    let pool = config::db::migrate_and_config_db(&db_url);
-
     HttpServer::new(move || {
         App::new()
             .wrap(Cors::new() // allowed_origin return access-control-allow-origin: * by default
@@ -64,7 +62,7 @@ async fn main() -> io::Result<()> {
                 .allowed_header(http::header::CONTENT_TYPE)
                 .max_age(3600)
                 .finish())
-            .data(pool.clone())
+            .data(config::db::migrate_and_config_db(&db_url))
             .wrap(actix_web::middleware::Logger::default())
             .wrap(crate::middleware::authen_middleware::Authentication)
             .wrap_fn(|req, srv| {

src/models/login_history.rs

@@ -3,7 +3,7 @@ use crate::{
     models::user::User,
     schema::login_history::{self, dsl::*},
 };
-use chrono::{DateTime, Utc};
+use chrono::{NaiveDateTime, Utc};
 use diesel::prelude::*;
 
 #[derive(Identifiable, Associations, Queryable)]
@@ -12,22 +12,23 @@ use diesel::prelude::*;
 pub struct LoginHistory {
     pub id: i32,
     pub user_id: i32,
-    pub login_timestamp: DateTime<Utc>,
+    pub login_timestamp: NaiveDateTime,
 }
 
 #[derive(Insertable)]
 #[table_name = "login_history"]
 pub struct LoginHistoryInsertableDTO {
     pub user_id: i32,
-    pub login_timestamp: DateTime<Utc>,
+    pub login_timestamp: NaiveDateTime,
 }
 
 impl LoginHistory {
     pub fn create(un: &str, conn: &Connection) -> Option<LoginHistoryInsertableDTO> {
         if let Ok(user) = User::find_user_by_username(un, conn) {
+            let now = Utc::now();
             Some(LoginHistoryInsertableDTO {
                 user_id: user.id,
-                login_timestamp: Utc::now(),
+                login_timestamp: now.naive_utc(),
             })
         } else {
             None

src/schema.rs

@@ -2,7 +2,7 @@ table! {
     login_history (id) {
         id -> Int4,
         user_id -> Int4,
-        login_timestamp -> Timestamptz,
+        login_timestamp -> Timestamp,
     }
 }