Wrote tests for config::new() which is used by pyo3 code

Author: Azrenbeth

src/lib.rs

@@ -1018,3 +1018,138 @@ mod lib_tests {
 
     //TODO: tests for correct SQL code produced by output_sql
 }
+
+#[cfg(test)]
+mod pyo3_tests {
+    use crate::{Config, LevelSizes};
+
+    #[test]
+    #[should_panic]
+    fn new_config_panics_if_no_db_url() {
+        let db_url = "".to_string();
+        let output_file = "".to_string();
+        let room_id = "[email protected]".to_string();
+        let min_state_group = "".to_string();
+        let groups_to_compress = "".to_string();
+        let min_saved_rows = "".to_string();
+        let level_sizes = "".to_string();
+        let transactions = false;
+        let graphs = false;
+
+        Config::new(
+            db_url,
+            output_file,
+            room_id,
+            min_state_group,
+            groups_to_compress,
+            min_saved_rows,
+            level_sizes,
+            transactions,
+            graphs,
+        );
+    }
+
+    #[test]
+    #[should_panic]
+    fn new_config_panics_if_no_room_id() {
+        let db_url = "postresql://homeserver.com/synapse".to_string();
+        let output_file = "".to_string();
+        let room_id = "".to_string();
+        let min_state_group = "".to_string();
+        let groups_to_compress = "".to_string();
+        let min_saved_rows = "".to_string();
+        let level_sizes = "".to_string();
+        let transactions = false;
+        let graphs = false;
+
+        Config::new(
+            db_url,
+            output_file,
+            room_id,
+            min_state_group,
+            groups_to_compress,
+            min_saved_rows,
+            level_sizes,
+            transactions,
+            graphs,
+        );
+    }
+
+    #[test]
+    fn new_config_correct_when_things_empty() {
+        // db_url and room_id have to be set or it will panic
+        let db_url = "postresql://homeserver.com/synapse".to_string();
+        let output_file = "".to_string();
+        let room_id = "room_id".to_string();
+        let min_state_group = "".to_string();
+        let groups_to_compress = "".to_string();
+        let min_saved_rows = "".to_string();
+        let level_sizes = "".to_string();
+        let transactions = false;
+        let graphs = false;
+
+        let config = Config::new(
+            db_url.clone(),
+            output_file,
+            room_id.clone(),
+            min_state_group,
+            groups_to_compress,
+            min_saved_rows,
+            level_sizes,
+            transactions,
+            graphs,
+        );
+
+        assert_eq!(config.db_url, db_url);
+        assert!(config.output_file.is_none());
+        assert_eq!(config.room_id, room_id);
+        assert!(config.min_state_group.is_none());
+        assert!(config.groups_to_compress.is_none());
+        assert!(config.min_saved_rows.is_none());
+        assert_eq!(
+            config.level_sizes,
+            "100,50,25".parse::<LevelSizes>().unwrap()
+        );
+        assert_eq!(config.transactions, transactions);
+        assert_eq!(config.graphs, graphs);
+    }
+
+    #[test]
+    fn new_config_correct_when_things_not_empty() {
+        // db_url and room_id have to be set or it will panic
+        let db_url = "postresql://homeserver.com/synapse".to_string();
+        let output_file = "/tmp/myFile".to_string();
+        let room_id = "room_id".to_string();
+        let min_state_group = "3225".to_string();
+        let groups_to_compress = "970".to_string();
+        let min_saved_rows = "500".to_string();
+        let level_sizes = "128,64,32".to_string();
+        let transactions = true;
+        let graphs = true;
+
+        let config = Config::new(
+            db_url.clone(),
+            output_file,
+            room_id.clone(),
+            min_state_group,
+            groups_to_compress,
+            min_saved_rows,
+            level_sizes,
+            transactions,
+            graphs,
+        );
+
+        assert_eq!(config.db_url, db_url);
+        assert!(!config.output_file.is_none());
+        assert_eq!(config.room_id, room_id);
+        assert_eq!(config.min_state_group, Some(3225));
+        assert_eq!(config.groups_to_compress, Some(970));
+        assert_eq!(config.min_saved_rows, Some(500));
+        assert_eq!(
+            config.level_sizes,
+            "128,64,32".parse::<LevelSizes>().unwrap()
+        );
+        assert_eq!(config.transactions, transactions);
+        assert_eq!(config.graphs, graphs);
+    }
+}