65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
""" test uti.py """
|
|
|
|
import pytest
|
|
|
|
import osxphotos.uti
|
|
from osxphotos.uti import (
|
|
_get_uti_from_mdls,
|
|
get_preferred_uti_extension,
|
|
get_uti_for_extension,
|
|
)
|
|
|
|
EXT_DICT = {"heic": "public.heic", "jpg": "public.jpeg", ".jpg": "public.jpeg"}
|
|
UTI_DICT = {"public.heic": "heic", "public.jpeg": "jpeg"}
|
|
|
|
|
|
def test_get_preferred_uti_extension():
|
|
"""test get_preferred_uti_extension"""
|
|
for uti in UTI_DICT:
|
|
assert get_preferred_uti_extension(uti) == UTI_DICT[uti]
|
|
|
|
|
|
def test_get_uti_for_extension():
|
|
"""get get_uti_for_extension"""
|
|
for ext in EXT_DICT:
|
|
assert get_uti_for_extension(ext) == EXT_DICT[ext]
|
|
|
|
|
|
def test_get_preferred_uti_extension_no_objc():
|
|
"""test get_preferred_uti_extension when running on macOS >= 12"""
|
|
OLD_VER = osxphotos.uti.OS_VER
|
|
osxphotos.uti.OS_VER = 12
|
|
for uti in UTI_DICT:
|
|
assert get_preferred_uti_extension(uti) == UTI_DICT[uti]
|
|
osxphotos.uti.OS_VER = OLD_VER
|
|
|
|
|
|
def test_get_uti_for_extension_no_objc():
|
|
"""get get_uti_for_extension when running on macOS >= 12"""
|
|
OLD_VER = osxphotos.uti.OS_VER
|
|
osxphotos.uti.OS_VER = 12
|
|
for ext in EXT_DICT:
|
|
assert get_uti_for_extension(ext) == EXT_DICT[ext]
|
|
osxphotos.uti.OS_VER = OLD_VER
|
|
|
|
|
|
def test_get_uti_from_mdls():
|
|
"""get _get_uti_from_mdls"""
|
|
for ext in EXT_DICT:
|
|
assert _get_uti_from_mdls(ext) == EXT_DICT[ext]
|
|
|
|
|
|
def test_get_uti_not_in_dict():
|
|
"""get UTI when objc is not available and it's not in the EXT_UTI_DICT"""
|
|
# monkey patch the EXT_UTI_DICT
|
|
OLD_VER = osxphotos.uti.OS_VER
|
|
osxphotos.uti.OS_VER = 12
|
|
osxphotos.uti.EXT_UTI_DICT = {}
|
|
osxphotos.uti.UTI_EXT_DICT = {}
|
|
for ext in EXT_DICT:
|
|
assert get_uti_for_extension(ext) == EXT_DICT[ext]
|
|
osxphotos.uti.OS_VER = OLD_VER
|
|
|
|
# re-initialize the cached dicts
|
|
osxphotos.uti._load_uti_dict()
|