42 lines
1 KiB
Python
42 lines
1 KiB
Python
|
import itertools
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from allauth.account.models import EmailAddress
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
|
||
|
|
from intent.orgs.models import OrgMembership, Role
|
||
|
|
from intent.orgs.services import create_org
|
||
|
|
|
||
|
|
_counter = itertools.count()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def make_user(db):
|
||
|
|
def make(username=None, *, email=None, is_staff=False):
|
||
|
|
username = username or f"user{next(_counter)}"
|
||
|
|
email = email or f"{username}@example.com"
|
||
|
|
user = get_user_model().objects.create_user(
|
||
|
|
username=username, email=email, is_staff=is_staff
|
||
|
|
)
|
||
|
|
EmailAddress.objects.create(user=user, email=email, verified=True, primary=True)
|
||
|
|
return user
|
||
|
|
|
||
|
|
return make
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def owner(make_user):
|
||
|
|
return make_user("olive")
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def org(owner):
|
||
|
|
return create_org(slug="catbriar", name="Catbriar", owner=owner)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def collaborator(make_user, org):
|
||
|
|
user = make_user("cole")
|
||
|
|
OrgMembership.objects.create(org=org, user=user, role=Role.COLLABORATOR)
|
||
|
|
return user
|