def test_load_all_athletes(self, mongo_mock):
mongo_mock.db.athletes.find.return_value = iter([
{'id': 123, 'name': 'foobar', 'sex': 'M'},
{'id': 456, 'name': 'baz', 'sex': 'F'}
])
result = load_athletes(mongo_mock)
self.assertEqual(len(result), 2)
self.assertItemsEqual(result.id, [123, 456])
self.assertItemsEqual(result.name, ['foobar', 'baz'])
self.assertItemsEqual(result.sex, ['M', 'F'])
def test_load_athletes_with_simple_filter(self, mongo_mock):
load_athletes(mongo_mock, sex='F')
mongo_mock.db.athletes.find.assert_called_with({'sex': 'F'})