diff --git a/crates/warpui_core/src/platform/file_picker.rs b/crates/warpui_core/src/platform/file_picker.rs index 57c00f99..9a202a9c 100644 --- a/crates/warpui_core/src/platform/file_picker.rs +++ b/crates/warpui_core/src/platform/file_picker.rs @@ -28,8 +28,8 @@ impl FileType { pub fn extensions(&self) -> &[&str] { match self { FileType::Image => &["png", "jpg", "jpeg"], - FileType::Yaml => &["yaml"], - FileType::Markdown => &["md"], + FileType::Yaml => &["yaml", "yml"], + FileType::Markdown => &["md", "markdown"], } } @@ -146,3 +146,7 @@ impl SaveFilePickerConfiguration { self } } + +#[cfg(test)] +#[path = "file_picker_tests.rs"] +mod tests; diff --git a/crates/warpui_core/src/platform/file_picker_tests.rs b/crates/warpui_core/src/platform/file_picker_tests.rs new file mode 100644 index 00000000..6f0bc3f7 --- /dev/null +++ b/crates/warpui_core/src/platform/file_picker_tests.rs @@ -0,0 +1,11 @@ +use super::*; + +#[test] +fn yaml_file_type_accepts_both_yaml_and_yml() { + assert_eq!(FileType::Yaml.extensions(), &["yaml", "yml"]); +} + +#[test] +fn markdown_file_type_accepts_md_and_markdown() { + assert_eq!(FileType::Markdown.extensions(), &["md", "markdown"]); +}