async def test_WAITING_FOR_REPLY() -> None: s = bcs.WAITING_FOR_REPLY("reqid") assert s.reply == None assert s.reqid == "reqid" r = await s.run(MockConnection(to_pop=None)) assert r == "_transition_to_disconnected" assert s.reply is None m = MockMessage() r = await s.run(MockConnection(to_pop=m)) assert r[0] == "_transition" assert isinstance(r[1], bcs.CONNECTED_AFTER_ACK) assert s.reply is m s._reqid = "nomatch" r = await s.run(MockConnection(to_pop=m)) assert r == "_next"
def test_WAITING_FOR_REPLY(): s = bcs.WAITING_FOR_REPLY("reqid") assert s.reply == None assert s.reqid == "reqid" r = s.run(_MockConnection(to_pop=None)) assert r.result() == "_transition_to_disconnected" assert s.reply is None m = _MockMessage() r = s.run(_MockConnection(to_pop=m)) res = r.result() assert res[0] == "_transition" assert isinstance(res[1], bcs.CONNECTED_AFTER_ACK) assert s.reply is m s._reqid = "nomatch" r = s.run(_MockConnection(to_pop=m)) assert r.result() == "_next"
async def test_WAITING_FOR_REPLY() -> None: s = bcs.WAITING_FOR_REPLY("reqid") assert s.reply is None assert s.reqid == "reqid" c = MockConnection() await s.run(c) assert c.state == "_transition_to_disconnected" assert s.reply is None m = MockMessage() c = MockConnection(to_pop=m) await s.run(c) assert c.state[0] == "_transition" assert isinstance(c.state[1], bcs.CONNECTED_AFTER_ACK) assert s.reply is m s._reqid = "nomatch" c = MockConnection(to_pop=m) await s.run(c) assert c.state == "_next"