Return multiple items from a mocked function with Python's mock.

gistfile1.py

import mock

def returnList(items):
  def func():
    for item in items:
      yield item
    yield mock.DEFAULT

  generator = func()

  def effect(*args, **kwargs):
    return generator.next()

  return effect


m = mock.Mock()
m.side_effect = returnList([1,2,3])
for i in ['a', 'b', 'c']:
    print i, m()

Turns out you can also just set a side effect to any iterable, including a list:

m = mock.Mock()
m.side_effect = [1, 2, 3]
for i in ['a', 'b', 'c']:
    print(i, m())

http://stackoverflow.com/questions/24897145/python-mock-multiple-return-values

Thanks for your great ideas, vpetro.
By the way, you can always use return_values.pop instead of lambda: return_values.pop().