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
}])
