from __future__ import annotations

import unittest

from support_ops_env.env import SupportOpsEnv
from support_ops_env.models import Action


class SupportOpsEnvTest(unittest.TestCase):
    def test_easy_task_can_score_perfectly(self) -> None:
        env = SupportOpsEnv("easy_account_takeover")
        env.reset()
        env.step(Action(action_type="request_context", target="T1", value="account_security"))
        env.step(Action(action_type="request_context", target="T1", value="billing_activity"))
        env.step(Action(action_type="set_priority", target="T1", value="urgent"))
        env.step(Action(action_type="set_route", target="T1", value="account_security"))
        env.step(
            Action(
                action_type="set_resolution",
                target="T1",
                value="temporary_lock_and_manual_recovery",
            )
        )
        _, _, done, info = env.step(
            Action(action_type="escalate", target="T1", value="security_specialist")
        )
        self.assertFalse(done)
        _, _, done, info = env.step(Action(action_type="finalize"))
        self.assertTrue(done)
        self.assertAlmostEqual(info["task_score"], 1.0, places=4)

    def test_hard_queue_ranking_is_scored(self) -> None:
        env = SupportOpsEnv("hard_queue_triage")
        env.reset()
        _, _, _, info = env.step(Action(action_type="rank_queue", value="T2,T3,T1"))
        self.assertGreater(info["task_score"], 0.0)


if __name__ == "__main__":
    unittest.main()