start making tests

tests/.coveragerc

@@ -0,0 +1,3 @@
+[run]
+source = src
+omit = tests/*, __init__.py

tests/helpers/folder_helper.py

@@ -0,0 +1,19 @@
+import os
+import shutil
+from pathlib import Path
+
+TEST_FOLDER = ".test_folder"
+
+
+def create_file(file):
+  Path(os.path.join(TEST_FOLDER, file)).touch()
+
+
+def delete_folder():
+  shutil.rmtree(TEST_FOLDER, ignore_errors=True)
+
+
+def create_folders():
+  delete_folder()
+  os.makedirs(os.path.join(TEST_FOLDER, '001', '001'))
+  os.makedirs(os.path.join(TEST_FOLDER, '002', '001'))

tests/pytest.ini

@@ -0,0 +1,3 @@
+[pytest]
+pythonpath = .
+addopts = --ignore-glob=**/tests/**

tests/test_scan_dir.py

@@ -0,0 +1,46 @@
+import unittest
+
+from helpers.folder_helper import delete_folder
+from helpers.folder_helper import create_folders
+from helpers.folder_helper import create_file
+from helpers.folder_helper import TEST_FOLDER
+from src.scan_folder import scan_folder
+from src.scan_folder import recursive_scan_folder
+
+
+class TestScanFolder(unittest.TestCase):
+  def test_scan_folder(self):
+    create_folders()
+    create_file(file='img_01.jpeg')
+    files = scan_folder(TEST_FOLDER)
+    delete_folder()
+    
+    assert len(files) == 1
+
+  def test_scan_empty_folder(self):
+    create_folders()
+    files = scan_folder(TEST_FOLDER)
+    delete_folder()
+
+    assert len(files) == 0
+
+  def test_scan_recursive_folder(self):
+    create_folders()
+    create_file(file='img_01.jpeg')
+    create_file(file='img_02.jpeg')
+    create_file(file='001/img_03.jpeg')
+    create_file(file='001/001/img_04.jpeg')
+    create_file(file='002/001/img_05.jpeg')
+
+    files = recursive_scan_folder(TEST_FOLDER)
+    delete_folder()
+
+    assert len(files) == 5
+
+  def test_scan_recursive_empty_folder(self):
+    create_folders()
+    files = recursive_scan_folder(TEST_FOLDER)
+    delete_folder()
+
+    assert len(files) == 0
+