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

udata/tests/helpers.py

@@ -1,37 +1,32 @@
-import mock
 import os
-
+from contextlib import contextmanager
+from datetime import timedelta
 from io import BytesIO
+from urllib.parse import parse_qs, urljoin, urlparse
 
+import mock
+from flask import current_app, json, request, url_for
+from flask_security.babel import FsDomain
 from PIL import Image
 
 from udata.mail import mail_sent
 
-from contextlib import contextmanager
-from datetime import timedelta
-from urllib.parse import urljoin, urlparse, parse_qs
-
-from flask import request, url_for, json, current_app
-from flask_security.babel import FsDomain
-
 
 def assert_equal_dates(datetime1, datetime2, limit=1):  # Seconds.
     """Lax date comparison, avoid comparing milliseconds and seconds."""
     __tracebackhide__ = True
-    delta = (datetime1 - datetime2)
+    delta = datetime1 - datetime2
     assert timedelta(seconds=-limit) <= delta <= timedelta(seconds=limit)
 
 
 def assert_starts_with(haystack, needle):
     __tracebackhide__ = True
-    msg = '{haystack} does not start with {needle}'.format(
-        haystack=haystack, needle=needle
-    )
+    msg = "{haystack} does not start with {needle}".format(haystack=haystack, needle=needle)
     assert haystack.startswith(needle), msg
 
 
 def assert_json_equal(first, second):
-    '''Ensure two dict produce the same JSON'''
+    """Ensure two dict produce the same JSON"""
     __tracebackhide__ = True
     json1 = json.loads(json.dumps(first))
     json2 = json.loads(json.dumps(second))
@@ -55,12 +50,12 @@ def mock_signals(callback, *signals):
 
     for signal, mock_handler in specs:
         signal.disconnect(mock_handler)
-        signal_name = getattr(signal, 'name', str(signal))
+        signal_name = getattr(signal, "name", str(signal))
         callback(signal_name, mock_handler)
 
 
 @contextmanager
-def assert_emit(*signals, assertions_callback = None):
+def assert_emit(*signals, assertions_callback=None):
     __tracebackhide__ = True
     msg = 'Signal "{0}" should have been emitted'
 
@@ -100,8 +95,8 @@ def capture_mails():
 
 
 REDIRECT_CODES = (301, 302, 303, 305, 307, 308)
