def side_effect(resp, *args, **kwargs): if resp is nio.RoomGetStateEventResponse: resp = nio.RoomGetStateEventResponse( {"twim": { "encrypted_val": "enceventid" }}, "", "", "") resp.transport_response = AsyncMock() resp.transport_response.status = 200 return resp elif resp is nio.RoomGetEventResponse: event = nio.Event({ "type": "dev.opsdroid.database", "event_id": "enceventid", "sender": "@someone:localhost", "origin_server_ts": "2005", "content": { "twim": { "hello": "world" } }, }) resp = nio.RoomGetEventResponse() resp.event = event return resp else: return nio.RoomSendResponse("enceventid", "!notaroomid")
def side_effect(resp, *args, **kwargs): if resp is nio.RoomGetStateEventResponse: resp = nio.RoomGetStateEventResponse({}, "", "", "") resp.transport_response = AsyncMock() resp.transport_response.status = 404 return resp else: return nio.RoomSendResponse("enceventid", "!notaroomid")
def side_effect(resp, *args, **kwargs): if resp is nio.RoomGetStateEventResponse: resp = nio.RoomGetStateEventResponse({"twim": "hello"}, "", "", "") return resp else: return nio.RoomSendResponse("enceventid", "!notaroomid")