from . import SVTestBase
from .. import options


class TestCropsanityRules(SVTestBase):
    options = {
        options.Cropsanity.internal_name: options.Cropsanity.option_enabled
    }

    def test_need_greenhouse_for_cactus(self):
        harvest_cactus = self.world.logic.region.can_reach_location("Harvest Cactus Fruit")
        self.assert_rule_false(harvest_cactus, self.multiworld.state)

        self.multiworld.state.collect(self.create_item("Cactus Seeds"))
        self.multiworld.state.collect(self.create_item("Shipping Bin"))
        self.multiworld.state.collect(self.create_item("Desert Obelisk"))
        self.assert_rule_false(harvest_cactus, self.multiworld.state)

        self.multiworld.state.collect(self.create_item("Greenhouse"))
        self.assert_rule_true(harvest_cactus, self.multiworld.state)