From 81e6172fb1d24f59d87e439a18d74474201b711b Mon Sep 17 00:00:00 2001 From: Blade Barringer Date: Sun, 17 Jan 2016 13:42:09 -0600 Subject: [PATCH] tests: Tighten up generateGroup members helper --- .../api-integration/v3/object-generators.js | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/test/helpers/api-integration/v3/object-generators.js b/test/helpers/api-integration/v3/object-generators.js index 81bb855d56..1758998788 100644 --- a/test/helpers/api-integration/v3/object-generators.js +++ b/test/helpers/api-integration/v3/object-generators.js @@ -72,17 +72,14 @@ export async function createAndPopulateGroup (settings = {}) { let groupLeader = await generateUser(leaderDetails); let group = await generateGroup(groupLeader, groupDetails); + const groupMembershipTypes = { + party: { 'party._id': group._id}, + guild: { guilds: [group._id] }, + }; + let members = await Q.all( - times(numberOfMembers, async () => { - let user = await generateUser(); - - if (group.type === 'party') { - await user.update({ 'party._id': group._id}); - } else { - await user.update({ guilds: [group._id] }); - } - - return user; + times(numberOfMembers, () => { + return generateUser(groupMembershipTypes[group.type]); }) );