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")

    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()