def test_retrive_role3(self): self.assertEquals(retrieve_role('sho_new_name'), ShoRole3)
def test_retrieve_unknowun_role(self): role = retrieve_role('unknowun_role') self.assertIsNone(role)
def save(self, *args, **kwargs): if self.account_type: role = retrieve_role(self.account_type) role.assign_role_to_user(self.user) super(UserProfile, self).save(*args, **kwargs)
def test_retrive_role2(self): self.assertEquals(retrieve_role('sho_role2'), ShoRole2)
def test_retrive_role1(self): self.assertEquals(retrieve_role("sho_role1"), ShoRole1)
def test_retrive_role2(self): self.assertEquals(retrieve_role('role2'), Role2)