92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
# Copyright (c) 2020, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# For license information, please see license.txt
|
|
|
|
|
|
import os
|
|
from contextlib import contextmanager
|
|
from typing import Any
|
|
|
|
import frappe
|
|
from frappe import set_user as _set_user
|
|
from frappe.model.document import Document
|
|
from frappe.tests.utils import FrappeTestCase
|
|
|
|
from press.utils import _get_current_team, _system_user
|
|
|
|
|
|
def doc_equal(self: Document, other: Document) -> bool:
|
|
"""Partial equality checking of Document object"""
|
|
if not isinstance(other, Document):
|
|
return False
|
|
if self.doctype == other.doctype and self.name == other.name:
|
|
return True
|
|
return False
|
|
|
|
|
|
def FrappeTestCase_setUp(self) -> None:
|
|
frappe.clear_cache()
|
|
frappe.db.truncate("Agent Request Failure")
|
|
frappe.local.conf.update({"throttle_user_limit": 600})
|
|
|
|
|
|
def execute():
|
|
settings = frappe.get_single("Press Settings")
|
|
if not (settings.stripe_secret_key and settings.stripe_publishable_key):
|
|
create_test_stripe_credentials()
|
|
import cssutils
|
|
|
|
# Silence the cssutils errors that are mostly pointless
|
|
cssutils.log.setLevel(50)
|
|
|
|
# Monkey patch certain methods for when tests are running
|
|
Document.__eq__ = doc_equal
|
|
|
|
FrappeTestCase.setUp = FrappeTestCase_setUp
|
|
FrappeTestCase.tearDown = lambda self: frappe.db.rollback()
|
|
FrappeTestCase.freeze_time = staticmethod(freeze_time)
|
|
|
|
# patch frappe.set_user that
|
|
frappe.set_user = set_user_with_current_team
|
|
|
|
# frappe.local.team helper
|
|
frappe.local.team = _get_current_team
|
|
frappe.local.system_user = _system_user
|
|
|
|
|
|
def set_user_with_current_team(user):
|
|
_set_user(user)
|
|
frappe.local._current_team = None
|
|
|
|
|
|
def create_test_stripe_credentials():
|
|
publishable_key = os.environ.get("STRIPE_PUBLISHABLE_KEY")
|
|
secret_key = os.environ.get("STRIPE_SECRET_KEY")
|
|
|
|
if publishable_key and secret_key:
|
|
frappe.db.set_single_value("Press Settings", "stripe_publishable_key", publishable_key)
|
|
frappe.db.set_single_value("Press Settings", "stripe_secret_key", secret_key)
|
|
|
|
|
|
@contextmanager
|
|
def freeze_time(time_to_freeze: Any, is_utc: bool = False, *args: Any, **kwargs: Any):
|
|
"""freeze time using freezegun, compatible with Python 3.10 and 3.11+."""
|
|
try:
|
|
from datetime import UTC
|
|
except ImportError:
|
|
from datetime import timezone
|
|
|
|
UTC = timezone.utc
|
|
|
|
from zoneinfo import ZoneInfo
|
|
|
|
from frappe.utils.data import get_datetime, get_system_timezone
|
|
from freezegun import freeze_time as freezegun_freeze_time
|
|
|
|
if not is_utc:
|
|
time_to_freeze = (
|
|
get_datetime(time_to_freeze).replace(tzinfo=ZoneInfo(get_system_timezone())).astimezone(UTC)
|
|
)
|
|
|
|
with freezegun_freeze_time(time_to_freeze, *args, **kwargs):
|
|
yield
|