def test_nominal(self):
epic = EpicFactory(ref="MY_EPIC_REF")
strategy = StrategyFactory(ref="MY_STRATEGY")
trade = TradeFactory(
ref="MY_TRADE",
epic=epic,
direction=TradeDirection.BUY,
open_datetime=arrow.get("2020-01-01 12:34:56"),
quantity=5,
status=TransactionStatus.REFUSED,
strategy=strategy,
)
assert trade.asdict() == {
"closed_quantities": 0,
"direction": TradeDirection.BUY,
"epic": "MY_EPIC_REF",
"open_date": "2020-01-01 12:34:56",
"open_quantity": 5,
"open_value": 101,
"ref": "MY_TRADE",
"result": -10,
"status": TransactionStatus.REFUSED,
"strategy": "MY_STRATEGY",
}
def test_no_strategy(self):
trade = TradeFactory(strategy=None)
assert trade.asdict()["strategy"] == "undefined" ""