71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
import pytest
|
|
from modules.organizer import (
|
|
categorizar_por_tipo,
|
|
criar_pasta_destino,
|
|
obter_data_ficheiro,
|
|
)
|
|
from datetime import datetime
|
|
# import tempfile
|
|
|
|
# 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()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"ficheiro, esperado",
|
|
[
|
|
("foto.jpeg", "imagem"),
|
|
("documento.docx", "documento"),
|
|
("video.mp4", "video"),
|
|
("musica.mp3", "audio"),
|
|
("ficheiro.xyz", "outros"),
|
|
],
|
|
)
|
|
def test_categorizar_por_tipo(ficheiro, esperado):
|
|
assert categorizar_por_tipo(ficheiro) == esperado
|
|
|
|
|
|
def test_obter_data_ficheiro(tmp_path):
|
|
ficheiro = tmp_path / "teste.txt"
|
|
ficheiro.write_text("conteudo")
|
|
|
|
# teste data de modificação
|
|
data_mod = obter_data_ficheiro(str(ficheiro), usar_criacao=False)
|
|
assert isinstance(data_mod, datetime)
|
|
|
|
# testar data de criação
|
|
data_criacao = obter_data_ficheiro(str(ficheiro), usar_criacao=True)
|
|
assert isinstance(data_criacao, datetime)
|
|
|
|
|
|
def test_criar_pasta_destino(tmp_path):
|
|
pasta = criar_pasta_destino(str(tmp_path), "documento")
|
|
assert pasta.exists()
|
|
assert pasta.is_dir()
|
|
|
|
pasta_com_data = criar_pasta_destino(
|
|
str(tmp_path), "imagem", datetime.now(), "%Y-%m-%d"
|
|
)
|
|
assert pasta_com_data.exists()
|
|
assert pasta_com_data.is_dir()
|