MrMeeb
d09b3b992c
Some checks failed
Build-Release-Image / Build-Image (linux/arm64) (push) Failing after 7m18s
Build-Release-Image / Build-Image (linux/amd64) (push) Has been cancelled
Build-Release-Image / Merge-Images (push) Has been cancelled
Build-Release-Image / Create-Release (push) Has been cancelled
Build-Release-Image / Notify (push) Has been cancelled
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
import arrow
|
|
|
|
from app.config import EMAIL_DOMAIN
|
|
from app.db import Session
|
|
from app.models import CoinbaseSubscription
|
|
from app.payments.coinbase import handle_coinbase_event
|
|
from tests.utils import create_new_user
|
|
|
|
|
|
def test_redirect_login_page(flask_client):
|
|
"""Start with a blank database."""
|
|
|
|
rv = flask_client.get("/")
|
|
assert rv.status_code == 302
|
|
assert rv.location == f"http://{EMAIL_DOMAIN}/auth/login"
|
|
|
|
|
|
def test_coinbase_webhook(flask_client):
|
|
r = flask_client.post("/coinbase")
|
|
assert r.status_code == 400
|
|
|
|
|
|
def test_handle_coinbase_event_new_subscription(flask_client):
|
|
user = create_new_user()
|
|
handle_coinbase_event(
|
|
{"data": {"code": "AAAAAA", "metadata": {"user_id": str(user.id)}}}
|
|
)
|
|
|
|
assert user.is_paid()
|
|
assert user.is_premium()
|
|
|
|
assert CoinbaseSubscription.get_by(user_id=user.id) is not None
|
|
|
|
|
|
def test_handle_coinbase_event_extend_subscription(flask_client):
|
|
user = create_new_user()
|
|
user.trial_end = None
|
|
Session.commit()
|
|
|
|
cb = CoinbaseSubscription.create(
|
|
user_id=user.id, end_at=arrow.now().shift(days=-400), commit=True
|
|
)
|
|
assert not cb.is_active()
|
|
|
|
assert not user.is_paid()
|
|
assert not user.is_premium()
|
|
|
|
handle_coinbase_event(
|
|
{"data": {"code": "AAAAAA", "metadata": {"user_id": str(user.id)}}}
|
|
)
|
|
|
|
assert user.is_paid()
|
|
assert user.is_premium()
|
|
|
|
assert CoinbaseSubscription.get_by(user_id=user.id) is not None
|