47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
import pytest
|
|
from modules.organizer import (
|
|
categorizar_por_tipo,
|
|
criar_pasta_destino,
|
|
obter_data_ficheiro,
|
|
)
|
|
from datetime import datetime
|
|
|
|
|
|
@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()
|