Assign interaction sim to fix broken null check (#114)

* Assign interaction sim to fix broken null check

* Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* Update src/control_any_sim/patches/small_business_income_data.py

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

Author: TitanNano

src/control_any_sim/main.py

@@ -19,6 +19,7 @@
 )
 
 import control_any_sim.cheats
+import control_any_sim.patches.small_business_income_data
 from control_any_sim import ts4_services
 from control_any_sim.services.integrity import IntegrityService
 from control_any_sim.services.interactions_service import InteractionsService
@@ -221,6 +222,7 @@ def canys_client_get_selector_visual_type(
     Override for Client::_get_selector_visual_type method.
 
     Clear selector visual type override for controled sims.
+
     """
     try:
         Logger.log("getting selector visual type")

src/control_any_sim/patches/__init__.py

@@ -0,0 +1 @@
+"""Patches for existing game classes."""

src/control_any_sim/patches/small_business_income_data.py

@@ -0,0 +1,46 @@
+"""Patches for the SmallBusinessIncomeData class."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any, Callable
+
+from small_business.small_business_income_data import SmallBusinessIncomeData
+
+from control_any_sim.util.inject import (
+    inject_method_to,
+)
+
+if TYPE_CHECKING:
+    from sims.sim import Sim
+
+
+@inject_method_to(SmallBusinessIncomeData, "_should_apply_markup_for_gain")
+def canys_should_apply_markup_for_gain(
+    original: Callable[[SmallBusinessIncomeData, Sim, Any, Any], bool],
+    self: SmallBusinessIncomeData,
+    target_sim: Sim,
+    interaction: Any,  # noqa: ANN401
+    tags: Any | None = None,  # noqa: ANN401
+) -> bool:
+    """
+    Override for SmallBusinessIncomeData::_should_apply_markup_for_gain method.
+
+    Fixes bug in the original implementation when the interaction.sim is not present.
+
+    Returns
+    -------
+        If the markup should be applied.
+
+    """
+    if target_sim is None:
+        return False
+
+    # set an interaction sim when none is present. The original function incorrectly checks for a null value and runs into an attribute error otherwise.
+    if (
+        interaction is not None
+        and self.is_sim_an_employee(target_sim.sim_info)
+        and isinstance(interaction.sim, property)
+    ):
+        interaction.sim = target_sim
+
+    return original(self, target_sim, interaction, tags)