def testGetRoles(self): test_ob = LocalRoles(None) test_ob.context = Mock() test_ob.context.members = set(['member1']) test_ob.context.managers = set(['manager1']) test_ob.context.invitedMembers = set(['invited1']) test_ob.context.invitedManagers = set(['invited3']) self.assertEquals( set(['Reader']), set(test_ob.getRoles('invited1'))) self.assertEquals( set(['Reader']), set(test_ob.getRoles('invited3'))) self.assertEquals( set(['Contributor']), set(test_ob.getRoles('member1'))) self.assertEquals( set(['Editor', 'Contributor']), set(test_ob.getRoles('manager1')))