Add membership backend

amanuensis/backend/membership.py

@@ -0,0 +1,32 @@
+"""
+Membership query interface
+"""
+
+from amanuensis.db import DbContext, Membership
+from amanuensis.errors import ArgumentError
+
+
+def create(
+    db: DbContext,
+    user_id: int,
+    lexicon_id: int,
+    is_editor: bool) -> Membership:
+    """
+    Create a new user membership in a lexicon.
+    """
+    # Quick argument verification
+    if not isinstance(user_id, int):
+        raise ArgumentError('user_id')
+    if not isinstance(lexicon_id, int):
+        raise ArgumentError('lexicon_id')
+    if not isinstance(is_editor, bool):
+        raise ArgumentError('is_editor')
+
+    new_membership = Membership(
+        user_id=user_id,
+        lexicon_id=lexicon_id,
+        is_editor=is_editor,
+    )
+    db.session.add(new_membership)
+    db.session.commit()
+    return new_membership

tests/test_membership.py

@@ -0,0 +1,28 @@
+from amanuensis.db import *
+import amanuensis.backend.lexicon as lexiq
+import amanuensis.backend.membership as memq
+import amanuensis.backend.user as userq
+
+from .test_db import db
+
+
+def test_create(db: DbContext):
+    """Test joining a game."""
+    # Set up a user and a lexicon
+    new_user = userq.create(db, 'username', 'password', 'user', 'a@b.c', False)
+    assert new_user.id, 'Failed to create user'
+    new_lexicon = lexiq.create(db, 'Test', None, 'prompt')
+    assert new_lexicon.id, 'Failed to create lexicon'
+
+    # Add the user to the lexicon as an editor
+    mem = memq.create(db, new_user.id, new_lexicon.id, True)
+    assert mem, 'Failed to create membership'
+
+    # Check that the user and lexicon are mutually visible in the ORM relationships
+    assert new_user.memberships, 'User memberships not updated'
+    assert new_lexicon.memberships, 'Lexicon memberships not updated'
+    assert new_user.memberships[0].lexicon_id == new_lexicon.id
+    assert new_lexicon.memberships[0].user_id == new_user.id
+
+    # Check that the editor flag was set properly
+    assert new_lexicon.memberships