import os
import sys
from pathlib import Path
from unittest.mock import MagicMock, patch
import pytest

PROJECT_ROOT = Path("c:/Projects/KieuStory")
sys.path.insert(0, str(PROJECT_ROOT / "05_Production_Pipeline"))
from antigravity_critic_gate import VideoCriticVerdict, ShotEvaluation

def test_mock_setup():
    verdict_json = '{"overall_score": 0.95, "approved": true, "suggested_action": "APPROVE", "critique_notes": "Valid"}'
    with patch.dict(os.environ, {"GEMINI_API_KEY": "fake_key"}):
        with patch("google.genai.Client") as mock_client:
            mock_inst = MagicMock()
            mock_client.return_value = mock_inst
            mock_resp = MagicMock()
            mock_resp.text = verdict_json
            mock_inst.models.generate_content.return_value = mock_resp
            assert mock_client is not None
            print("Mock client verified")

if __name__ == "__main__":
    test_mock_setup()