from . import LingoTestBase


class TestRequiredRoomLogic(LingoTestBase):
    options = {
        "shuffle_doors": "complex",
        "shuffle_colors": "false",
    }

    def test_pilgrim_first(self) -> None:
        self.assertFalse(self.multiworld.state.can_reach("The Seeker", "Region", self.player))
        self.assertFalse(self.multiworld.state.can_reach("Pilgrim Antechamber", "Region", self.player))
        self.assertFalse(self.multiworld.state.can_reach("Pilgrim Room", "Region", self.player))
        self.assertFalse(self.can_reach_location("The Seeker - Achievement"))

        self.collect_by_name("Pilgrim Room - Sun Painting")
        self.assertFalse(self.multiworld.state.can_reach("The Seeker", "Region", self.player))
        self.assertTrue(self.multiworld.state.can_reach("Pilgrim Antechamber", "Region", self.player))
        self.assertTrue(self.multiworld.state.can_reach("Pilgrim Room", "Region", self.player))
        self.assertFalse(self.can_reach_location("The Seeker - Achievement"))

        self.collect_by_name("Pilgrim Room - Shortcut to The Seeker")
        self.assertTrue(self.multiworld.state.can_reach("The Seeker", "Region", self.player))
        self.assertTrue(self.multiworld.state.can_reach("Pilgrim Room", "Region", self.player))
        self.assertFalse(self.can_reach_location("The Seeker - Achievement"))

        self.collect_by_name("Starting Room - Back Right Door")
        self.assertTrue(self.can_reach_location("The Seeker - Achievement"))

    def test_hidden_first(self) -> None:
        self.assertFalse(self.multiworld.state.can_reach("The Seeker", "Region", self.player))
        self.assertFalse(self.multiworld.state.can_reach("Pilgrim Room", "Region", self.player))
        self.assertFalse(self.can_reach_location("The Seeker - Achievement"))

        self.collect_by_name("Starting Room - Back Right Door")
        self.assertFalse(self.multiworld.state.can_reach("The Seeker", "Region", self.player))
        self.assertFalse(self.multiworld.state.can_reach("Pilgrim Room", "Region", self.player))
        self.assertFalse(self.can_reach_location("The Seeker - Achievement"))

        self.collect_by_name("Pilgrim Room - Shortcut to The Seeker")
        self.assertFalse(self.multiworld.state.can_reach("The Seeker", "Region", self.player))
        self.assertFalse(self.multiworld.state.can_reach("Pilgrim Room", "Region", self.player))
        self.assertFalse(self.can_reach_location("The Seeker - Achievement"))

        self.collect_by_name("Pilgrim Room - Sun Painting")
        self.assertTrue(self.multiworld.state.can_reach("The Seeker", "Region", self.player))
        self.assertTrue(self.multiworld.state.can_reach("Pilgrim Room", "Region", self.player))
        self.assertTrue(self.can_reach_location("The Seeker - Achievement"))


class TestRequiredDoorLogic(LingoTestBase):
    options = {
        "shuffle_doors": "complex",
        "shuffle_colors": "false",
    }

    def test_through_rhyme(self) -> None:
        self.assertFalse(self.can_reach_location("Rhyme Room - Circle/Looped Square Wall"))

        self.collect_by_name("Starting Room - Rhyme Room Entrance")
        self.assertFalse(self.can_reach_location("Rhyme Room - Circle/Looped Square Wall"))

        self.collect_by_name("Rhyme Room (Looped Square) - Door to Circle")
        self.assertTrue(self.can_reach_location("Rhyme Room - Circle/Looped Square Wall"))

    def test_through_hidden(self) -> None:
        self.assertFalse(self.can_reach_location("Rhyme Room - Circle/Looped Square Wall"))

        self.collect_by_name("Starting Room - Rhyme Room Entrance")
        self.assertFalse(self.can_reach_location("Rhyme Room - Circle/Looped Square Wall"))

        self.collect_by_name("Starting Room - Back Right Door")
        self.assertFalse(self.can_reach_location("Rhyme Room - Circle/Looped Square Wall"))

        self.collect_by_name("Hidden Room - Rhyme Room Entrance")
        self.assertTrue(self.can_reach_location("Rhyme Room - Circle/Looped Square Wall"))


class TestSimpleDoors(LingoTestBase):
    options = {
        "shuffle_doors": "simple",
        "shuffle_colors": "false",
    }

    def test_requirement(self):
        self.assertFalse(self.multiworld.state.can_reach("Outside The Wanderer", "Region", self.player))
        self.assertFalse(self.multiworld.state.can_reach("Orange Tower Third Floor", "Region", self.player))

        self.collect_by_name("Rhyme Room Doors")
        self.assertTrue(self.multiworld.state.can_reach("Outside The Wanderer", "Region", self.player))
        self.assertTrue(self.multiworld.state.can_reach("Orange Tower Third Floor", "Region", self.player))