def _addUsersAndFriends(self):
from lifescore.models import User, Friend
user_1 = User(1, None, None, 'user_1', None, None, 850)
user_2 = User(2, None, None, 'user_2', None, None, 840)
friends_1 = [Friend(i, 'name_%d' % i) for i in range(2, 32)]
friends_2 = [Friend(i, 'name_%d' % i) for i in range(32, 51)]
user_1.friends = friends_1
user_2.friends = friends_2
self.session.add(user_1)
self.session.add(user_2)
self.session.flush()
