42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
from datetime import timedelta
|
||
|
|
from unittest import mock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from django.db.backends.postgresql.psycopg_any import DateRange
|
||
|
|
from django.utils import timezone
|
||
|
|
from membership.models import Member
|
||
|
|
from membership.models import Membership
|
||
|
|
from membership.models import MembershipType
|
||
|
|
from membership.models import SubscriptionPeriod
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def membership_type():
|
||
|
|
return MembershipType.objects.create(name="Test Membership Type")
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def current_period():
|
||
|
|
SubscriptionPeriod.objects.create(
|
||
|
|
period=DateRange(timezone.now().date() - timedelta(days=182), timezone.now().date() + timedelta(days=183))
|
||
|
|
)
|
||
|
|
return SubscriptionPeriod.objects.current()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def active_membership(membership_type, current_period):
|
||
|
|
member = Member.objects.create_user("test", "lala@adas.com", "1234")
|
||
|
|
membership = Membership.objects.create(
|
||
|
|
user=member,
|
||
|
|
membership_type=membership_type,
|
||
|
|
period=current_period,
|
||
|
|
activated=True,
|
||
|
|
)
|
||
|
|
return membership
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
def mock_matrix_notify():
|
||
|
|
with mock.patch("utils.matrix.httpx.post"):
|
||
|
|
yield
|