from cicd_debug_env.env import CICDDebugEnv
from cicd_debug_env.models import Action

def test_env_reset():
    env = CICDDebugEnv()
    obs = env.reset()
    assert obs.pipeline_yaml != ""
    assert isinstance(obs.step_blame_scores, dict)

def test_env_step():
    env = CICDDebugEnv()
    env.reset()
    action = Action("analyze_error", {}, 0.9, "testing")
    obs, reward, done, info = env.step(action)
    assert not done
    assert "reward_breakdown" in info
    assert len(obs.episode_history) > 0
    
def test_submit_solution_ends_episode():
    env = CICDDebugEnv()
    env.reset()
    action = Action("submit_solution", {}, 0.9, "testing")
    obs, reward, done, info = env.step(action)
    assert done