diff --git a/tests/helpers/folder_helper.py b/tests/helpers/folder_helper.py index 7cd8521..609b1a7 100644 --- a/tests/helpers/folder_helper.py +++ b/tests/helpers/folder_helper.py @@ -2,6 +2,8 @@ import os import shutil from pathlib import Path +from src.scan_folder import recursive_scan_folder + TEST_FOLDER = ".test_folder" TEST_IMAGES = "tests/test_files" TEST_TEMP_FOLDER = os.path.join(TEST_FOLDER, 'Temp') @@ -28,3 +30,15 @@ def copy_test_images(): os.makedirs(TEST_IMAGE_FOLDER) shutil.copy(src=os.path.join(TEST_IMAGES, 'test_image_001.JPG'), dst=TEST_TEMP_FOLDER) shutil.copy(src=os.path.join(TEST_IMAGES, 'test_image_002.JPG'), dst=TEST_TEMP_FOLDER) + + +def copy_images(brand:str, model:str): + delete_folder() + create_folders() + files = recursive_scan_folder(path=TEST_IMAGES) + for file in files: + file_name = file.split("/")[2:][0] + file_name = file_name.split("_") + if file_name[0] == brand and file_name[1] == model: + shutil.copy(src=file, dst=TEST_FOLDER) + diff --git a/tests/test_files/samsung_a54_001.jpg b/tests/test_files/samsung_a54_001.jpg new file mode 100644 index 0000000..6e58122 Binary files /dev/null and b/tests/test_files/samsung_a54_001.jpg differ diff --git a/tests/test_files/samsung_a54_002.jpg b/tests/test_files/samsung_a54_002.jpg new file mode 100644 index 0000000..995b0ea Binary files /dev/null and b/tests/test_files/samsung_a54_002.jpg differ diff --git a/tests/test_files/samsung_a54_003.jpg b/tests/test_files/samsung_a54_003.jpg new file mode 100644 index 0000000..59dd271 Binary files /dev/null and b/tests/test_files/samsung_a54_003.jpg differ diff --git a/tests/test_files/samsung_a54_004.jpg b/tests/test_files/samsung_a54_004.jpg new file mode 100644 index 0000000..1e3e7be Binary files /dev/null and b/tests/test_files/samsung_a54_004.jpg differ diff --git a/tests/test_files/samsung_a54_005.jpg b/tests/test_files/samsung_a54_005.jpg new file mode 100644 index 0000000..795fb30 Binary files /dev/null and b/tests/test_files/samsung_a54_005.jpg differ diff --git a/tests/test_files/samsung_a54_006.jpg b/tests/test_files/samsung_a54_006.jpg new file mode 100644 index 0000000..3cb68fe Binary files /dev/null and b/tests/test_files/samsung_a54_006.jpg differ diff --git a/tests/test_files/samsung_a54_007.jpg b/tests/test_files/samsung_a54_007.jpg new file mode 100644 index 0000000..513522d Binary files /dev/null and b/tests/test_files/samsung_a54_007.jpg differ diff --git a/tests/test_with_real_data.py b/tests/test_with_real_data.py new file mode 100644 index 0000000..19038ed --- /dev/null +++ b/tests/test_with_real_data.py @@ -0,0 +1,29 @@ +import unittest + +from helpers.folder_helper import TEST_FOLDER +from helpers.folder_helper import copy_images + +from src.meta_data_handler import get_meta_data +from src.scan_folder import recursive_scan_folder + + +class TestSamsung(unittest.TestCase): + def test_a54(self): + copy_images(brand="samsung", model="a54") + files = recursive_scan_folder(path=TEST_FOLDER) + meta_data = get_meta_data(images=files) + for image in meta_data: + assert image.make == "samsung" + + def test_a52s(self): + pass + + def test_a14(self): + pass + + +@unittest.skip("") +class TestApple(unittest.TestCase): + def test_iphone(self): + pass +