Add option to commit changes to the database automatically (#53)

compressor_integration_tests/tests/compressor_config_tests.rs

@@ -1,5 +1,10 @@
 use compressor_integration_tests::{
-    add_contents_to_database, empty_database, map_builder::line_with_state, DB_URL,
+    add_contents_to_database, database_collapsed_states_match_map, database_structure_matches_map,
+    empty_database,
+    map_builder::{
+        compressed_3_3_from_0_to_13_with_state, line_segments_with_state, line_with_state,
+    },
+    DB_URL,
 };
 use serial_test::serial;
 use synapse_compress_state::{run, Config};
@@ -36,6 +41,7 @@ fn run_succeeds_without_crashing() {
     let level_sizes = "3,3".to_string();
     let transactions = true;
     let graphs = false;
+    let commit_changes = false;
 
     let config = Config::new(
         db_url.clone(),
@@ -48,8 +54,74 @@ fn run_succeeds_without_crashing() {
         level_sizes,
         transactions,
         graphs,
+        commit_changes,
     )
     .unwrap();
 
     run(config);
 }
+
+#[test]
+#[serial(db)]
+fn changes_commited_if_no_min_saved_rows() {
+    // This starts with the following structure
+    //
+    // 0-1-2 3-4-5 6-7-8 9-10-11 12-13
+    //
+    // Each group i has state:
+    //     ('node','is',      i)
+    //     ('group',  j, 'seen') - for all j less than i
+    let initial = line_segments_with_state(0, 13);
+
+    // Place this initial state into an empty database
+    empty_database();
+    add_contents_to_database("room1", &initial);
+
+    // set up the config options
+    let db_url = DB_URL.to_string();
+    let room_id = "room1".to_string();
+    let output_file = Some("./tests/tmp/changes_commited_if_no_min_saved_rows.sql".to_string());
+    let min_state_group = None;
+    let min_saved_rows = None;
+    let groups_to_compress = None;
+    let max_state_group = None;
+    let level_sizes = "3,3".to_string();
+    let transactions = true;
+    let graphs = false;
+    let commit_changes = true;
+
+    let config = Config::new(
+        db_url,
+        room_id,
+        output_file,
+        min_state_group,
+        groups_to_compress,
+        min_saved_rows,
+        max_state_group,
+        level_sizes,
+        transactions,
+        graphs,
+        commit_changes,
+    )
+    .unwrap();
+
+    // Run the compressor with those settings
+    run(config);
+
+    // This should have created the following structure in the database
+    // i.e. groups 6 and 9 should have changed from before
+    // N.B. this saves 11 rows
+    //
+    // 0  3\      12
+    // 1  4 6\    13
+    // 2  5 7 9
+    //      8 10
+    //        11
+    let expected = compressed_3_3_from_0_to_13_with_state();
+
+    // Check that the database still gives correct states for each group!
+    assert!(database_collapsed_states_match_map(&initial));
+
+    // Check that the structure of the database matches the expected structure
+    assert!(database_structure_matches_map(&expected))
+}