r-database/src/tests/table_test.rs

#[cfg(test)]
mod tests {
    use std::{fs, path::Path};

    use crate::database::{table::Table, row::Row};

    #[test]
    fn should_find_two_rows() {
        let mut table: Table = Table::new("Cats", "id");
        table.insert_row(Row::new(hashmap!["id" => 1, "name" => "Ozzy"]));
        table.insert_row(Row::new(hashmap!["id" => 2, "name" => "Simon"]));

        let row = table.find_by_pk(&1u64);
        assert_eq!(row.is_err(), false);
        let pretty_print = serde_json::to_string_pretty(row.unwrap());

        println!("Found Row: {}", pretty_print.unwrap());
        let row = table.find_by_pk(&2u64);
        let pretty_print = serde_json::to_string_pretty(row.unwrap());

        println!("Found Row: {}", pretty_print.unwrap());
    }

    #[test]
    fn should_fail_to_find_row() {
        let mut table: Table = Table::new("Cats", "id");
        table.insert_row(Row::new(hashmap!["id" => 1, "name" => "Ozzy"]));
        table.insert_row(Row::new(hashmap!["id" => 2, "name" => "Simon"]));

        let row = table.find_by_pk(&3u64);
        assert_eq!(row.is_err(), true);
    }


    #[test]
    fn should_write_and_read_to_file() {
        let mut table: Table = Table::new("Cats", "id");
        table.insert_row(Row::new(hashmap!["id" => 1, "name" => "Ozzy"]));
        table.insert_row(Row::new(hashmap!["id" => 2, "name" => "Simon"]));
        table.to_file("./db.json");

        let exists = Path::try_exists(Path::new("./db.json"));
        assert_eq!(exists.unwrap(), true);

        let table = Table::from_file("./db.json".to_owned());
        assert_eq!(table.is_ok(), true);

        let binding = table.unwrap();
        let row = binding.find_by_pk(&1u64);
        assert_eq!(row.is_ok(), true);

        // Cleanup file
        fs::remove_file("./db.json");
    }
}