udata 9.1.2.dev30355__py2.py3-none-any.whl → 9.1.2.dev30454__py2.py3-none-any.whl

udata/harvest/tests/factories.py

@@ -1,64 +1,63 @@
 import factory
 import pytest
-
 from factory.fuzzy import FuzzyChoice
 from flask.signals import Namespace
 
-from udata.factories import ModelFactory
 from udata.core.dataset.factories import DatasetFactory
 from udata.core.dataset.models import Dataset
+from udata.factories import ModelFactory
 
 from .. import backends
-from ..models import HarvestItem, HarvestSource, HarvestJob
+from ..models import HarvestItem, HarvestJob, HarvestSource
 
 
 def dtfactory(start, end):
-    return factory.Faker('date_time_between', start_date=start, end_date=end)
+    return factory.Faker("date_time_between", start_date=start, end_date=end)
 
 
 class HarvestSourceFactory(ModelFactory):
     class Meta:
         model = HarvestSource
 
-    name = factory.Faker('name')
-    url = factory.Faker('url')
-    description = factory.Faker('text')
-    backend = 'factory'
+    name = factory.Faker("name")
+    url = factory.Faker("url")
+    description = factory.Faker("text")
+    backend = "factory"
 
 
 class HarvestJobFactory(ModelFactory):
     class Meta:
         model = HarvestJob
 
-    created = dtfactory('-3h', '-2h')
-    started = dtfactory('-2h', '-1h')
-    ended = dtfactory('-1h', 'now')
+    created = dtfactory("-3h", "-2h")
+    started = dtfactory("-2h", "-1h")
+    ended = dtfactory("-1h", "now")
     status = FuzzyChoice(HarvestJob.status.choices)
     source = factory.SubFactory(HarvestSourceFactory)
 
 
 ns = Namespace()
 
-mock_initialize = ns.signal('backend:initialize')
-mock_process = ns.signal('backend:process')
+mock_initialize = ns.signal("backend:initialize")
+mock_process = ns.signal("backend:process")
 
 DEFAULT_COUNT = 3
 
 
 class FactoryBackend(backends.BaseBackend):
-    name = 'factory'
+    name = "factory"
     filters = (
-        backends.HarvestFilter('Test', 'test', int, 'An integer'),
-        backends.HarvestFilter('Tag', 'tag', str),
+        backends.HarvestFilter("Test", "test", int, "An integer"),
+        backends.HarvestFilter("Tag", "tag", str),
     )
     features = (
-        backends.HarvestFeature('test', 'Test'),
-        backends.HarvestFeature('toggled', 'Toggled', 'A togglable', True),
+        backends.HarvestFeature("test", "Test"),
+        backends.HarvestFeature("toggled", "Toggled", "A togglable", True),
     )
 
     def inner_harvest(self):
         mock_initialize.send(self)
-        for i in range(self.config.get('count', DEFAULT_COUNT)):
+        for i in range(self.config.get("count", DEFAULT_COUNT)):
             self.process_dataset(str(i))
             if self.is_done():
                 return
@@ -67,15 +66,15 @@ class FactoryBackend(backends.BaseBackend):
         mock_process.send(self, item=item.remote_id)
 
         dataset = self.get_dataset(item.remote_id)
-        dataset.title = f'dataset-{item.remote_id}'
+        dataset.title = f"dataset-{item.remote_id}"
 
         return dataset
 
 
 class MockBackendsMixin(object):
-    '''A mixin mocking the harvest backend'''
+    """A mixin mocking the harvest backend"""
+
     @pytest.fixture(autouse=True)
     def mock_backend(self, mocker):
-        return_value = {'factory': FactoryBackend}
-        mocker.patch('udata.harvest.backends.get_all',
-                     return_value=return_value)
+        return_value = {"factory": FactoryBackend}
+        mocker.patch("udata.harvest.backends.get_all", return_value=return_value)