import unittest
import tempfile
from pathlib import Path
from modules.scanner import listar_ficheiros


class TestScanner(unittest.TestCase):
    def setUp(self) -> None:
        # criar derectorio temporario
        self.temp_dir = tempfile.TemporaryDirectory()
        (Path(self.temp_dir.name) / "teste.txt").touch()
        (Path(self.temp_dir.name) / "imagem.jpg").touch()

    def tearDown(self) -> None:
        self.temp_dir.cleanup()

    def test_listar_ficheiros(self):
        ficheiros = listar_ficheiros(self.temp_dir.name)
        self.assertEqual(len(ficheiros), 2)
        for f in ficheiros:
            self.assertTrue(Path(f).exists())


if __name__ == "__main__":
    unittest.main()