testes unitários

tests/test_organizer.py

@@ -1,6 +1,11 @@
 import unittest
-from organizer import categorizar_por_tipo
-from pathlib import Path
+import tempfile
+from datetime import datetime
+from modules.organizer import (
+    categorizar_por_tipo,
+    criar_pasta_destino,
+    obter_data_ficheiro,
+)
 
 
 class TestOrganizer(unittest.TestCase):
@@ -8,7 +13,18 @@ class TestOrganizer(unittest.TestCase):
         self.assertEqual(categorizar_por_tipo("imagem.jpg"), "imagem")
         self.assertEqual(categorizar_por_tipo("documento.pdf"), "documento")
         self.assertEqual(categorizar_por_tipo("video.mp4"), "video")
-        self.assertEqual(categorizar_por_tipo("ficheiro.zip"), "outro")
+        self.assertEqual(categorizar_por_tipo("ficheiro.zip"), "outros")
+
+    def test_criar_pasta_destino(self):
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            path = criar_pasta_destino(tmp_dir, "imagem")
+            self.assertTrue(path.exists())
+            self.assertTrue(path.is_dir())
+
+    def test_obter_data_ficheiro(self):
+        with tempfile.NamedTemporaryFile() as tmp_file:
+            data_mod = obter_data_ficheiro(tmp_file.name, usar_criacao=False)
+            self.assertIsInstance(data_mod, datetime)
 
 
 if __name__ == "__main__":