def test_select_rooms_ok(service: UpdateReservationPrices, context: Context,
house, room):
service._rooms_repo = Mock(select=Mock(return_value=[room]))
context.house = house
result = service.select_rooms(context)
assert is_successful(result)
assert result.unwrap().rooms == {room.id: room}
def test_select_rooms_error(service: UpdateReservationPrices, context: Context,
house):
service._rooms_repo = Mock(select=Mock(side_effect=RuntimeError('ERR')))
context.house = house
result = service.select_rooms(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == 'ERR'