start making tests

src/main.py

@@ -3,7 +3,7 @@ import logging
 
 from meta_data_handler import MetadataHandler
 from file_handler import sort_pictures
-from folder_handler import *
+from scan_folder import *
 
 sys.path.append("../")
 log_folder = "."
@@ -29,7 +29,7 @@ def start_process():
     raise err
 
 
-files = recursive_scan_dir(src)
+files = recursive_scan_folder(src)
 if len(files) > 0:
   start_process()
 else:

src/scan_folder.py

@@ -1,11 +1,11 @@
 import os
 
 
-def scan_dir(path:str):
+def scan_folder(path:str):
   return next(os.walk(path))[2]
 
 
-def recursive_scan_dir(path:str):
+def recursive_scan_folder(path:str):
   results = []
   for root, folders, files in os.walk(path):
     list_files = os.listdir(root)

tests/.coveragerc

@@ -1,6 +1,3 @@
 [run]
 source = src
 omit = tests/*, __init__.py
-
-[report]
-include_namespace_packages = True

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/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
+