def test_contains(self): """ Fetcher contains object """ user = User() group1 = Group(id='xxxx-xxxx-xxx', name="group1") group2 = Group(id='yyyy-yyyy-yyy', name="group2") group3 = Group(id='zzzz-zzzz-zzz', name="group3") user.add_child(group1) user.add_child(group2) self.assertEquals(group1 in user.groups, True) self.assertEquals(group2 in user.groups, True) self.assertEquals(group3 in user.groups, False)
def test_index(self): """ Fetcher index object """ user = User() group1 = Group(id='xxxx-xxxx-xxx', name="group1") group2 = Group(id='yyyy-yyyy-yyy', name="group2") group3 = Group(id='zzzz-zzzz-zzz', name="group3") user.add_child(group1) user.add_child(group2) self.assertEquals(user.groups.index(group1), 0) self.assertEquals(user.groups.index(group2), 1) with self.assertRaises(ValueError): user.groups.index(group3)
def test_flush(self): """ Flush fetcher """ user = User() group1 = Group(name="group1") group2 = Group(name="group2") group3 = Group(name="group3") user.add_child(group1) user.add_child(group2) user.groups.append(group3) self.assertEquals(user.groups, [group1, group2, group3]) user.groups.flush() self.assertEquals(user.groups, [])
def test_parents_relationship(self): """ Test is parent obects are correctly set""" user = User() group1 = Group(id='xxxx-xxxx-xxx', name="group1") user.add_child(group1) self.assertEquals(group1.parent_object, user) user.remove_child(group1) self.assertEquals(group1.parent_object, None) user.add_child(group1) updated_group = Group(id='xxxx-xxxx-xxx', name="group-updated") user.update_child(updated_group) self.assertEquals(user.groups[0], updated_group) self.assertEquals(user.groups[0].name, 'group-updated') with self.assertRaises(InternalConsitencyError): user.add_child(user)