def test_accept_bid_calls_market_method_and_publishes_response(self): bid = Bid("b_id", now(), 12, 13, "b_buyer") payload = {"data": json.dumps({ "seller": "mykonos", "energy": 12, "bid": bid.to_json_string(), "transaction_uuid": "trans_id" }) } trade = Trade(id="trade_id", time=now(), offer_bid=bid, seller="trade_seller", buyer="trade_buyer") self.market.accept_bid = MagicMock(return_value=trade) self.subscriber._accept_bid(payload) sleep(.3) self.subscriber.market.accept_bid.assert_called_once() self.subscriber.redis_db.publish.assert_called_once_with( "id/ACCEPT_BID/RESPONSE", json.dumps({ "status": "ready", "trade": trade.to_json_string(), "transaction_uuid": "trans_id" }) )
def test_accept_offer_calls_market_method_and_publishes_response(self): offer = Offer("o_id", now(), 12, 13, "o_seller") payload = {"data": json.dumps({ "buyer": "mykonos", "energy": 12, "offer_or_id": offer.to_json_string(), "transaction_uuid": "trans_id" }) } trade = Trade(id="trade_id", time=now(), offer_bid=offer, seller="trade_seller", buyer="trade_buyer") self.market.accept_offer = MagicMock(return_value=trade) self.subscriber._accept_offer(payload) sleep(0.1) self.subscriber.market.accept_offer.assert_called_once_with( offer_or_id=offer, buyer="mykonos", energy=12 ) self.subscriber.redis_db.publish.assert_called_once_with( "id/ACCEPT_OFFER/RESPONSE", json.dumps({ "status": "ready", "trade": trade.to_json_string(), "transaction_uuid": "trans_id" }) )