servala-portal/src/tests/conftest.py

import pytest

from servala.core.models import (
    Organization,
    OrganizationMembership,
    OrganizationOrigin,
    User,
)


@pytest.fixture
def origin():
    return OrganizationOrigin.objects.create(name="TESTORIGIN")


@pytest.fixture
def organization(origin):
    return Organization.objects.create(
        name="Test Org", namespace="test-org", origin=origin
    )


@pytest.fixture
def other_organization(origin):
    return Organization.objects.create(
        name="Test Org Alternate", namespace="test-org-alternate", origin=origin
    )


@pytest.fixture
def org_owner(organization):
    user = User.objects.create(email="user@example.org", password="example")
    OrganizationMembership.objects.create(
        organization=organization, user=user, role="owner"
    )
    return user