Stardew Valley: Remove Rarecrow Locations from Night Market when Museumsanity is Disabled (#5146)

worlds/stardew_valley/test/rules/TestMuseum.py

@@ -1,12 +1,16 @@
 from collections import Counter
+from unittest.mock import patch
 
 from ..bases import SVTestBase
-from ...options import Museumsanity
+from ..options import presets
+from ... import options, StardewLogic, StardewRule
+from ...logic.museum_logic import MuseumLogic
+from ...stardew_rule import true_, LiteralStardewRule
 
 
 class TestMuseumMilestones(SVTestBase):
     options = {
-        Museumsanity.internal_name: Museumsanity.option_milestones
+        options.Museumsanity: options.Museumsanity.option_milestones
     }
 
     def test_50_milestone(self):
@@ -14,3 +18,45 @@ class TestMuseumMilestones(SVTestBase):
 
         milestone_rule = self.world.logic.museum.can_find_museum_items(50)
         self.assert_rule_false(milestone_rule, self.multiworld.state)
+
+
+class DisabledMuseumRule(LiteralStardewRule):
+    value = False
+
+    def __or__(self, other) -> StardewRule:
+        return other
+
+    def __and__(self, other) -> StardewRule:
+        return self
+
+    def __repr__(self):
+        return "Disabled Museum Rule"
+
+
+class TestMuseumsanityDisabledExcludesMuseumDonationsFromOtherLocations(SVTestBase):
+    options = {
+        **presets.allsanity_mods_6_x_x(),
+        options.Museumsanity.internal_name: options.Museumsanity.option_none
+    }
+
+    def test_museum_donations_are_never_required_in_any_locations(self):
+        with patch("worlds.stardew_valley.logic.museum_logic.MuseumLogic") as MockMuseumLogic:
+            museum_logic: MuseumLogic = MockMuseumLogic.return_value
+            museum_logic.can_donate_museum_items.return_value = DisabledMuseumRule()
+            museum_logic.can_donate_museum_artifacts.return_value = DisabledMuseumRule()
+            museum_logic.can_find_museum_artifacts.return_value = DisabledMuseumRule()
+            museum_logic.can_find_museum_minerals.return_value = DisabledMuseumRule()
+            museum_logic.can_find_museum_items.return_value = DisabledMuseumRule()
+            museum_logic.can_complete_museum.return_value = DisabledMuseumRule()
+            museum_logic.can_donate.return_value = DisabledMuseumRule()
+            # Allowing calls to museum rules since a lot of other logic depends on it, for minerals for instance.
+            museum_logic.can_find_museum_item.return_value = true_
+
+            regions = {region.name for region in self.multiworld.regions}
+            self.world.logic = StardewLogic(self.player, self.world.options, self.world.content, regions)
+            self.world.set_rules()
+
+            self.collect_everything()
+            for location in self.get_real_locations():
+                with self.subTest(location.name):
+                    self.assert_can_reach_location(location)