import pytest from sqlalchemy import select from amanuensis.db import * from amanuensis.errors import ArgumentError import amanuensis.backend.membership as memq def test_create_membership(db: DbContext, make): """Test joining a game.""" # Set up a user and a lexicon new_user = make.user() assert new_user.id, "Failed to create user" new_lexicon = make.lexicon() 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 any(map(lambda mem: mem.lexicon == new_lexicon, new_user.memberships)) assert any(map(lambda mem: mem.user == new_user, new_lexicon.memberships)) # Check that the editor flag was set properly editor = db( select(User) .join(User.memberships) .join(Membership.lexicon) .where(Lexicon.id == new_lexicon.id) .where(Membership.is_editor == True) ).scalar_one() assert editor is not None assert isinstance(editor, User) assert editor.id == new_user.id # Check that joining twice is not allowed with pytest.raises(ArgumentError): mem2 = memq.create(db, new_user.id, new_lexicon.id, False) assert mem2