Address comments on code review

test/test_api_usage.py

@@ -166,6 +166,13 @@ class ApiTestCase(unittest.TestCase):
     parsed = py_json.loads(data)
     return parsed
 
+  def assertInTeam(self, data, membername):
+    for memberData in data['members']:
+      if memberData['name'] == membername:
+        return
+
+    self.fail(membername + ' not found in team: ' + json.dumps(data))
+
   def login(self, username, password='password'):
     return self.postJsonResponse(Signin, data=dict(username=username, password=password))
 
@@ -378,6 +385,28 @@ class TestCreateNewUser(ApiTestCase):
                              expected_code=201)
     self.assertEquals('"Created"', data)
 
+  def test_createuser_withteaminvite(self):
+    inviter = model.get_user(ADMIN_ACCESS_USER)
+    team = model.get_organization_team(ORGANIZATION, 'owners')
+    invite = model.add_or_invite_to_team(inviter, team, None, 'foo@example.com')
+     
+    details = {
+      'inviteCode': invite.invite_token
+    }
+    details.update(NEW_USER_DETAILS);
+
+    data = self.postResponse(User,
+                             data=details,
+                             expected_code=201)
+    self.assertEquals('"Created"', data)
+
+    # Make sure the user was added to the team.
+    self.login(ADMIN_ACCESS_USER)
+    json = self.getJsonResponse(TeamMemberList,
+                                params=dict(orgname=ORGANIZATION,
+                                            teamname='owners'))
+    self.assertInTeam(json, NEW_USER_DETAILS['username'])
+
 
 class TestSignout(ApiTestCase):
   def test_signout(self):
@@ -743,13 +772,6 @@ class TestGetOrganizationTeamMembers(ApiTestCase):
 
 
 class TestUpdateOrganizationTeamMember(ApiTestCase):
-  def assertInTeam(self, data, membername):
-    for memberData in data['members']:
-      if memberData['name'] == membername:
-        return
-
-    self.fail(membername + ' not found in team: ' + json.dumps(data))
-
   def test_addmember_alreadyteammember(self):
     self.login(ADMIN_ACCESS_USER)