Merge pull request #323 from jimfingal/jim/cli-tooling

Add some extra Python / Docker cli tooling

bookwyrm/management/commands/initdb.py

@@ -0,0 +1,100 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth.models import Group, Permission
+from django.contrib.contenttypes.models import ContentType
+
+from bookwyrm.models import Connector, User
+from bookwyrm.settings import DOMAIN
+
+def init_groups():
+    groups = ['admin', 'moderator', 'editor']
+    for group in groups:
+        Group.objects.create(name=group)
+
+def init_permissions():
+    permissions = [{
+            'codename': 'edit_instance_settings',
+            'name': 'change the instance info',
+            'groups': ['admin',]
+        }, {
+            'codename': 'set_user_group',
+            'name': 'change what group a user is in',
+            'groups': ['admin', 'moderator']
+        }, {
+            'codename': 'control_federation',
+            'name': 'control who to federate with',
+            'groups': ['admin', 'moderator']
+        }, {
+            'codename': 'create_invites',
+            'name': 'issue invitations to join',
+            'groups': ['admin', 'moderator']
+        }, {
+            'codename': 'moderate_user',
+            'name': 'deactivate or silence a user',
+            'groups': ['admin', 'moderator']
+        }, {
+            'codename': 'moderate_post',
+            'name': 'delete other users\' posts',
+            'groups': ['admin', 'moderator']
+        }, {
+            'codename': 'edit_book',
+            'name': 'edit book info',
+            'groups': ['admin', 'moderator', 'editor']
+        }]
+
+    content_type = ContentType.objects.get_for_model(User)
+    for permission in permissions:
+        permission_obj = Permission.objects.create(
+            codename=permission['codename'],
+            name=permission['name'],
+            content_type=content_type,
+        )
+        # add the permission to the appropriate groups
+        for group_name in permission['groups']:
+            Group.objects.get(name=group_name).permissions.add(permission_obj)
+
+    # while the groups and permissions shouldn't be changed because the code
+    # depends on them, what permissions go with what groups should be editable
+
+
+def init_connectors():
+    Connector.objects.create(
+        identifier=DOMAIN,
+        name='Local',
+        local=True,
+        connector_file='self_connector',
+        base_url='https://%s' % DOMAIN,
+        books_url='https://%s/book' % DOMAIN,
+        covers_url='https://%s/images/covers' % DOMAIN,
+        search_url='https://%s/search?q=' % DOMAIN,
+        priority=1,
+    )
+
+    Connector.objects.create(
+        identifier='bookwyrm.social',
+        name='BookWyrm dot Social',
+        connector_file='bookwyrm_connector',
+        base_url='https://bookwyrm.social' ,
+        books_url='https://bookwyrm.social/book',
+        covers_url='https://bookwyrm.social/images/covers',
+        search_url='https://bookwyrm.social/search?q=',
+        priority=2,
+    )
+
+    Connector.objects.create(
+        identifier='openlibrary.org',
+        name='OpenLibrary',
+        connector_file='openlibrary',
+        base_url='https://openlibrary.org',
+        books_url='https://openlibrary.org',
+        covers_url='https://covers.openlibrary.org',
+        search_url='https://openlibrary.org/search?q=',
+        priority=3,
+    )
+
+class Command(BaseCommand):
+    help = 'Initializes the database with starter data'
+
+    def handle(self, *args, **options):
+        init_groups()
+        init_permissions()
+        init_connectors()

bookwyrm/tests/test_signing.py

@@ -6,6 +6,8 @@ import pathlib
 import json
 import responses
 
+import pytest
+
 from django.test import TestCase, Client
 from django.utils.http import http_date
 
@@ -167,6 +169,7 @@ class Signature(TestCase):
         response = self.send_test_request(sender=self.fake_remote)
         self.assertEqual(response.status_code, 401)
 
+    @pytest.mark.integration
     def test_changed_data(self):
         '''Message data must match the digest header.'''
         response = self.send_test_request(
@@ -174,12 +177,14 @@ class Signature(TestCase):
             send_data=get_follow_data(self.mouse, self.cat))
         self.assertEqual(response.status_code, 401)
 
+    @pytest.mark.integration
     def test_invalid_digest(self):
         response = self.send_test_request(
             self.mouse,
             digest='SHA-256=AAAAAAAAAAAAAAAAAA')
         self.assertEqual(response.status_code, 401)
 
+    @pytest.mark.integration
     def test_old_message(self):
         '''Old messages should be rejected to prevent replay attacks.'''
         response = self.send_test_request(