Fixed unit tests

- Fixed relative imports
- Removed import handling via ImportError exception
- Fixed CI by running via module, instead of file
- Added / improved handling in __init__.py for unit tests

.github/workflows/ci.yml

@@ -53,6 +53,10 @@ jobs:
           python3 .github/scripts/check_gamefixes.py
           python3 .github/scripts/check_verbs.py
 
+      # We need a known parent package in the unit tests, so we need to change to the parent dir
+      # As "umu-protonfixes" is not a valid package name, we create a symbolic link to "protonfixes"
       - name: Test with unittest
         run: |
-          python3 protonfixes_test.py
+          cd ..
+          ln -s umu-protonfixes protonfixes
+          python3 -m protonfixes.protonfixes_test

__init__.py

@@ -1,18 +1,40 @@
-"""Starts the protonfix module"""
+"""Starts the protonfix module and runs fixes after pre-flight-checks"""
 
 import os
 import sys
+import traceback
 
-RUN_CONDITIONS = [
-    'STEAM_COMPAT_DATA_PATH' in os.environ,
-    'PROTONFIXES_DISABLE' not in os.environ,
-    'waitforexitandrun' in sys.argv[1],
-]
+from . import fix
+from .logger import log
 
-if all(RUN_CONDITIONS):
-    import traceback
-    from . import fix
 
+def check_conditions() -> bool:
+    """Determine, if the actual game was executed and protonfixes isn't deactivated.
+
+    Returns:
+        bool: True, if the fix should be executed.
+    """    
+    return len(sys.argv) >= 1 and \
+        'STEAM_COMPAT_DATA_PATH' in os.environ and \
+        'PROTONFIXES_DISABLE' not in os.environ and \
+        'waitforexitandrun' in sys.argv[1]
+
+
+def check_iscriptevaluator() -> bool:
+    """Determine, if we were invoked while running "iscriptevaluator.exe".
+
+    Returns:
+        bool: True, if we were invoked while running "iscriptevaluator.exe".
+    """    
+    return len(sys.argv) >= 2 and \
+        'iscriptevaluator.exe' in sys.argv[2]
+
+
+if check_iscriptevaluator():
+    log.debug('Skipping fix execution. We are running "iscriptevaluator.exe".')
+elif not check_conditions():
+    log.warn('Skipping fix execution. We are probably running an unit test.')
+else:
     try:
         fix.main()
 

checks.py

@@ -1,9 +1,6 @@
 """Run some tests and generate warnings for proton configuration issues"""
 
-try:
-    from .logger import log
-except ImportError:
-    from logger import log
+from .logger import log
 
 
 def esync_file_limits() -> bool:

config.py

@@ -1,9 +1,11 @@
 """Load configuration settings for protonfixes"""
 
-from config_base import ConfigBase
 from dataclasses import dataclass
 from pathlib import Path
 
+from .config_base import ConfigBase
+
+
 class Config(ConfigBase):
     """Configuration for umu-protonfix"""
 

config_base.py

@@ -5,11 +5,11 @@
 from configparser import ConfigParser
 from dataclasses import is_dataclass
 from pathlib import Path
-
 from typing import Any
 from collections.abc import Callable
 
-from logger import log, LogLevel
+from .logger import log, LogLevel
+
 
 class ConfigBase:
     """Base class for configuration objects.

engine.py

@@ -2,6 +2,7 @@
 
 import os
 import sys
+
 from .logger import log, LogLevel
 
 

fix.py

@@ -4,20 +4,15 @@
 import re
 import sys
 import csv
+
 from functools import lru_cache
 from importlib import import_module
 from typing import Optional
 
-try:
-    from .config import config
-    from .util import proton_version
-    from .checks import run_checks
-    from .logger import log
-except ImportError:
-    from config import config
-    from util import proton_version
-    from checks import run_checks
-    from logger import log
+from .config import config
+from .util import proton_version
+from .checks import run_checks
+from .logger import log
 
 try:
     import __main__ as protonmain

protonfixes_test.py

@@ -1,11 +1,13 @@
-import unittest
+import io
 import os
 import tempfile
+import unittest
+import urllib.request
+
 from pathlib import Path
 from unittest.mock import patch, mock_open
-import io
-import urllib.request
-import fix
+
+from . import fix
 
 
 class TestProtonfixes(unittest.TestCase):
@@ -315,7 +317,7 @@ def testGetGameNameDBFileNotFound(self):
 
         with patch('builtins.open', mock_open()) as mocked_open:
             mocked_open.side_effect = FileNotFoundError
-            with patch('fix.log') as mocked_log:  # Mock the logger separately
+            with patch('protonfixes.fix.log') as mocked_log:  # Mock the logger separately
                 func = fix.get_game_name.__wrapped__  # Do not reference the cache
                 result = func()
                 self.assertEqual(result, 'UNKNOWN')
@@ -329,7 +331,7 @@ def testGetGameNameDbOS(self):
 
         with patch('builtins.open', mock_open()) as mocked_open:
             mocked_open.side_effect = OSError
-            with patch('fix.log') as mocked_log:  # Mock the logger separately
+            with patch('protonfixes.fix.log') as mocked_log:  # Mock the logger separately
                 func = fix.get_game_name.__wrapped__  # Do not reference the cache
                 result = func()
                 self.assertEqual(result, 'UNKNOWN')
@@ -357,7 +359,7 @@ def testGetGameNameDbUnicode(self):
 
         with patch('builtins.open', mock_open()) as mocked_open:
             mocked_open.side_effect = UnicodeDecodeError('utf-8', b'', 0, 1, '')
-            with patch('fix.log') as mocked_log:  # Mock the logger separately
+            with patch('protonfixes.fix.log') as mocked_log:  # Mock the logger separately
                 func = fix.get_game_name.__wrapped__  # Do not reference the cache
                 result = func()
                 self.assertEqual(result, 'UNKNOWN')

util.py

@@ -20,14 +20,9 @@
 from typing import Any, Union, Optional
 from collections.abc import Mapping, Callable
 
-try:
-    from .logger import log
-    from .config import config
-    from .steamhelper import install_app
-except ImportError:
-    from logger import log
-    from config import config
-    from steamhelper import install_app
+from .logger import log
+from .config import config
+from .steamhelper import install_app
 
 try:
     import __main__ as protonmain