import tempfile
import unittest
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent))

from rotation_worker import (
    ACTIVE,
    HIDDEN_ROTATION,
    VIOLATION_HIDDEN,
    RotationConfig,
    account_publish_limit,
    apply_exhibition_cap,
    hide_violation,
    load_config,
    restore_permanent,
)


class RotationWorkerTest(unittest.TestCase):
    def test_account_publish_limit_changes_after_first_day(self):
        now = 2_000_000_000
        self.assertEqual(account_publish_limit({"id": "a", "createdAt": now}, now), 5)
        self.assertEqual(account_publish_limit({"id": "a", "createdAt": now - 25 * 60 * 60 * 1000}, now), 10)

    def test_exhibition_cap_hides_old_excess_objects(self):
        state = {
            "assets": [],
            "placed": [
                {"id": "old", "assetId": "a", "x": 1, "y": 1, "publishedAt": 10},
                {"id": "new", "assetId": "b", "x": 2, "y": 2, "publishedAt": 20},
            ],
            "dynamicSummons": [],
            "objectVotes": {},
        }
        summary = apply_exhibition_cap(state, RotationConfig(display_limit=1, newest_slots=1, revival_slots=0), seed=1, now=30)
        self.assertEqual(summary["active"], 1)
        self.assertEqual(state["placed"][1]["status"], ACTIVE)
        self.assertEqual(state["placed"][0]["status"], HIDDEN_ROTATION)

    def test_display_limit_caps_slot_total(self):
        config = RotationConfig(display_limit=3, newest_slots=5, revival_slots=5)
        self.assertEqual(config.newest_slots + config.revival_slots, 3)

    def test_violation_hide_and_restore(self):
        state = {"placed": [{"id": "obj", "assetId": "a"}], "dynamicSummons": []}
        self.assertEqual(hide_violation(state, ["obj"], now=1), ["obj"])
        self.assertEqual(state["placed"][0]["status"], VIOLATION_HIDDEN)
        self.assertEqual(restore_permanent(state, ["obj"], now=2), ["obj"])
        self.assertEqual(state["placed"][0]["status"], HIDDEN_ROTATION)

    def test_load_config_from_json_with_override(self):
        with tempfile.TemporaryDirectory() as temp:
            path = Path(temp) / "policy.json"
            path.write_text('{"display_limit": 40, "newest_slots": 30}', encoding="utf-8")
            config = load_config(path, {"revival_slots": 7})
        self.assertEqual(config.display_limit, 40)
        self.assertEqual(config.newest_slots, 30)
        self.assertEqual(config.revival_slots, 7)


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