Add more hooks to active management

core/tests/trading/test_checks.py

@@ -13,13 +13,12 @@ from core.models import (
     TradingTime,
     User,
 )
+from core.tests.helpers import StrategyMixin
 from core.trading import checks
 
 
-class ChecksTestCase(TestCase):
+class ChecksTestCase(StrategyMixin, TestCase):
     def setUp(self):
-        self.time_8 = time(8, 0, 0)
-        self.time_16 = time(16, 0, 0)
         self.user = User.objects.create_user(
             username="testuser", email="test@example.com", password="test"
         )
@@ -28,36 +27,7 @@ class ChecksTestCase(TestCase):
             name="Test Account",
             exchange="fake",
         )
-        self.order_settings = OrderSettings.objects.create(
-            user=self.user, name="Default"
-        )
-        self.trading_time_now = TradingTime.objects.create(
-            user=self.user,
-            name="Test Trading Time",
-            start_day=1,  # Monday
-            start_time=self.time_8,
-            end_day=1,  # Monday
-            end_time=self.time_16,
-        )
-        self.risk_model = RiskModel.objects.create(
-            user=self.user,
-            name="Test Risk Model",
-            max_loss_percent=50,
-            max_risk_percent=10,
-            max_open_trades=10,
-            max_open_trades_per_symbol=5,
-        )
-
-        self.strategy = Strategy.objects.create(
-            user=self.user,
-            name="Test Strategy",
-            account=self.account,
-            order_settings=self.order_settings,
-            risk_model=self.risk_model,
-            active_management_enabled=True,
-        )
-        self.strategy.trading_times.set([self.trading_time_now])
-        self.strategy.save()
+        super().setUp()
 
     @freezegun.freeze_time("2023-02-13T09:00:00")  # Monday at 09:00
     def test_within_trading_times_pass(self):