test: Refactor function 'can_parse_language_from_valid_filename'

Author: utilForever

src/file_manager/models.rs

@@ -77,34 +77,17 @@ impl Language {
 #[cfg(test)]
 mod tests {
     use super::*;
-
-    #[test]
-    fn can_parse_language_from_valid_filename() {
-        assert_eq!(
-            Language::from_filename("solution.py").unwrap(),
-            Language::Python
-        );
-        assert_eq!(
-            Language::from_filename("test.PY").unwrap(),
-            Language::Python
-        );
-
-        assert_eq!(
-            Language::from_filename("Solution.java").unwrap(),
-            Language::Java
-        );
-
-        assert_eq!(Language::from_filename("code.cpp").unwrap(), Language::Cpp);
-
-        assert_eq!(
-            Language::from_filename("input.in").unwrap(),
-            Language::Input
-        );
-
-        assert_eq!(
-            Language::from_filename("output.out").unwrap(),
-            Language::Output
-        );
+    use rstest::rstest;
+
+    #[rstest]
+    #[case("solution.py", Language::Python)]
+    #[case("test.PY", Language::Python)]
+    #[case("Solution.java", Language::Java)]
+    #[case("code.cpp", Language::Cpp)]
+    #[case("input.in", Language::Input)]
+    #[case("output.out", Language::Output)]
+    fn can_parse_language_from_valid_filename(#[case] filename: &str, #[case] expected: Language) {
+        assert_eq!(Language::from_filename(filename).unwrap(), expected);
     }
 
     #[test]