test: Move unit tests for Language into models.rs

Author: reddevilmidzy

src/file_manager/models.rs

@@ -112,3 +112,108 @@ impl Language {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_language_from_filename_success() {
+        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
+        );
+    }
+
+    #[test]
+    fn test_language_from_filename_errors() {
+        assert_eq!(
+            Language::from_filename(".java").unwrap_err(),
+            LanguageError::InvalidFilename
+        );
+
+        assert_eq!(
+            Language::from_filename("py").unwrap_err(),
+            LanguageError::InvalidFilename
+        );
+
+        assert_eq!(
+            Language::from_filename("file.").unwrap_err(),
+            LanguageError::InvalidFilename
+        );
+
+        assert_eq!(
+            Language::from_filename("run.exe").unwrap_err(),
+            LanguageError::UnsupportedExtension("exe".to_string())
+        );
+
+        assert_eq!(
+            Language::from_filename("").unwrap_err(),
+            LanguageError::InvalidFilename
+        );
+    }
+
+    #[test]
+    fn test_language_from_extension_success() {
+        assert_eq!(Language::from_extension("py").unwrap(), Language::Python);
+        assert_eq!(Language::from_extension("rs").unwrap(), Language::Rust);
+        assert_eq!(Language::from_extension("java").unwrap(), Language::Java);
+        assert_eq!(Language::from_extension("cpp").unwrap(), Language::Cpp);
+        assert_eq!(
+            Language::from_extension("js").unwrap(),
+            Language::Javascript
+        );
+        assert_eq!(
+            Language::from_extension("ts").unwrap(),
+            Language::Typescript
+        );
+        assert_eq!(Language::from_extension("c").unwrap(), Language::C);
+        assert_eq!(Language::from_extension("cs").unwrap(), Language::Csharp);
+        assert_eq!(Language::from_extension("go").unwrap(), Language::Go);
+        assert_eq!(Language::from_extension("kt").unwrap(), Language::Kotlin);
+        assert_eq!(Language::from_extension("rb").unwrap(), Language::Ruby);
+        assert_eq!(Language::from_extension("php").unwrap(), Language::Php);
+        assert_eq!(Language::from_extension("pl").unwrap(), Language::Perl);
+        assert_eq!(Language::from_extension("pas").unwrap(), Language::Pascal);
+        assert_eq!(Language::from_extension("ml").unwrap(), Language::Ocaml);
+        assert_eq!(Language::from_extension("scala").unwrap(), Language::Scala);
+        assert_eq!(Language::from_extension("d").unwrap(), Language::D);
+        assert_eq!(Language::from_extension("txt").unwrap(), Language::Text);
+        assert_eq!(Language::from_extension("md").unwrap(), Language::Markdown);
+        assert_eq!(Language::from_extension("in").unwrap(), Language::Input);
+        assert_eq!(Language::from_extension("out").unwrap(), Language::Output);
+    }
+
+    #[test]
+    fn test_language_from_extension_errors() {
+        assert_eq!(
+            Language::from_extension("").unwrap_err(),
+            LanguageError::UnsupportedExtension("".to_string())
+        );
+        assert_eq!(
+            Language::from_extension("no").unwrap_err(),
+            LanguageError::UnsupportedExtension("no".to_string())
+        );
+    }
+}

tests/file_manager/language.rs

@@ -1,2 +1 @@
 pub mod handlers;
-pub mod language;