def teardown_method(self, method): self.object_tank = None context.objects = ObjectsTank() super(ObjectTankTestCase, self).teardown_method(method)
def setup_method(self, method): super(ObjectTankTestCase, self).setup_method(method) context.objects = None self.object_tank = ObjectsTank()
class ObjectTankTestCase(BaseTestCase): def setup_method(self, method): super(ObjectTankTestCase, self).setup_method(method) context.objects = None self.object_tank = ObjectsTank() def teardown_method(self, method): self.object_tank = None context.objects = ObjectsTank() super(ObjectTankTestCase, self).teardown_method(method) def test_init(self): assert_that(self.object_tank, is_not(equal_to(None))) def test_register(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) registered_obj = self.object_tank.objects[1] assert_that(registered_obj, equal_to(dummy_obj)) assert_that(self.object_tank.relations, has_length(1)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) assert_that(self.object_tank.objects_by_type['dummy'], has_length(1)) def test_register_multi(self): dummy_obj = DummyObject() dummy_obj_2 = DummyObject() obj_id_1 = self.object_tank.register(dummy_obj) obj_id_2 = self.object_tank.register(dummy_obj_2) assert_that(self.object_tank.objects, has_length(2)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) for obj_id in (obj_id_1, obj_id_2): assert_that(self.object_tank.objects, has_key(obj_id)) assert_that(self.object_tank.relations, has_key(obj_id)) assert_that(self.object_tank.relations[obj_id], has_length(0)) assert_that(self.object_tank.objects_by_type['dummy'], has_item(obj_id)) def test_register_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) child_obj_id_2 = self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) assert_that(self.object_tank.objects, has_length(3)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) for obj_id in (parent_obj_id, child_obj_id_1, child_obj_id_2): assert_that(self.object_tank.objects, has_key(obj_id)) assert_that(self.object_tank.relations, has_key(obj_id)) assert_that(self.object_tank.objects_by_type['dummy'], has_item(obj_id)) assert_that(self.object_tank.relations[parent_obj_id], has_length(2)) for obj_id in (child_obj_id_1, child_obj_id_2): assert_that(self.object_tank.relations[parent_obj_id], has_item(obj_id)) def test_unregister(self): # TODO: Do some more unregister tests. dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) registered_obj = self.object_tank.objects[1] assert_that(registered_obj, equal_to(dummy_obj)) assert_that(self.object_tank.relations, has_length(1)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) assert_that(self.object_tank.objects_by_type['dummy'], has_length(1)) self.object_tank.unregister(obj_id=1) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(0)) assert_that(self.object_tank.relations, has_length(0)) assert_that(self.object_tank.objects_by_type, has_length(1)) # We don't actually delete the empty list. assert_that(self.object_tank.objects_by_type, has_key('dummy')) assert_that(self.object_tank.objects_by_type['dummy'], has_length(0)) def test_find_one(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj = self.object_tank.find_one(obj_id=1) assert_that(found_obj, equal_to(dummy_obj)) def test_find_all_single(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj_list = self.object_tank.find_all(obj_id=1) assert_that(found_obj_list, equal_to([dummy_obj])) def test_find_all_single_with_children(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj_list = self.object_tank.find_all(obj_id=1, children=True) assert_that(found_obj_list, equal_to([dummy_obj])) def test_find_all_single_by_type(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj_list = self.object_tank.find_all(types=('dummy', )) assert_that(found_obj_list, equal_to([dummy_obj])) def test_find_all_with_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) found_obj_list = self.object_tank.find_all(obj_id=parent_obj_id, children=True) assert_that(found_obj_list, has_length(3)) for obj in (dummy_obj, dummy_child_1, dummy_child_2): assert_that(found_obj_list, has_item(obj)) def test_find_all_only_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) found_obj_list = self.object_tank.find_all(parent_id=parent_obj_id) assert_that(found_obj_list, has_length(2)) for obj in (dummy_child_1, dummy_child_2): assert_that(found_obj_list, has_item(obj)) def test_tree_single(self): dummy_obj = DummyObject() obj_id = self.object_tank.register(dummy_obj) tree = self.object_tank.tree(obj_id) assert_that(tree, equal_to({'object': dummy_obj, 'children': []})) def test_tree_with_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) tree = self.object_tank.tree(parent_obj_id) assert_that( tree, equal_to({ 'object': dummy_obj, 'children': [{ 'object': dummy_child_1, 'children': [] }, { 'object': dummy_child_2, 'children': [] }] })) def test_sub_tree(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) tree = self.object_tank.tree(child_obj_id_1) assert_that(tree, equal_to({'object': dummy_child_1, 'children': []})) def test_nested_tree(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) self.object_tank.register(dummy_child_2, parent_id=child_obj_id_1) tree = self.object_tank.tree(parent_obj_id) assert_that( tree, equal_to({ 'object': dummy_obj, 'children': [{ 'object': dummy_child_1, 'children': [{ 'object': dummy_child_2, 'children': [] }] }] }))
def _setup_object_tank(self): from amplify.agent.tanks.objects import ObjectsTank self.objects = ObjectsTank() self.top_object = self.objects.root_object self.top_object_id = self.objects.root_id
class ObjectTankTestCase(BaseTestCase): def setup_method(self, method): super(ObjectTankTestCase, self).setup_method(method) context.objects = None self.object_tank = ObjectsTank() def teardown_method(self, method): self.object_tank = None context.objects = ObjectsTank() super(ObjectTankTestCase, self).teardown_method(method) def test_init(self): assert_that(self.object_tank, is_not(equal_to(None))) def test_register(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) registered_obj = self.object_tank.objects[1] assert_that(registered_obj, equal_to(dummy_obj)) assert_that(self.object_tank.relations, has_length(1)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) assert_that(self.object_tank.objects_by_type['dummy'], has_length(1)) def test_register_multi(self): dummy_obj = DummyObject() dummy_obj_2 = DummyObject() obj_id_1 = self.object_tank.register(dummy_obj) obj_id_2 = self.object_tank.register(dummy_obj_2) assert_that(self.object_tank.objects, has_length(2)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) for obj_id in (obj_id_1, obj_id_2): assert_that(self.object_tank.objects, has_key(obj_id)) assert_that(self.object_tank.relations, has_key(obj_id)) assert_that(self.object_tank.relations[obj_id], has_length(0)) assert_that(self.object_tank.objects_by_type['dummy'], has_item(obj_id)) def test_register_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) child_obj_id_2 = self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) assert_that(self.object_tank.objects, has_length(3)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) for obj_id in (parent_obj_id, child_obj_id_1, child_obj_id_2): assert_that(self.object_tank.objects, has_key(obj_id)) assert_that(self.object_tank.relations, has_key(obj_id)) assert_that(self.object_tank.objects_by_type['dummy'], has_item(obj_id)) assert_that(self.object_tank.relations[parent_obj_id], has_length(2)) for obj_id in (child_obj_id_1, child_obj_id_2): assert_that(self.object_tank.relations[parent_obj_id], has_item(obj_id)) def test_unregister(self): # TODO: Do some more unregister tests. dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) registered_obj = self.object_tank.objects[1] assert_that(registered_obj, equal_to(dummy_obj)) assert_that(self.object_tank.relations, has_length(1)) assert_that(self.object_tank.objects_by_type, has_length(1)) assert_that(self.object_tank.objects_by_type, has_key('dummy')) assert_that(self.object_tank.objects_by_type['dummy'], has_length(1)) self.object_tank.unregister(obj_id=1) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(0)) assert_that(self.object_tank.relations, has_length(0)) assert_that(self.object_tank.objects_by_type, has_length(1)) # We don't actually delete the empty list. assert_that(self.object_tank.objects_by_type, has_key('dummy')) assert_that(self.object_tank.objects_by_type['dummy'], has_length(0)) def test_find_one(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj = self.object_tank.find_one(obj_id=1) assert_that(found_obj, equal_to(dummy_obj)) def test_find_all_single(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj_list = self.object_tank.find_all(obj_id=1) assert_that(found_obj_list, equal_to([dummy_obj])) def test_find_all_single_with_children(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj_list = self.object_tank.find_all(obj_id=1, children=True) assert_that(found_obj_list, equal_to([dummy_obj])) def test_find_all_single_by_type(self): dummy_obj = DummyObject() self.object_tank.register(dummy_obj) assert_that(self.object_tank._ID_SEQUENCE, not_(equal_to(0))) assert_that(self.object_tank.objects, has_length(1)) assert_that(self.object_tank.objects, has_key(1)) found_obj_list = self.object_tank.find_all(types=('dummy',)) assert_that(found_obj_list, equal_to([dummy_obj])) def test_find_all_with_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) child_obj_id_2 = self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) found_obj_list = self.object_tank.find_all(obj_id=parent_obj_id, children=True) assert_that(found_obj_list, has_length(3)) for obj in (dummy_obj, dummy_child_1, dummy_child_2): assert_that(found_obj_list, has_item(obj)) def test_find_all_only_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) child_obj_id_2 = self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) found_obj_list = self.object_tank.find_all(parent_id=parent_obj_id) assert_that(found_obj_list, has_length(2)) for obj in (dummy_child_1, dummy_child_2): assert_that(found_obj_list, has_item(obj)) def test_tree_single(self): dummy_obj = DummyObject() obj_id = self.object_tank.register(dummy_obj) tree = self.object_tank.tree(obj_id) assert_that(tree, equal_to({'object': dummy_obj, 'children': []})) def test_tree_with_children(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) child_obj_id_2 = self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) tree = self.object_tank.tree(parent_obj_id) assert_that(tree, equal_to( { 'object': dummy_obj, 'children': [ { 'object': dummy_child_1, 'children': [] }, { 'object': dummy_child_2, 'children': [] } ] } )) def test_sub_tree(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) child_obj_id_2 = self.object_tank.register(dummy_child_2, parent_id=parent_obj_id) tree = self.object_tank.tree(child_obj_id_1) assert_that(tree, equal_to( { 'object': dummy_child_1, 'children': [] } )) def test_nested_tree(self): dummy_obj = DummyObject() dummy_child_1 = DummyObject() dummy_child_2 = DummyObject() parent_obj_id = self.object_tank.register(dummy_obj) child_obj_id_1 = self.object_tank.register(dummy_child_1, parent_id=parent_obj_id) child_obj_id_2 = self.object_tank.register(dummy_child_2, parent_id=child_obj_id_1) tree = self.object_tank.tree(parent_obj_id) assert_that(tree, equal_to( { 'object': dummy_obj, 'children': [ { 'object': dummy_child_1, 'children': [ { 'object': dummy_child_2, 'children': [] } ] } ] } ))