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

udata/tests/api/test_follow_api.py

@@ -1,10 +1,9 @@
 from flask import url_for
 
 from udata.api import api
-from udata.models import db, Follow
-
 from udata.core.followers.api import FollowAPI
 from udata.core.followers.signals import on_follow, on_unfollow
+from udata.models import Follow, db
 
 from . import APITestCase
 
@@ -16,7 +15,7 @@ class FakeModel(db.Document):
         pass
 
 
-@api.route('/fake/<id>/follow/', endpoint='follow_fake')
+@api.route("/fake/<id>/follow/", endpoint="follow_fake")
 class FollowFakeAPI(FollowAPI):
     model = FakeModel
 
@@ -30,34 +29,33 @@ class FollowAPITest(APITestCase):
         self.signal_emitted = True
 
     def test_follow(self):
-        '''It should follow on POST'''
+        """It should follow on POST"""
         user = self.login()
         to_follow = FakeModel.objects.create()
 
         with on_follow.connected_to(self.handler):
-            response = self.post(url_for('api.follow_fake', id=to_follow.id))
+            response = self.post(url_for("api.follow_fake", id=to_follow.id))
 
         self.assert201(response)
 
         nb_followers = Follow.objects.followers(to_follow).count()
 
-        self.assertEqual(response.json['followers'], nb_followers)
+        self.assertEqual(response.json["followers"], nb_followers)
         self.assertEqual(Follow.objects.following(to_follow).count(), 0)
         self.assertEqual(nb_followers, 1)
-        self.assertIsInstance(Follow.objects.followers(to_follow).first(),
-                              Follow)
+        self.assertIsInstance(Follow.objects.followers(to_follow).first(), Follow)
         self.assertEqual(Follow.objects.following(user).count(), 1)
         self.assertEqual(Follow.objects.followers(user).count(), 0)
         self.assertTrue(self.signal_emitted)
 
     def test_follow_already_followed(self):
-        '''It should do nothing when following an already followed object'''
+        """It should do nothing when following an already followed object"""
         user = self.login()
         to_follow = FakeModel.objects.create()
         Follow.objects.create(follower=user, following=to_follow)
 
         with on_follow.connected_to(self.handler):
-            response = self.post(url_for('api.follow_fake', id=to_follow.id))
+            response = self.post(url_for("api.follow_fake", id=to_follow.id))
 
         self.assertStatus(response, 200)
 
@@ -68,19 +66,19 @@ class FollowAPITest(APITestCase):
         self.assertFalse(self.signal_emitted)
 
     def test_unfollow(self):
-        '''It should unfollow on DELETE'''
+        """It should unfollow on DELETE"""
         user = self.login()
         to_follow = FakeModel.objects.create()
         Follow.objects.create(follower=user, following=to_follow)
 
         with on_unfollow.connected_to(self.handler):
-            response = self.delete(url_for('api.follow_fake', id=to_follow.id))
+            response = self.delete(url_for("api.follow_fake", id=to_follow.id))
 
         self.assert200(response)
 
         nb_followers = Follow.objects.followers(to_follow).count()
 
-        self.assertEqual(response.json['followers'], nb_followers)
+        self.assertEqual(response.json["followers"], nb_followers)
 
         self.assertEqual(Follow.objects.following(to_follow).count(), 0)
         self.assertEqual(nb_followers, 0)
@@ -89,9 +87,9 @@ class FollowAPITest(APITestCase):
         self.assertTrue(self.signal_emitted)
 
     def test_unfollow_not_existing(self):
-        '''It should raise 404 when trying to unfollow a not followed object'''
+        """It should raise 404 when trying to unfollow a not followed object"""
         self.login()
         to_follow = FakeModel.objects.create()
 
-        response = self.delete(url_for('api.follow_fake', id=to_follow.id))
+        response = self.delete(url_for("api.follow_fake", id=to_follow.id))
         self.assert404(response)

udata/tests/api/test_me_api.py

@@ -2,18 +2,15 @@ from datetime import datetime
 
 from flask import url_for
 
-from udata.models import Discussion, Follow, Member, User
-from udata.core.discussions.models import Message as DiscMsg
-from udata.core.dataset.factories import (
-    CommunityResourceFactory,
-    DatasetFactory
-)
 from udata.core.dataset.activities import UserCreatedDataset
