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