def test_add_orders_to_strategy(self):
"""Strategy object can have orders associated."""
strategy = StrategyFactory(
user=self.user,
asset=self.asset,
)
order = OrderFactory(user=self.user,
asset_id=self.asset,
strategy=strategy)
self.assertTrue(strategy.orders.filter(pk=order.pk).exists())
def order():
return OrderFactory()
def setUp(self):
self.user = UserFactory()
self.asset = AssetFactory(symbol="AAPL")
self.strategy = StrategyFactory(asset=self.asset)
self.order = OrderFactory(strategy=self.strategy, asset_id=self.asset)