+from udata.core.dataset.factories import CommunityResourceFactory, DatasetFactory
 from udata.core.discussions.factories import DiscussionFactory
-from udata.core.reuse.factories import ReuseFactory
+from udata.core.discussions.models import Message as DiscMsg
 from udata.core.organization.factories import OrganizationFactory
+from udata.core.reuse.factories import ReuseFactory
 from udata.core.user.factories import UserFactory
 from udata.i18n import _
+from udata.models import Discussion, Follow, Member, User
 from udata.tests.helpers import capture_mails
 from udata.utils import faker
 
@@ -24,173 +21,163 @@ class MeAPITest(APITestCase):
     modules = []
 
     def test_get_profile(self):
-        '''It should fetch my user data on GET'''
+        """It should fetch my user data on GET"""
         self.login()
-        response = self.get(url_for('api.me'))
+        response = self.get(url_for("api.me"))
         self.assert200(response)
-        self.assertEqual(response.json['email'], self.user.email)
-        self.assertEqual(response.json['first_name'], self.user.first_name)
-        self.assertEqual(response.json['roles'], [])
+        self.assertEqual(response.json["email"], self.user.email)
+        self.assertEqual(response.json["first_name"], self.user.first_name)
+        self.assertEqual(response.json["roles"], [])
 
     def test_get_profile_with_deleted_org(self):
-        '''It should not display my membership to deleted organizations'''
+        """It should not display my membership to deleted organizations"""
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         org = OrganizationFactory(members=[member])
-        deleted_org = OrganizationFactory(members=[member],
-                                          deleted=datetime.utcnow())
-        response = self.get(url_for('api.me'))
+        deleted_org = OrganizationFactory(members=[member], deleted=datetime.utcnow())
+        response = self.get(url_for("api.me"))
         self.assert200(response)
-        orgs = [o['id'] for o in response.json['organizations']]
+        orgs = [o["id"] for o in response.json["organizations"]]
         self.assertIn(str(org.id), orgs)
         self.assertNotIn(str(deleted_org.id), orgs)
 
     def test_get_profile_401(self):
-        '''It should raise a 401 on GET /me if no user is authenticated'''
-        response = self.get(url_for('api.me'))
+        """It should raise a 401 on GET /me if no user is authenticated"""
+        response = self.get(url_for("api.me"))
         self.assert401(response)
 
     def test_update_profile(self):
-        '''It should update my profile from the API'''
+        """It should update my profile from the API"""
         self.login()
         data = self.user.to_dict()
         self.assertTrue(self.user.active)
-        data['about'] = 'new about'
-        data['active'] = False
-        response = self.put(url_for('api.me'), data)
+        data["about"] = "new about"
+        data["active"] = False
+        response = self.put(url_for("api.me"), data)
         self.assert200(response)
         self.assertEqual(User.objects.count(), 1)
         self.user.reload()
-        self.assertEqual(self.user.about, 'new about')
+        self.assertEqual(self.user.about, "new about")
         self.assertTrue(self.user.active)
 
     def test_my_metrics(self):
         self.login()
-        response = self.get(url_for('api.my_metrics'))
+        response = self.get(url_for("api.my_metrics"))
         self.assert200(response)
-        self.assertEqual(response.json['resources_availability'], 100.)
-        self.assertEqual(response.json['datasets_org_count'], 0)
-        self.assertEqual(response.json['followers_org_count'], 0)
-        self.assertEqual(response.json['datasets_count'], 0)
-        self.assertEqual(response.json['followers_count'], 0)
+        self.assertEqual(response.json["resources_availability"], 100.0)
+        self.assertEqual(response.json["datasets_org_count"], 0)
+        self.assertEqual(response.json["followers_org_count"], 0)
+        self.assertEqual(response.json["datasets_count"], 0)
+        self.assertEqual(response.json["followers_count"], 0)
 
     def test_my_reuses(self):
         user = self.login()
         reuses = [ReuseFactory(owner=user) for _ in range(2)]
 
