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)