Allows to pass the group when creating users

HISTORY.rst

@@ -1,5 +1,9 @@
 Changelog
 ---------
+
+- Allows to pass the group when creating users.
+  [msom]
+
 0.15.0 (2017-08-17)
 ~~~~~~~~~~~~~~~~~~~
 

onegov/user/collections/user.py

@@ -90,7 +90,7 @@ def query(self):
 
     def add(self, username, password, role,
             data=None, second_factor=None, active=True, realname=None,
-            signup_token=None):
+            signup_token=None, group=None):
         """ Add a user to the collection.
 
             The arguments given to this function are the attributes of the
@@ -109,7 +109,8 @@ def add(self, username, password, role,
             second_factor=second_factor,
             active=active,
             realname=realname,
-            signup_token=signup_token
+            signup_token=signup_token,
+            group_id=group.id if group else None
         )
 
         self.session.add(user)

onegov/user/tests/test_collections.py

@@ -19,6 +19,12 @@ def test_user_add(session):
 
     assert users.by_username(user.username).username == 'dmr@example.org'
 
+    groups = UserGroupCollection(session)
+    group = groups.add(name='group')
+
+    user = users.add('abc@example.org', 'p@ssw0rd', 'root', group=group)
+    assert users.by_username(user.username).group.name == 'group'
+
 
 def test_user_add_conflict(session):