from ckan.new_tests.helpers import reset_db, call_action from ckan.new_tests.factories import Organization, Group from ckan import model from ckan.plugins import toolkit from ckanext.harvest.tests.factories import (HarvestSourceObj, HarvestJobObj, HarvestObjectObj) from ckanext.harvest.tests.lib import run_harvest import ckanext.harvest.model as harvest_model from ckanext.harvest.harvesters.base import HarvesterBase from ckanext.harvest.harvesters.ckanharvester import CKANHarvester import mock_ckan # Start CKAN-alike server we can test harvesting against it mock_ckan.serve() def was_last_job_considered_error_free(): last_job = model.Session.query(harvest_model.HarvestJob) \ .order_by(harvest_model.HarvestJob.created.desc()) \ .first() job = MagicMock() job.source = last_job.source job.id = '' return bool(HarvesterBase.last_error_free_job(job)) class TestCkanHarvester(object): @classmethod def setup(cls):
except ImportError: from ckan.new_tests.helpers import reset_db, call_action from ckan.new_tests.factories import Organization, Group from ckan import model from ckan.plugins import toolkit from ckanext.harvest.tests.factories import (HarvestSourceObj, HarvestJobObj, HarvestObjectObj) from ckanext.harvest.tests.lib import run_harvest import ckanext.harvest.model as harvest_model from ckanext.harvest.harvesters.ckanharvester import CKANHarvester import mock_ckan # Start CKAN-alike server we can test harvesting against it mock_ckan.serve() def was_last_job_considered_error_free(): last_job = model.Session.query(harvest_model.HarvestJob) \ .order_by(harvest_model.HarvestJob.created.desc()) \ .first() job = MagicMock() job.source = last_job.source job.id = '' return bool(CKANHarvester._last_error_free_job(job)) class TestCkanHarvester(object): @classmethod def setup(cls):