-REDIRECT_MSG = 'HTTP Status {} expected but got {{}}'.format(
-    ', '.join(str(code) for code in REDIRECT_CODES)
+REDIRECT_MSG = "HTTP Status {} expected but got {{}}".format(
+    ", ".join(str(code) for code in REDIRECT_CODES)
 )
 
 
@@ -132,12 +127,11 @@ def assert_status(response, status_code, message=None):
     """
     __tracebackhide__ = True
 
-    message = message or 'HTTP Status %s expected but got %s' \
-                         % (status_code, response.status_code)
-    if response.mimetype == 'application/json':
+    message = message or "HTTP Status %s expected but got %s" % (status_code, response.status_code)
+    if response.mimetype == "application/json":
         try:
-            second_line = 'Response content is {0}'.format(response.json)
-            message = '\n'.join((message, second_line))
+            second_line = "Response content is {0}".format(response.json)
+            message = "\n".join((message, second_line))
         except Exception:
             pass
     assert response.status_code == status_code, message
@@ -189,18 +183,18 @@ def assert500(response):
 
 
 def full_url(*args, **kwargs):
-    '''Build a full URL'''
+    """Build a full URL"""
     return urljoin(request.url_root, url_for(*args, **kwargs))
 
 
 def data_path(filename):
-    '''Get a test data path'''
-    return os.path.join(os.path.dirname(__file__), 'data', filename)
+    """Get a test data path"""
+    return os.path.join(os.path.dirname(__file__), "data", filename)
 
 
 def assert_command_ok(result):
     __tracebackhide__ = True
-    msg = 'Command failed with exit code {0.exit_code} and output:\n{0.output}'
+    msg = "Command failed with exit code {0.exit_code} and output:\n{0.output}"
     assert result.exit_code == 0, msg.format(result)
 
 
@@ -208,28 +202,29 @@ def assert_urls_equal(url1, url2):
     __tracebackhide__ = True
     p1 = urlparse(url1)
     p2 = urlparse(url2)
-    assert p1.scheme == p2.scheme, 'Scheme does not match'
-    assert p1.netloc == p2.netloc, 'Network location does not match'
-    assert p1.path == p2.path, 'Path does not match'
+    assert p1.scheme == p2.scheme, "Scheme does not match"
+    assert p1.netloc == p2.netloc, "Network location does not match"
+    assert p1.path == p2.path, "Path does not match"
     q1 = parse_qs(p1.query)
     q2 = parse_qs(p2.query)
-    assert q1 == q2, 'Query does not match'
-    assert p1.fragment == p2.fragment, 'Fragment does not match'
+    assert q1 == q2, "Query does not match"
+    assert p1.fragment == p2.fragment, "Fragment does not match"
 
 
 def assert_cors(response):
-    '''CORS headers presence assertion'''
+    """CORS headers presence assertion"""
     __tracebackhide__ = True
-    assert 'Access-Control-Allow-Origin' in response.headers
+    assert "Access-Control-Allow-Origin" in response.headers
 
 
 def create_test_image():
     file = BytesIO()
-    image = Image.new('RGBA', size=(50, 50), color=(155, 0, 0))
-    image.save(file, 'png')
-    file.name = 'test.png'
+    image = Image.new("RGBA", size=(50, 50), color=(155, 0, 0))
+    image.save(file, "png")
+    file.name = "test.png"
     file.seek(0)
     return file
 
+
 def security_gettext(string):
     return FsDomain(current_app).gettext(string)

udata/tests/models.py

@@ -1,11 +1,11 @@
 # -*- coding: utf-8 -*-
 from __future__ import unicode_literals
 
-from udata.models import db, WithMetrics
+from udata.models import WithMetrics, db
 
 
 class FakeModel(WithMetrics, db.Document):
     name = db.StringField()
 
     def __unicode__(self):
-        return self.name or ''
+        return self.name or ""

udata/tests/organization/test_csv_adapter.py

@@ -1,7 +1,6 @@
-
 import pytest
 
-from udata.core.dataset.csv import ResourcesCsvAdapter, DatasetCsvAdapter
+from udata.core.dataset.csv import DatasetCsvAdapter, ResourcesCsvAdapter
 from udata.core.dataset.factories import DatasetFactory, ResourceFactory
 from udata.core.dataset.models import Dataset
 from udata.core.organization.csv import OrganizationCsvAdapter
@@ -10,22 +9,28 @@ from udata.core.organization.models import Organization
 
 
 @pytest.mark.frontend
-@pytest.mark.usefixtures('clean_db')
+@pytest.mark.usefixtures("clean_db")
 class OrganizationCSVAdapterTest:
-
     def test_organization_downloads_counts(self):
         org_with_dataset = OrganizationFactory()
         org_without_dataset = OrganizationFactory()
 
-        DatasetFactory(organization=org_with_dataset, resources=[
-            ResourceFactory(metrics={
-                'views': 42,
-            }),
-            ResourceFactory(metrics={
-                'views': 1337,
-            }),
-            ResourceFactory(),
-        ])
+        DatasetFactory(
+            organization=org_with_dataset,
+            resources=[
+                ResourceFactory(
+                    metrics={
+                        "views": 42,
+                    }
+                ),
+                ResourceFactory(
+                    metrics={
+                        "views": 1337,
+                    }
+                ),
+                ResourceFactory(),
+            ],
+        )
         DatasetFactory(organization=org_with_dataset, resources=[])
         adapter = OrganizationCsvAdapter(Organization.objects.all())
 
@@ -33,10 +38,10 @@ class OrganizationCSVAdapterTest:
         csv = {}
         for row in adapter.rows():
             values = dict(zip(adapter.header(), row))
-            csv[values['id']] = values
+            csv[values["id"]] = values
 
         org_values = csv[str(org_with_dataset.id)]
-        assert org_values['downloads'] == 1337 + 42
+        assert org_values["downloads"] == 1337 + 42
 
         org_values = csv[str(org_without_dataset.id)]
-        assert org_values['downloads'] == 0
+        assert org_values["downloads"] == 0

udata/tests/organization/test_notifications.py

@@ -1,27 +1,20 @@
 import pytest
 
-from udata.models import MembershipRequest, Member
-
-from udata.core.user.factories import UserFactory
 from udata.core.organization.factories import OrganizationFactory
-from udata.core.organization.notifications import (
-    membership_request_notifications
-)
-
+from udata.core.organization.notifications import membership_request_notifications
+from udata.core.user.factories import UserFactory
+from udata.models import Member, MembershipRequest
 from udata.tests.helpers import assert_equal_dates
 
 
-@pytest.mark.usefixtures('clean_db')
+@pytest.mark.usefixtures("clean_db")
 class OrganizationNotificationsTest:
     def test_pending_membership_requests(self):
         admin = UserFactory()
         editor = UserFactory()
         applicant = UserFactory()
-        request = MembershipRequest(user=applicant, comment='test')
-        members = [
-            Member(user=editor, role='editor'),
-            Member(user=admin, role='admin')
-        ]
+        request = MembershipRequest(user=applicant, comment="test")
+        members = [Member(user=editor, role="editor"), Member(user=admin, role="admin")]
         org = OrganizationFactory(members=members, requests=[request])
 
         assert len(membership_request_notifications(applicant)) is 0
@@ -31,8 +24,8 @@ class OrganizationNotificationsTest:
         assert len(notifications) is 1
         dt, details = notifications[0]
         assert_equal_dates(dt, request.created)
-        assert details['id'] == request.id
-        assert details['organization'] == org.id
-        assert details['user']['id'] == applicant.id
-        assert details['user']['fullname'] == applicant.fullname
-        assert details['user']['avatar'] == str(applicant.avatar)
+        assert details["id"] == request.id
+        assert details["organization"] == org.id
+        assert details["user"]["id"] == applicant.id
+        assert details["user"]["fullname"] == applicant.fullname
+        assert details["user"]["avatar"] == str(applicant.avatar)

udata/tests/organization/test_organization_model.py

@@ -1,18 +1,17 @@
 from datetime import datetime
 
 from udata.core.dataset.factories import DatasetFactory, HiddenDatasetFactory
-from udata.core.organization.factories import OrganizationFactory
 from udata.core.followers.signals import on_follow, on_unfollow
+from udata.core.organization.factories import OrganizationFactory
 from udata.core.reuse.factories import ReuseFactory, VisibleReuseFactory
 from udata.core.user.factories import UserFactory
-from udata.models import Dataset, Reuse, Follow, Member
+from udata.models import Dataset, Follow, Member, Reuse
 from udata.tests.helpers import assert_emit
 
-from .. import TestCase, DBTestMixin
+from .. import DBTestMixin, TestCase
 
 
 class OrganizationModelTest(TestCase, DBTestMixin):
-
     # Load metrics
     import udata.core.organization.metrics  # noqa
     import udata.core.followers.metrics  # noqa
@@ -20,7 +19,7 @@ class OrganizationModelTest(TestCase, DBTestMixin):
     def test_organization_metrics(self):
         # Members count update are in API calls, thus being tested in API dedicated tests
 
-        member = Member(user=UserFactory(), role='admin')
+        member = Member(user=UserFactory(), role="admin")
         org = OrganizationFactory(members=[member])
 
         with assert_emit(Reuse.on_create):
@@ -30,12 +29,13 @@ class OrganizationModelTest(TestCase, DBTestMixin):
             dataset = DatasetFactory(organization=org)
             HiddenDatasetFactory(organization=org)
         with assert_emit(on_follow):
-            follow = Follow.objects.create(following=org, follower=UserFactory(),
-                                           since=datetime.utcnow())
+            follow = Follow.objects.create(
+                following=org, follower=UserFactory(), since=datetime.utcnow()
+            )
 
-        assert org.get_metrics()['datasets'] == 1
-        assert org.get_metrics()['reuses'] == 1
-        assert org.get_metrics()['followers'] == 1
+        assert org.get_metrics()["datasets"] == 1
+        assert org.get_metrics()["reuses"] == 1
+        assert org.get_metrics()["followers"] == 1
 
         with assert_emit(Reuse.on_delete):
             reuse.deleted = datetime.utcnow()
@@ -47,6 +47,6 @@ class OrganizationModelTest(TestCase, DBTestMixin):
             follow.until = datetime.utcnow()
             follow.save()
 
-        assert org.get_metrics()['datasets'] == 0
-        assert org.get_metrics()['reuses'] == 0
-        assert org.get_metrics()['followers'] == 0
+        assert org.get_metrics()["datasets"] == 0
+        assert org.get_metrics()["reuses"] == 0
+        assert org.get_metrics()["followers"] == 0

udata/tests/organization/test_organization_rdf.py

@@ -1,16 +1,15 @@
 from flask import url_for
-
-from rdflib import URIRef, Literal, BNode
-from rdflib.namespace import RDF, FOAF, RDFS
+from rdflib import BNode, Literal, URIRef
+from rdflib.namespace import FOAF, RDF, RDFS
 from rdflib.resource import Resource as RdfResource
 
 from udata import api
-from udata.rdf import DCAT, DCT, HYDRA
-from udata.tests import TestCase, DBTestMixin
-from udata.core.organization.factories import OrganizationFactory
-from udata.core.organization.rdf import organization_to_rdf, build_org_catalog
 from udata.core.dataset.factories import DatasetFactory
 from udata.core.dataset.models import Dataset
+from udata.core.organization.factories import OrganizationFactory
+from udata.core.organization.rdf import build_org_catalog, organization_to_rdf
+from udata.rdf import DCAT, DCT, HYDRA
+from udata.tests import DBTestMixin, TestCase
 from udata.utils import faker
 
 
@@ -36,9 +35,7 @@ class OrganizationToRdfTest(DBTestMixin, TestCase):
 
     def test_all_fields(self):
         org = OrganizationFactory(url=faker.uri())
-        org_url = url_for('api.organization',
-                          org=org.id,
-                          _external=True)
+        org_url = url_for("api.organization", org=org.id, _external=True)
         o = organization_to_rdf(org)
         g = o.graph
 
@@ -55,7 +52,7 @@ class OrganizationToRdfTest(DBTestMixin, TestCase):
 
     def test_catalog(self):
         origin_org = OrganizationFactory()
-        uri = url_for('api.organization_rdf', org=origin_org.id, _external=True)
+        uri = url_for("api.organization_rdf", org=origin_org.id, _external=True)
 
         datasets = DatasetFactory.create_batch(3, organization=origin_org)
         catalog = build_org_catalog(origin_org, datasets)
@@ -88,16 +85,28 @@ class OrganizationToRdfTest(DBTestMixin, TestCase):
         origin_org = OrganizationFactory()
         page_size = 3
         total = 4
-        uri = url_for('api.organization_rdf', org=origin_org.id, _external=True)
-        uri_first = url_for('api.organization_rdf_format', org=origin_org.id, format='json',
-                            page=1, page_size=page_size, _external=True)
-        uri_last = url_for('api.organization_rdf_format', org=origin_org.id, format='json',
-                           page=2, page_size=page_size, _external=True)
+        uri = url_for("api.organization_rdf", org=origin_org.id, _external=True)
+        uri_first = url_for(
+            "api.organization_rdf_format",
+            org=origin_org.id,
+            format="json",
+            page=1,
+            page_size=page_size,
+            _external=True,
+        )
+        uri_last = url_for(
+            "api.organization_rdf_format",
+            org=origin_org.id,
+            format="json",
+            page=2,
+            page_size=page_size,
+            _external=True,
+        )
         DatasetFactory.create_batch(total, organization=origin_org)
 
         # First page
         datasets = Dataset.objects.paginate(1, page_size)
-        catalog = build_org_catalog(origin_org, datasets, format='json')
+        catalog = build_org_catalog(origin_org, datasets, format="json")
         graph = catalog.graph
 
         self.assertIsInstance(catalog, RdfResource)
@@ -110,8 +119,7 @@ class OrganizationToRdfTest(DBTestMixin, TestCase):
 
         self.assertEqual(len(list(catalog.objects(DCAT.dataset))), page_size)
 
-        paginations = list(graph.subjects(RDF.type,
-                                          HYDRA.PartialCollectionView))
+        paginations = list(graph.subjects(RDF.type, HYDRA.PartialCollectionView))
         self.assertEqual(len(paginations), 1)
         pagination = graph.resource(paginations[0])
         self.assertEqual(pagination.identifier, URIRef(uri_first))
@@ -122,7 +130,7 @@ class OrganizationToRdfTest(DBTestMixin, TestCase):
 
         # Second page
         datasets = Dataset.objects.paginate(2, page_size)
-        catalog = build_org_catalog(origin_org, datasets, format='json')
+        catalog = build_org_catalog(origin_org, datasets, format="json")
         graph = catalog.graph
 
         self.assertIsInstance(catalog, RdfResource)
@@ -135,8 +143,7 @@ class OrganizationToRdfTest(DBTestMixin, TestCase):
 
         self.assertEqual(len(list(catalog.objects(DCAT.dataset))), 1)
 
-        paginations = list(graph.subjects(RDF.type,
-                                          HYDRA.PartialCollectionView))
+        paginations = list(graph.subjects(RDF.type, HYDRA.PartialCollectionView))
         self.assertEqual(len(paginations), 1)
         pagination = graph.resource(paginations[0])
         self.assertEqual(pagination.identifier, URIRef(uri_last))

udata/tests/organization/test_organization_tasks.py

@@ -3,9 +3,9 @@ from flask import url_for
 from udata.api.oauth2 import OAuth2Client
 from udata.core import storages
 from udata.core.dataset.factories import DatasetFactory, ResourceFactory
-from udata.core.user.factories import AdminFactory
 from udata.core.organization import tasks
-from udata.models import Dataset, Organization, Transfer, Member, ContactPoint
+from udata.core.user.factories import AdminFactory
+from udata.models import ContactPoint, Dataset, Member, Organization, Transfer
 from udata.tests.api import APITestCase
 from udata.tests.helpers import create_test_image
 
@@ -13,14 +13,14 @@ from udata.tests.helpers import create_test_image
 class OrganizationTasksTest(APITestCase):
     def test_purge_organizations(self):
         self.login()
-        member = Member(user=self.user, role='admin')
-        org = Organization.objects.create(name='delete me', description='XXX', members=[member])
+        member = Member(user=self.user, role="admin")
+        org = Organization.objects.create(name="delete me", description="XXX", members=[member])
         data = {
-            'email': 'mooneywayne@cobb-cochran.com',
-            'name': 'Martin Schultz',
-            'organization': str(org.id)
+            "email": "mooneywayne@cobb-cochran.com",
+            "name": "Martin Schultz",
+            "organization": str(org.id),
         }
-        response = self.post(url_for('api.contact_points'), data)
+        response = self.post(url_for("api.contact_points"), data)
         self.assert201(response)
 
         self.assertEqual(ContactPoint.objects().count(), 1)
@@ -33,33 +33,32 @@ class OrganizationTasksTest(APITestCase):
         user = AdminFactory()
         self.login(user)
         response = self.post(
-            url_for('api.organization_logo', org=org),
-            {'file': (file, 'test.png')},
-            json=False)
+            url_for("api.organization_logo", org=org), {"file": (file, "test.png")}, json=False
+        )
         self.assert200(response)
 
         transfer_to_org = Transfer.objects.create(
             owner=user,
             recipient=org,
             subject=dataset,
-            comment='comment',
+            comment="comment",
         )
         transfer_from_org = Transfer.objects.create(
             owner=org,
             recipient=user,
             subject=dataset,
-            comment='comment',
+            comment="comment",
         )
 
         oauth_client = OAuth2Client.objects.create(
-            name='test-client',
+            name="test-client",
             owner=user,
             organization=org,
-            redirect_uris=['https://test.org/callback'],
+            redirect_uris=["https://test.org/callback"],
         )
 
         # Delete organization
-        response = self.delete(url_for('api.organization', org=org))
+        response = self.delete(url_for("api.organization", org=org))
         self.assert204(response)
 
         tasks.purge_organizations()
@@ -79,5 +78,5 @@ class OrganizationTasksTest(APITestCase):
         dataset = Dataset.objects(id=dataset.id).first()
         self.assertIsNone(dataset.organization)
 
-        organization = Organization.objects(name='delete me').first()
+        organization = Organization.objects(name="delete me").first()
         self.assertIsNone(organization)