def test_create_many_users(self):
config = Mock()
users = [{'uniqueid' : sentinel.uniqueid1,
'email' : sentinel.email1,
'firstname': sentinel.firstname1,
'lastname' : sentinel.lastname1,
'plan_id' : sentinel.plan_id1,
},
{'uniqueid' : sentinel.uniqueid2,
'email' : sentinel.email2,
'firstname': sentinel.firstname2,
'lastname' : sentinel.lastname2,
'plan_id' : sentinel.plan_id2,
},
{'uniqueid' : sentinel.uniqueid3,
'email' : sentinel.email3,
'firstname': sentinel.firstname3,
'lastname' : sentinel.lastname3,
'plan_id' : sentinel.plan_id3,
},]
returns = [{'success' : True,
'avatar_id' : sentinel.avatar_id1,
'account_id': sentinel.account_id1,
'server_assigned_password' : sentinel.s_a_p1,
},
{'success' : True,
'avatar_id' : sentinel.avatar_id2,
'account_id': sentinel.account_id2,
'server_assigned_password' : sentinel.s_a_p2,
},
{'success' : True,
'avatar_id' : sentinel.avatar_id3,
'account_id': sentinel.account_id3,
'server_assigned_password' : sentinel.s_a_p3,
},]
def side_effect(*args):
return returns.pop(0)
self.api_iface_create.side_effect = side_effect
results = group_manager._api_create_users(config, users)
self.assertEqual(len(results), 3)
self.assertIs(results[0]['uniqueid'], sentinel.uniqueid1)
self.assertIs(results[0]['avatar_id'], sentinel.avatar_id1)
self.assertIs(results[1]['uniqueid'], sentinel.uniqueid2)
self.assertIs(results[1]['avatar_id'], sentinel.avatar_id2)
self.assertIs(results[2]['uniqueid'], sentinel.uniqueid3)
self.assertIs(results[2]['avatar_id'], sentinel.avatar_id3)
def test_create_one_user(self):
config = Mock()
users = [{'uniqueid' : sentinel.uniqueid,
'email' : sentinel.email,
'firstname': sentinel.firstname,
'lastname' : sentinel.lastname,
'plan_id' : sentinel.plan_id,
}]
server_ok = {'success' : True,
'avatar_id' : sentinel.avatar_id,
'account_id': sentinel.account_id,
'server_assigned_password': sentinel.server_assigned_password,
}
self.api_iface_create.return_value = server_ok
results = group_manager._api_create_users(config, users)
self.assertEqual(len(results), 1)
self.assertIs(results[0]['uniqueid'], sentinel.uniqueid)
self.assertIs(results[0]['avatar_id'], sentinel.avatar_id)
def test_create_no_users(self):
config = Mock()
users = []
self.assertEqual(group_manager._api_create_users(config, users),
[])
