def get_valid_group(id, name): """ Returns a valid group object """ group = Group() group.id = id group.name = name return group
def test_to_dict(self): """ Get object as dictionary """ enterprise = Enterprise() enterprise.id = 3 enterprise.name ="NewEnterprise" # List of objects admingroup = Group() admingroup.name = "Admins" othergroup = Group() othergroup.name = "Others" enterprise.groups = [admingroup, othergroup] # Object ceo = User() ceo.firstname = 'John' ceo.lastname = 'Doe' enterprise.ceo = ceo to_dict = enterprise.to_dict() self.assertEquals(sorted(to_dict.keys()), sorted(['groups', 'token', 'lastUpdatedDate', 'allowedForwardingClasses', 'name', 'ceo', 'parentType', 'parentID', 'owner', 'creationDate', 'ID', 'description'])) self.assertEquals(to_dict['name'], 'NewEnterprise') self.assertEquals(to_dict['ID'], 3) #self.assertEquals(to_dict['externalID'], None) #self.assertEquals(to_dict['localID'], None) self.assertEquals(to_dict['parentID'], None) self.assertEquals(to_dict['parentType'], None) self.assertEquals(to_dict['owner'], None) self.assertEquals(to_dict['creationDate'], None) self.assertEquals(to_dict['ceo'], { 'APIKey': None, 'APIKeyExpiry': None, 'ID': None, 'avatarData': None, 'avatarType': None, 'creationDate': None, 'email': None, 'enterpriseID': None, 'enterpriseName': None, 'firstName': 'John', 'lastName': 'Doe', 'owner': None, 'parentID': None, 'parentType': None, 'password': None, 'role': None, 'userName': None, 'lastUpdatedDate': None }) self.assertEquals(to_dict['groups'], [{ 'ID': None, 'creationDate': None, 'lastUpdatedDate': None, 'name': 'Admins', 'owner': None, 'parentID': None, 'parentType': None }, { 'ID': None, 'creationDate': None, 'lastUpdatedDate': None, 'name': 'Others', 'owner': None, 'parentID': None, 'parentType': None }])