chore: add test

Author: appflowy

collab-database/src/fields/type_option/checkbox_type_option.rs

@@ -49,7 +49,7 @@ impl TypeOptionCellWriter for CheckboxTypeOption {
       Value::Number(n) => Some(n.to_string()),
       _ => None,
     } {
-      cell.insert(CELL_DATA.into(), bool_from_str(&data).into());
+      cell.insert(CELL_DATA.into(), bool_from_str(&data).to_string().into());
     }
     cell
   }
@@ -75,3 +75,85 @@ fn bool_from_str(s: &str) -> bool {
     _ => false,
   }
 }
+
+#[cfg(test)]
+mod tests {
+  use super::*;
+
+  #[test]
+  fn test_checkbox_type_option_json_cell() {
+    let option = CheckboxTypeOption::new();
+    let mut cell = new_cell_builder(FieldType::Checkbox);
+    cell.insert(CELL_DATA.into(), "true".into());
+
+    // Convert cell to JSON
+    let value = option.json_cell(&cell);
+    assert_eq!(value, Value::String("true".to_string()));
+
+    // Test with empty data
+    let empty_cell = new_cell_builder(FieldType::Checkbox);
+    let empty_value = option.json_cell(&empty_cell);
+    assert_eq!(empty_value, Value::String("".to_string()));
+  }
+
+  #[test]
+  fn test_checkbox_type_option_numeric_cell() {
+    let option = CheckboxTypeOption::new();
+
+    let mut true_cell = new_cell_builder(FieldType::Checkbox);
+    true_cell.insert(CELL_DATA.into(), "true".into());
+    assert_eq!(option.numeric_cell(&true_cell), Some(1.0));
+
+    let mut false_cell = new_cell_builder(FieldType::Checkbox);
+    false_cell.insert(CELL_DATA.into(), "false".into());
+    assert_eq!(option.numeric_cell(&false_cell), Some(0.0));
+
+    let mut invalid_cell = new_cell_builder(FieldType::Checkbox);
+    invalid_cell.insert(CELL_DATA.into(), "invalid".into());
+    assert_eq!(option.numeric_cell(&invalid_cell), Some(0.0));
+  }
+
+  #[test]
+  fn test_checkbox_type_option_write_json() {
+    let option = CheckboxTypeOption::new();
+
+    // Write a string
+    let value = Value::String("true".to_string());
+    let cell = option.write_json(value);
+    assert_eq!(cell.get_as::<String>(CELL_DATA).unwrap(), "true");
+
+    // Write a boolean
+    let value = Value::Bool(true);
+    let cell = option.write_json(value);
+    assert_eq!(cell.get_as::<String>(CELL_DATA).unwrap(), "true");
+
+    // Write a number
+    let value = Value::Number(1.into());
+    let cell = option.write_json(value);
+    assert_eq!(cell.get_as::<String>(CELL_DATA).unwrap(), "true");
+  }
+
+  #[test]
+  fn test_checkbox_type_option_raw_conversion() {
+    let option = CheckboxTypeOption::new();
+    assert_eq!(
+      option.convert_raw_cell_data("raw data"),
+      "raw data".to_string()
+    );
+  }
+
+  #[test]
+  fn test_bool_from_str() {
+    assert!(bool_from_str("true"));
+    assert!(bool_from_str("1"));
+    assert!(bool_from_str("yes"));
+
+    assert!(!bool_from_str("false"));
+    assert!(!bool_from_str("0"));
+    assert!(!bool_from_str("no"));
+
+    // Invalid inputs default to false
+    assert!(!bool_from_str("invalid"));
+    assert!(!bool_from_str(""));
+  }
+}

collab-database/src/fields/type_option/checklist_type_option.rs

@@ -69,3 +69,73 @@ impl From<ChecklistCellData> for Cell {
     cell
   }
 }
+#[cfg(test)]
+mod checklist_type_option_tests {
+  use super::*;
+  use serde_json::json;
+
+  #[test]
+  fn test_json_cell_conversion() {
+    let checklist_option = ChecklistTypeOption;
+
+    let cell_data = ChecklistCellData::from((
+      vec!["Opt1".to_string(), "Opt2".to_string()],
+      vec!["Opt1".to_string()],
+    ));
+    let cell: Cell = cell_data.clone().into();
+
+    let json_value = checklist_option.json_cell(&cell);
+    let restored_data: ChecklistCellData =
+      serde_json::from_value(json_value).expect("Valid JSON value");
+
+    assert_eq!(restored_data.options.len(), 2);
+    assert_eq!(restored_data.selected_option_ids.len(), 1);
+  }
+
+  #[test]
+  fn test_numeric_cell_conversion() {
+    let checklist_option = ChecklistTypeOption;
+
+    let cell_data = ChecklistCellData::from((
+      vec!["Opt1".to_string(), "Opt2".to_string()],
+      vec!["Opt1".to_string()],
+    ));
+    let cell: Cell = cell_data.clone().into();
+
+    let numeric_value = checklist_option.numeric_cell(&cell);
+    assert!(numeric_value.is_none());
+  }
+
+  #[test]
+  fn test_raw_cell_data_conversion() {
+    let checklist_option = ChecklistTypeOption;
+
+    let cell_data = ChecklistCellData::from((
+      vec!["OptA".to_string(), "OptB".to_string()],
+      vec!["OptA".to_string()],
+    ));
+    let cell_data_json = serde_json::to_string(&cell_data).expect("Valid serialization");
+
+    let converted_data = checklist_option.convert_raw_cell_data(&cell_data_json);
+    assert_eq!(converted_data, "OptA,OptB");
+  }
+
+  #[test]
+  fn test_write_json_to_cell() {
+    let checklist_option = ChecklistTypeOption;
+
+    let json_value = json!({
+        "options": [
+            { "id": "1", "name": "Option1", "color": 0 },
+            { "id": "2", "name": "Option2", "color": 1 }
+        ],
+        "selected_option_ids": ["1"]
+    });
+
+    let cell = checklist_option.write_json(json_value);
+    let restored_data = ChecklistCellData::from(&cell);
+
+    assert_eq!(restored_data.options.len(), 2);
+    assert_eq!(restored_data.selected_option_ids.len(), 1);
+  }
+}