def set_mock_items(*items): db_api.__db_items = copy.copy(items) db_api.get_items.side_effect = ( tools.get_db_api_get_items(*db_api.__db_items)) db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*db_api.__db_items)) db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*db_api.__db_items)) db_api.get_items_ids.side_effect = ( tools.get_db_api_get_items_ids(*db_api.__db_items))
def set_mock_db_items(self, *items): self._db_items = copy.copy(items) self.db_api.get_items.side_effect = (tools.get_db_api_get_items( *self._db_items)) self.db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*self._db_items)) self.db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*self._db_items)) self.db_api.get_items_ids.side_effect = ( tools.get_db_api_get_items_ids(*self._db_items))
def set_mock_items(*items): db_api.__db_items = copy.copy(items) db_api.get_items.side_effect = (tools.get_db_api_get_items( *db_api.__db_items)) db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*db_api.__db_items)) db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*db_api.__db_items)) db_api.get_items_ids.side_effect = (tools.get_db_api_get_items_ids( *db_api.__db_items))
def set_mock_db_items(self, *items): self._db_items = copy.copy(items) self.db_api.get_items.side_effect = ( tools.get_db_api_get_items(*self._db_items)) self.db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*self._db_items)) self.db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*self._db_items)) self.db_api.get_items_ids.side_effect = ( tools.get_db_api_get_items_ids(*self._db_items))
def test_get_db_items(self, db_api): describer = image_api.ImageDescriber() describer.context = base.create_context() # NOTE(ft): the first requested image appears is user owend and public, # the second is absent db_api.get_items.side_effect = (tools.get_db_api_get_items()) db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(fakes.DB_IMAGE_1)) db_api.get_public_items.side_effect = [[fakes.DB_IMAGE_1], [], []] describer.ids = set([fakes.ID_EC2_IMAGE_1, fakes.ID_EC2_IMAGE_2]) self.assertRaises(exception.InvalidAMIIDNotFound, describer.get_db_items)
def test_get_db_items(self, db_api): describer = image_api.ImageDescriber() describer.context = base.create_context() # NOTE(ft): the first requested image appears is user owend and public, # the second is absent db_api.get_items.side_effect = ( tools.get_db_api_get_items()) db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(fakes.DB_IMAGE_1)) db_api.get_public_items.side_effect = [ [fakes.DB_IMAGE_1], [], []] describer.ids = set([fakes.ID_EC2_IMAGE_1, fakes.ID_EC2_IMAGE_2]) self.assertRaises(exception.InvalidAMIIDNotFound, describer.get_db_items)