[AI] fix

tests/test_main_integration.py

@@ -0,0 +1,133 @@
+import json
+import os
+import tempfile
+from unittest.mock import patch, MagicMock
+import tkinter as tk
+from time_tracker.main import WorkTracker, ConfigWrapper
+
+
+def test_tracker_initialization():
+    """Test that WorkTracker initializes without errors"""
+    # Create a temporary config file to avoid creating one in the current directory
+    with tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False) as tmp:
+        json.dump({
+            "tracking_enabled": True,
+            "current_task_id": None,
+            "window_rules": {
+                "work": ["PyCharm", "VS Code"],
+                "distraction": ["YouTube", "Twitter"]
+            }
+        }, tmp)
+        tmp_path = tmp.name
+
+    original_config_path = "config.json"
+
+    # Temporarily replace the config file
+    if os.path.exists(original_config_path):
+        os.rename(original_config_path, f"{original_config_path}.backup")
+
+    os.rename(tmp_path, original_config_path)
+
+    try:
+        # Initialize the tracker
+        tracker = WorkTracker()
+
+        # Verify that config is properly loaded as a ConfigWrapper object
+        assert hasattr(tracker, 'config')
+        assert isinstance(tracker.config, ConfigWrapper)
+        assert hasattr(tracker.config, 'tracking_enabled')
+        assert hasattr(tracker.config, 'current_task_id')
+        assert hasattr(tracker.config, 'window_rules')
+        assert tracker.config.tracking_enabled is True
+        assert tracker.config.current_task_id is None
+
+    finally:
+        # Restore original config file
+        os.remove(original_config_path)
+        if os.path.exists(f"{original_config_path}.backup"):
+            os.rename(f"{original_config_path}.backup", original_config_path)
+
+
+def test_config_wrapper_functionality():
+    """Test that ConfigWrapper provides attribute access to config values"""
+    config_dict = {
+        "tracking_enabled": True,
+        "current_task_id": "TEST-123",
+        "window_rules": {
+            "work": ["PyCharm", "VS Code"],
+            "distraction": ["YouTube", "Twitter"]
+        }
+    }
+
+    wrapper = ConfigWrapper(config_dict)
+
+    # Test attribute access
+    assert wrapper.tracking_enabled is True
+    assert wrapper.current_task_id == "TEST-123"
+    assert "PyCharm" in wrapper.window_rules["work"]
+
+    # Test updating config
+    wrapper.tracking_enabled = False
+    assert wrapper.tracking_enabled is False
+    # Note: Direct assignment doesn't update the original dict, but the wrapper still works
+
+
+def test_tracker_config_attribute_access():
+    """Test that config can be accessed as an object with attributes"""
+
+    # Create a mock config object that behaves like the real config
+    class MockConfig:
+        def __init__(self):
+            self.tracking_enabled = True
+            self.current_task_id = "TEST-123"
+
+    config = MockConfig()
+    assert hasattr(config, 'tracking_enabled')
+    assert config.tracking_enabled is True
+    assert config.current_task_id == "TEST-123"
+
+
+def test_overlay_compatible_config():
+    """Test that the config structure is compatible with overlay expectations"""
+    from time_tracker.core.config import load_config
+
+    # Create a temporary config file
+    with tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False) as tmp:
+        json.dump({
+            "tracking_enabled": True,
+            "current_task_id": "TEST-123",
+            "window_rules": {
+                "work": ["PyCharm", "VS Code"],
+                "distraction": ["YouTube", "Twitter"]
+            }
+        }, tmp)
+        tmp_path = tmp.name
+
+    original_config_path = "config.json"
+
+    # Temporarily replace the config file
+    if os.path.exists(original_config_path):
+        os.rename(original_config_path, f"{original_config_path}.backup")
+
+    os.rename(tmp_path, original_config_path)
+
+    try:
+        # Load config using the actual function
+        config = load_config()
+
+        # Verify structure
+        assert isinstance(config, dict)
+        assert config.get("tracking_enabled") is True
+        assert config.get("current_task_id") == "TEST-123"
+
+        # Create a compatible object that supports attribute access
+        wrapped_config = ConfigWrapper(config)
+        assert hasattr(wrapped_config, 'tracking_enabled')
+        assert wrapped_config.tracking_enabled is True
+        assert wrapped_config.current_task_id == "TEST-123"
+
+    finally:
+        # Restore original config file
+        os.remove(original_config_path)
+        if os.path.exists(f"{original_config_path}.backup"):
+            os.rename(f"{original_config_path}.backup", original_config_path)