-        response = self.get(url_for('api.my_reuses'))
+        response = self.get(url_for("api.my_reuses"))
         self.assert200(response)
 
         self.assertEqual(len(response.json), len(reuses))
 
     def test_my_org_datasets(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
-        community_resources = [
-            DatasetFactory(owner=user) for _ in range(2)]
-        org_datasets = [
-            DatasetFactory(organization=organization)
-            for _ in range(2)]
+        community_resources = [DatasetFactory(owner=user) for _ in range(2)]
+        org_datasets = [DatasetFactory(organization=organization) for _ in range(2)]
 
-        response = self.get(url_for('api.my_org_datasets'))
+        response = self.get(url_for("api.my_org_datasets"))
         self.assert200(response)
-        self.assertEqual(
-            len(response.json),
-            len(community_resources) + len(org_datasets))
+        self.assertEqual(len(response.json), len(community_resources) + len(org_datasets))
 
     def test_my_org_datasets_with_search(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
         datasets = [
-            DatasetFactory(owner=user, title='foô'),
+            DatasetFactory(owner=user, title="foô"),
         ]
         org_datasets = [
-            DatasetFactory(organization=organization, title='foô'),
+            DatasetFactory(organization=organization, title="foô"),
         ]
 
         # Should not be listed.
         DatasetFactory(owner=user)
         DatasetFactory(organization=organization)
 
-        response = self.get(url_for('api.my_org_datasets'),
-                            qs={'q': 'foô'})
+        response = self.get(url_for("api.my_org_datasets"), qs={"q": "foô"})
         self.assert200(response)
         self.assertEqual(len(response.json), len(datasets) + len(org_datasets))
 
     def test_my_org_community_resources(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
-        community_resources = [
-            CommunityResourceFactory(owner=user) for _ in range(2)]
+        community_resources = [CommunityResourceFactory(owner=user) for _ in range(2)]
         org_community_resources = [
-            CommunityResourceFactory(organization=organization)
-            for _ in range(2)]
+            CommunityResourceFactory(organization=organization) for _ in range(2)
+        ]
 
-        response = self.get(url_for('api.my_org_community_resources'))
+        response = self.get(url_for("api.my_org_community_resources"))
         self.assert200(response)
         self.assertEqual(
-            len(response.json),
-            len(community_resources) + len(org_community_resources))
+            len(response.json), len(community_resources) + len(org_community_resources)
+        )
 
     def test_my_org_community_resources_with_search(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
         community_resources = [
-            CommunityResourceFactory(owner=user, title='foô'),
+            CommunityResourceFactory(owner=user, title="foô"),
         ]
         org_community_resources = [
-            CommunityResourceFactory(organization=organization, title='foô'),
+            CommunityResourceFactory(organization=organization, title="foô"),
         ]
 
         # Should not be listed.
         CommunityResourceFactory(owner=user)
         CommunityResourceFactory(organization=organization)
 
-        response = self.get(url_for('api.my_org_community_resources'),
-                            qs={'q': 'foô'})
+        response = self.get(url_for("api.my_org_community_resources"), qs={"q": "foô"})
         self.assert200(response)
         self.assertEqual(
-            len(response.json),
-            len(community_resources) + len(org_community_resources))
+            len(response.json), len(community_resources) + len(org_community_resources)
+        )
 
     def test_my_org_reuses(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
         reuses = [ReuseFactory(owner=user) for _ in range(2)]
-        org_reuses = [ReuseFactory(organization=organization)
-                      for _ in range(2)]
+        org_reuses = [ReuseFactory(organization=organization) for _ in range(2)]
 
-        response = self.get(url_for('api.my_org_reuses'))
+        response = self.get(url_for("api.my_org_reuses"))
         self.assert200(response)
         self.assertEqual(len(response.json), len(reuses) + len(org_reuses))
 
     def test_my_org_reuses_with_search(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
         reuses = [
-            ReuseFactory(owner=user, title='foô'),
+            ReuseFactory(owner=user, title="foô"),
         ]
         org_reuses = [
-            ReuseFactory(organization=organization, title='foô'),
+            ReuseFactory(organization=organization, title="foô"),
         ]
 
         # Should not be listed.
         ReuseFactory(owner=user)
         ReuseFactory(organization=organization)
 
-        response = self.get(url_for('api.my_org_reuses'), qs={'q': 'foô'})
+        response = self.get(url_for("api.my_org_reuses"), qs={"q": "foô"})
         self.assert200(response)
         self.assertEqual(len(response.json), len(reuses) + len(org_reuses))
 
     def test_my_org_discussions(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
         reuse = ReuseFactory(owner=user)
         org_reuse = ReuseFactory(organization=organization)
@@ -198,23 +185,23 @@ class MeAPITest(APITestCase):
         org_dataset = DatasetFactory(organization=organization)
 
         discussions = [
-            Discussion.objects.create(subject=dataset, title='', user=user),
-            Discussion.objects.create(subject=org_dataset, title='', user=user),
-            Discussion.objects.create(subject=reuse, title='', user=user),
-            Discussion.objects.create(subject=org_reuse, title='', user=user),
+            Discussion.objects.create(subject=dataset, title="", user=user),
+            Discussion.objects.create(subject=org_dataset, title="", user=user),
+            Discussion.objects.create(subject=reuse, title="", user=user),
+            Discussion.objects.create(subject=org_reuse, title="", user=user),
         ]
 
         # Should not be listed
-        Discussion.objects.create(subject=DatasetFactory(), title='', user=user)
-        Discussion.objects.create(subject=ReuseFactory(), title='', user=user)
+        Discussion.objects.create(subject=DatasetFactory(), title="", user=user)
+        Discussion.objects.create(subject=ReuseFactory(), title="", user=user)
 
-        response = self.get(url_for('api.my_org_discussions'))
+        response = self.get(url_for("api.my_org_discussions"))
         self.assert200(response)
         self.assertEqual(len(response.json), len(discussions))
 
     def test_my_org_discussions_with_search(self):
         user = self.login()
-        member = Member(user=user, role='editor')
+        member = Member(user=user, role="editor")
         organization = OrganizationFactory(members=[member])
         reuse = ReuseFactory(owner=user)
         org_reuse = ReuseFactory(organization=organization)
@@ -222,68 +209,68 @@ class MeAPITest(APITestCase):
         org_dataset = DatasetFactory(organization=organization)
 
         discussions = [
-            Discussion.objects.create(subject=dataset, title='foô', user=user),
-            Discussion.objects.create(subject=org_reuse, title='foô', user=user),
+            Discussion.objects.create(subject=dataset, title="foô", user=user),
+            Discussion.objects.create(subject=org_reuse, title="foô", user=user),
         ]
 
         # Should not be listed.
-        Discussion.objects.create(subject=reuse, title='', user=user),
-        Discussion.objects.create(subject=org_dataset, title='', user=user),
+        (Discussion.objects.create(subject=reuse, title="", user=user),)
+        (Discussion.objects.create(subject=org_dataset, title="", user=user),)
 
         # Should really not be listed.
-        Discussion.objects.create(subject=DatasetFactory(), title='foô', user=user)
-        Discussion.objects.create(subject=ReuseFactory(), title='foô', user=user)
+        Discussion.objects.create(subject=DatasetFactory(), title="foô", user=user)
+        Discussion.objects.create(subject=ReuseFactory(), title="foô", user=user)
 
-        response = self.get(url_for('api.my_org_discussions'), qs={'q': 'foô'})
+        response = self.get(url_for("api.my_org_discussions"), qs={"q": "foô"})
         self.assert200(response)
         self.assertEqual(len(response.json), len(discussions))
 
     def test_my_reuses_401(self):
-        response = self.get(url_for('api.my_reuses'))
+        response = self.get(url_for("api.my_reuses"))
         self.assert401(response)
 
     def test_generate_apikey(self):
-        '''It should generate an API Key on POST'''
+        """It should generate an API Key on POST"""
         self.login()
-        response = self.post(url_for('api.my_apikey'))
+        response = self.post(url_for("api.my_apikey"))
         self.assert201(response)
-        self.assertIsNotNone(response.json['apikey'])
+        self.assertIsNotNone(response.json["apikey"])
 
         self.user.reload()
         self.assertFalse(self.user.apikey.startswith("b'"))
         self.assertIsNotNone(self.user.apikey)
-        self.assertEqual(self.user.apikey, response.json['apikey'])
+        self.assertEqual(self.user.apikey, response.json["apikey"])
 
     def test_regenerate_apikey(self):
-        '''It should regenerate an API Key on POST'''
+        """It should regenerate an API Key on POST"""
         self.login()
         self.user.generate_api_key()
         self.user.save()
 
         apikey = self.user.apikey
-        response = self.post(url_for('api.my_apikey'))
+        response = self.post(url_for("api.my_apikey"))
         self.assert201(response)
-        self.assertIsNotNone(response.json['apikey'])
+        self.assertIsNotNone(response.json["apikey"])
 
         self.user.reload()
         self.assertIsNotNone(self.user.apikey)
         self.assertNotEqual(self.user.apikey, apikey)
-        self.assertEqual(self.user.apikey, response.json['apikey'])
+        self.assertEqual(self.user.apikey, response.json["apikey"])
 
     def test_clear_apikey(self):
-        '''It should clear an API Key on DELETE'''
+        """It should clear an API Key on DELETE"""
         self.login()
         self.user.generate_api_key()
         self.user.save()
 
-        response = self.delete(url_for('api.my_apikey'))
+        response = self.delete(url_for("api.my_apikey"))
         self.assert204(response)
 
         self.user.reload()
         self.assertIsNone(self.user.apikey)
 
     def test_delete(self):
-        '''It should delete the connected user'''
+        """It should delete the connected user"""
         user = self.login()
         user_email = user.email
         self.assertIsNone(user.deleted)
@@ -291,13 +278,10 @@ class MeAPITest(APITestCase):
         members = [Member(user=user), Member(user=other_user)]
         organization = OrganizationFactory(members=members)
         disc_msg_content = faker.sentence()
-        disc_msg = DiscMsg(content=disc_msg_content,
-                           posted_by=user)
+        disc_msg = DiscMsg(content=disc_msg_content, posted_by=user)
         other_disc_msg_content = faker.sentence()
-        other_disc_msg = DiscMsg(content=other_disc_msg_content,
-                                 posted_by=other_user)
-        discussion = DiscussionFactory(user=user,
-                                       discussion=[disc_msg, other_disc_msg])
+        other_disc_msg = DiscMsg(content=other_disc_msg_content, posted_by=other_user)
+        discussion = DiscussionFactory(user=user, discussion=[disc_msg, other_disc_msg])
         dataset = DatasetFactory(owner=user)
         reuse = ReuseFactory(owner=user)
         resource = CommunityResourceFactory(owner=user)
@@ -307,10 +291,10 @@ class MeAPITest(APITestCase):
         followed = Follow.objects().create(follower=other_user, following=user)
 
         with capture_mails() as mails:
-            response = self.delete(url_for('api.me'))
+            response = self.delete(url_for("api.me"))
         self.assertEqual(len(mails), 1)
         self.assertEqual(mails[0].send_to, set([user_email]))
-        self.assertEqual(mails[0].subject, _('Account deletion'))
+        self.assertEqual(mails[0].subject, _("Account deletion"))
         self.assert204(response)
 
         user.reload()
@@ -329,11 +313,11 @@ class MeAPITest(APITestCase):
             followed.reload()
 
         # The personal data of the user are anonymized
-        self.assertEqual(user.email, '{}@deleted'.format(user.id))
+        self.assertEqual(user.email, "{}@deleted".format(user.id))
         self.assertEqual(user.password, None)
         self.assertEqual(user.active, False)
-        self.assertEqual(user.first_name, 'DELETED')
-        self.assertEqual(user.last_name, 'DELETED')
+        self.assertEqual(user.first_name, "DELETED")
+        self.assertEqual(user.last_name, "DELETED")
         self.assertFalse(bool(user.avatar))
         self.assertEqual(user.avatar_url, None)
         self.assertEqual(user.website, None)
@@ -348,9 +332,8 @@ class MeAPITest(APITestCase):
 
         # The discussions are kept but the messages are anonymized
         self.assertEqual(len(discussion.discussion), 2)
-        self.assertEqual(discussion.discussion[0].content, 'DELETED')
-        self.assertEqual(discussion.discussion[1].content,
-                         other_disc_msg_content)
+        self.assertEqual(discussion.discussion[0].content, "DELETED")
+        self.assertEqual(discussion.discussion[1].content, other_disc_msg_content)
 
         # The datasets are unchanged
         self.assertEqual(dataset.owner, user)