33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import unittest
|
|
from modules.organizer import (
|
|
categorizar_por_tipo,
|
|
criar_pasta_destino,
|
|
obter_data_ficheiro,
|
|
)
|
|
import tempfile
|
|
from datetime import datetime
|
|
|
|
|
|
class TestOrganizer(unittest.TestCase):
|
|
def test_categorizar_por_tipo(self):
|
|
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"), "outros")
|
|
self.assertEqual(categorizar_por_tipo("audio.mp3"), "audio")
|
|
|
|
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__":
|
|
unittest.main()
|